mcp

type CapabilityConfig struct {
	Tools       bool
	Resources   bool
	Prompts     bool
	Completions bool
	Tasks       bool
}

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

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

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

type CompletionHook func(ctx context.Context, req CompletionRequest) (*CompletionResult, error)

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

type CompletionRequest struct {
	SessionID string             `json:"sessionId"`
	Ref       CompletionRef      `json:"ref"`
	Argument  CompletionArgument `json:"argument"`
	Context   CompletionContext  `json:"context,omitempty"`
}

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

type ContentBlock struct {
	Type string `json:"type"` // "text", "image", "audio", "resource_link", "resource"

	// Text content (type = "text").
	Text string `json:"text,omitempty"`

	// Image/audio content (type = "image" or "audio").
	Data     string `json:"data,omitempty"`     // base64-encoded
	MimeType string `json:"mimeType,omitempty"` // e.g. "image/png"

	// Resource link content (type = "resource_link").
	URI         string `json:"uri,omitempty"`
	Name        string `json:"name,omitempty"`
	Title       string `json:"title,omitempty"`
	Description string `json:"description,omitempty"`
	Size        int64  `json:"size,omitempty"`

	// Embedded resource content (type = "resource").
	Resource *ResourceContent `json:"resource,omitempty"`
}

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

type DiscoverResult struct {
	SupportedVersions []string       `json:"supportedVersions"`
	Capabilities      map[string]any `json:"capabilities"`
	Meta              map[string]any `json:"_meta"`
}

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

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

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

type Icon struct {
	Src      string   `json:"src"`
	MimeType string   `json:"mimeType,omitempty"`
	Sizes    []string `json:"sizes,omitempty"`
	Theme    string   `json:"theme,omitempty"` // "light" or "dark"
}

type InitialSessionListenerBudgetOptions struct {
	SafetyBuffer time.Duration
	MaxDuration  time.Duration
}

type InputRequest struct {
	Method string `json:"method"`
	Params any    `json:"params,omitempty"`
}

type InputRequiredResult struct {
	ResultType    ResultType              `json:"resultType,omitempty"`
	InputRequests map[string]InputRequest `json:"inputRequests,omitempty"`
	RequestState  string                  `json:"requestState,omitempty"`
	Meta          map[string]any          `json:"_meta,omitempty"`
}

type LoggingLevel string

type LoggingLevelHook func(ctx context.Context, req LoggingLevelRequest) error

type LoggingLevelRequest struct {
	SessionID string       `json:"sessionId"`
	Level     LoggingLevel `json:"level"`
}

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

type MemorySessionStoreOption func(*MemorySessionStore)

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

type MemoryStreamStoreOption func(*MemoryStreamStore)

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

type OriginValidator func(origin string) bool

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

type PromptDef struct {
	Name        string           `json:"name"`
	Title       string           `json:"title,omitempty"`
	Description string           `json:"description,omitempty"`
	Arguments   []PromptArgument `json:"arguments,omitempty"`
}

type PromptHandler func(ctx context.Context, args json.RawMessage) (*PromptResult, error)

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

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

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

type ProtocolShape string

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

type RelatedTaskMetadata struct {
	TaskID string `json:"taskId"`
}

type Request struct {
	JSONRPC string          `json:"jsonrpc"`
	ID      any             `json:"id,omitempty"`
	Method  string          `json:"method"`
	Params  json.RawMessage `json:"params,omitempty"`
}

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

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

type ResourceHandler func(ctx context.Context) ([]ResourceContent, error)

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

type ResourceSubscription struct {
	SessionID string `json:"sessionId"`
	URI       string `json:"uri"`
}

type ResourceSubscriptionHook func(ctx context.Context, sub ResourceSubscription) error

type ResourceTemplateDef struct {
	URITemplate string `json:"uriTemplate"`
	Name        string `json:"name"`
	Title       string `json:"title,omitempty"`
	Description string `json:"description,omitempty"`
	MimeType    string `json:"mimeType,omitempty"`
}

type Response struct {
	JSONRPC string    `json:"jsonrpc"`
	ID      any       `json:"id"`
	Result  any       `json:"result,omitempty"`
	Error   *RPCError `json:"error,omitempty"`
}

type ResultType string

type SSEEvent struct {
	Data any
}

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

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

type ServerOption func(*Server)

type Session struct {
	ID        string            `json:"id"`
	CreatedAt time.Time         `json:"createdAt"`
	ExpiresAt time.Time         `json:"expiresAt"`
	Data      map[string]string `json:"data,omitempty"`
}

type SessionStore interface {
	Get(ctx context.Context, id string) (*Session, error)
	Put(ctx context.Context, session *Session) error
	Delete(ctx context.Context, id string) error
}

type StreamEvent struct {
	ID   string
	Data json.RawMessage
}

type StreamStore interface {
	Create(ctx context.Context, sessionID string) (streamID string, err error)
	Append(ctx context.Context, sessionID, streamID string, data json.RawMessage) (eventID string, err error)
	Close(ctx context.Context, sessionID, streamID string) error

	// Subscribe streams events after afterEventID. If afterEventID is empty,
	// it streams from the beginning. If afterEventID is present, it must belong
	// to the requested stream.
	Subscribe(ctx context.Context, sessionID, streamID, afterEventID string) (<-chan StreamEvent, error)

	// StreamForEvent returns the stream id that the given event id belongs to.
	StreamForEvent(ctx context.Context, sessionID, eventID string) (streamID string, err error)

	// DeleteSession removes all stream state for a session.
	DeleteSession(ctx context.Context, sessionID string) error
}

type StreamingToolHandler func(ctx context.Context, args json.RawMessage, emit func(SSEEvent)) (*ToolResult, error)

type Task struct {
	TaskID        string     `json:"taskId"`
	Status        TaskStatus `json:"status"`
	StatusMessage string     `json:"statusMessage,omitempty"`
	CreatedAt     time.Time  `json:"createdAt"`
	LastUpdatedAt time.Time  `json:"lastUpdatedAt"`
	TTL           *int64     `json:"ttl"`
	PollInterval  *int64     `json:"pollInterval,omitempty"`
}

type TaskListRequest struct {
	SessionID string `json:"sessionId"`
	Cursor    string `json:"cursor,omitempty"`
	Limit     int    `json:"limit,omitempty"`
}

type TaskListResult struct {
	Tasks      []Task `json:"tasks"`
	NextCursor string `json:"nextCursor,omitempty"`
}

type TaskLookup struct {
	SessionID string `json:"sessionId"`
	TaskID    string `json:"taskId"`
}

type TaskMetadata struct {
	TTL *int64 `json:"ttl,omitempty"`
}

type TaskRecord struct {
	SessionID string          `json:"sessionId"`
	Method    string          `json:"method"`
	ToolName  string          `json:"toolName,omitempty"`
	Task      Task            `json:"task"`
	Result    json.RawMessage `json:"result,omitempty"`
	Error     *RPCError       `json:"error,omitempty"`
}

type TaskRuntimeOptions struct {
	Store                  TaskStore
	DefaultTTL             time.Duration
	MaxTTL                 time.Duration
	PollInterval           time.Duration
	ListLimit              int
	ModelImmediateResponse string
}

type TaskStatus string

type TaskStore interface {
	Create(ctx context.Context, task TaskRecord) (*TaskRecord, error)
	Get(ctx context.Context, lookup TaskLookup) (*TaskRecord, error)
	Update(ctx context.Context, task TaskRecord) (*TaskRecord, error)
	List(ctx context.Context, req TaskListRequest) (*TaskListResult, error)
	Cancel(ctx context.Context, lookup TaskLookup) (*TaskRecord, error)
	DeleteSession(ctx context.Context, sessionID string) error
}

type TaskSupport string

type ToolAnnotations struct {
	Title           string `json:"title,omitempty"`
	ReadOnlyHint    *bool  `json:"readOnlyHint,omitempty"`
	DestructiveHint *bool  `json:"destructiveHint,omitempty"`
	IdempotentHint  *bool  `json:"idempotentHint,omitempty"`
	OpenWorldHint   *bool  `json:"openWorldHint,omitempty"`
}

type ToolDef struct {
	Name        string           `json:"name"`
	Title       string           `json:"title,omitempty"`
	Description string           `json:"description,omitempty"`
	Annotations *ToolAnnotations `json:"annotations,omitempty"`

	Icons        []Icon          `json:"icons,omitempty"`
	Execution    *ToolExecution  `json:"execution,omitempty"`
	InputSchema  json.RawMessage `json:"inputSchema"`
	OutputSchema json.RawMessage `json:"outputSchema,omitempty"`
}

type ToolExecution struct {
	// TaskSupport indicates if the tool supports task-augmented execution.
	// Values: "forbidden", "optional", "required".
	TaskSupport TaskSupport `json:"taskSupport,omitempty"`
}

type ToolHandler func(ctx context.Context, args json.RawMessage) (*ToolResult, error)

type ToolInput struct {
	InputResponses map[string]any
	RequestState   string
}

type ToolLifecycleFinish struct {
	Name              string               `json:"name"`
	StartedAt         time.Time            `json:"startedAt"`
	FinishedAt        time.Time            `json:"finishedAt"`
	Duration          time.Duration        `json:"duration"`
	Outcome           ToolLifecycleOutcome `json:"outcome"`
	JSONRPCErrorCode  int                  `json:"jsonrpcErrorCode,omitempty"`
	ResultMarkedError bool                 `json:"resultMarkedError,omitempty"`
}

type ToolLifecycleOptions[Args any] struct {
	Name        string
	Timeout     time.Duration
	NoArgs      bool
	StrictJSON  bool
	Validate    func(context.Context, Args) error
	HandleError func(context.Context, error) (*ToolResult, bool)
	Telemetry   ToolLifecycleTelemetry
	Clock       apptheory.Clock
}

type ToolLifecycleOutcome string

type ToolLifecycleStart struct {
	Name      string    `json:"name"`
	StartedAt time.Time `json:"startedAt"`
}

type ToolLifecycleTelemetry struct {
	Start  func(context.Context, ToolLifecycleStart)
	Finish func(context.Context, ToolLifecycleFinish)
}

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

type ToolResult struct {
	Content           []ContentBlock          `json:"content"`
	IsError           bool                    `json:"isError,omitempty"`
	StructuredContent map[string]any          `json:"structuredContent,omitempty"`
	ResultType        ResultType              `json:"resultType,omitempty"`
	InputRequests     map[string]InputRequest `json:"inputRequests,omitempty"`
	RequestState      string                  `json:"requestState,omitempty"`
}