protocol

type Annotated struct {
	Annotations *Annotations `json:"annotations,omitempty"`
}

type Annotations struct {
	Audience []Role  `json:"audience,omitempty"`
	Priority float64 `json:"priority,omitempty"`
}

type BlobResourceContents struct {
	URI      string `json:"uri"`
	Blob     []byte `json:"blob"`
	MimeType string `json:"mimeType,omitempty"`
}

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

type CallToolResult struct {
	Content []Content `json:"content"`
	IsError bool      `json:"isError,omitempty"`
}

type CancelledNotification struct {
	RequestID RequestID `json:"requestId"`
	Reason    string    `json:"reason,omitempty"`
}

type ClientCapabilities struct {
}

type ClientNotify interface{}

type ClientRequest interface{}

type ClientResponse interface{}

type Complete struct {
	Values  []string `json:"values"`
	HasMore bool     `json:"hasMore,omitempty"`
	Total   int      `json:"total,omitempty"`
}

type CompleteRequest struct {
	Argument struct {
		Name  string `json:"name"`
		Value string `json:"value"`
	} `json:"argument"`
	Ref interface{} `json:"ref"` // Can be PromptReference or ResourceReference
}

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

type Content interface {
	GetType() string
}

type CreateMessageOption func(*CreateMessageRequest)

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

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

type EmbeddedResource struct {
	Type        string           `json:"type"` // Must be "resource"
	Resource    ResourceContents `json:"resource"`
	Annotations *Annotations     `json:"annotations,omitempty"`
}

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

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

type ImageContent struct {
	Annotated
	Type     string `json:"type"`
	Data     []byte `json:"data"`
	MimeType string `json:"mimeType"`
}

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

type InitializeRequest struct {
	ClientInfo      Implementation     `json:"clientInfo"`
	Capabilities    ClientCapabilities `json:"capabilities"`
	ProtocolVersion string             `json:"protocolVersion"`
}

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

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

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

type InputSchemaType string

type JSONRPCNotification struct {
	JSONRPC   string          `json:"jsonrpc"`
	Method    Method          `json:"method"`
	Params    interface{}     `json:"params,omitempty"`
	RawParams json.RawMessage `json:"-"`
}

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

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

type ListPromptsRequest struct{}

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

type ListResourceTemplatesRequest struct{}

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

type ListResourcesRequest struct{}

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

type ListRootsRequest struct{}

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

type ListToolsRequest struct{}

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

type LogMessageNotification struct {
	Level   LoggingLevel           `json:"level"`
	Message string                 `json:"message"`
	Meta    map[string]interface{} `json:"meta,omitempty"`
}

type LoggingLevel string

type Method string

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

type ModelPreferences struct {
	CostPriority         float64     `json:"costPriority,omitempty"`
	IntelligencePriority float64     `json:"intelligencePriority,omitempty"`
	SpeedPriority        float64     `json:"speedPriority,omitempty"`
	Hints                []ModelHint `json:"hints,omitempty"`
}

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

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

type PingRequest struct{}

type PingResult struct{}

type ProgressNotification struct {
	ProgressToken ProgressToken `json:"progressToken"`
	Progress      float64       `json:"progress"`
	Total         float64       `json:"total,omitempty"`
}

type ProgressToken interface{} // can be string or integer

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 PromptListChangedNotification struct {
	Meta map[string]interface{} `json:"_meta,omitempty"`
}

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

type PromptReference struct {
	Type string `json:"type"`
	Name string `json:"name"`
}

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

type ReadResourceRequest struct {
	URI       string                 `json:"uri"`
	Arguments map[string]interface{} `json:"-"`
}

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

type RequestID interface{} // 字符串/数值

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

type ResourceContents interface {
	GetURI() string
	GetMimeType() string
}

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

type ResourceReference struct {
	Type string `json:"type"`
	URI  string `json:"uri"`
}

type ResourceTemplate struct {
	Annotated
	Name              string                `json:"name"`
	URITemplate       string                `json:"uriTemplate"`
	URITemplateParsed *uritemplate.Template `json:"-"`
	Description       string                `json:"description,omitempty"`
	MimeType          string                `json:"mimeType,omitempty"`
}

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

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

type Role string

type Root struct {
	Name string `json:"name,omitempty"`
	URI  string `json:"uri"`
}

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

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

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

type ServerCapabilities struct {
	// Experimental map[string]interface{} `json:"experimental,omitempty"`
	// Logging      interface{}            `json:"logging,omitempty"`
	Prompts   *PromptsCapability   `json:"prompts,omitempty"`
	Resources *ResourcesCapability `json:"resources,omitempty"`
	Tools     *ToolsCapability     `json:"tools,omitempty"`
}

type ServerNotify interface{}

type ServerRequest interface{}

type ServerResponse interface{}

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

type SetLoggingLevelResult struct {
	Success bool `json:"success"`
}

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

type SubscribeResult struct{}

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

type TextResourceContents struct {
	URI      string `json:"uri"`
	Text     string `json:"text"`
	MimeType string `json:"mimeType,omitempty"`
}

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

	// Description is a human-readable description of the tool
	Description string `json:"description,omitempty"`

	// InputSchema defines the expected parameters for the tool using JSON Schema
	InputSchema InputSchema `json:"inputSchema"`
}

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

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

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

type UnsubscribeResult struct{}