浏览代码

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

jihuaqiang 5 天之前
父节点
当前提交
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: