Test update

microsoft · Aug 6, 2024 · dc006a5 · dc006a5
1 parent b5662f0
commit dc006a5
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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