Add langchain-astradb package (part2) (#323)

Author: nicoloboschi

pyproject.toml

@@ -19,6 +19,7 @@ llama-parse = { version = "0.1.4" }
 langchain = "0.1.12"
 langchain-core = "0.1.31"
 langchain-community = "0.0.28"
+langchain-astradb = "0.0.1"
 langchain-openai = "0.0.8"
 langchain-google-genai = { version = "0.0.9", optional = true }
 langchain-google-vertexai = { version = "0.1.0", 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-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-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

@@ -42,16 +42,16 @@ 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.1.12"
+langchain-astradb = "0.0.1"
 langchain-core = "0.1.31"
 langchain-community = "0.0.28"
 langchain-openai = "0.0.8"
 langchain-google-genai = "0.0.9"
 langchain-google-vertexai = "0.1.0"
 langchain-nvidia-ai-endpoints = "0.0.3"
 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