protocol

type Annotations struct {
	Audience []string `json:"audience,omitempty"` // Describes who the intended customer is
	Priority *float64 `json:"priority,omitempty"` // Importance of the data (0-1)
}

type AudioContent struct {
	Type        string              `json:"type"` // Should always be "audio"
	Data        string              `json:"data"`
	MediaType   string              `json:"mediaType"`
	Annotations *ContentAnnotations `json:"annotations,omitempty"`
}

type AudioResourceContents struct {
	ContentType string `json:"mimeType"` // e.g., "audio/mpeg", "audio/wav"
	Audio       string `json:"audio"`    // Base64 encoded string
	URI         string `json:"uri"`      // Required by schema
}

type BlobResourceContents struct {
	ContentType string `json:"mimeType"` // e.g., "image/png", "application/octet-stream"
	Blob        string `json:"blob"`     // Base64 encoded string
	URI         string `json:"uri"`      // Required by schema
}

type CallToolRequestParams struct {
	// V2025 format
	ToolCall *ToolCall    `json:"tool_call,omitempty"`
	Meta     *RequestMeta `json:"_meta,omitempty"`

	// V2024 format (for backward compatibility)
	Name      string          `json:"name,omitempty"`
	Arguments json.RawMessage `json:"arguments,omitempty"`
}

type CallToolResult struct {
	ToolCallID string          `json:"tool_call_id"`
	Output     json.RawMessage `json:"output,omitempty"`
	Error      *ToolError      `json:"error,omitempty"`
	Meta       *RequestMeta    `json:"_meta,omitempty"`
}

type CallToolResultV2024 struct {
	ToolCallID string       `json:"tool_call_id"`      // Assuming tool_call_id was implicitly present or intended
	Content    []Content    `json:"content"`           // Result content (SLICE of structured parts)
	IsError    bool         `json:"isError,omitempty"` // Indicates if the tool call failed
	Meta       *RequestMeta `json:"_meta,omitempty"`   // Propagate meta if needed
}

type CancelledParams struct {
	ID     interface{} `json:"id"`
	Reason *string     `json:"reason,omitempty"` // Added for 2025-03-26
}

type ClientCapabilities struct {
	Experimental map[string]interface{} `json:"experimental,omitempty"`
	Logging      *struct{}              `json:"logging,omitempty"` // Added based on ServerCapabilities
	Prompts      *struct {
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"prompts,omitempty"` // Added based on ServerCapabilities
	Resources *struct {
		Subscribe   bool `json:"subscribe,omitempty"`
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"resources,omitempty"` // Added based on ServerCapabilities
	Tools *struct {
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"tools,omitempty"` // Added based on ServerCapabilities
	Roots *struct {
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"roots,omitempty"`
	Sampling      *SamplingCapability `json:"sampling,omitempty"` // Corrected type
	Authorization *struct {
	} `json:"authorization,omitempty"`
}

type CompleteRequest struct {
	Ref      CompletionReference `json:"ref"`
	Argument CompletionArgument  `json:"argument"`
}

type CompleteResult struct {
	Completion Completion `json:"completion"`
}

type Completion struct {
	Values  []string `json:"values"`            // Array of completion suggestions (max 100).
	Total   *int     `json:"total,omitempty"`   // Optional total number of available matches.
	HasMore *bool    `json:"hasMore,omitempty"` // Optional flag indicating if more results exist beyond the returned Values.
}

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

type CompletionReference struct {
	Type ReferenceType `json:"type"`
	Name string        `json:"name,omitempty"` // Used for RefTypePrompt
	URI  string        `json:"uri,omitempty"`  // Used for RefTypeResource
}

type Content interface {
	GetType() string
}

type ContentAnnotations struct {
	Title    *string  `json:"title,omitempty"`
	Audience []string `json:"audience,omitempty"`
	Priority *float64 `json:"priority,omitempty"`
}

type ContentTypeDetect struct {
	Type     string            `json:"type,omitempty"`
	Text     string            `json:"text,omitempty"`
	Tool     json.RawMessage   `json:"tool,omitempty"`
	Data     json.RawMessage   `json:"data,omitempty"`
	Parts    []json.RawMessage `json:"parts,omitempty"`
	Function json.RawMessage   `json:"function,omitempty"`
}

type CreateMessageRequestParams struct {
	Context     []SamplingMessage `json:"context"` // Renamed to 'messages' in 2025-03-26
	Preferences *ModelPreferences `json:"preferences,omitempty"`
}

type CreateMessageRequestParamsV20250326 struct {
	Messages         []SamplingMessage          `json:"messages"`                   // Renamed from 'context'
	ModelPreferences *ModelPreferencesV20250326 `json:"modelPreferences,omitempty"` // Use new struct
	SystemPrompt     *string                    `json:"systemPrompt,omitempty"`     // Added field
	MaxTokens        *int                       `json:"maxTokens,omitempty"`        // Added field
}

type CreateMessageResult struct {
	Message   SamplingMessage `json:"message"`
	ModelHint *ModelHint      `json:"modelHint,omitempty"`
}

type CreateMessageResultV20250326 struct {
	Role       string    `json:"role"`                 // Added field (e.g., "assistant")
	Content    []Content `json:"content"`              // Added field (replaces Message.Content)
	Model      *string   `json:"model,omitempty"`      // Added field (replaces ModelHint.ModelURI)
	StopReason *string   `json:"stopReason,omitempty"` // Added field (replaces ModelHint.FinishReason)

}

type EmbeddedResourceContent struct {
	Type        string              `json:"type"` // Should always be "resource"
	Resource    Resource            `json:"resource"`
	Annotations *ContentAnnotations `json:"annotations,omitempty"`
}

type ErrorCode int

type ErrorMessage struct {
	Payload ErrorPayload `json:"error"` // Field name MUST be "error" for JSON-RPC compliance
}

type ErrorPayload struct {
	Code    ErrorCode   `json:"code"`           // Use ErrorCode type
	Message string      `json:"message"`        // Short error description
	Data    interface{} `json:"data,omitempty"` // Optional additional error details
}

type GetPromptRequestParams struct {
	URI       string                 `json:"uri"`
	Name      string                 `json:"name,omitempty"` // Added for 2025-03-26 schema
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

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

type ImageContent struct {
	Type        string              `json:"type"` // Should always be "image"
	Data        string              `json:"data"`
	MediaType   string              `json:"mediaType"`
	Annotations *ContentAnnotations `json:"annotations,omitempty"`
}

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

type InitializeRequestParams struct {
	ProtocolVersion  string             `json:"protocolVersion"`
	Capabilities     ClientCapabilities `json:"capabilities"`
	ClientInfo       Implementation     `json:"clientInfo"`
	Trace            *string            `json:"trace,omitempty"`
	WorkspaceFolders []WorkspaceFolder  `json:"workspaceFolders,omitempty"`
}

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

type InitializedNotificationParams struct{}

type JSONRPCNotification struct {
	JSONRPC string      `json:"jsonrpc"`          // MUST be "2.0"
	Method  string      `json:"method"`           // Method name (e.g., "initialized", "notifications/...")
	Params  interface{} `json:"params,omitempty"` // Parameters (struct or array)

}

type JSONRPCRequest struct {
	JSONRPC string      `json:"jsonrpc"`          // MUST be "2.0"
	ID      interface{} `json:"id"`               // Request ID (string, number, or null)
	Method  string      `json:"method"`           // Method name (e.g., "initialize", "tools/call")
	Params  interface{} `json:"params,omitempty"` // Parameters (struct or array)
}

type JSONRPCResponse struct {
	JSONRPC string        `json:"jsonrpc"`          // MUST be "2.0"
	ID      interface{}   `json:"id"`               // MUST be the same as the request ID (or null if error before ID parsing)
	Result  interface{}   `json:"result,omitempty"` // Result object (on success)
	Error   *ErrorPayload `json:"error,omitempty"`  // Error object (on failure)
}

type ListPromptsRequestParams struct {
	Filter map[string]interface{} `json:"filter,omitempty"`
	Cursor string                 `json:"cursor,omitempty"`
}

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

type ListResourceTemplatesRequestParams struct{}

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

type ListResourcesRequestParams struct {
	Filter map[string]interface{} `json:"filter,omitempty"`
	Cursor string                 `json:"cursor,omitempty"`
}

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

type ListRootsRequestParams struct{}

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

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

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

type LoggingLevel string

type LoggingMessageParams struct {
	Level   LoggingLevel `json:"level"`
	Message string       `json:"message,omitempty"` // Kept for 2024-11-05 compatibility
	Logger  *string      `json:"logger,omitempty"`  // Added for 2025-03-26
	Data    interface{}  `json:"data,omitempty"`    // Added for 2025-03-26 (JSON object)
}

type MCPError struct {
	ErrorPayload
}

type ModelHint struct {
	ModelURI     string  `json:"modelUri"`
	InputTokens  *int    `json:"inputTokens,omitempty"`
	OutputTokens *int    `json:"outputTokens,omitempty"`
	FinishReason *string `json:"finishReason,omitempty"`
}

type ModelHintV20250326 struct {
	Name string `json:"name"`
}

type ModelPreferences struct {
	ModelURI    string   `json:"modelUri,omitempty"`
	Temperature *float64 `json:"temperature,omitempty"`
	TopP        *float64 `json:"topP,omitempty"`
	TopK        *int     `json:"topK,omitempty"`
}

type ModelPreferencesV20250326 struct {
	Hints                []ModelHintV20250326 `json:"hints,omitempty"`
	CostPriority         *float64             `json:"costPriority,omitempty"`         // 0-1
	SpeedPriority        *float64             `json:"speedPriority,omitempty"`        // 0-1
	IntelligencePriority *float64             `json:"intelligencePriority,omitempty"` // 0-1
}

type ProgressParams struct {
	Token   string      `json:"token"`
	Value   interface{} `json:"value"`
	Message *string     `json:"message,omitempty"` // Ensure omitempty is present (relevant for 2025-03-26)
}

type Prompt struct {
	URI         string                 `json:"uri"`
	Name        string                 `json:"name"`                  // Human-readable name
	Description string                 `json:"description,omitempty"` // Longer description
	Arguments   []PromptArgument       `json:"arguments,omitempty"`
	Messages    []PromptMessage        `json:"messages"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type PromptArgument struct {
	Name        string `json:"name"`
	Description string `json:"description,omitempty"`
	Type        string `json:"type"` // e.g., "string", "number", "boolean"
	Required    bool   `json:"required,omitempty"`
}

type PromptMessage struct {
	Role    string  `json:"role"`    // Must be "user" or "assistant" per schema
	Content Content `json:"content"` // Must be a single Content object per schema
}

type PromptReference struct {
	URI       string                 `json:"uri"`
	Arguments map[string]interface{} `json:"arguments,omitempty"`
}

type PromptsListChangedParams struct{}

type PropertyDetail struct {
	Type        string        `json:"type"`
	Description string        `json:"description,omitempty"`
	Enum        []interface{} `json:"enum,omitempty"`   // Possible values for the property
	Format      string        `json:"format,omitempty"` // Specific format (e.g., "date-time", "email")
}

type ReadResourceRequestParams struct {
	URI     string `json:"uri"`
	Version string `json:"version,omitempty"`
}

type ReadResourceResult struct {
	Resource Resource           `json:"resource"`
	Contents []ResourceContents `json:"contents"` // Actual content (Text, Blob, or Audio) - Array for 2024-11-05 and 2025-03-26
}

type ReferenceType string

type RequestMeta struct {
	// Use interface{} to accept string or number, per spec.
	ProgressToken interface{} `json:"progressToken,omitempty"`
}

type Resource struct {
	URI         string                 `json:"uri"`                   // Unique identifier (e.g., "file:///path/to/file", "git://...?rev=...")
	Kind        string                 `json:"kind,omitempty"`        // e.g., "file", "git_commit", "api_spec"
	Name        string                 `json:"name"`                  // Human-readable name (required by MCP spec)
	Description string                 `json:"description,omitempty"` // Longer description
	Version     string                 `json:"version,omitempty"`     // Opaque version string (changes when content changes)
	Metadata    map[string]interface{} `json:"metadata,omitempty"`    // Additional arbitrary metadata
	Size        *int                   `json:"size,omitempty"`        // Optional size in bytes (added in 2025-03-26)
	Annotations Annotations            `json:"annotations,omitempty"` // Added in 2025-03-26
}

type ResourceContents interface {
	GetContentType() string
}

type ResourceKind string

type ResourceTemplate struct {
	Kind        string                 `json:"kind"`                  // Kind of resource the template creates
	Name        string                 `json:"name"`                  // Human-readable name (required by MCP spec)
	Description string                 `json:"description,omitempty"` // Longer description
	Metadata    map[string]interface{} `json:"metadata,omitempty"`    // Default metadata or parameters needed
	URITemplate string                 `json:"uriTemplate"`           // URI template for constructing resource URLs
}

type ResourceUpdatedParams struct {
	Resource Resource `json:"resource"`
}

type ResourcesListChangedParams struct{}

type Root struct {
	URI         string                 `json:"uri"`
	Kind        string                 `json:"kind,omitempty"`
	Name        string                 `json:"name"` // Human-readable name
	Description string                 `json:"description,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type RootsListChangedParams struct{}

type SamplingCapability struct {
	Enabled bool `json:"enabled,omitempty"` // Indicates if the client supports handling sampling/request

}

type SamplingMessage struct {
	Role    string    `json:"role"`
	Content []Content `json:"content"`
	Name    *string   `json:"name,omitempty"`
}

type SamplingRequestParams struct {
	Messages         []SamplingMessage          `json:"messages"`
	ModelPreferences *ModelPreferencesV20250326 `json:"modelPreferences,omitempty"`
	SystemPrompt     *string                    `json:"systemPrompt,omitempty"`
	MaxTokens        *int                       `json:"maxTokens,omitempty"`
	Meta             *RequestMeta               `json:"_meta,omitempty"` // Allow progress reporting
}

type SamplingResult struct {
	Role       string       `json:"role"`
	Content    []Content    `json:"content"`
	Model      *string      `json:"model,omitempty"`
	StopReason *string      `json:"stopReason,omitempty"`
	Meta       *RequestMeta `json:"_meta,omitempty"` // Allow progress reporting
}

type ServerCapabilities struct {
	Experimental map[string]interface{} `json:"experimental,omitempty"`
	Logging      *struct{}              `json:"logging,omitempty"`
	Prompts      *struct {
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"prompts,omitempty"`
	Resources *struct {
		Subscribe   bool `json:"subscribe,omitempty"`
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"resources,omitempty"`
	Tools *struct {
		ListChanged bool `json:"listChanged,omitempty"`
	} `json:"tools,omitempty"`
	Authorization *struct {
	} `json:"authorization,omitempty"`
	Completions *struct {
	} `json:"completions,omitempty"`
}

type SetLevelRequestParams struct {
	Level LoggingLevel `json:"level"`
}

type SubscribeResourceParams struct {
	URIs []string `json:"uris"`
}

type SubscribeResourceResult struct{}

type TextContent struct {
	Type        string              `json:"type"` // Should always be "text"
	Text        string              `json:"text"`
	Annotations *ContentAnnotations `json:"annotations,omitempty"`
}

type TextResourceContents struct {
	ContentType string `json:"mimeType"` // e.g., "text/plain", "application/json"
	Text        string `json:"text"`     // Changed from Content to Text with json tag "text"
	URI         string `json:"uri"`      // Required by schema
}

type Tool struct {
	Name        string          `json:"name"`
	Description string          `json:"description,omitempty"`
	InputSchema ToolInputSchema `json:"inputSchema"`
	Annotations ToolAnnotations `json:"annotations,omitempty"`
}

type ToolAnnotations struct {
	Title           string `json:"title,omitempty"`           // Optional human-readable title for the tool.
	ReadOnlyHint    *bool  `json:"readOnlyHint,omitempty"`    // Use pointer for optional boolean
	DestructiveHint *bool  `json:"destructiveHint,omitempty"` // Use pointer for optional boolean
	IdempotentHint  *bool  `json:"idempotentHint,omitempty"`  // Use pointer for optional boolean
	OpenWorldHint   *bool  `json:"openWorldHint,omitempty"`   // Use pointer for optional boolean
}

type ToolCall struct {
	ID       string          `json:"id"`
	ToolName string          `json:"tool_name"`
	Input    json.RawMessage `json:"input,omitempty"`
}

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

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

type ToolsListChangedParams struct{}

type UnsubscribeResourceParams struct {
	URIs []string `json:"uris"`
}

type UnsubscribeResourceResult struct{}

type WorkspaceFolder struct {
	URI  string `json:"uri"`
	Name string `json:"name"`
}