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")