Special handling for Azure OpenAI: fix test

microsoft · Sep 25, 2024 · c7298c5 · c7298c5
1 parent 3a5f8be
commit c7298c5
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/python/tests/unit/connectors/open_ai/services/test_azure_chat_completion.py b/python/tests/unit/connectors/open_ai/services/test_azure_chat_completion.py
@@ -787,6 +787,5 @@ async def test_cmc_streaming(
     mock_create.assert_awaited_once_with(
         model=azure_openai_unit_test_env["AZURE_OPENAI_CHAT_DEPLOYMENT_NAME"],
         stream=True,
-        stream_options={"include_usage": True},
         messages=azure_chat_completion._prepare_chat_history_for_request(chat_history),
     )