jihuaqiang 1 місяць тому
батько
коміт
6c94dfbedd
1 змінених файлів з 4 додано та 0 видалено
  1. 4 0
      agent.py

+ 4 - 0
agent.py

@@ -51,6 +51,10 @@ class AgentState(TypedDict):
     error: Optional[str]
     status: str
 
+class ExpandRequest(BaseModel):
+    requestId: str = Field(..., description="请求ID")
+    query: str = Field(..., description="查询词")
+
 # 请求模型
 class TriggerRequest(BaseModel):
     requestId: str = Field(..., description="请求ID")