Merge pull request #11 from TogetherCrew/feat/upgrade-llama-index-dep…

…endency feat: Added legacy llama-index library!
TogetherCrew · Mar 12, 2024 · 04f67ce · 04f67ce
2 parents 2315053 + 7339f7b
commit 04f67ce
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -9,7 +9,7 @@ neo4j>=5.8.0, <6.0.0
 llama-index-core==0.10.17
 llama-index-embeddings-openai==0.1.6
 llama-index-llms-openai==0.1.7
-llama-index-vector-stores-postgres==0.1.2
+llama-index-legacy
 sqlalchemy[asyncio]
 cohere>=4.39, <5.0.0
 pgvector

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 
 setup(
     name="tc-hivemind-backend",
-    version="1.1.4",
+    version="1.1.5",
     author="Mohammad Amin Dadgar, TogetherCrew",
     maintainer="Mohammad Amin Dadgar",
     maintainer_email="[email protected]",

diff --git a/tc_hivemind_backend/pg_vector_access.py b/tc_hivemind_backend/pg_vector_access.py
@@ -7,7 +7,7 @@
 from llama_index.core.node_parser import SimpleNodeParser
 from llama_index.core.node_parser.interface import MetadataAwareTextSplitter
 from llama_index.core.schema import BaseNode
-from llama_index.vector_stores.postgres import PGVectorStore
+from llama_index.legacy.vector_stores import PGVectorStore
 from tc_hivemind_backend.db.credentials import load_postgres_credentials
 from tc_hivemind_backend.db.utils.delete_data import delete_data
 from tc_hivemind_backend.db.utils.model_hyperparams import load_model_hyperparams