Add MCP Apps to NativeAOT coverage (#1711)
Co-authored-by: Loi Nguyen <loinguyen@local> Co-authored-by: Jeff Handley <jeffhandley@users.noreply.github.com>
This commit is contained in:
parent
9f845f8ca5
commit
dfb2205cac
1 changed files with 31 additions and 13 deletions
|
|
@ -1,17 +1,20 @@
|
|||
using Microsoft.Extensions.DependencyInjection;
|
||||
using ModelContextProtocol.Client;
|
||||
using ModelContextProtocol.Extensions.Apps;
|
||||
using ModelContextProtocol.Protocol;
|
||||
using ModelContextProtocol.Server;
|
||||
using System.IO.Pipelines;
|
||||
|
||||
Pipe clientToServerPipe = new(), serverToClientPipe = new();
|
||||
|
||||
// Create a server using a stream-based transport over an in-memory pipe.
|
||||
await using McpServer server = McpServer.Create(
|
||||
new StreamServerTransport(clientToServerPipe.Reader.AsStream(), serverToClientPipe.Writer.AsStream()),
|
||||
new McpServerOptions()
|
||||
{
|
||||
ToolCollection = [McpServerTool.Create((string arg) => $"Echo: {arg}", new() { Name = "Echo" })]
|
||||
});
|
||||
var services = new ServiceCollection();
|
||||
services.AddMcpServer()
|
||||
.WithStreamServerTransport(clientToServerPipe.Reader.AsStream(), serverToClientPipe.Writer.AsStream())
|
||||
.WithTools<AotTools>()
|
||||
.WithMcpApps();
|
||||
|
||||
await using var serviceProvider = services.BuildServiceProvider();
|
||||
var server = serviceProvider.GetRequiredService<McpServer>();
|
||||
_ = server.RunAsync();
|
||||
|
||||
// Connect a client using a stream-based transport over the same in-memory pipe.
|
||||
|
|
@ -20,13 +23,18 @@ await using McpClient client = await McpClient.CreateAsync(
|
|||
|
||||
// List all tools.
|
||||
var tools = await client.ListToolsAsync();
|
||||
if (tools.Count == 0)
|
||||
var echo = tools.FirstOrDefault(t => t.Name == "Echo");
|
||||
if (echo is null)
|
||||
{
|
||||
throw new Exception("Expected at least one tool.");
|
||||
throw new Exception("Expected the Echo tool.");
|
||||
}
|
||||
|
||||
var ui = echo.ProtocolTool.Meta?["ui"]?.AsObject();
|
||||
if (ui?["resourceUri"]?.GetValue<string>() != "ui://aot/echo")
|
||||
{
|
||||
throw new Exception($"Unexpected app UI metadata: {ui}");
|
||||
}
|
||||
|
||||
// Invoke a tool.
|
||||
var echo = tools.First(t => t.Name == "Echo");
|
||||
var result = await echo.InvokeAsync(new() { ["arg"] = "Hello World" });
|
||||
if (result is null || !result.ToString()!.Contains("Echo: Hello World"))
|
||||
{
|
||||
|
|
@ -34,3 +42,11 @@ if (result is null || !result.ToString()!.Contains("Echo: Hello World"))
|
|||
}
|
||||
|
||||
Console.WriteLine("Success!");
|
||||
|
||||
[McpServerToolType]
|
||||
internal sealed class AotTools
|
||||
{
|
||||
[McpServerTool(Name = "Echo")]
|
||||
[McpAppUi(ResourceUri = "ui://aot/echo")]
|
||||
public static string Echo(string arg) => $"Echo: {arg}";
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue