Add langchain-astradb package (#314)

Author: cbornet

pyproject.toml

@@ -13,13 +13,14 @@ packages = [{ include = "ragstack" }]
 python = ">=3.9,<3.12"
 astrapy = "~0.7.0"
 cassio = "~0.1.3"
-unstructured = { version = "0.12.5" }
+unstructured = "0.12.5"
 llama-index = { version = "0.9.48", extras = ["langchain"] }
-llama-parse = { version = "0.1.4" }
-langchain = { version = "0.1.4" }
+llama-parse = "0.1.4"
+langchain = "0.1.4"
+langchain-astradb = "0.0.1"
 langchain-core = "0.1.16"
 langchain-community = "0.0.16"
-langchain-openai = { version = "0.0.3" }
+langchain-openai = "0.0.3"
 langchain-google-genai = { version = "0.0.6", optional = true }
 langchain-google-vertexai = { version = "0.0.3", optional = true }
 langchain-nvidia-ai-endpoints = { version = "0.0.1", optional = true }

ragstack-e2e-tests/pyproject.langchain.toml

@@ -27,8 +27,9 @@ beautifulsoup4 = "^4"
 langchain = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/langchain", extras = [
     "openai",
 ] }
-langchain_core = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/core" }
-langchain_community = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/community" }
+langchain-astradb = { git = "https://github.com/langchain-ai/langchain-datastax.git", branch = "main", subdirectory = "libs/astradb" }
+langchain-core = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/core" }
+langchain-community = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/community" }
 langchain-openai = { git = "https://github.com/langchain-ai/langchain.git", branch = "master", subdirectory = "libs/partners/openai" }
 langchain-google-genai = { git = "https://github.com/langchain-ai/langchain-google.git", branch = "main", subdirectory = "libs/genai" }
 langchain-google-vertexai = { git = "https://github.com/langchain-ai/langchain-google.git", branch = "main", subdirectory = "libs/vertexai" }

ragstack-e2e-tests/pyproject.llamaindex.toml

@@ -41,17 +41,18 @@ 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 = { version = "0.1.2" }
+langchain = "0.1.2"
+langchain-astradb = "0.0.1"
 langchain-core = "0.1.15"
 langchain-community = "0.0.15"
-langchain-openai = { version = "0.0.3" }
-langchain-google-genai = { version = "0.0.6" }
-langchain-google-vertexai = { version = "0.0.3" }
-langchain-nvidia-ai-endpoints = { version = "0.0.1" }
+langchain-openai = "0.0.3"
+langchain-google-genai = "0.0.6"
+langchain-google-vertexai = "0.0.3"
+langchain-nvidia-ai-endpoints = "0.0.1"
 astrapy = "~0.7.0"
 # we need this specific feature from cassio: https://github.com/CassioML/cassio/pull/128
 cassio = "~0.1.4"
-unstructured = { version = "0.12.5" }
+unstructured = "0.12.5"
 
 [build-system]
 requires = ["poetry-core"]

ragstack-e2e-tests/pyproject.ragstack-ai.toml

@@ -28,6 +28,9 @@ beautifulsoup4 = "^4"
 # we need this specific feature from cassio: https://github.com/CassioML/cassio/pull/128
 cassio = "~0.1.4"
 
+# Remove when langchain-astradb is part of ragstack-ai package
+langchain-astradb = "0.0.1"
+
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"

tests/unit-tests/test_ragstack.py

@@ -2,6 +2,7 @@ def test_import():
     from llama_index.vector_stores import AstraDBVectorStore  # noqa
     from llama_index.vector_stores import CassandraVectorStore  # noqa
     from langchain.vectorstores import AstraDB  # noqa
+    from langchain_astradb import AstraDBVectorStore  # noqa
     import langchain_core  # noqa
     import langsmith  # noqa
     import astrapy  # noqa