67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
|
from typing import List, Optional, Any, Dict
|
|||
|
|
from pydantic import BaseModel, Field
|
|||
|
|
|
|||
|
|
# 聊天请求
|
|||
|
|
class ChatRequest(BaseModel):
|
|||
|
|
query: str = Field(..., description="用户输入的消息")
|
|||
|
|
system_user_id: int = Field(..., description="系统用户ID（当前登录用户）")
|
|||
|
|
context_id: Optional[str] = Field(None, description="会话上下文ID")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
json_schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"query": "查看所有可用代理",
|
|||
|
|
"system_user_id": 1000000001,
|
|||
|
|
"context_id": "session-1234567890"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 聊天响应
|
|||
|
|
class ChatResponse(BaseModel):
|
|||
|
|
content: str = Field(..., description="AI 回复内容")
|
|||
|
|
context_id: str = Field(..., description="会话上下文ID")
|
|||
|
|
task_id: Optional[str] = Field(None, description="任务ID")
|
|||
|
|
status: str = Field(default="success", description="响应状态")
|
|||
|
|
|
|||
|
|
class Config:
|
|||
|
|
json_schema_extra = {
|
|||
|
|
"example": {
|
|||
|
|
"content": "系统中共有 4 个可用代理...",
|
|||
|
|
"context_id": "session-1234567890",
|
|||
|
|
"task_id": "task-abc123",
|
|||
|
|
"status": "success"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 错误响应
|
|||
|
|
class ErrorResponse(BaseModel):
|
|||
|
|
error: str = Field(..., description="错误类型")
|
|||
|
|
message: str = Field(..., description="错误消息")
|
|||
|
|
details: Optional[Dict[str, Any]] = Field(None, description="错误详情")
|
|||
|
|
|
|||
|
|
# A2A 消息部分
|
|||
|
|
class A2AMessagePart(BaseModel):
|
|||
|
|
kind: str = "text"
|
|||
|
|
text: str
|
|||
|
|
|
|||
|
|
# A2A 消息
|
|||
|
|
class A2AMessage(BaseModel):
|
|||
|
|
context_id: str
|
|||
|
|
role: str = "user"
|
|||
|
|
parts: List[A2AMessagePart]
|
|||
|
|
message_id: str
|
|||
|
|
|
|||
|
|
# A2A 请求
|
|||
|
|
class A2ARequest(BaseModel):
|
|||
|
|
jsonrpc: str = "2.0"
|
|||
|
|
method: str = "message/send"
|
|||
|
|
params: Dict[str, A2AMessage]
|
|||
|
|
id: int
|
|||
|
|
|
|||
|
|
# A2A 响应
|
|||
|
|
class A2AResponse(BaseModel):
|
|||
|
|
id: int
|
|||
|
|
jsonrpc: str
|
|||
|
|
result: Dict[str, Any]
|
|||
|
|
|