Microsoft.Extensions.AI.Abstractions integration (#25)
* MEAI integration * README and xmldocs
This commit is contained in:
parent
502a09d9fc
commit
89c4913ed7
1 changed files with 589 additions and 1 deletions
119
tests/McpDotNet.Extensions.AI.Tests/IntegrationTests.cs
Normal file
119
tests/McpDotNet.Extensions.AI.Tests/IntegrationTests.cs
Normal file
|
|
@ -0,0 +1,119 @@
|
|||
using McpDotNet.Client;
|
||||
using McpDotNet.Configuration;
|
||||
using Microsoft.Extensions.AI;
|
||||
using Microsoft.Extensions.Logging.Abstractions;
|
||||
using OpenAI;
|
||||
|
||||
namespace McpDotNet.Extensions.AI.Tests;
|
||||
|
||||
public class IntegrationTests
|
||||
{
|
||||
private const string OpenAIKey = ""; // Provide your own key when running the test. Do not commit it.
|
||||
|
||||
private static McpServerConfig GetEverythingServerConfig()
|
||||
{
|
||||
var config = new McpServerConfig
|
||||
{
|
||||
Id = "everything",
|
||||
Name = "Everything",
|
||||
TransportType = "stdio",
|
||||
TransportOptions = new Dictionary<string, string>
|
||||
{
|
||||
["command"] = "npx",
|
||||
["arguments"] = "-y @modelcontextprotocol/server-everything",
|
||||
}
|
||||
};
|
||||
|
||||
return config;
|
||||
}
|
||||
|
||||
private static async Task<IMcpClient> GetMcpClientAsync()
|
||||
{
|
||||
McpClientOptions options = new()
|
||||
{
|
||||
ClientInfo = new() { Name = "McpDotNet.Extensions.AI.Tests", Version = "1.0.0" }
|
||||
};
|
||||
|
||||
var factory = new McpClientFactory(
|
||||
[GetEverythingServerConfig()],
|
||||
options,
|
||||
NullLoggerFactory.Instance
|
||||
);
|
||||
|
||||
return await factory.GetClientAsync("everything");
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IntegrateWithMeai_UsingEverythingServer_ToolsAreProperlyCalled()
|
||||
{
|
||||
var client = await GetMcpClientAsync();
|
||||
var tools = await client.ListToolsAsync();
|
||||
var mappedTools = tools.Tools.Select(t => t.ToAITool(client)).ToList();
|
||||
|
||||
IChatClient openaiClient = new OpenAIClient(OpenAIKey)
|
||||
.AsChatClient("gpt-4o-mini");
|
||||
|
||||
IChatClient chatClient = new ChatClientBuilder(openaiClient)
|
||||
.UseFunctionInvocation()
|
||||
.Build();
|
||||
|
||||
// Create message list
|
||||
IList<ChatMessage> messages = new List<ChatMessage>();
|
||||
// Add a system message
|
||||
messages.Add(new(ChatRole.System, "You are a helpful assistant, helping us test MCP server functionality."));
|
||||
// If MCP server provides instructions, add them as an additional system message (you could also add it as a content part)
|
||||
if (!string.IsNullOrEmpty(client.ServerInstructions))
|
||||
{
|
||||
messages.Add(new(ChatRole.System, client.ServerInstructions));
|
||||
}
|
||||
// Add a user message
|
||||
messages.Add(new(ChatRole.User, "Please call the echo tool with the string 'Hello MCP!' and output the response ad verbatim."));
|
||||
|
||||
// Call the chat client
|
||||
Console.WriteLine("Asking GPT-4o-mini to call the Echo Tool...");
|
||||
var response = await chatClient.GetResponseAsync(
|
||||
messages,
|
||||
new() { Tools = mappedTools, Temperature = 0 });
|
||||
|
||||
// Assert
|
||||
Assert.Equal(4, messages.Count); // 1 system messages, 1 user message, 1 tool message, 1 response message
|
||||
Assert.Equal("Echo: Hello MCP!", response.Message.Contents[0].ToString());
|
||||
}
|
||||
|
||||
[Fact]
|
||||
public async Task IntegrateWithMeai_UsingEverythingServer_AndSessionScope_ToolsAreProperlyCalled()
|
||||
{
|
||||
await using var sessionScope = await McpSessionScope.CreateAsync(GetEverythingServerConfig());
|
||||
|
||||
IChatClient openaiClient = new OpenAIClient(OpenAIKey)
|
||||
.AsChatClient("gpt-4o-mini");
|
||||
|
||||
IChatClient chatClient = new ChatClientBuilder(openaiClient)
|
||||
.UseFunctionInvocation()
|
||||
.Build();
|
||||
|
||||
// Create message list
|
||||
IList<ChatMessage> messages =
|
||||
[
|
||||
// Add a system message
|
||||
new(ChatRole.System, "You are a helpful assistant, helping us test MCP server functionality."),
|
||||
];
|
||||
// If MCP server provides instructions, add them as an additional system message (you could also add it as a content part)
|
||||
foreach (var serverInstruction in sessionScope.ServerInstructions)
|
||||
{
|
||||
messages.Add(new(ChatRole.System, serverInstruction));
|
||||
}
|
||||
// Add a user message
|
||||
messages.Add(new(ChatRole.User, "Please call the echo tool with the string 'Hello MCP!' and output the response ad verbatim."));
|
||||
|
||||
// Call the chat client
|
||||
Console.WriteLine("Asking GPT-4o-mini to call the Echo Tool...");
|
||||
var response = await chatClient.GetResponseAsync(
|
||||
messages,
|
||||
new() { Tools = sessionScope.Tools, Temperature = 0 });
|
||||
|
||||
// Assert
|
||||
Assert.Equal(4, messages.Count); // 1 system messages, 1 user message, 1 tool message, 1 response message
|
||||
Assert.Equal("Echo: Hello MCP!", response.Message.Contents[0].ToString());
|
||||
}
|
||||
}
|
||||
Loading…
Reference in a new issue