jihuaqiang 1 week ago
parent
commit
6c94dfbedd
1 changed files with 4 additions and 0 deletions
  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")