fix(vertex_ai): mark the Lyria 3 catalog entries text-only

`vertex_ai/lyria-3-clip-preview` and `vertex_ai/lyria-3-pro-preview` were
registered with `supports_vision`, `supports_image_input`, and an `image`
modality, which contradicts their `gemini/lyria-3-*` siblings and makes
/model/info advertise image input on text-to-music models.
This commit is contained in:
mateo-berri 2026-09-05 23:00:01 -07:00
parent 02b44820c4
commit 11e45ad953
3 changed files with 17 additions and 22 deletions

View file

@ -45513,7 +45513,7 @@
},
"vertex_ai/lyria-002": {
"litellm_provider": "vertex_ai",
"mode": "audio_speech",
"mode": "audio_speech",
"output_cost_per_image": 0.06,
"source": "https://cloud.google.com/gemini-enterprise-agent-platform/generative-ai/pricing#lyria",
"supported_audio_formats": [
@ -45549,8 +45549,7 @@
"/v1/audio/speech"
],
"supported_modalities": [
"text",
"image"
"text"
],
"supported_output_modalities": [
"audio"
@ -45561,11 +45560,10 @@
"supports_audio_input": false,
"supports_audio_output": true,
"supports_function_calling": false,
"supports_image_input": true,
"supports_prompt_caching": false,
"supports_response_schema": false,
"supports_system_messages": false,
"supports_vision": true,
"supports_vision": false,
"supports_web_search": false,
"vertex_ai_audio_api": "lyria_interactions"
},
@ -45588,8 +45586,7 @@
"/v1/audio/speech"
],
"supported_modalities": [
"text",
"image"
"text"
],
"supported_output_modalities": [
"audio"
@ -45600,11 +45597,10 @@
"supports_audio_input": false,
"supports_audio_output": true,
"supports_function_calling": false,
"supports_image_input": true,
"supports_prompt_caching": false,
"supports_response_schema": false,
"supports_system_messages": false,
"supports_vision": true,
"supports_vision": false,
"supports_web_search": false,
"vertex_ai_audio_api": "lyria_interactions"
},

View file

@ -45513,7 +45513,7 @@
},
"vertex_ai/lyria-002": {
"litellm_provider": "vertex_ai",
"mode": "audio_speech",
"mode": "audio_speech",
"output_cost_per_image": 0.06,
"source": "https://cloud.google.com/gemini-enterprise-agent-platform/generative-ai/pricing#lyria",
"supported_audio_formats": [
@ -45549,8 +45549,7 @@
"/v1/audio/speech"
],
"supported_modalities": [
"text",
"image"
"text"
],
"supported_output_modalities": [
"audio"
@ -45561,11 +45560,10 @@
"supports_audio_input": false,
"supports_audio_output": true,
"supports_function_calling": false,
"supports_image_input": true,
"supports_prompt_caching": false,
"supports_response_schema": false,
"supports_system_messages": false,
"supports_vision": true,
"supports_vision": false,
"supports_web_search": false,
"vertex_ai_audio_api": "lyria_interactions"
},
@ -45588,8 +45586,7 @@
"/v1/audio/speech"
],
"supported_modalities": [
"text",
"image"
"text"
],
"supported_output_modalities": [
"audio"
@ -45600,11 +45597,10 @@
"supports_audio_input": false,
"supports_audio_output": true,
"supports_function_calling": false,
"supports_image_input": true,
"supports_prompt_caching": false,
"supports_response_schema": false,
"supports_system_messages": false,
"supports_vision": true,
"supports_vision": false,
"supports_web_search": false,
"vertex_ai_audio_api": "lyria_interactions"
},

View file

@ -2887,14 +2887,17 @@ def test_vertex_ai_lyria_models_in_cost_map():
"/v1beta/interactions",
"/v1/audio/speech",
]
assert clip["supported_modalities"] == ["text", "image"]
assert pro["supported_modalities"] == ["text", "image"]
assert clip["supported_modalities"] == ["text"]
assert pro["supported_modalities"] == ["text"]
assert clip["supports_vision"] is False
assert pro["supports_vision"] is False
assert "supports_image_input" not in clip
assert "supports_image_input" not in pro
assert clip["supported_regions"] == ["global"]
assert pro["supported_regions"] == ["global"]
assert clip["supports_audio_output"] is True
assert pro["supports_audio_output"] is True
assert clip["supports_image_input"] is True
assert pro["supports_image_input"] is True
def test_model_info_for_fireworks_short_form_models():
"""