From 718ed2ab8b36a02af17b623048d47cdf8d1aa4ad Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Nicol=C3=B2=20Boschi?= Date: Tue, 13 Feb 2024 10:44:16 +0100 Subject: [PATCH] fix llama-index parameter --- .../e2e_tests/llama_index/test_compatibility_rag.py | 2 +- ragstack-e2e-tests/e2e_tests/test_utils/vector_store_handler.py | 1 - 2 files changed, 1 insertion(+), 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 index 44786f080..6f28b1f37 100644 --- 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 index a90ed58b7..8fdcf7960 100644 --- 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