jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 05:07:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Swap the Models label for the model name not it's display name, (#147918)

Browse Source 
Swap display name for name.
This commit is contained in:
Ivan Lopez Hernandez 2025-07-02 04:12:58 +00:00 committed by GitHub
parent 6842bfae4c
commit 2e7113d881
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
homeassistant/components/google_generative_ai_conversation/config_flow.py
View File

@ -330,13 +330,14 @@ async def google_generative_ai_config_option_schema(
api_models = [api_model async for api_model in api_models_pager] api_models = [api_model async for api_model in api_models_pager]
models = [ models = [
SelectOptionDict( SelectOptionDict(
label=api_model.display_name, label=api_model.name.lstrip("models/"),
value=api_model.name, value=api_model.name,
) )
for api_model in sorted(api_models, key=lambda x: x.display_name or "") for api_model in sorted(
api_models, key=lambda x: x.name.lstrip("models/") or ""
)
if ( if (
api_model.display_name api_model.name
and api_model.name
and ("tts" in api_model.name) == (subentry_type == "tts") and ("tts" in api_model.name) == (subentry_type == "tts")
and "vision" not in api_model.name and "vision" not in api_model.name
and api_model.supported_actions and api_model.supported_actions

4
tests/components/google_generative_ai_conversation/test_config_flow.py
View File

@ -43,25 +43,21 @@ from tests.common import MockConfigEntry
def get_models_pager(): def get_models_pager():
"""Return a generator that yields the models.""" """Return a generator that yields the models."""
model_25_flash = Mock( model_25_flash = Mock(
display_name="Gemini 2.5 Flash",
supported_actions=["generateContent"], supported_actions=["generateContent"],
) )
model_25_flash.name = "models/gemini-2.5-flash" model_25_flash.name = "models/gemini-2.5-flash"
model_20_flash = Mock( model_20_flash = Mock(
display_name="Gemini 2.0 Flash",
supported_actions=["generateContent"], supported_actions=["generateContent"],
) )
model_20_flash.name = "models/gemini-2.0-flash" model_20_flash.name = "models/gemini-2.0-flash"
model_15_flash = Mock( model_15_flash = Mock(
display_name="Gemini 1.5 Flash",
supported_actions=["generateContent"], supported_actions=["generateContent"],
) )
model_15_flash.name = "models/gemini-1.5-flash-latest" model_15_flash.name = "models/gemini-1.5-flash-latest"
model_15_pro = Mock( model_15_pro = Mock(
display_name="Gemini 1.5 Pro",
supported_actions=["generateContent"], supported_actions=["generateContent"],
) )
model_15_pro.name = "models/gemini-1.5-pro-latest" model_15_pro.name = "models/gemini-1.5-pro-latest"