a2a

type AgentCard struct {
	Name               string       `json:"name"`
	Description        string       `json:"description"`
	URL                string       `json:"url"`
	Provider           Provider     `json:"provider"`
	Version            string       `json:"version"`
	Capabilities       Capabilities `json:"capabilities"`
	DefaultInputModes  []string     `json:"defaultInputModes"`
	DefaultOutputModes []string     `json:"defaultOutputModes"`
	Skills             []Skill      `json:"skills,omitempty"`
}

type Artifact struct {
	Name  string `json:"name"`
	Parts []Part `json:"parts"`
}

type Capabilities struct {
	Streaming              bool `json:"streaming"`
	PushNotifications      bool `json:"pushNotifications"`
	StateTransitionHistory bool `json:"stateTransitionHistory"`
}

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

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

type JSONRPCRequest struct {
	JSONRPC string `json:"jsonrpc"` // 固定为 "2.0"
	ID      any    `json:"id"`      // 字符串或数字
	Method  string `json:"method"`
	Params  any    `json:"params,omitempty"`
}

type JSONRPCResponse struct {
	JSONRPC string    `json:"jsonrpc"` // 固定为 "2.0"
	ID      any       `json:"id"`
	Result  any       `json:"result,omitempty"`
	Error   *RPCError `json:"error,omitempty"`
}

type Message struct {
	MessageID        string   `json:"messageId"`
	Role             string   `json:"role"` // "user" 或 "agent"
	Parts            []Part   `json:"parts"`
	Kind             string   `json:"kind"` // 固定为 "message"
	ContextID        string   `json:"contextId,omitempty"`
	TaskID           string   `json:"taskId,omitempty"`
	ReferenceTaskIDs []string `json:"referenceTaskIds,omitempty"`
}

type MessageSendParams struct {
	Message   Message  `json:"message"`
	ContextID string   `json:"contextId,omitempty"`
	Metadata  Metadata `json:"metadata,omitempty"`
}

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

type MessageStreamParams struct {
	Message   Message  `json:"message"`
	ContextID string   `json:"contextId,omitempty"`
	Metadata  Metadata `json:"metadata,omitempty"`
}

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

type Metadata map[string]any

type Part struct {
	Kind string `json:"kind"` // "text", "file", "data"
	Text string `json:"text,omitempty"`
	Data any    `json:"data,omitempty"`
	// File 相关字段
	Name     string `json:"name,omitempty"`
	MimeType string `json:"mimeType,omitempty"`
	URL      string `json:"url,omitempty"`
}

type Provider struct {
	Organization string `json:"organization"`
	URL          string `json:"url"`
}

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

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

type Skill struct {
	ID          string   `json:"id"`
	Name        string   `json:"name"`
	Description string   `json:"description"`
	Tags        []string `json:"tags,omitempty"`
}

type Task struct {
	ID        string     `json:"id"`
	ContextID string     `json:"contextId"`
	Status    TaskStatus `json:"status"`
	Artifacts []Artifact `json:"artifacts,omitempty"`
	History   []Message  `json:"history,omitempty"`
	Metadata  Metadata   `json:"metadata,omitempty"`
	Kind      string     `json:"kind"` // 固定为 "task"
}

type TaskState string

type TaskStatus struct {
	State     TaskState `json:"state"`
	Timestamp string    `json:"timestamp"` // ISO 8601 格式
	Message   *Message  `json:"message,omitempty"`
}

type TaskStore interface {
	// Load 加载任务
	Load(agentID, taskID string) (*Task, error)

	// Save 保存任务
	Save(agentID string, task *Task) error

	// Delete 删除任务
	Delete(agentID, taskID string) error

	// List 列出 Agent 的所有任务
	List(agentID string) ([]*Task, error)

	// AddCancellation 添加取消信号
	AddCancellation(taskID string)

	// RemoveCancellation 移除取消信号
	RemoveCancellation(taskID string)

	// IsCanceled 检查是否已取消
	IsCanceled(taskID string) bool
}

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

type TasksCancelResult struct {
	Success bool   `json:"success"`
	Message string `json:"message,omitempty"`
}

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

type TasksGetResult struct {
	Task *Task `json:"task"`
}