Test update

tests/unit_tests/dotenv_data/dotenv_with_embedding_dependency_1

@@ -0,0 +1,37 @@
+# Chat
+DEBUG=True
+DATASOURCE_TYPE="AzureCognitiveSearch"
+AZURE_OPENAI_RESOURCE=
+AZURE_OPENAI_MODEL=my_model
+AZURE_OPENAI_KEY=dummy
+AZURE_OPENAI_MODEL_NAME=model_name
+AZURE_OPENAI_TEMPERATURE=0
+AZURE_OPENAI_TOP_P=1.0
+AZURE_OPENAI_MAX_TOKENS=1000
+AZURE_OPENAI_STOP_SEQUENCE=
+AZURE_OPENAI_SYSTEM_MESSAGE=You are an AI assistant that helps people find information.
+AZURE_OPENAI_PREVIEW_API_VERSION=2024-05-01-preview
+AZURE_OPENAI_STREAM=False
+AZURE_OPENAI_ENDPOINT=https://dummy.openai.azure.com/
+AZURE_OPENAI_EMBEDDING_NAME=embedding_model
+AZURE_OPENAI_EMBEDDING_ENDPOINT=
+AZURE_OPENAI_EMBEDDING_KEY=
+# Chat with data: common settings
+SEARCH_TOP_K=5
+SEARCH_STRICTNESS=3
+SEARCH_ENABLE_IN_DOMAIN=True
+# Chat with data: Azure AI Search
+AZURE_SEARCH_SERVICE=search_service
+AZURE_SEARCH_INDEX=search_index
+AZURE_SEARCH_KEY=dummy
+AZURE_SEARCH_SEMANTIC_SEARCH_CONFIG=
+AZURE_SEARCH_TOP_K=5
+AZURE_SEARCH_ENABLE_IN_DOMAIN=true
+AZURE_SEARCH_CONTENT_COLUMNS=content1,content2
+AZURE_SEARCH_FILENAME_COLUMN=filepath
+AZURE_SEARCH_TITLE_COLUMN=title
+AZURE_SEARCH_URL_COLUMN=url
+AZURE_SEARCH_VECTOR_COLUMNS=vector1
+AZURE_SEARCH_QUERY_TYPE=simple
+AZURE_SEARCH_PERMITTED_GROUPS_COLUMN=
+AZURE_SEARCH_STRICTNESS=3

tests/unit_tests/dotenv_data/dotenv_with_embedding_dependency_2

@@ -0,0 +1,37 @@
+# Chat
+DEBUG=True
+DATASOURCE_TYPE="AzureCognitiveSearch"
+AZURE_OPENAI_RESOURCE=
+AZURE_OPENAI_MODEL=my_model
+AZURE_OPENAI_KEY=dummy
+AZURE_OPENAI_MODEL_NAME=model_name
+AZURE_OPENAI_TEMPERATURE=0
+AZURE_OPENAI_TOP_P=1.0
+AZURE_OPENAI_MAX_TOKENS=1000
+AZURE_OPENAI_STOP_SEQUENCE=
+AZURE_OPENAI_SYSTEM_MESSAGE=You are an AI assistant that helps people find information.
+AZURE_OPENAI_PREVIEW_API_VERSION=2024-05-01-preview
+AZURE_OPENAI_STREAM=False
+AZURE_OPENAI_ENDPOINT=https://dummy.openai.azure.com/
+AZURE_OPENAI_EMBEDDING_NAME=
+AZURE_OPENAI_EMBEDDING_ENDPOINT=https://embeddings.openai.azure.com
+AZURE_OPENAI_EMBEDDING_KEY=dummy
+# Chat with data: common settings
+SEARCH_TOP_K=5
+SEARCH_STRICTNESS=3
+SEARCH_ENABLE_IN_DOMAIN=True
+# Chat with data: Azure AI Search
+AZURE_SEARCH_SERVICE=search_service
+AZURE_SEARCH_INDEX=search_index
+AZURE_SEARCH_KEY=dummy
+AZURE_SEARCH_SEMANTIC_SEARCH_CONFIG=
+AZURE_SEARCH_TOP_K=5
+AZURE_SEARCH_ENABLE_IN_DOMAIN=true
+AZURE_SEARCH_CONTENT_COLUMNS=content1,content2
+AZURE_SEARCH_FILENAME_COLUMN=filepath
+AZURE_SEARCH_TITLE_COLUMN=title
+AZURE_SEARCH_URL_COLUMN=url
+AZURE_SEARCH_VECTOR_COLUMNS=vector1
+AZURE_SEARCH_QUERY_TYPE=simple
+AZURE_SEARCH_PERMITTED_GROUPS_COLUMN=
+AZURE_SEARCH_STRICTNESS=3

tests/unit_tests/dotenv_data/dotenv_with_embedding_dependency_3

@@ -0,0 +1,37 @@
+# Chat
+DEBUG=True
+DATASOURCE_TYPE="AzureCognitiveSearch"
+AZURE_OPENAI_RESOURCE=
+AZURE_OPENAI_MODEL=my_model
+AZURE_OPENAI_KEY=dummy
+AZURE_OPENAI_MODEL_NAME=model_name
+AZURE_OPENAI_TEMPERATURE=0
+AZURE_OPENAI_TOP_P=1.0
+AZURE_OPENAI_MAX_TOKENS=1000
+AZURE_OPENAI_STOP_SEQUENCE=
+AZURE_OPENAI_SYSTEM_MESSAGE=You are an AI assistant that helps people find information.
+AZURE_OPENAI_PREVIEW_API_VERSION=2024-05-01-preview
+AZURE_OPENAI_STREAM=False
+AZURE_OPENAI_ENDPOINT=https://dummy.openai.azure.com/
+AZURE_OPENAI_EMBEDDING_NAME=
+AZURE_OPENAI_EMBEDDING_ENDPOINT=https://embeddings.openai.azure.com
+AZURE_OPENAI_EMBEDDING_KEY=
+# Chat with data: common settings
+SEARCH_TOP_K=5
+SEARCH_STRICTNESS=3
+SEARCH_ENABLE_IN_DOMAIN=True
+# Chat with data: Azure AI Search
+AZURE_SEARCH_SERVICE=search_service
+AZURE_SEARCH_INDEX=search_index
+AZURE_SEARCH_KEY=dummy
+AZURE_SEARCH_SEMANTIC_SEARCH_CONFIG=
+AZURE_SEARCH_TOP_K=5
+AZURE_SEARCH_ENABLE_IN_DOMAIN=true
+AZURE_SEARCH_CONTENT_COLUMNS=content1,content2
+AZURE_SEARCH_FILENAME_COLUMN=filepath
+AZURE_SEARCH_TITLE_COLUMN=title
+AZURE_SEARCH_URL_COLUMN=url
+AZURE_SEARCH_VECTOR_COLUMNS=vector1
+AZURE_SEARCH_QUERY_TYPE=simple
+AZURE_SEARCH_PERMITTED_GROUPS_COLUMN=
+AZURE_SEARCH_STRICTNESS=3

tests/unit_tests/test_settings.py

@@ -68,5 +68,42 @@ def test_dotenv_with_elasticsearch_success(app_settings):
     print(payload)
 
 
+def test_dotenv_with_embedding_dependency_1(app_settings):
+    # Validate model object
+    assert app_settings.azure_openai is not None
+    assert app_settings.azure_openai.embedding_name == "embedding_model"
+
+    # Validate API payload structure
+    payload = app_settings.datasource.construct_payload_configuration()
+    assert payload["parameters"]["embedding_dependency"]["type"] == "deployment_name"
+    assert payload["parameters"]["embedding_dependency"]["deployment_name"] == "embedding_model"
+    print(payload)
+
+
+def test_dotenv_with_embedding_dependency_2(app_settings):
+    # Validate model object
+    assert app_settings.azure_openai is not None
+    assert app_settings.azure_openai.embedding_endpoint == "https://embeddings.openai.azure.com"
+    assert app_settings.azure_openai.embedding_key == "dummy"
+
+    # Validate API payload structure
+    payload = app_settings.datasource.construct_payload_configuration()
+    assert payload["parameters"]["embedding_dependency"]["type"] == "api_key"
+    assert payload["parameters"]["embedding_dependency"]["key"] == "dummy"
+    print(payload)
+
+
+def test_dotenv_with_embedding_dependency_3(app_settings):
+    # Validate model object
+    assert app_settings.azure_openai is not None
+    assert app_settings.azure_openai.embedding_endpoint == "https://embeddings.openai.azure.com"
+    assert app_settings.azure_openai.embedding_key == None
+
+    # Validate API payload structure
+    payload = app_settings.datasource.construct_payload_configuration()
+    assert payload["parameters"]["embedding_dependency"]["type"] == "system_assigned_managed_identity"
+    print(payload)
+
+