Merge pull request #294 from NomaDamas/Feature/#293

version update and fix langchain VectorStore location for proper use
Marker-Inc-Korea · Oct 29, 2023 · e1a8212 · e1a8212
2 parents 515a6c4 + 2dd4ac8
commit e1a8212
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/RAGchain/retrieval/vectordb_retrieval.py b/RAGchain/retrieval/vectordb_retrieval.py
@@ -2,7 +2,7 @@
 from uuid import UUID
 
 from langchain.schema import Document
-from langchain.vectorstores import VectorStore
+from langchain.schema.vectorstore import VectorStore
 
 from RAGchain.retrieval.base import BaseRetrieval
 from RAGchain.schema import Passage

diff --git a/setup.py b/setup.py
@@ -22,8 +22,8 @@
     license='Apache License 2.0',
     py_modules=[splitext(basename(path))[0] for path in glob('./*.py')],
     install_requires=[
-        'langchain==0.0.309',
-        'chromadb==0.4.5',
+        'langchain==0.0.324',
+        'chromadb>=0.4.15',
         'urllib3',
         'pdfminer.six',
         'click',