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

fix(gemini/transformation.py): handle file_data being passed in (#9786)

Browse source
This commit is contained in:
Krish Dholakia 2025-04-07 16:32:08 -07:00 • committed by GitHub
parent fcb0d45930
commit 1c63e3dad0
2 changed files with 22 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/llm_translation/base_llm_unit_tests.py
View file

@ -928,7 +928,8 @@ class BaseLLMChatTest(ABC):
assert response._hidden_params["response_cost"] > 0
@pytest.mark.parametrize("input_type", ["input_audio", "audio_url"])
def test_supports_audio_input(self, input_type):
@pytest.mark.parametrize("format_specified", [True, False])
def test_supports_audio_input(self, input_type, format_specified):
from litellm.utils import return_raw_request, supports_audio_input
from litellm.types.utils import CallTypes
os.environ["LITELLM_LOCAL_MODEL_COST_MAP"] = "True"
@ -958,10 +959,16 @@ class BaseLLMChatTest(ABC):
test_file_id = "gs://bucket/file.wav"
if input_type == "input_audio":
audio_content.append({
"type": "input_audio",
"input_audio": {"data": encoded_string, "format": audio_format},
})
if format_specified:
audio_content.append({
"type": "input_audio",
"input_audio": {"data": encoded_string, "format": audio_format},
})
else:
audio_content.append({
"type": "input_audio",
"input_audio": {"data": encoded_string},
})
elif input_type == "audio_url":
audio_content.append(
{