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