fix(drivers-prompt-azure-openai): fix AzureOpenAiChatPromptDriver by …

…removing unsupported "modalities" (#1694)
griptape-ai · Feb 11, 2025 · 573fb4e · 573fb4e
1 parent 7a095b6
commit 573fb4e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/griptape/drivers/prompt/azure_openai_chat_prompt_driver.py b/griptape/drivers/prompt/azure_openai_chat_prompt_driver.py
@@ -63,5 +63,6 @@ def _base_params(self, prompt_stack: PromptStack) -> dict:
                 del params["parallel_tool_calls"]
 
         # TODO: Add once Azure supports modalities
-        del params["modalities"]
+        if "modalities" in params:
+            del params["modalities"]
         return params
diff --git a/tests/unit/drivers/prompt/test_azure_openai_chat_prompt_driver.py b/tests/unit/drivers/prompt/test_azure_openai_chat_prompt_driver.py
@@ -143,7 +143,6 @@ def test_try_run(
             model="gpt-4",
             use_native_tools=use_native_tools,
             structured_output_strategy=structured_output_strategy,
-            modalities=modalities,
             extra_params={"foo": "bar"},
             api_version=api_version,
         )
@@ -157,7 +156,6 @@ def test_try_run(
             temperature=driver.temperature,
             user=driver.user,
             messages=messages,
-            modalities=driver.modalities,
             **{
                 "audio": driver.audio,
             }
@@ -231,7 +229,6 @@ def test_try_stream_run(
             stream=True,
             use_native_tools=use_native_tools,
             structured_output_strategy=structured_output_strategy,
-            modalities=modalities,
             extra_params={"foo": "bar"},
             api_version=api_version,
         )
@@ -250,7 +247,6 @@ def test_try_stream_run(
             }
             if "audio" in driver.modalities
             else {},
-            modalities=driver.modalities,
             stream=True,
             messages=messages,
             **{