protocol

type Annotation struct {
	Audience     []Role  `json:"audience,omitempty"`     // Target audience (user, assistant)
	Priority     float64 `json:"priority,omitempty"`     // Priority (0.0-1.0)
	LastModified string  `json:"lastModified,omitempty"` // Last modified time (ISO 8601)
}

type AudioContent struct {
	Type        ContentType `json:"type"`
	Data        string      `json:"data"`     // Base64 encoded audio data
	MimeType    string      `json:"mimeType"` // e.g., audio/wav, audio/mp3
	Annotations *Annotation `json:"annotations,omitempty"`
}

type CallToolParams struct {
	Meta      map[string]any `json:"_meta,omitempty"`
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments"`
	Task      *TaskMetadata  `json:"task,omitempty"` // MCP 2025-11-25: Task metadata for task-augmented requests
}

type CallToolRequest struct {
	Name      string         `json:"name"`
	Arguments map[string]any `json:"arguments,omitempty"`
	Task      *TaskMetadata  `json:"task,omitempty"` // MCP 2025-11-25: Task metadata for task-augmented requests
}

type CallToolResult struct {
	Content           []Content      `json:"content"`
	IsError           bool           `json:"isError,omitempty"`
	StructuredContent any            `json:"structuredContent,omitempty"` // MCP 2025-06-18
	Meta              map[string]any `json:"_meta,omitempty"`             // MCP 2025-06-18: Extended metadata
}

type CancelTaskParams struct {
	Meta   map[string]any `json:"_meta,omitempty"`
	TaskID string         `json:"taskId"`
	Reason string         `json:"reason,omitempty"`
}

type CancelTaskResult struct {
	Task
}

type CancelledNotificationParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	// Request ID to cancel
	RequestID any `json:"requestId"`
	// Optional cancellation reason description
	Reason string `json:"reason,omitempty"`
}

type ClientCapabilities struct {
	Roots        *RootsCapability       `json:"roots,omitempty"`
	Sampling     *SamplingCapability    `json:"sampling,omitempty"`
	Elicitation  *ElicitationCapability `json:"elicitation,omitempty"`
	Tasks        *ClientTasksCapability `json:"tasks,omitempty"` // MCP 2025-11-25
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

type ClientInfo struct {
	Name       string `json:"name"`
	Title      string `json:"title,omitempty"`
	Version    string `json:"version"`
	WebsiteURL string `json:"websiteUrl,omitempty"`
	Icons      []Icon `json:"icons,omitempty"`
}

type ClientTaskRequestsCapability struct {
	// Sampling specifies task support for sampling-related requests
	Sampling *SamplingTaskCapability `json:"sampling,omitempty"`
	// Elicitation specifies task support for elicitation-related requests
	Elicitation *ElicitationTaskCapability `json:"elicitation,omitempty"`
}

type ClientTasksCapability struct {
	// List indicates client supports the tasks/list operation
	List *struct{} `json:"list,omitempty"`
	// Cancel indicates client supports the tasks/cancel operation
	Cancel *struct{} `json:"cancel,omitempty"`
	// Requests specifies which request types support task augmentation
	Requests *ClientTaskRequestsCapability `json:"requests,omitempty"`
}

type CompleteRequest struct {
	Ref      map[string]any     `json:"ref"`               // Reference (PromptReference or ResourceReference)
	Argument CompletionArgument `json:"argument"`          // Argument to complete
	Context  *CompletionContext `json:"context,omitempty"` // Optional context
}

type CompleteResult struct {
	Completion CompletionResult `json:"completion"` // Completion result
}

type CompletionArgument struct {
	Name  string `json:"name"`
	Value string `json:"value"`
}

type CompletionCapability struct{}

type CompletionContext struct {
	Arguments map[string]string `json:"arguments,omitempty"`
}

type CompletionReference interface {
	GetType() ReferenceType
}

type CompletionResult struct {
	Values  []string `json:"values"`          // Completion suggestions (max 100)
	Total   *int     `json:"total,omitempty"` // Optional: total matches count
	HasMore bool     `json:"hasMore"`         // Whether there are more results
}

type Content interface {
	GetType() ContentType
}

type ContentBlock struct {
	Type     ContentType `json:"type"`
	Text     string      `json:"text,omitempty"`
	Data     string      `json:"data,omitempty"`
	MimeType string      `json:"mimeType,omitempty"`
}

type ContentType string

type CreateMessageParams = CreateMessageRequest

type CreateMessageRequest struct {
	Meta             map[string]any         `json:"_meta,omitempty"`
	Messages         []SamplingMessage      `json:"messages"`
	ModelPreferences *ModelPreferences      `json:"modelPreferences,omitempty"`
	SystemPrompt     string                 `json:"systemPrompt,omitempty"`
	IncludeContext   IncludeContext         `json:"includeContext,omitempty"`
	Temperature      *float64               `json:"temperature,omitempty"` // 0.0-1.0
	MaxTokens        int                    `json:"maxTokens"`             // Required
	StopSequences    []string               `json:"stopSequences,omitempty"`
	Metadata         map[string]interface{} `json:"metadata,omitempty"`
	// Tools available for the LLM to use (MCP 2025-11-25)
	Tools []SamplingTool `json:"tools,omitempty"`
	// ToolChoice controls tool selection behavior (MCP 2025-11-25)
	ToolChoice *ToolChoice `json:"toolChoice,omitempty"`
	// Task metadata for task-augmented requests (MCP 2025-11-25)
	Task *TaskMetadata `json:"task,omitempty"`
}

type CreateMessageResult struct {
	Role       Role       `json:"role"`
	Content    Content    `json:"content"`
	Model      string     `json:"model"`
	StopReason StopReason `json:"stopReason"`
}

type CreateTaskResult struct {
	Meta map[string]any `json:"_meta,omitempty"`
	Task Task           `json:"task"`
}

type ElicitationAction string

type ElicitationCapability struct{}

type ElicitationCompleteNotificationParams struct {
	ElicitationID string `json:"elicitationId"`
}

type ElicitationCreateParams struct {
	Message         string     `json:"message"`
	RequestedSchema JSONSchema `json:"requestedSchema"`
}

type ElicitationResult struct {
	Action  ElicitationAction `json:"action"`
	Content interface{}       `json:"content,omitempty"`
}

type ElicitationTaskCapability struct {
	// Create indicates client supports task-augmented elicitation/create requests
	Create *struct{} `json:"create,omitempty"`
}

type EmbeddedResourceContent struct {
	Type     ContentType      `json:"type"`
	Resource ResourceContents `json:"resource"`
}

type EmptyResult struct{}

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

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

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

type GetTaskParams struct {
	Meta   map[string]any `json:"_meta,omitempty"`
	TaskID string         `json:"taskId"`
}

type GetTaskResult struct {
	Task
}

type Icon struct {
	// Source is the URI pointing to the icon resource (required), can be:
	// - HTTP/HTTPS URL pointing to an image file
	// - data URI containing base64 encoded image data
	Source string `json:"src"`
	// MIMEType is an optional MIME type
	MIMEType string `json:"mimeType,omitempty"`
	// Sizes is an optional size specification (e.g., ["48x48"], ["any"] for scalable formats like SVG)
	Sizes []string `json:"sizes,omitempty"`
	// Theme is an optional theme, such as "light" or "dark"
	Theme string `json:"theme,omitempty"`
}

type ImageContent struct {
	Type        ContentType `json:"type"`
	Data        string      `json:"data"`
	MimeType    string      `json:"mimeType"`
	Annotations *Annotation `json:"annotations,omitempty"` // MCP 2025-06-18
}

type IncludeContext string

type InitializeParams struct {
	Meta            map[string]any     `json:"_meta,omitempty"`
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ClientCapabilities `json:"capabilities"`
	ClientInfo      ClientInfo         `json:"clientInfo"`
}

type InitializeResult struct {
	ProtocolVersion string             `json:"protocolVersion"`
	Capabilities    ServerCapabilities `json:"capabilities"`
	ServerInfo      ServerInfo         `json:"serverInfo"`
	Instructions    string             `json:"instructions,omitempty"`
}

type InitializedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

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

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

type JSONSchema map[string]interface{}

type ListPromptsParams struct {
	Cursor string `json:"cursor,omitempty"`
}

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

type ListPromptsResult struct {
	Prompts []Prompt `json:"prompts"`
	PaginatedResult
}

type ListResourceTemplatesParams = ListResourceTemplatesRequest

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

type ListResourceTemplatesResult struct {
	ResourceTemplates []ResourceTemplate `json:"resourceTemplates"`
	PaginatedResult
}

type ListResourcesParams struct {
	Cursor string `json:"cursor,omitempty"`
}

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

type ListResourcesResult struct {
	Resources []Resource `json:"resources"`
	PaginatedResult
}

type ListRootsParams struct {
}

type ListRootsRequest struct {
}

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

type ListTasksParams struct {
	Meta   map[string]any `json:"_meta,omitempty"`
	Cursor string         `json:"cursor,omitempty"`
}

type ListTasksResult struct {
	Meta       map[string]any `json:"_meta,omitempty"`
	Tasks      []Task         `json:"tasks"`
	NextCursor *string        `json:"nextCursor,omitempty"`
}

type ListToolsParams struct {
	Cursor string `json:"cursor,omitempty"`
}

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

type ListToolsResult struct {
	Tools []Tool `json:"tools"`
	PaginatedResult
}

type LoggingCapability struct{}

type LoggingLevel string

type LoggingMessageParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	// Data to log, such as a string message or object
	// Allows any JSON-serializable type
	Data any `json:"data"`
	// Severity level of this log message
	Level LoggingLevel `json:"level"`
	// Optional name of the logger that emitted this message
	Logger string `json:"logger,omitempty"`
}

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

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

type ModelPreferences struct {
	Hints                []ModelHint `json:"hints,omitempty"`
	CostPriority         *float64    `json:"costPriority,omitempty"`         // 0-1, cost priority
	SpeedPriority        *float64    `json:"speedPriority,omitempty"`        // 0-1, speed priority
	IntelligencePriority *float64    `json:"intelligencePriority,omitempty"` // 0-1, intelligence priority
}

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

type PaginationParams struct {
	Cursor string `json:"cursor,omitempty"`
}

type PingParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

type ProgressNotificationParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	// Progress token to associate this notification with an ongoing request
	ProgressToken any `json:"progressToken"`
	// Current progress value, should increase with each progress update
	Progress float64 `json:"progress"`
	// Total progress value (if known), 0 indicates unknown
	Total float64 `json:"total,omitempty"`
	// Optional progress description message
	Message string `json:"message,omitempty"`
}

type Prompt struct {
	Name        string           `json:"name"`
	Title       string           `json:"title,omitempty"` // MCP 2025-06-18: Human-friendly title
	Description string           `json:"description,omitempty"`
	Arguments   []PromptArgument `json:"arguments,omitempty"`
	Meta        map[string]any   `json:"_meta,omitempty"` // MCP 2025-06-18: Extended metadata
}

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

type PromptListChangedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

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

type PromptReference struct {
	Type ReferenceType `json:"type"` // Must be "ref/prompt"
	Name string        `json:"name"` // Prompt name
}

type PromptsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type PromptsListChangedNotification struct{}

type ReadResourceParams struct {
	URI string `json:"uri"`
}

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

type ReadResourceResult struct {
	Contents []ResourceContents `json:"contents"`
}

type ReferenceType string

type Resource struct {
	URI         string         `json:"uri"`
	Name        string         `json:"name"`
	Description string         `json:"description,omitempty"`
	MimeType    string         `json:"mimeType,omitempty"`
	Meta        map[string]any `json:"_meta,omitempty"`
}

type ResourceContents struct {
	URI         string      `json:"uri"`
	Title       string      `json:"title,omitempty"`
	MimeType    string      `json:"mimeType,omitempty"`
	Text        string      `json:"text,omitempty"`
	Blob        string      `json:"blob,omitempty"`
	Annotations *Annotation `json:"annotations,omitempty"`
}

type ResourceLinkContent struct {
	Type        ContentType `json:"type"`
	URI         string      `json:"uri"`
	Name        string      `json:"name,omitempty"`
	Description string      `json:"description,omitempty"`
	MimeType    string      `json:"mimeType,omitempty"`
	Annotations *Annotation `json:"annotations,omitempty"`
}

type ResourceListChangedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

type ResourceReference struct {
	Type ReferenceType `json:"type"` // Must be "ref/resource"
	URI  string        `json:"uri"`  // Resource URI (may contain template variables)
}

type ResourceTemplate struct {
	URITemplate string         `json:"uriTemplate"`
	Name        string         `json:"name"`
	Description string         `json:"description,omitempty"`
	MimeType    string         `json:"mimeType,omitempty"`
	Meta        map[string]any `json:"_meta,omitempty"`
}

type ResourceTemplateListChangedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

type ResourceTemplatesListChangedNotification struct{}

type ResourceUpdatedNotificationParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	// Updated resource URI
	URI string `json:"uri"`
}

type ResourcesCapability struct {
	Subscribe   bool `json:"subscribe,omitempty"`
	ListChanged bool `json:"listChanged,omitempty"`
	Templates   bool `json:"templates,omitempty"`
}

type ResourcesListChangedNotification struct{}

type Role string

type Root struct {
	URI  string `json:"uri"`            // Root directory URI, must use file:// protocol
	Name string `json:"name,omitempty"` // Optional human-readable name
}

type RootsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type RootsListChangedNotification struct{}

type RootsListChangedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

type SamplingCapability struct {
	// Tools indicates client supports tool use in sampling requests (MCP 2025-11-25)
	Tools *struct{} `json:"tools,omitempty"`
}

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

type SamplingTaskCapability struct {
	// CreateMessage indicates client supports task-augmented sampling/createMessage requests
	CreateMessage *struct{} `json:"createMessage,omitempty"`
}

type SamplingTool struct {
	Name        string     `json:"name"`
	Description string     `json:"description,omitempty"`
	InputSchema JSONSchema `json:"inputSchema"`
}

type ServerCapabilities struct {
	Tools        *ToolsCapability       `json:"tools,omitempty"`
	Resources    *ResourcesCapability   `json:"resources,omitempty"`
	Prompts      *PromptsCapability     `json:"prompts,omitempty"`
	Logging      *LoggingCapability     `json:"logging,omitempty"`
	Completion   *CompletionCapability  `json:"completion,omitempty"` // MCP 2025-06-18: Parameter auto-completion
	Tasks        *TasksCapability       `json:"tasks,omitempty"`      // MCP 2025-11-25
	Experimental map[string]interface{} `json:"experimental,omitempty"`
}

type ServerInfo struct {
	Name       string `json:"name"`
	Title      string `json:"title,omitempty"`
	Version    string `json:"version"`
	WebsiteURL string `json:"websiteUrl,omitempty"`
	Icons      []Icon `json:"icons,omitempty"`
}

type ServerTaskRequestsCapability struct {
	// Tools specifies task support for tool-related requests
	Tools *ToolsTaskCapability `json:"tools,omitempty"`
}

type SetLoggingLevelParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	// The log level the client wishes to receive from the server
	// The server should send all logs at this level and higher (i.e., more severe) to the client
	Level LoggingLevel `json:"level"`
}

type StopReason string

type SubscribeParams struct {
	URI string `json:"uri"`
}

type Task struct {
	// TaskID is the unique identifier for this task
	TaskID string `json:"taskId"`
	// Status is the current state of the task execution
	Status TaskStatus `json:"status"`
	// StatusMessage is an optional human-readable message providing additional details
	StatusMessage string `json:"statusMessage,omitempty"`
	// CreatedAt is the ISO 8601 timestamp when the task was created
	CreatedAt string `json:"createdAt"`
	// LastUpdatedAt is the ISO 8601 timestamp when the task status was last updated
	LastUpdatedAt string `json:"lastUpdatedAt"`
	// TTL is the time-to-live in milliseconds from creation before task may be deleted
	// null indicates no TTL is set
	TTL *int `json:"ttl"`
	// PollInterval is the suggested time in milliseconds between status checks
	PollInterval *int `json:"pollInterval,omitempty"`
}

type TaskMetadata struct {
	// TTL specifies the retention duration of a task in milliseconds
	TTL *int `json:"ttl,omitempty"`
}

type TaskResultParams struct {
	Meta   map[string]any `json:"_meta,omitempty"`
	TaskID string         `json:"taskId"`
}

type TaskStatus string

type TaskStatusNotificationParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
	Task
}

type TaskSupport string

type TasksCapability struct {
	// List indicates server supports the tasks/list operation
	List *struct{} `json:"list,omitempty"`
	// Cancel indicates server supports the tasks/cancel operation
	Cancel *struct{} `json:"cancel,omitempty"`
	// Requests specifies which request types support task augmentation
	Requests *ServerTaskRequestsCapability `json:"requests,omitempty"`
}

type TextContent struct {
	Type        ContentType `json:"type"`
	Text        string      `json:"text"`
	Annotations *Annotation `json:"annotations,omitempty"` // MCP 2025-06-18
}

type Tool struct {
	Name         string         `json:"name"`
	Title        string         `json:"title,omitempty"` // MCP 2025-06-18: Human-friendly title
	Description  string         `json:"description,omitempty"`
	InputSchema  JSONSchema     `json:"inputSchema"`
	OutputSchema JSONSchema     `json:"outputSchema,omitempty"` // MCP 2025-06-18
	Execution    *ToolExecution `json:"execution,omitempty"`    // MCP 2025-11-25: Execution behavior
	Meta         map[string]any `json:"_meta,omitempty"`        // MCP 2025-06-18: Extended metadata
}

type ToolChoice struct {
	Mode ToolChoiceMode `json:"mode,omitempty"`
}

type ToolChoiceMode string

type ToolExecution struct {
	// TaskSupport indicates the level of task support for this tool
	// Can be "required", "optional", or "forbidden"
	TaskSupport TaskSupport `json:"taskSupport,omitempty"`
}

type ToolList struct {
	Tools []Tool `json:"tools"`
}

type ToolListChangedParams struct {
	Meta map[string]any `json:"_meta,omitempty"`
}

type ToolParameter struct {
	Name        string     `json:"name"`
	Description string     `json:"description,omitempty"`
	Required    bool       `json:"required,omitempty"`
	Schema      JSONSchema `json:"schema,omitempty"`
}

type ToolResultContent struct {
	Type              ContentType            `json:"type"`
	ToolUseID         string                 `json:"toolUseId"`
	Content           []ContentBlock         `json:"content"`
	IsError           bool                   `json:"isError,omitempty"`
	StructuredContent map[string]interface{} `json:"structuredContent,omitempty"`
	Meta              map[string]any         `json:"_meta,omitempty"`
}

type ToolUseContent struct {
	Type  ContentType            `json:"type"`
	ID    string                 `json:"id"`
	Name  string                 `json:"name"`
	Input map[string]interface{} `json:"input"`
	Meta  map[string]any         `json:"_meta,omitempty"`
}

type ToolsCapability struct {
	ListChanged bool `json:"listChanged,omitempty"`
}

type ToolsListChangedNotification struct{}

type ToolsTaskCapability struct {
	// Call indicates server supports task-augmented tools/call requests
	Call *struct{} `json:"call,omitempty"`
}

type UnsubscribeParams struct {
	URI string `json:"uri"`
}