Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
make mcp types internal
  • Loading branch information
christothes committed Apr 30, 2025
commit 12e8a87b56e32bf0750f62a70272ff80adde8d73
2 changes: 1 addition & 1 deletion src/Utility/ChatTools.cs
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ public ChatTools(Type tool, params Type[] additionalTools) : this((EmbeddingClie
internal override ChatTool MethodInfoToTool(MethodInfo methodInfo) =>
ChatTool.CreateFunctionTool(methodInfo.Name, GetMethodDescription(methodInfo), BuildParametersJson(methodInfo.GetParameters()));

protected override async Task Add(BinaryData toolDefinitions, McpClient client)
internal override async Task Add(BinaryData toolDefinitions, McpClient client)
{
using var document = JsonDocument.Parse(toolDefinitions);
if (!document.RootElement.TryGetProperty("tools", out JsonElement toolsElement))
Expand Down
2 changes: 1 addition & 1 deletion src/Utility/MCP/McpClient.cs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@

namespace OpenAI;

public class McpClient
internal class McpClient
{
private readonly McpSession _session;
private readonly ClientPipeline _pipeline = ClientPipeline.Create();
Expand Down
2 changes: 1 addition & 1 deletion src/Utility/ResponseTools.cs
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ public ResponseTools(Type tool, params Type[] additionalTools) : this((Embedding
internal override ResponseTool MethodInfoToTool(MethodInfo methodInfo) =>
ResponseTool.CreateFunctionTool(methodInfo.Name, GetMethodDescription(methodInfo), BuildParametersJson(methodInfo.GetParameters()));

protected override async Task Add(BinaryData toolDefinitions, McpClient client)
internal override async Task Add(BinaryData toolDefinitions, McpClient client)
{
using var document = JsonDocument.Parse(toolDefinitions);
if (!document.RootElement.TryGetProperty("tools", out JsonElement toolsElement))
Expand Down
8 changes: 4 additions & 4 deletions src/Utility/ToolsBase.cs
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,8 @@ public abstract class ToolsBase<TTool> where TTool : class
protected readonly EmbeddingClient _client;
protected readonly List<VectorbaseEntry> _entries = [];

protected readonly List<McpClient> _mcpClients = [];
protected readonly Dictionary<string, McpClient> _mcpClientsByEndpoint = [];
internal readonly List<McpClient> _mcpClients = [];
internal readonly Dictionary<string, McpClient> _mcpClientsByEndpoint = [];
protected const string _mcpToolSeparator = "_-_";

protected ToolsBase(EmbeddingClient client = null)
Expand All @@ -44,7 +44,7 @@ public void AddLocalTools(params Type[] tools)
Add(functionHolder);
}

public async Task AddMcpServerAsync(McpClient client)
internal async Task AddMcpServerAsync(McpClient client)
{
if (client == null) throw new ArgumentNullException(nameof(client));
_mcpClientsByEndpoint[client.ServerEndpoint.AbsoluteUri] = client;
Expand Down Expand Up @@ -80,7 +80,7 @@ public void Add(MethodInfo function)

internal abstract TTool MethodInfoToTool(MethodInfo methodInfo);

protected abstract Task Add(BinaryData toolDefinitions, McpClient client);
internal abstract Task Add(BinaryData toolDefinitions, McpClient client);

public string Call(string name, object[] arguments)
{
Expand Down