Browse Source

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

jihuaqiang 1 tháng trước cách đây
mục cha
commit
db20240c72
2 tập tin đã thay đổi với 2 bổ sung1 xóa
  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: