Fixing WithPromptsFromAssembly (#174)

* Fixing WithPromptsFromAssembly

Was looking for tools

* Fixing test
This commit is contained in:
Aaron Powell 2025-04-01 11:59:31 +11:00 • committed by GitHub
commit b9bb871787
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -176,7 +176,7 @@ public static partial class McpServerBuilderExtensions
}
/// <summary>
/// Adds types marked with the <see cref="McpServerToolTypeAttribute"/> attribute from the given assembly as prompts to the server.
/// Adds types marked with the <see cref="McpServerPromptTypeAttribute"/> attribute from the given assembly as prompts to the server.
/// </summary>
/// <param name="builder">The builder instance.</param>
/// <param name="promptAssembly">The assembly to load the types from. Null to get the current assembly</param>
@ -190,7 +190,7 @@ public static partial class McpServerBuilderExtensions
return builder.WithPrompts(
from t in promptAssembly.GetTypes()
where t.GetCustomAttribute<McpServerToolTypeAttribute>() is not null
where t.GetCustomAttribute<McpServerPromptTypeAttribute>() is not null
select t);
}
#endregion