feat: add ids to documents

SverreNystad · Jan 15, 2025 · 47d06be · 47d06be
1 parent 8c6b8f7
commit 47d06be
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rules-engine/src/knowledge_base/agent/rag_service.py b/rules-engine/src/knowledge_base/agent/rag_service.py
@@ -65,7 +65,7 @@ def __init__(self):
         for i in range(0, 90, batch_size):
             print(f"Batch: {i}")
             batch = splits[i:i+batch_size]
-            self.parent_doc_retriever.add_documents(batch, ids=None)
+            self.parent_doc_retriever.add_documents(batch, ids=i)
 
         #self.parent_doc_retriever.add_documents(splits, ids=None)