Upgrade llama-index and reactivate Bedrock tests

datastax · Dec 16, 2023 · 8aafee2 · 8aafee2
1 parent c9a6584
commit 8aafee2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,7 +14,7 @@ python = ">=3.9,<3.12"
 astrapy = "~0.6.2"
 cassio = "~0.1.3"
 unstructured = "^0.10"
-llama-index = { version = "0.9.14", extras = ["langchain"] }
+llama-index = { version = "0.9.15.post2", extras = ["langchain"] }
 langchain = { version = "0.0.350", extras = ["openai"] }
 langchain-core = "0.1.1"
 langchain-community = "0.0.3"

diff --git a/ragstack-e2e-tests/e2e_tests/llama_index/test_compatibility_rag.py b/ragstack-e2e-tests/e2e_tests/llama_index/test_compatibility_rag.py
@@ -183,10 +183,8 @@ def huggingface_hub_embedding():
         ("openai_embedding", "openai_llm"),
         ("azure_openai_embedding", "azure_openai_llm"),
         ("vertex_embedding", "vertex_llm"),
-        # ("bedrock_titan_embedding", "bedrock_anthropic_llm"),
-        # Deactivated for now because of
-        # https://github.com/run-llama/llama_index/pull/9396
-        # ("bedrock_cohere_embedding", "bedrock_meta_llm"),
+        ("bedrock_titan_embedding", "bedrock_anthropic_llm"),
+        ("bedrock_cohere_embedding", "bedrock_meta_llm"),
         ("huggingface_hub_embedding", "huggingface_hub_llm"),
     ],
 )

diff --git a/ragstack-e2e-tests/pyproject.langchain.toml b/ragstack-e2e-tests/pyproject.langchain.toml
@@ -23,7 +23,7 @@ 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" }
-llama-index = { version = "0.9.14", extras = ["langchain"] }
+llama-index = { version = "0.9.15.post2", extras = ["langchain"] }
 astrapy = "^0.6"
 cassio = "^0.1"