Upgrade llama-index and reactivate Bedrock tests (#159)

datastax · Jan 11, 2024 · 9dd5288 · 9dd5288
1 parent 45ad913
commit 9dd5288
Showing 1 changed file with 2 additions and 4 deletions.
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
@@ -273,10 +273,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"),
     ],
 )