diff --git a/tests/integration_tests/dotenv_templates/dotenv.jinja2 b/tests/integration_tests/dotenv_templates/dotenv.jinja2 index ecca5f4b5c..a6ca554058 100644 --- a/tests/integration_tests/dotenv_templates/dotenv.jinja2 +++ b/tests/integration_tests/dotenv_templates/dotenv.jinja2 @@ -27,7 +27,7 @@ AZURE_COSMOSDB_CONVERSATIONS_CONTAINER={{ AZURE_COSMOSDB_CONVERSATIONS_CONTAINER AZURE_COSMOSDB_ACCOUNT_KEY={{ AZURE_COSMOSDB_ACCOUNT_KEY }} AZURE_COSMOSDB_ENABLE_FEEDBACK={{ AZURE_COSMOSDB_ENABLE_FEEDBACK }} {% endif %} -{% if DATASOURCE_TYPE == "AZURE_COGNITIVE_SEARCH" %} +{% if DATASOURCE_TYPE == "AzureCognitiveSearch" %} AZURE_SEARCH_SERVICE={{ AZURE_SEARCH_SERVICE }} AZURE_SEARCH_INDEX={{ AZURE_SEARCH_INDEX }} AZURE_SEARCH_KEY={{ AZURE_SEARCH_KEY }} @@ -39,7 +39,7 @@ AZURE_SEARCH_URL_COLUMN= AZURE_SEARCH_VECTOR_COLUMNS=contentVector AZURE_SEARCH_QUERY_TYPE={{ AZURE_SEARCH_QUERY_TYPE }} AZURE_SEARCH_PERMITTED_GROUPS_COLUMN={{ AZURE_SEARCH_PERMITTED_GROUPS_COLUMN }} -{% elif DATASOURCE_TYPE == "ELASTICSEARCH" %} +{% elif DATASOURCE_TYPE == "Elasticsearch" %} ELASTICSEARCH_ENDPOINT={{ ELASTICSEARCH_ENDPOINT }} ELASTICSEARCH_ENCODED_API_KEY={{ ELASTICSEARCH_ENCODED_API_KEY }} ELASTICSEARCH_INDEX={{ ELASTICSEARCH_INDEX }} @@ -52,7 +52,7 @@ ELASTICSEARCH_VECTOR_COLUMNS=text_embedding.predicted_value {% if USE_ELASTICSEARCH_EMBEDDINGS and ELASTICSEARCH_EMBEDDING_MODEL_ID %} ELASTICSEARCH_EMBEDDING_MODEL_ID={{ ELASTICSEARCH_EMBEDDING_MODEL_ID }} {% endif %} -{% elif DATASOURCE_TYPE == "AZURE_COSMOS_DB" %} +{% elif DATASOURCE_TYPE == "AzureCosmosDB" %} AZURE_COSMOSDB_MONGO_VCORE_CONNECTION_STRING={{ AZURE_COSMOSDB_MONGO_VCORE_CONNECTION_STRING }} AZURE_COSMOSDB_MONGO_VCORE_DATABASE={{ AZURE_COSMOSDB_MONGO_VCORE_DATABASE }} AZURE_COSMOSDB_MONGO_VCORE_CONTAINER={{ AZURE_COSMOSDB_MONGO_VCORE_CONTAINER }} @@ -62,7 +62,7 @@ AZURE_COSMOSDB_MONGO_VCORE_FILENAME_COLUMN= AZURE_COSMOSDB_MONGO_VCORE_TITLE_COLUMN= AZURE_COSMOSDB_MONGO_VCORE_URL_COLUMN= AZURE_COSMOSDB_MONGO_VCORE_VECTOR_COLUMNS={{ AZURE_COSMOSDB_MONGO_VCORE_VECTOR_COLUMNS }} -{% elif DATASOURCE_TYPE == "PINECONE" %} +{% elif DATASOURCE_TYPE == "Pinecone" %} PINECONE_ENVIRONMENT={{ PINECONE_ENVIRONMENT }} PINECONE_API_KEY={{ PINECONE_API_KEY }} PINECONE_INDEX_NAME={{ PINECONE_INDEX_NAME }} @@ -71,7 +71,7 @@ PINECONE_FILENAME_COLUMN= PINECONE_TITLE_COLUMN= PINECONE_URL_COLUMN= PINECONE_VECTOR_COLUMNS={{ PINECONE_VECTOR_COLUMNS }} -{% elif DATASOURCE_TYPE == "AZURE_ML_INDEX" %} +{% elif DATASOURCE_TYPE == "AzureMLIndex" %} AZURE_MLINDEX_NAME={{ AZURE_ML_INDEX_NAME }} AZURE_MLINDEX_VERSION={{ AZURE_ML_INDEX_VERSION }} AZURE_ML_PROJECT_RESOURCE_ID={{ AZURE_ML_PROJECT_RESOURCE_ID }} diff --git a/tests/integration_tests/test_datasources.py b/tests/integration_tests/test_datasources.py index 8b9571709f..aa3b90be4a 100644 --- a/tests/integration_tests/test_datasources.py +++ b/tests/integration_tests/test_datasources.py @@ -146,6 +146,7 @@ async def test_dotenv(test_app: Quart, dotenv_template_params: dict[str, str]): } ] } + test_client = test_app.test_client() response = await test_client.post(request_path, json=request_data) assert response.status_code == 200