瀏覽代碼

clean_agent

丁云鹏 1 月之前
父節點
當前提交
066de21727
共有 1 個文件被更改,包括 11 次插入0 次删除
  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: