set allow_dangerous_deserialization

bioimage-io · Mar 12, 2024 · 6a059f0 · 6a059f0
1 parent 2915045
commit 6a059f0
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/bioimageio_chatbot/knowledge_base.py b/bioimageio_chatbot/knowledge_base.py
@@ -27,7 +27,7 @@ def load_docs_store(db_path, collection_name):
 
     # Load from vector store
     embeddings = OpenAIEmbeddings()
-    docs_store = FAISS.load_local(index_name=collection_name, folder_path=db_path, embeddings=embeddings)
+    docs_store = FAISS.load_local(index_name=collection_name, folder_path=db_path, embeddings=embeddings, allow_dangerous_deserialization=True)
     return docs_store
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 
 [project]
 name = "bioimageio-chatbot"
-version = "0.1.75"
+version = "0.1.76"
 readme = "README.md"
 description = "Your Personal Assistant in BioImage Analysis."
 dependencies = [
@@ -20,7 +20,8 @@ dependencies = [
   "duckduckgo-search>=5.0",
   "rank-bm25",
   "langchain-openai",
-  "langchain-community",
+  "langchain-core>=0.1.31",
+  "langchain-community>=0.0.27",
   "html2text",
 ]
 

diff --git a/requirements.txt b/requirements.txt
@@ -6,7 +6,9 @@ hypha>=0.15.50
 tqdm
 aiofiles
 serpapi
-langchain>=0.1.10
+langchain>=0.1.11
+langchain-community==0.0.27
+langchain-core==0.1.31
 beautifulsoup4
 pandas
 duckduckgo-search==5.0

diff --git a/tests/test_knowledge_base.py b/tests/test_knowledge_base.py
@@ -3,7 +3,7 @@
 
 def test_knowledge_base():
     """Test the knowledge base"""
-    vectordb = FAISS.load_local(folder_path="./bioimageio-knowledge-base", index_name="bioimage.io", embeddings=OpenAIEmbeddings())
+    vectordb = FAISS.load_local(folder_path="./bioimageio-knowledge-base", index_name="bioimage.io", embeddings=OpenAIEmbeddings(), allow_dangerous_deserialization=True)
     retriever = vectordb.as_retriever(score_threshold=0.4)
     items = retriever.get_relevant_documents("community partner", verbose=True)
     assert len(items) > 0