modelcontextprotocol/csharp-sdk
Watch
1
0
Fork
You've already forked csharp-sdk
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Add tools support to sampling (#976)

Browse source
This commit is contained in:
Stephen Toub 2025-11-20 09:52:23 -05:00 • committed by GitHub
commit 15139cc569
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1623 additions and 425 deletions
Download patch file Download diff file Expand all files Collapse all files

174
tests/ModelContextProtocol.Tests/Protocol/CreateMessageRequestParamsTests.cs Normal file
View file

@ -0,0 +1,174 @@
using ModelContextProtocol.Protocol;
using System.Text.Json;
namespace ModelContextProtocol.Tests.Protocol;
public class CreateMessageRequestParamsTests
{
[Fact]
public void WithTools_SerializationRoundtrips()
{
CreateMessageRequestParams requestParams = new()
{
MaxTokens = 1000,
Messages =
[
new SamplingMessage
{
Role = Role.User,
Content = [new TextContentBlock { Text = "What's the weather in Paris?" }]
}
],
Tools =
[
new Tool
{
Name = "get_weather",
Description = "Get weather for a city",
InputSchema = JsonElement.Parse("""
{
"type": "object",
"properties": {
"city": { "type": "string" }
},
"required": ["city"]
}
""")
}
],
ToolChoice = new ToolChoice { Mode = "auto" }
};
var json = JsonSerializer.Serialize(requestParams, McpJsonUtilities.DefaultOptions);
var deserialized = JsonSerializer.Deserialize<CreateMessageRequestParams>(json, McpJsonUtilities.DefaultOptions);
Assert.NotNull(deserialized);
Assert.Equal(1000, deserialized.MaxTokens);
Assert.NotNull(deserialized.Messages);
Assert.Single(deserialized.Messages);
Assert.Equal(Role.User, deserialized.Messages[0].Role);
Assert.Single(deserialized.Messages[0].Content);
var textContent = Assert.IsType<TextContentBlock>(deserialized.Messages[0].Content[0]);
Assert.Equal("What's the weather in Paris?", textContent.Text);
Assert.NotNull(deserialized.Tools);
Assert.Single(deserialized.Tools);
Assert.Equal("get_weather", deserialized.Tools[0].Name);
Assert.Equal("Get weather for a city", deserialized.Tools[0].Description);
Assert.Equal("object", deserialized.Tools[0].InputSchema.GetProperty("type").GetString());
Assert.True(deserialized.Tools[0].InputSchema.GetProperty("properties").TryGetProperty("city", out var cityProp));
Assert.Equal("string", cityProp.GetProperty("type").GetString());
Assert.Single(deserialized.Tools[0].InputSchema.GetProperty("required").EnumerateArray());
Assert.Equal("city", deserialized.Tools[0].InputSchema.GetProperty("required")[0].GetString());
Assert.NotNull(deserialized.ToolChoice);
Assert.Equal("auto", deserialized.ToolChoice.Mode);
}
[Fact]
public void WithToolChoiceRequired_SerializationRoundtrips()
{
CreateMessageRequestParams requestParams = new()
{
MaxTokens = 1000,
Messages =
[
new SamplingMessage
{
Role = Role.User,
Content = [new TextContentBlock { Text = "What's the weather?" }]
}
],
Tools =
[
new Tool
{
Name = "get_weather",
Description = "Get weather for a city",
InputSchema = JsonElement.Parse("""
{
"type": "object",
"properties": { "city": { "type": "string" } },
"required": ["city"]
}
""")
}
],
ToolChoice = new ToolChoice { Mode = "required" }
};
var json = JsonSerializer.Serialize(requestParams, McpJsonUtilities.DefaultOptions);
var deserialized = JsonSerializer.Deserialize<CreateMessageRequestParams>(json, McpJsonUtilities.DefaultOptions);
Assert.NotNull(deserialized);
Assert.Equal(1000, deserialized.MaxTokens);
Assert.NotNull(deserialized.Messages);
Assert.Single(deserialized.Messages);
Assert.Equal(Role.User, deserialized.Messages[0].Role);
Assert.Single(deserialized.Messages[0].Content);
var textContent = Assert.IsType<TextContentBlock>(deserialized.Messages[0].Content[0]);
Assert.Equal("What's the weather?", textContent.Text);
Assert.NotNull(deserialized.Tools);
Assert.Single(deserialized.Tools);
Assert.Equal("get_weather", deserialized.Tools[0].Name);
Assert.Equal("Get weather for a city", deserialized.Tools[0].Description);
Assert.Equal("object", deserialized.Tools[0].InputSchema.GetProperty("type").GetString());
Assert.NotNull(deserialized.ToolChoice);
Assert.Equal("required", deserialized.ToolChoice.Mode);
}
[Fact]
public void WithToolChoiceNone_SerializationRoundtrips()
{
CreateMessageRequestParams requestParams = new()
{
MaxTokens = 1000,
Messages =
[
new SamplingMessage
{
Role = Role.User,
Content = [new TextContentBlock { Text = "What's the weather in Paris?" }]
}
],
Tools =
[
new Tool
{
Name = "get_weather",
Description = "Get weather for a city",
InputSchema = JsonElement.Parse("""
{
"type": "object",
"properties": { "city": { "type": "string" } },
"required": ["city"]
}
""")
}
],
ToolChoice = new ToolChoice { Mode = "none" }
};
var json = JsonSerializer.Serialize(requestParams, McpJsonUtilities.DefaultOptions);
var deserialized = JsonSerializer.Deserialize<CreateMessageRequestParams>(json, McpJsonUtilities.DefaultOptions);
Assert.NotNull(deserialized);
Assert.Equal(1000, deserialized.MaxTokens);
Assert.NotNull(deserialized.Messages);
Assert.Single(deserialized.Messages);
Assert.Equal(Role.User, deserialized.Messages[0].Role);
Assert.Single(deserialized.Messages[0].Content);
var textContent = Assert.IsType<TextContentBlock>(deserialized.Messages[0].Content[0]);
Assert.Equal("What's the weather in Paris?", textContent.Text);
Assert.NotNull(deserialized.Tools);
Assert.Single(deserialized.Tools);
Assert.Equal("get_weather", deserialized.Tools[0].Name);
Assert.Equal("Get weather for a city", deserialized.Tools[0].Description);
Assert.Equal("object", deserialized.Tools[0].InputSchema.GetProperty("type").GetString());
Assert.NotNull(deserialized.ToolChoice);
Assert.Equal("none", deserialized.ToolChoice.Mode);
}
}