fix llama-index parameter

datastax · Feb 13, 2024 · 718ed2a · 718ed2a
1 parent 0d393ff
commit 718ed2a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 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
@@ -102,7 +102,7 @@ def bedrock_meta_llm():
         model="meta.llama2-13b-chat-v1",
         aws_access_key_id=get_required_env("AWS_ACCESS_KEY_ID"),
         aws_secret_access_key=get_required_env("AWS_SECRET_ACCESS_KEY"),
-        aws_region_name=get_required_env("BEDROCK_AWS_REGION"),
+        region_name=get_required_env("BEDROCK_AWS_REGION"),
     )
 
 

diff --git a/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py b/ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py
@@ -4,7 +4,6 @@
 
 from langchain_core.chat_history import BaseChatMessageHistory
 from langchain_core.vectorstores import VectorStore as LangChainVectorStore
-from llama_index.vector_stores.types import VectorStore as LLamaIndexVectorStore
 
 from e2e_tests.test_utils import skip_test_due_to_implementation_not_supported