mcp

type CallToolParams struct {
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments,omitempty"`
}

type Client struct {
	ServerURL  string
	AuthToken  string
	HTTPClient *http.Client
}

type Error struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
	Data    any    `json:"data,omitempty"`
}

type MCPClient interface {
	// SetTimeout sets the HTTP client timeout for MCP requests
	SetTimeout(timeout time.Duration)

	// CreateTool creates a new MCP tool with schema generation from a Go struct
	CreateTool(name, description, mcpToolName string, inputSchema any) (*Tool, error)

	// CallTool calls an MCP tool with the given arguments and returns the result
	CallTool(ctx context.Context, tool *Tool, arguments map[string]any) (*ToolResult, error)

	// CallToolByName calls an MCP tool by its MCP tool name directly
	CallToolByName(ctx context.Context, mcpToolName string, arguments map[string]any) (*ToolResult, error)

	// Ping sends a ping request to the MCP server to check connectivity
	Ping(ctx context.Context) error

	// ListTools lists available tools from the MCP server
	ListTools(ctx context.Context) ([]map[string]any, error)
}

type Manager struct {
	// contains filtered or unexported fields
}

type MultiManager struct {
	// contains filtered or unexported fields
}

type Request struct {
	JSONRPC string `json:"jsonrpc"`
	ID      int    `json:"id"`
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

type Response struct {
	JSONRPC string `json:"jsonrpc"`
	ID      int    `json:"id"`
	Result  any    `json:"result,omitempty"`
	Error   *Error `json:"error,omitempty"`
}

type Tool struct {
	Name        string         `json:"name"`
	Description string         `json:"description"`
	InputSchema map[string]any `json:"inputSchema"`
	MCPToolName string         `json:"mcpToolName"` // The actual tool name in MCP server
}

type ToolManager interface {
	// AddTool adds a tool to the managed collection
	AddTool(tool *Tool) error

	// RemoveTool removes a tool from the managed collection
	RemoveTool(name string) error

	// GetTool retrieves a tool by name
	GetTool(name string) (*Tool, bool)

	// GetAllTools returns all managed tools
	GetAllTools() []*Tool

	// CallTool calls a managed tool by name
	CallTool(ctx context.Context, name string, arguments map[string]any) (*ToolResult, error)
}

type ToolResult struct {
	Content   string `json:"content"`
	IsError   bool   `json:"isError"`
	ErrorCode int    `json:"errorCode,omitempty"`
}