Fix document retrieval

talkdai · May 11, 2024 · 4e8de79 · 4e8de79
1 parent 7e1b776
commit 4e8de79
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/dialog_lib/embeddings/retrievers.py b/dialog_lib/embeddings/retrievers.py
@@ -29,8 +29,7 @@ def _get_relevant_documents(self, query, *, run_manager):
         )
         return [
             Document(
-                document_id=content.id,
-                document=f"{content.question}\n\n{content.content}",
+                page_content=f"{content.question}\n\n{content.content}",
                 metadata={
                     "title": content.question,
                     "category": content.category,

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dialog-lib"
-version = "0.0.1.17"
+version = "0.0.1.18"
 description = ""
 authors = ["Talkd.AI <foss@talkd.ai>"]
 license = "MIT"