Bump langchain versions (#584)

* Bump langchain versions * Use langchain GraphVectorStore * Pin ragstack-langchain version in notebook * Restore integration_tests/test_graph_store.py * Update extractor tests * Bump versions again to get LangChain extractors * Remove reimports
datastax · Jul 22, 2024 · 146f987 · 146f987
1 parent 8b50c12
commit 146f987
Show file tree

Hide file tree

Showing 21 changed files with 44 additions and 1,417 deletions.
diff --git a/libs/e2e-tests/pyproject.llamaindex.toml b/libs/e2e-tests/pyproject.llamaindex.toml
@@ -42,9 +42,9 @@ llama-index-multi-modal-llms-gemini = { git = "https://github.com/run-llama/llam
 
 llama-parse = { git = "https://github.com/run-llama/llama_parse.git", branch = "main" }
 
-langchain = "0.2.5"
-langchain-core = "0.2.9"
-langchain-community = "0.2.5"
+langchain = "0.2.10"
+langchain-core = "0.2.22"
+langchain-community = "0.2.9"
 langchain-astradb = "0.3.3"
 langchain-openai = "0.1.8"
 langchain-google-genai = { version = "1.0.6" }

diff --git a/libs/knowledge-store/notebooks/astra_support.ipynb b/libs/knowledge-store/notebooks/astra_support.ipynb
@@ -14,7 +14,7 @@
    "outputs": [],
    "source": [
     "%pip install -q \\\n",
-    "    ragstack-ai-langchain[knowledge-store] \\\n",
+    "    ragstack-ai-langchain[knowledge-store]==1.3.0 \\\n",
     "    beautifulsoup4 markdownify python-dotenv"
    ]
   },

diff --git a/libs/langchain/pyproject.toml b/libs/langchain/pyproject.toml
@@ -18,9 +18,9 @@ ragstack-ai-colbert = { version = "1.0.5", optional = true }
 ragstack-ai-knowledge-store = { version = "0.1.0", optional = true }
 
 # langchain
-langchain = "0.2.5"
-langchain-core = "0.2.9"
-langchain-community = "0.2.5"
+langchain = "0.2.10"
+langchain-core = "0.2.22"
+langchain-community = "0.2.9"
 langchain-astradb = "0.3.3"
 langchain-openai = "0.1.8"
 langchain-google-genai = { version = "1.0.6", optional = true }

diff --git a/libs/langchain/ragstack_langchain/graph_store/__init__.py b/libs/langchain/ragstack_langchain/graph_store/__init__.py
@@ -1,4 +1 @@
-from .base import GraphStore, Node
-from .cassandra import CassandraGraphStore
 
-__all__ = ["CassandraGraphStore", "GraphStore", "Node"]