tests updated

microsoft · Oct 30, 2024 · 3611774 · 3611774
1 parent 71615a2
commit 3611774
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/packages/ai/tests/ai/planners/test_assistants_planner.py b/python/packages/ai/tests/ai/planners/test_assistants_planner.py
@@ -560,7 +560,7 @@ async def test_create_openai_assistant(self, mock_async_openai):
         params = beta.AssistantCreateParams(model="123")
 
         assistant = await AssistantsPlanner.create_assistant(
-            api_key="", api_version="", organization="", endpoint="", request=params
+            api_key="", azure_ad_token_provider=None, api_version="", organization="", endpoint="", request=params
         )
 
         self.assertTrue(mock_async_openai.called)
@@ -573,6 +573,7 @@ async def test_create_azure_openai_assistant(self, mock_async_azure_openai):
 
         assistant = await AssistantsPlanner.create_assistant(
             api_key="",
+            azure_ad_token_provider=None,
             api_version="",
             organization="",
             endpoint="this is my endpoint",