mcp

type AgentTool struct {
	Type     string `json:"type"`
	Function struct {
		Name        string      `json:"name"`
		Description string      `json:"description"`
		Parameters  interface{} `json:"parameters"`
	} `json:"function"`
}

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

type EnvVarTemplate struct {
	Name        string `json:"name"`
	Description string `json:"description"`
	Required    bool   `json:"required"`
	Secret      bool   `json:"secret"`
	Default     string `json:"default"`
}

type InvalidArgsError struct {
	// Tool is the name of the tool that failed validation
	Tool string
	// Server is the MCP server name hosting the tool
	Server string
	// Failures contains the individual validation errors
	Failures []ValidationFailure
	// contains filtered or unexported fields
}

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

type MCPConfig struct {
	Enabled      bool                       `json:"enabled"`
	Servers      map[string]MCPServerConfig `json:"servers"`
	AutoStart    bool                       `json:"auto_start"`
	AutoDiscover bool                       `json:"auto_discover"`
	Timeout      time.Duration              `json:"timeout"`
}

type MCPContent struct {
	Type        string                 `json:"type"`
	Text        string                 `json:"text,omitempty"`
	Data        string                 `json:"data,omitempty"`
	MimeType    string                 `json:"mimeType,omitempty"`
	Annotations map[string]interface{} `json:"annotations,omitempty"`
}

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

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

type MCPManager interface {
	// AddServer adds a new MCP server
	AddServer(config MCPServerConfig) error

	// RemoveServer removes an MCP server
	RemoveServer(name string) error

	// GetServer gets an MCP server by name
	GetServer(name string) (MCPServer, bool)

	// ListServers lists all registered servers
	ListServers() []MCPServer

	// StartAll starts all configured servers
	StartAll(ctx context.Context) error

	// StopAll stops all running servers
	StopAll(ctx context.Context) error

	// GetAllTools gets all tools from all running servers
	GetAllTools(ctx context.Context) ([]MCPTool, error)

	// CallTool calls a tool on the appropriate server
	CallTool(ctx context.Context, serverName, toolName string, args map[string]interface{}) (*MCPToolCallResult, error)
}

type MCPMessage struct {
	JSONRPC string      `json:"jsonrpc"`
	ID      interface{} `json:"id,omitempty"`
	Method  string      `json:"method,omitempty"`
	Params  interface{} `json:"params,omitempty"`
	Result  interface{} `json:"result,omitempty"`
	Error   *MCPError   `json:"error,omitempty"`
}

type MCPPrompt struct {
	Name        string              `json:"name"`
	Description string              `json:"description,omitempty"`
	Arguments   []MCPPromptArgument `json:"arguments,omitempty"`
	ServerName  string              `json:"server_name"`
}

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

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

type MCPServer interface {
	// Start starts the MCP server process
	Start(ctx context.Context) error

	// Stop stops the MCP server process
	Stop(ctx context.Context) error

	// IsRunning checks if the server is running
	IsRunning() bool

	// GetName returns the server name
	GetName() string

	// GetConfig returns the server configuration
	GetConfig() MCPServerConfig

	// Initialize sends initialize request to server
	Initialize(ctx context.Context) error

	// ListTools lists available tools from the server
	ListTools(ctx context.Context) ([]MCPTool, error)

	// CallTool calls a tool on the server
	CallTool(ctx context.Context, request MCPToolCallRequest) (*MCPToolCallResult, error)

	// ListResources lists available resources from the server
	ListResources(ctx context.Context) ([]MCPResource, error)

	// ReadResource reads a resource from the server
	ReadResource(ctx context.Context, uri string) (*MCPContent, error)

	// ListPrompts lists available prompts from the server
	ListPrompts(ctx context.Context) ([]MCPPrompt, error)

	// GetPrompt gets a prompt from the server
	GetPrompt(ctx context.Context, name string, args map[string]interface{}) (*MCPContent, error)
}

type MCPServerConfig struct {
	Name        string            `json:"name"`
	Type        string            `json:"type,omitempty"`        // "stdio" or "http"
	Command     string            `json:"command,omitempty"`     // For stdio servers
	Args        []string          `json:"args,omitempty"`        // For stdio servers
	URL         string            `json:"url,omitempty"`         // For HTTP servers
	Env         map[string]string `json:"env,omitempty"`         // Non-secret environment variables
	Credentials map[string]string `json:"credentials,omitempty"` // Secret credentials (env var name -> placeholder)
	WorkingDir  string            `json:"working_dir,omitempty"` // For stdio servers
	Timeout     time.Duration     `json:"timeout,omitempty"`
	AutoStart   bool              `json:"auto_start"`
	MaxRestarts int               `json:"max_restarts"`
}

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

type MCPServerTemplate struct {
	ID          string           `json:"id"`
	Name        string           `json:"name"`
	Description string           `json:"description"`
	Type        string           `json:"type"` // "stdio", "http"
	URL         string           `json:"url"`  // For HTTP servers
	Command     string           `json:"command"`
	Args        []string         `json:"args"`
	EnvVars     []EnvVarTemplate `json:"env_vars"`
	Timeout     time.Duration    `json:"timeout"`
	Features    []string         `json:"features"`
	AuthType    string           `json:"auth_type"`
	Docs        string           `json:"docs"`
}

type MCPTemplatesConfig struct {
	Templates map[string]MCPServerTemplate `json:"templates"`
}

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

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

type MCPToolCallResult struct {
	Content []MCPContent `json:"content"`
	IsError bool         `json:"isError"`
}

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

type Parameters struct {
	Args   []string
	Kwargs map[string]interface{}
}

type Result struct {
	Success       bool                   `json:"success"`
	Output        interface{}            `json:"output"`
	Errors        []string               `json:"errors"`
	Metadata      map[string]interface{} `json:"metadata"`
	ExecutionTime time.Duration          `json:"execution_time"`
}

type ValidationFailure struct {
	// Path is the JSON pointer path to the failing field (e.g., ".query", ".filters[0]")
	Path string
	// Reason is a human-readable explanation of why validation failed
	Reason string
}