jihuaqiang 1 mēnesi atpakaļ
vecāks
revīzija
6c94dfbedd
1 mainītis faili ar 4 papildinājumiem un 0 dzēšanām
  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")