Просмотр исходного кода

fix: 深采入库不传 parents,仅传 children 数据

- parents 改为 None 不传,children.parent_id 随意填写
- sample_center_client.batch_import 支持 parents=None 时不传该字段
kinglee 6 дней назад
Родитель
Сommit
8e23f7b32b
2 измененных файлов с 6 добавлено и 13 удалено
  1. 1 7
      app/routes/deep_routes.py
  2. 5 6
      app/sample_center_client.py

+ 1 - 7
app/routes/deep_routes.py

@@ -306,12 +306,6 @@ def import_to_knowledge():
 
     parent_id = f"parent_{deep_id}"
 
-    parents = [{
-        "index": 0,
-        "parent_id": parent_id,
-        "text": "",
-        "metadata": {"title": item.title or "", "url": item.url or ""},
-    }]
     children = [{
         "index": 0,
         "parent_id": parent_id,
@@ -331,7 +325,7 @@ def import_to_knowledge():
     result = client.batch_import(
         kb_id=kb_id,
         task_no=task_no,
-        parents=parents,
+        parents=None,
         children=children,
         callback_url=url_for('knowledge.knowledge_import_callback', _external=True),
     )

+ 5 - 6
app/sample_center_client.py

@@ -105,20 +105,19 @@ class SampleCenterClient:
         resp = requests.get(url, headers=self._headers(), timeout=15)
         return self._parse(resp)
 
-    def batch_import(self, kb_id, task_no, parents, children=None, callback_url=None):
+    def batch_import(self, kb_id, task_no, parents=None, children=None, callback_url=None):
         """POST /api/v1/knowledge-bases/{kb_id}/batch-import -- 批量入库。"""
         url = f"{self.base_url}/api/v1/knowledge-bases/{kb_id}/batch-import"
-        payload = {
-            "task_no": task_no,
-            "parents": parents,
-        }
+        payload = {"task_no": task_no}
+        if parents:
+            payload["parents"] = parents
         if children:
             payload["children"] = children
         if callback_url:
             payload["callback_url"] = callback_url
         logger.info(
             f"请求批量入库: url={url}, task_no={task_no}, "
-            f"parents_count={len(parents)}, children_count={len(children) if children else 0}, "
+            f"parents_count={len(parents) if parents else 0}, children_count={len(children) if children else 0}, "
             f"parents={parents!r}, children={children!r}, callback_url={callback_url}"
         )
         resp = requests.post(