unittestcase

Author: Pavan-Microsoft

documents/CustomizingAzdParameters.md

@@ -14,7 +14,7 @@ By default this template will use the environment name as the prefix to prevent
 | `AZURE_CONTENT_UNDERSTANDING_LOCATION`    | string  | `swedencentral`          | Specifies the region for content understanding resources.                  |
 | `AZURE_SECONDARY_LOCATION`                | string  | `eastus2`                | Specifies a secondary Azure region.                                        |
 | `AZURE_OPENAI_MODEL_DEPLOYMENT_TYPE`     | string  | `GlobalStandard`         | Defines the model deployment type (allowed: `Standard`, `GlobalStandard`). |
-| `AZURE_OPEN_AI_DEPLOYMENT_MODEL`          | string  | `gpt-4o-mini`            | Specifies the GPT model name (e.g., `gpt-4`, `gpt-4o-mini`).               |
+| `AZURE_OPENAI_DEPLOYMENT_MODEL`          | string  | `gpt-4o-mini`            | Specifies the GPT model name (e.g., `gpt-4`, `gpt-4o-mini`).               |
 | `AZURE_ENV_MODEL_VERSION`                 | string  | `2024-07-18`             | Sets the Azure model version (allowed: `2024-08-06`, etc.).                |
 | `AZURE_OPENAI_API_VERSION`            | string  | `2025-01-01-preview`     | Specifies the API version for Azure OpenAI.                                |
 | `AZURE_OPENAI_DEPLOYMENT_MODEL_CAPACITY` | integer | `30`                     | Sets the GPT model capacity.                                               |

src/api/.env.sample

@@ -18,5 +18,7 @@ DISPLAY_CHART_DEFAULT="False"
 REACT_APP_LAYOUT_CONFIG="{\n  \"appConfig\": {\n    \"THREE_COLUMN\": {\n      \"DASHBOARD\": 50,\n      \"CHAT\": 33,\n      \"CHATHISTORY\": 17\n    },\n    \"TWO_COLUMN\": {\n      \"DASHBOARD_CHAT\": {\n        \"DASHBOARD\": 65,\n        \"CHAT\": 35\n      },\n      \"CHAT_CHATHISTORY\": {\n        \"CHAT\": 80,\n        \"CHATHISTORY\": 20\n      }\n    }\n  },\n  \"charts\": [\n    {\n      \"id\": \"SATISFIED\",\n      \"name\": \"Satisfied\",\n      \"type\": \"card\",\n      \"layout\": { \"row\": 1, \"column\": 1, \"height\": 11 }\n    },\n    {\n      \"id\": \"TOTAL_CALLS\",\n      \"name\": \"Total Calls\",\n      \"type\": \"card\",\n      \"layout\": { \"row\": 1, \"column\": 2, \"span\": 1 }\n    },\n    {\n      \"id\": \"AVG_HANDLING_TIME\",\n      \"name\": \"Average Handling Time\",\n      \"type\": \"card\",\n      \"layout\": { \"row\": 1, \"column\": 3, \"span\": 1 }\n    },\n    {\n      \"id\": \"SENTIMENT\",\n      \"name\": \"Topics Overview\",\n      \"type\": \"donutchart\",\n      \"layout\": { \"row\": 2, \"column\": 1, \"width\": 40, \"height\": 44.5 }\n    },\n    {\n      \"id\": \"AVG_HANDLING_TIME_BY_TOPIC\",\n      \"name\": \"Average Handling Time By Topic\",\n      \"type\": \"bar\",\n      \"layout\": { \"row\": 2, \"column\": 2, \"row-span\": 2, \"width\": 60 }\n    },\n    {\n      \"id\": \"TOPICS\",\n      \"name\": \"Trending Topics\",\n      \"type\": \"table\",\n      \"layout\": { \"row\": 3, \"column\": 1, \"span\": 2 }\n    },\n    {\n      \"id\": \"KEY_PHRASES\",\n      \"name\": \"Key Phrases\",\n      \"type\": \"wordcloud\",\n      \"layout\": { \"row\": 3, \"column\": 2, \"height\": 44.5 }\n    }\n  ]\n}"
 SQLDB_DATABASE=
 SQLDB_SERVER=
+SQLDB_USER_MID=
+SQLDB_USERNAME=
 USE_AI_PROJECT_CLIENT="False"
 USE_CHAT_HISTORY_ENABLED="True"

src/api/common/config/config.py

@@ -6,6 +6,8 @@
 """
 
 import os
+from dotenv import load_dotenv
+load_dotenv()
 
 
 class Config:

src/tests/api/helpers/test_chat_helper copy.py

@@ -0,0 +1,45 @@
+from unittest.mock import patch, MagicMock
+import pytest
+import sys
+import os
+
+sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "../../../../api")))
+
+import helpers.azure_openai_helper as azure_openai_helper
+
+class TestAzureOpenAIHelper:
+    @patch("helpers.azure_openai_helper.openai.AzureOpenAI")
+    @patch("helpers.azure_openai_helper.get_bearer_token_provider")
+    @patch("helpers.azure_openai_helper.DefaultAzureCredential")
+    @patch("helpers.azure_openai_helper.Config")
+    def test_get_azure_openai_client(
+        self, mock_config, mock_credential, mock_token_provider, mock_azure_openai
+    ):
+        """Test that get_azure_openai_client returns a properly configured client."""
+        # Arrange
+        mock_config_instance = MagicMock()
+        mock_config_instance.azure_openai_endpoint = "https://test-endpoint"
+        mock_config_instance.azure_openai_api_version = "2024-01-01"
+        mock_config.return_value = mock_config_instance
+
+        mock_token = MagicMock()
+        mock_token_provider.return_value = mock_token
+
+        mock_client = MagicMock()
+        mock_azure_openai.return_value = mock_client
+
+        # Act
+        client = azure_openai_helper.get_azure_openai_client()
+
+        # Assert
+        mock_config.assert_called_once()
+        mock_credential.assert_called_once()
+        mock_token_provider.assert_called_once_with(
+            mock_credential.return_value, "https://cognitiveservices.azure.com/.default"
+        )
+        mock_azure_openai.assert_called_once_with(
+            azure_endpoint="https://test-endpoint",
+            api_version="2024-01-01",
+            azure_ad_token_provider=mock_token,
+        )
+        assert client == mock_client