wehub-resource-sync
57 lines
2.3 KiB
Go
57 lines
2.3 KiB
Go
package mcp
|
|
|
|
import (
|
|
"github.com/mudler/LocalAI/core/config"
|
|
"github.com/mudler/LocalAI/pkg/functions"
|
|
)
|
|
|
|
// MCPToolRequest is the NATS request-reply payload for executing an MCP tool
|
|
// on an agent worker. The frontend serializes the model's MCP server config
|
|
// so the worker can create sessions and execute the tool.
|
|
type MCPToolRequest struct {
|
|
ModelName string `json:"model_name"`
|
|
ToolName string `json:"tool_name"`
|
|
Arguments map[string]any `json:"arguments,omitempty"`
|
|
RemoteServers config.MCPGenericConfig[config.MCPRemoteServers] `json:"remote_servers"`
|
|
StdioServers config.MCPGenericConfig[config.MCPSTDIOServers] `json:"stdio_servers"`
|
|
}
|
|
|
|
// MCPToolResponse is the NATS reply for an MCP tool execution.
|
|
type MCPToolResponse struct {
|
|
Result string `json:"result,omitempty"`
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
// MCPDiscoveryRequest is the NATS request-reply payload for discovering
|
|
// available MCP tools, prompts, and resources from a model's MCP servers.
|
|
type MCPDiscoveryRequest struct {
|
|
ModelName string `json:"model_name"`
|
|
RemoteServers config.MCPGenericConfig[config.MCPRemoteServers] `json:"remote_servers"`
|
|
StdioServers config.MCPGenericConfig[config.MCPSTDIOServers] `json:"stdio_servers"`
|
|
}
|
|
|
|
// MCPDiscoveryResponse is the NATS reply for an MCP discovery request.
|
|
type MCPDiscoveryResponse struct {
|
|
Servers []MCPServerInfo `json:"servers,omitempty"`
|
|
Tools []MCPToolDef `json:"tools,omitempty"` // flattened tool list with functions
|
|
Error string `json:"error,omitempty"`
|
|
}
|
|
|
|
// MCPServerInfo describes an MCP server and its available capabilities.
|
|
type MCPServerInfo struct {
|
|
Name string `json:"name"`
|
|
Type string `json:"type"`
|
|
Tools []string `json:"tools"`
|
|
Prompts []string `json:"prompts,omitempty"`
|
|
Resources []string `json:"resources,omitempty"`
|
|
}
|
|
|
|
// MCPToolDef is a serializable tool definition (function schema) that can
|
|
// travel over NATS. Unlike MCPToolInfo which holds a live session pointer,
|
|
// this is pure data.
|
|
type MCPToolDef struct {
|
|
ServerName string `json:"server_name"`
|
|
ToolName string `json:"tool_name"`
|
|
Function functions.Function `json:"function"`
|
|
}
|