38 lines
1.1 KiB
C#
38 lines
1.1 KiB
C#
using BotSharp.Abstraction.Functions.Models;
|
|
using ModelContextProtocol.Client;
|
|
using ModelContextProtocol.Protocol.Types;
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Text.Json;
|
|
|
|
namespace BotSharp.Core.MCP;
|
|
|
|
internal static class AIFunctionUtilities
|
|
{
|
|
public static FunctionDef MapToFunctionDef(McpClientTool tool)
|
|
{
|
|
if (tool == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(tool));
|
|
}
|
|
|
|
var properties = tool.JsonSchema.GetProperty("properties");
|
|
var required = tool.JsonSchema.GetProperty("required");
|
|
|
|
FunctionDef funDef = new FunctionDef
|
|
{
|
|
Name = tool.Name,
|
|
Description = tool.Description ?? string.Empty,
|
|
Type = "function",
|
|
Parameters = new FunctionParametersDef
|
|
{
|
|
Type = "object",
|
|
Properties = JsonDocument.Parse(properties.GetRawText()),
|
|
Required = JsonSerializer.Deserialize<List<string>>(required.GetRawText())
|
|
}
|
|
};
|
|
|
|
return funDef;
|
|
}
|
|
}
|