Microsoft.Extensions.AI.Abstractions integration (#25)

* MEAI integration * README and xmldocs
2025-02-23 13:23:56 +01:00 · 2025-02-23 13:23:56 +01:00 · 89c4913ed7
commit 89c4913ed7
parent 502a09d9fc
1 changed files with 589 additions and 1 deletions
--- a/tests/McpDotNet.Extensions.AI.Tests/AIFunctionUtilitiesTests.cs
+++ b/tests/McpDotNet.Extensions.AI.Tests/AIFunctionUtilitiesTests.cs
@ -0,0 +1,95 @@
+using McpDotNet.Protocol.Types;
+using System.Text.Json;
+
+namespace McpDotNet.Extensions.AI.Tests;
+
+public class AIFunctionUtilitiesTests
+{
+    [Fact]
+    public void MapToJsonElement_WithRequiredProperties_CreatesCorrectSchema()
+    {
+        // Arrange
+        var tool = new Tool()
+        {
+            Name = "calculator",
+            Description = "A simple calculator",
+            InputSchema = new JsonSchema
+            {
+                Properties = new Dictionary<string, JsonSchemaProperty>
+                {
+                    ["a"] = new JsonSchemaProperty { Type = "number" },
+                    ["b"] = new JsonSchemaProperty { Type = "number" }
+                },
+                Required = ["a", "b"]
+            }
+        };
+
+        // Act
+        var result = AIFunctionUtilities.MapToJsonElement(tool);
+
+        // Assert
+        var resultObj = JsonSerializer.Deserialize<JsonElement>(result.GetRawText());
+        Assert.Equal("object", resultObj.GetProperty("type").GetString());
+        Assert.Equal("calculator", resultObj.GetProperty("title").GetString());
+        Assert.Equal("A simple calculator", resultObj.GetProperty("description").GetString());
+        // Add more detailed property assertions
+    }
+
+    [Fact]
+    public void MapToJsonElement_WithOptionalProperties_CreatesCorrectSchema()
+    {
+        // Arrange
+        var tool = new Tool()
+        {
+            Name = "format",
+            Description = "Text formatter",
+            InputSchema = new JsonSchema
+            {
+                Properties = new Dictionary<string, JsonSchemaProperty>
+                {
+                    ["text"] = new JsonSchemaProperty { Type = "string" },
+                    ["style"] = new JsonSchemaProperty { Type = "string" }
+                },
+                Required = ["text"]  // only text is required
+            }
+        };
+
+        // Act
+        var result = AIFunctionUtilities.MapToJsonElement(tool);
+
+        // Assert
+        var resultObj = JsonSerializer.Deserialize<JsonElement>(result.GetRawText());
+        Assert.Equal("object", resultObj.GetProperty("type").GetString());
+        Assert.Equal("format", resultObj.GetProperty("title").GetString());
+        Assert.Equal("Text formatter", resultObj.GetProperty("description").GetString());
+        Assert.Contains("text", resultObj.GetProperty("required").EnumerateArray().Select(x => x.GetString()));
+        Assert.DoesNotContain("style", resultObj.GetProperty("required").EnumerateArray().Select(x => x.GetString()));
+    }
+
+    [Fact]
+    public void MapToJsonElement_WithNullProperties_HandlesGracefully()
+    {
+        // Arrange
+        var tool = new Tool()
+        {
+            Name = "minimal",
+            Description = "Minimal tool",
+            InputSchema = new JsonSchema
+            {
+                Properties = null,
+                Required = null
+            }
+        };
+
+        // Act
+        var result = AIFunctionUtilities.MapToJsonElement(tool);
+
+        // Assert
+        var resultObj = JsonSerializer.Deserialize<JsonElement>(result.GetRawText());
+        Assert.Equal("object", resultObj.GetProperty("type").GetString());
+        Assert.Equal("minimal", resultObj.GetProperty("title").GetString());
+        Assert.Equal("Minimal tool", resultObj.GetProperty("description").GetString());
+        Assert.True(resultObj.GetProperty("properties").GetRawText() == "{}");
+        Assert.Empty(resultObj.GetProperty("required").EnumerateArray());
+    }
+}