入参调整

agent.py

@@ -64,6 +64,7 @@ class TriggerResponse(BaseModel):
 
 class ExpandRequest(BaseModel):
     requestId: str = Field(..., description="扩展查询请求ID")
+    query: str = Field(..., description="扩展查询词")  # 可选
 
 # 全局变量
 identify_tool = None
@@ -612,7 +613,7 @@ async def extract(requestId: str, query: str):
         raise HTTPException(status_code=500, detail=f"执行Agent时出错: {str(e)}")
 
 @app.post("/expand")
-async def expand(request: ExpandRequest, query: str, background_tasks: BackgroundTasks):
+async def expand(request: ExpandRequest, background_tasks: BackgroundTasks):
     """
     执行扩展查询处理
     
@@ -628,7 +629,7 @@ async def expand(request: ExpandRequest, query: str, background_tasks: Backgroun
         _update_expansion_status(request.requestId, 1)
         
         # 添加后台任务
-        background_tasks.add_task(execute_expand_agent_with_api, request.requestId, query)
+        background_tasks.add_task(execute_expand_agent_with_api, request.requestId, request.query)
         
         # 立即返回状态
         return {"status": 1, "requestId": request.requestId, "message": "扩展查询处理已启动"}