a2a

type A2AClient interface {
	// 发现从远程特工处取回特工卡
	Discover(ctx context.Context, url string) (*AgentCard, error)
	// 发送消息同步并等待回复.
	Send(ctx context.Context, msg *A2AMessage) (*A2AMessage, error)
	// SendAsync 同步发送消息并返回任务ID.
	SendAsync(ctx context.Context, msg *A2AMessage) (string, error)
	// GetResult通过任务ID检索一个同步任务的结果.
	GetResult(ctx context.Context, taskID string) (*A2AMessage, error)
}

type A2AMessage struct {
	// ID是此消息的唯一标识符 。
	ID string `json:"id"`
	// 类型表示信件类型(任务,结果,出错,状态,取消).
	Type A2AMessageType `json:"type"`
	// 从是发件人的标识符.
	From string `json:"from"`
	// 为接收代理的识别符.
	To string `json:"to"`
	// 有效载荷包含消息数据.
	Payload any `json:"payload"`
	// 时间戳是消息创建时.
	Timestamp time.Time `json:"timestamp"`
	// PresidentTo 是此回复信件的ID( 可选) 。
	ReplyTo string `json:"reply_to,omitempty"`
}

type A2AMessageType string

type A2AServer interface {
	// 注册代理在服务器上注册本地代理 。
	RegisterAgent(agent agent.Agent) error
	// Unregister Agent 从服务器中删除一个代理 。
	UnregisterAgent(agentID string) error
	// ServiHTTP 执行 http. 服务A2A请求的掌上电脑
	ServeHTTP(w http.ResponseWriter, r *http.Request)
	// Get AgentCard为注册代理人取回代理卡.
	GetAgentCard(agentID string) (*AgentCard, error)
}

type AgentCard struct {
	// 名称是此代理的唯一标识符 。
	Name string `json:"name"`
	// 描述提供了一种人能读取的关于该剂目的的描述.
	Description string `json:"description"`
	// URL是能够到达此代理的终点 。
	URL string `json:"url"`
	// 版本表示代理的版本.
	Version string `json:"version"`
	// 能力列出这个代理提供的能力.
	Capabilities []Capability `json:"capabilities"`
	// 输入Schema定义了代理商的预期输入格式的JSON Schema.
	InputSchema *structured.JSONSchema `json:"input_schema,omitempty"`
	// OutsionSchema定义了代理输出格式的JSON Schema.
	OutputSchema *structured.JSONSchema `json:"output_schema,omitempty"`
	// 工具列出这个代理可以使用或曝光的工具.
	Tools []ToolDefinition `json:"tools,omitempty"`
	// 元数据包含额外的可扩展性的密钥值对.
	Metadata map[string]string `json:"metadata,omitempty"`
}

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

type AgentConfigProvider interface {
	ID() string
	Name() string
	Type() AgentType
	Description() string
	Tools() []string
	Metadata() map[string]string
}

type AgentType string

type AsyncResponse struct {
	TaskID        string `json:"task_id"`
	Status        string `json:"status"`
	Message       string `json:"message,omitempty"`
	EstimatedTime int    `json:"estimated_time,omitempty"` // seconds
}

type Capability struct {
	// 名称是此能力的唯一标识符 。
	Name string `json:"name"`
	// 描述为人们提供了这种能力的可读性描述.
	Description string `json:"description"`
	// 类型表示能力类型(任务,查询,流).
	Type CapabilityType `json:"type"`
}

type CapabilityType string

type ClientConfig struct {
	// 超时是HTTP请求的默认超时.
	Timeout time.Duration
	// RetryCount 是失败请求的重试次数 。
	RetryCount int
	// RetryDelay是重试之间的延迟.
	RetryDelay time.Duration
	// 信头是请求中要包含的额外信头 。
	Headers map[string]string
	// AgentID 是本地代理提出请求的标识符 。
	AgentID string
}

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

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

type ServerConfig struct {
	// BaseURL 是此服务器可访问的基础 URL 。
	BaseURL string
	// 默认代理ID是在没有特定代理目标时使用的代理ID.
	DefaultAgentID string
	// 请求超时是处理请求的超时.
	RequestTimeout time.Duration
	// StrictRouting 启用严格路由：目标代理不存在时直接返回错误，不回退默认代理。
	StrictRouting bool
	// 启用 Auth 允许对收到的请求进行认证 。
	EnableAuth bool
	// AuthToken 是预期的认证符( 如果 EullAuth 是真实的) 。
	AuthToken string
	// logger 是日志实例 。
	Logger *zap.Logger
}

type SimpleAgentConfig struct {
	AgentID          string
	AgentName        string
	AgentType        AgentType
	AgentDescription string
	AgentTools       []string
	AgentMetadata    map[string]string
}

type ToolDefinition struct {
	// 名称是此工具的唯一标识符 。
	Name string `json:"name"`
	// 描述提供了一种人类可以读取的描述,说明这个工具是做什么的.
	Description string `json:"description"`
	// 参数定义了该工具的输入参数的JSON Schema.
	Parameters *structured.JSONSchema `json:"parameters"`
	// Version 是工具的版本标识符.
	Version string `json:"version,omitempty"`
}

type ToolSchemaProvider interface {
	GetAllowedTools(agentID string) []types.ToolSchema
}