mcp

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

type ConfigLoader struct{}

type JSONRPCError struct {
	Code    int             `json:"code"`
	Message string          `json:"message"`
	Data    json.RawMessage `json:"data,omitempty"`
}

type JSONRPCMessage struct {
	ID     interface{}      `json:"id"`
	Method string           `json:"method"`
	Params *json.RawMessage `json:"params,omitempty"`
}

type JSONRPCRequest struct {
	JSONRPC string          `json:"jsonrpc"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
	ID      interface{}     `json:"id,omitempty"`
}

type JSONRPCResponse struct {
	JSONRPC string          `json:"jsonrpc"`
	Result  json.RawMessage `json:"result,omitempty"`
	Error   *JSONRPCError   `json:"error,omitempty"`
	ID      interface{}     `json:"id,omitempty"`
}

type MCPMessage struct {
	Method string          `json:"method"`
	Params json.RawMessage `json:"params,omitempty"`
}

type MCPServer struct {
	// Server configuration
	Port       int
	Path       string
	AuthToken  string
	EnableCORS bool

	// Handler for MCP requests
	Handler MCPServerHandler
	// contains filtered or unexported fields
}

type MCPServerHandler interface {
	// HandleToolCall handles a tool call request
	HandleToolCall(ctx context.Context, tool string, args map[string]interface{}) (json.RawMessage, error)

	// HandleResourceRead reads a resource
	HandleResourceRead(ctx context.Context, uri string) (json.RawMessage, error)

	// HandlePromptGet retrieves a prompt
	HandlePromptGet(ctx context.Context, name string, args map[string]interface{}) (string, error)

	// ListTools returns available tools
	ListTools() []Tool

	// ListResources returns available resources
	ListResources() []Resource

	// ListPrompts returns available prompts
	ListPrompts() []Prompt

	// HandleSamplingRequest handles LLM sampling requests from MCP clients
	HandleSamplingRequest(ctx context.Context, req *SamplingRequest) (*SamplingResponse, error)
}

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

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

type Prompt struct {
	Name        string           `json:"name"`
	Description string           `json:"description,omitempty"`
	Arguments   []PromptArgument `json:"arguments,omitempty"`
}

type PromptArgument struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Required    bool   `json:"required,omitempty"`
}

type Resource struct {
	URI         string `json:"uri"`
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	MimeType    string `json:"mimeType,omitempty"`
}

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

type SamplingMessage struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type SamplingRequest struct {
	Method       string            `json:"method,omitempty"`
	Messages     []SamplingMessage `json:"messages"`
	MaxTokens    int               `json:"maxTokens,omitempty"`
	Temperature  float64           `json:"temperature,omitempty"`
	SystemPrompt string            `json:"systemPrompt,omitempty"`
	Tools        []Tool            `json:"tools,omitempty"`
}

type SamplingResponse struct {
	Content string `json:"content"`
	Stopped string `json:"stoppedReason,omitempty"`
}

type ServerConfig struct {
	Command   string   `json:"command"`
	Args      []string `json:"args"`
	Env       []string `json:"env,omitempty"`
	Transport string   `json:"transport"`     // "stdio" or "sse"
	URL       string   `json:"url,omitempty"` // for SSE transport
}

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

type Tool struct {
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	InputSchema map[string]interface{} `json:"inputSchema"`
}

type Transport interface {
	Send(ctx context.Context, req *JSONRPCRequest) (*JSONRPCResponse, error)
	Close() error
}