fix: fix azure config tests

TensorOpsAI · Dec 18, 2024 · 8ea43c6 · 8ea43c6
1 parent 95c1723
commit 8ea43c6
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/libs/core/tests/unit_tests/conftest.py b/libs/core/tests/unit_tests/conftest.py
@@ -75,5 +75,10 @@ def _provider_config_name():
 def mock_azure_provider():
     config = MagicMock()
     config.id = "mock_azure_provider"
-    base_url = "mock_url.com"
-    return MockAzureProvider(config=config, base_url=base_url)
+    api_key="key"
+    api_endpoint="http://azureopenai.com"
+    api_version="2025-01-01-preview"
+    return MockAzureProvider(config=config, 
+                             api_endpoint=api_endpoint,
+                             api_key=api_key,
+                             api_version=api_version)