|
@@ -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:
|