Browse Source

clean_agent

丁云鹏 1 week ago
parent
commit
066de21727
1 changed files with 11 additions and 0 deletions
  1. 11 0
      agent.py

+ 11 - 0
agent.py

@@ -64,11 +64,15 @@ class TriggerResponse(BaseModel):
 
 class ExpandRequest(BaseModel):
     requestId: str = Field(..., description="扩展查询请求ID")
+<<<<<<< HEAD
     query: str = Field(..., description="扩展查询词")  # 可选
 
 class ExtractRequest(BaseModel):
     requestId: str = Field(..., description="请求ID")
     query: str = Field(..., description="查询词")
+=======
+    query: str = Field(..., description="查询内容")
+>>>>>>> 1c36a03 (clean_agent)
 
 # 全局变量
 identify_tool = None
@@ -594,10 +598,17 @@ async def process_request_background(request_id: str):
 extraction_requests: set = set()
 
 @app.post("/extract")
+<<<<<<< HEAD
 async def extract(request: ExtractRequest):
     try:
         requestId = request.requestId
         query = request.query
+=======
+async def extract(request: ExpandRequest, background_tasks: BackgroundTasks):
+    try:
+        request_id = request.requestId
+        query_word = request.query
+>>>>>>> 1c36a03 (clean_agent)
         # 检查请求是否已经在处理中
         async with RUNNING_LOCK:
             if requestId in extraction_requests: