not found key kb_id

isthaison · Dec 4, 2024 · 3260642 · 3260642
1 parent 2d6cf81
commit 3260642
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rag/svr/task_executor.py b/rag/svr/task_executor.py
@@ -269,7 +269,7 @@ def build_chunks(task, progress_callback):
 
 def init_kb(row, vector_size: int):
     idxnm = search.index_name(row["tenant_id"])
-    return settings.docStoreConn.createIdx(idxnm, row["kb_id"], vector_size)
+    return settings.docStoreConn.createIdx(idxnm, row.get("kb_id",""), vector_size)
 
 
 def embedding(docs, mdl, parser_config=None, callback=None):