소스 검색

Merge branch 'main' of https://git.yishihui.com/ai/knowledge-agent

jihuaqiang 1 개월 전
부모
커밋
db20240c72
2개의 변경된 파일2개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 0
      agent.py
  2. 1 1
      tools/agent_tools.py

+ 1 - 0
agent.py

@@ -609,6 +609,7 @@ async def extract(request: ExtractRequest):
             update_extract_status(requestId, 1)
             # 执行Agent
             result = execute_agent_with_api(json.dumps({"query_word": query, "request_id": requestId}))
+            update_extract_status(requestId, 2)
         finally:
             # 无论成功失败,都从运行集合中移除
             async with RUNNING_LOCK:

+ 1 - 1
tools/agent_tools.py

@@ -124,7 +124,7 @@ class QueryDataTool:
 
         results: List[Dict[str, Any]] = []
         for row in rows:
-            data_cell = row[0]
+            data_cell = row
             if not data_cell:
                 continue
             try: