types

type BlobResourceContents struct {
	ResourceContents
	Blob string `json:"blob"` // base64-encoded data
}

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

type CallToolResult struct {
	Content []interface{} `json:"content"` // Can be TextContent, ImageContent, or EmbeddedResource
	IsError bool          `json:"isError,omitempty"`
}

type ClientCapabilities struct {
	// Experimental features support
	Experimental map[string]interface{} `json:"experimental,omitempty"`

	// Roots capability
	Roots *RootsClientCapabilities `json:"roots,omitempty"`

	// Sampling capability
	Sampling *SamplingClientCapabilities `json:"sampling,omitempty"`
}

type CreateMessageRequest struct {
	Method           string            `json:"method"`
	Messages         []SamplingMessage `json:"messages"`
	ModelPreferences *ModelPreferences `json:"modelPreferences,omitempty"`
	SystemPrompt     string            `json:"systemPrompt,omitempty"`
	IncludeContext   string            `json:"includeContext,omitempty"`
	Temperature      float64           `json:"temperature,omitempty"`
	MaxTokens        int               `json:"maxTokens"`
	StopSequences    []string          `json:"stopSequences,omitempty"`
	Metadata         interface{}       `json:"metadata,omitempty"`
}

type CreateMessageResult struct {
	Role       Role           `json:"role"`
	Content    MessageContent `json:"content"` // Using the same MessageContent interface from prompts
	Model      string         `json:"model"`
	StopReason string         `json:"stopReason,omitempty"`
}

type Cursor string

type EmbeddedResource struct {
	Type     string           `json:"type"`
	Resource ResourceContents `json:"resource"`
}

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

type GetPromptRequest struct {
	Method    string            `json:"method"`
	Name      string            `json:"name"`
	Arguments map[string]string `json:"arguments,omitempty"`
}

type GetPromptResult struct {
	Description string          `json:"description,omitempty"`
	Messages    []PromptMessage `json:"messages"`
}

type ID = jsonrpc2.ID // This is typically a string or number

type ImageContent struct {
	Type     string `json:"type"`
	Data     string `json:"data"` // base64-encoded
	MimeType string `json:"mimeType"`
}

type Implementation struct {
	Name    string `json:"name"`
	Version string `json:"version"`
}

type InitializeRequest struct {
	// The latest version of MCP that the client supports
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ClientCapabilities `json:"capabilities"`
	ClientInfo      Implementation     `json:"clientInfo"`
}

type InitializeResult struct {
	// The version of MCP that the server will use
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ServerCapabilities `json:"capabilities"`
	ServerInfo      Implementation     `json:"serverInfo"`
	// Optional instructions for using the server
	Instructions string `json:"instructions,omitempty"`
}

type InitializedNotification struct {
	Method string `json:"method"`
}

type ListPromptsRequest struct {
	Method string  `json:"method"`
	Cursor *Cursor `json:"cursor,omitempty"`
}

type ListPromptsResult struct {
	Prompts    []Prompt `json:"prompts"`
	NextCursor *Cursor  `json:"nextCursor,omitempty"`
}

type ListResourceTemplatesRequest struct {
	Method string  `json:"method"`
	Cursor *Cursor `json:"cursor,omitempty"`
}

type ListResourceTemplatesResult struct {
	ResourceTemplates []ResourceTemplate `json:"resourceTemplates"`
	NextCursor        *Cursor            `json:"nextCursor,omitempty"`
}

type ListResourcesRequest struct {
	Method string  `json:"method"`
	Cursor *Cursor `json:"cursor,omitempty"`
}

type ListResourcesResult struct {
	Resources  []Resource `json:"resources"`
	NextCursor *Cursor    `json:"nextCursor,omitempty"`
}

type ListRootsRequest struct {
	Method string `json:"method"`
}

type ListRootsResult struct {
	Roots []Root `json:"roots"`
}

type ListToolsRequest struct {
	Method string  `json:"method"`
	Cursor *Cursor `json:"cursor,omitempty"`
}

type ListToolsResult struct {
	Tools      []Tool  `json:"tools"`
	NextCursor *Cursor `json:"nextCursor,omitempty"`
}

type LoggingServerCapabilities struct {
}

type McpTool interface {
	GetName() string
	GetDescription() string
	GetDefinition() Tool
	GetHandler() ToolHandler
}

type Message struct {
	JSONRPC string           `json:"jsonrpc"`
	ID      *ID              `json:"id,omitempty"`
	Method  string           `json:"method,omitempty"`
	Params  *json.RawMessage `json:"params,omitempty"`
	Result  *json.RawMessage `json:"result,omitempty"`
	Error   *ErrorResponse   `json:"error,omitempty"`
}

type MessageContent interface {
	// contains filtered or unexported methods
}

type ModelHint struct {
	Name string `json:"name,omitempty"`
}

type ModelPreferences struct {
	// Optional hints for model selection
	Hints []ModelHint `json:"hints,omitempty"`

	// Priority values between 0 and 1
	CostPriority         float64 `json:"costPriority,omitempty"`
	SpeedPriority        float64 `json:"speedPriority,omitempty"`
	IntelligencePriority float64 `json:"intelligencePriority,omitempty"`
}

type Notification struct {
	Method string            `json:"method"`
	Params *json.RawMessage  `json:"params,omitempty"`
	Meta   *NotificationMeta `json:"_meta,omitempty"`
}

type NotificationMeta map[string]interface{}

type PaginatedRequest struct {
	Cursor *Cursor `json:"cursor,omitempty"`
}

type PaginatedResult struct {
	NextCursor *Cursor `json:"nextCursor,omitempty"`
}

type ProgressToken interface{} // string or number

type Prompt struct {
	// Name uniquely identifies the prompt
	Name string `json:"name"`

	// Optional description
	Description string `json:"description,omitempty"`

	// Optional arguments for templating
	Arguments []PromptArgument `json:"arguments,omitempty"`
}

type PromptArgument struct {
	// Name of the argument
	Name string `json:"name"`

	// Optional description
	Description string `json:"description,omitempty"`

	// Whether this argument is required
	Required bool `json:"required,omitempty"`
}

type PromptListChangedNotification struct {
	Method string `json:"method"`
}

type PromptMessage struct {
	Role    Role           `json:"role"`
	Content MessageContent `json:"content"`
}

type PromptsServerCapabilities struct {
	// Whether the server supports notifications for changes to the prompt list
	ListChanged bool `json:"listChanged,omitempty"`
}

type ReadResourceRequest struct {
	Method string `json:"method"`
	URI    string `json:"uri"`
}

type ReadResourceResult struct {
	Contents []ResourceContent `json:"contents"` // Can be TextResourceContents or BlobResourceContents
}

type Request struct {
	Method string           `json:"method"`
	Params *json.RawMessage `json:"params,omitempty"`
	Meta   *RequestMeta     `json:"_meta,omitempty"`
}

type RequestMeta struct {
	ProgressToken ProgressToken `json:"progressToken,omitempty"`
}

type Resource struct {
	// URI identifying this resource
	URI string `json:"uri"`

	// Human-readable name
	Name string `json:"name"`

	// Optional description
	Description string `json:"description,omitempty"`

	// Optional MIME type
	MimeType string `json:"mimeType,omitempty"`
}

type ResourceContent interface {
	// contains filtered or unexported methods
}

type ResourceContents struct {
	// URI identifying this resource
	URI string `json:"uri"`

	// Optional MIME type
	MimeType string `json:"mimeType,omitempty"`
}

type ResourceListChangedNotification struct {
	Method string `json:"method"`
}

type ResourceTemplate struct {
	// URI template for constructing resource URIs
	URITemplate string `json:"uriTemplate"`

	// Human-readable name
	Name string `json:"name"`

	// Optional description
	Description string `json:"description,omitempty"`

	// Optional MIME type
	MimeType string `json:"mimeType,omitempty"`
}

type ResourceUpdatedNotification struct {
	Method string `json:"method"`
	URI    string `json:"uri"`
}

type ResourcesServerCapabilities struct {
	// Whether the server supports subscribing to resource updates
	Subscribe bool `json:"subscribe,omitempty"`

	// Whether the server supports notifications for changes to the resource list
	ListChanged bool `json:"listChanged,omitempty"`
}

type Result struct {
	Meta *ResultMeta `json:"_meta,omitempty"`
}

type ResultMeta map[string]interface{}

type Role string

type Root struct {
	// URI identifying the root. Must start with file:// for now
	URI string `json:"uri"`

	// Optional name for the root
	Name string `json:"name,omitempty"`
}

type RootsClientCapabilities struct {
	// Whether the client supports notifications for changes to the roots list
	ListChanged bool `json:"listChanged,omitempty"`
}

type RootsListChangedNotification struct {
	Method string `json:"method"`
}

type SamplingClientCapabilities struct {
}

type SamplingHandler func(ctx context.Context, req *CreateMessageRequest) (*CreateMessageResult, error)

type SamplingMessage struct {
	Role    Role           `json:"role"`
	Content MessageContent `json:"content"` // Using the same MessageContent interface
}

type ServerCapabilities struct {
	// Experimental features support
	Experimental map[string]interface{} `json:"experimental,omitempty"`

	// Logging capability
	Logging *LoggingServerCapabilities `json:"logging,omitempty"`

	// Prompts capability
	Prompts *PromptsServerCapabilities `json:"prompts,omitempty"`

	// Resources capability
	Resources *ResourcesServerCapabilities `json:"resources,omitempty"`

	// Tools capability
	Tools *ToolsServerCapabilities `json:"tools,omitempty"`
}

type SubscribeRequest struct {
	Method string `json:"method"`
	URI    string `json:"uri"`
}

type TextContent struct {
	Type string `json:"type"`
	Text string `json:"text"`
}

type TextResourceContents struct {
	ResourceContents
	Text string `json:"text"`
}

type Tool struct {
	// Name of the tool
	Name string `json:"name"`

	// Optional description
	Description string `json:"description,omitempty"`

	// JSON Schema defining expected parameters
	InputSchema ToolInputSchema `json:"inputSchema"`
}

type ToolHandler func(ctx context.Context, arguments map[string]interface{}) (*CallToolResult, error)

type ToolInputSchema struct {
	Type       string                 `json:"type"`
	Properties map[string]interface{} `json:"properties,omitempty"`
	Required   []string               `json:"required,omitempty"`
}

type ToolListChangedNotification struct {
	Method string `json:"method"`
}

type ToolsServerCapabilities struct {
	// Whether the server supports notifications for changes to the tool list
	ListChanged bool `json:"listChanged,omitempty"`
}

type TypedTool[T any] struct {
	// contains filtered or unexported fields
}

type TypedToolHandler[T any] func(ctx context.Context, input T) (*CallToolResult, error)

type UnsubscribeRequest struct {
	Method string `json:"method"`
	URI    string `json:"uri"`
}