phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(azure/gpt_transformation.py): add azure audio model support

Browse source 
Closes https://github.com/BerriAI/litellm/issues/6305
This commit is contained in:
Krrish Dholakia 2025-03-19 16:20:30 -07:00
parent ef1bde153e
commit fe24b9d90b
3 changed files with 14 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/llms/azure/chat/gpt_transformation.py
View file

@ -99,6 +99,8 @@ class AzureOpenAIConfig(BaseConfig):
"extra_headers", "extra_headers",
"parallel_tool_calls", "parallel_tool_calls",
"prediction", "prediction",
"modalities",
"audio",
] ]
def _is_response_format_supported_model(self, model: str) -> bool: def _is_response_format_supported_model(self, model: str) -> bool:

1
litellm/llms/openai/chat/gpt_transformation.py
View file

@ -125,6 +125,7 @@ class OpenAIGPTConfig(BaseLLMModelInfo, BaseConfig):
"max_retries", "max_retries",
"extra_headers", "extra_headers",
"parallel_tool_calls", "parallel_tool_calls",
"audio",
] # works across all models ] # works across all models
model_specific_params = [] model_specific_params = []

11
tests/llm_translation/test_optional_params.py
View file

@ -1368,3 +1368,14 @@ def test_bedrock_invoke_anthropic_max_tokens():
assert "max_tokens_to_sample" not in optional_params assert "max_tokens_to_sample" not in optional_params
assert optional_params["max_tokens"] == 1024 assert optional_params["max_tokens"] == 1024
def test_azure_modalities_param():
optional_params = get_optional_params(
model="chatgpt-v2",
custom_llm_provider="azure",
modalities=["text", "audio"],
audio={"type": "audio_input", "input": "test.wav"},
)
assert optional_params["modalities"] == ["text", "audio"]
assert optional_params["audio"] == {"type": "audio_input", "input": "test.wav"}