Adds correct imports from langchain_core for LCEL Retriever

talkdai · May 11, 2024 · 7e1b776 · 7e1b776
1 parent 995c92c
commit 7e1b776
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/dialog_lib/embeddings/retrievers.py b/dialog_lib/embeddings/retrievers.py
@@ -3,7 +3,8 @@
 from dialog_lib.db.models import CompanyContent
 from sqlalchemy.orm import DeclarativeBase, Session
 
-from langchain.schema.retriever import BaseRetriever, Document
+from langchain_core.retrievers import BaseRetriever
+from langchain_core.documents import Document
 from dialog_lib.embeddings.generate import get_most_relevant_contents_from_message
 
 class DialogRetriever(BaseRetriever):

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