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

Default to gemini-2.5-flash (#147334)

Browse Source
This commit is contained in:
tronikos 2025-06-23 11:11:16 -07:00 committed by GitHub
parent 442fb88011
commit 2833e97625
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/google_generative_ai_conversation/const.py
View File

@ -9,7 +9,7 @@ CONF_PROMPT = "prompt"
ATTR_MODEL = "model" ATTR_MODEL = "model"
CONF_RECOMMENDED = "recommended" CONF_RECOMMENDED = "recommended"
CONF_CHAT_MODEL = "chat_model" CONF_CHAT_MODEL = "chat_model"
RECOMMENDED_CHAT_MODEL = "models/gemini-2.0-flash" RECOMMENDED_CHAT_MODEL = "models/gemini-2.5-flash"
RECOMMENDED_TTS_MODEL = "gemini-2.5-flash-preview-tts" RECOMMENDED_TTS_MODEL = "gemini-2.5-flash-preview-tts"
CONF_TEMPERATURE = "temperature" CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0 RECOMMENDED_TEMPERATURE = 1.0

2
tests/components/google_generative_ai_conversation/snapshots/test_diagnostics.ambr
View File

@ -5,7 +5,7 @@
'api_key': '**REDACTED**', 'api_key': '**REDACTED**',
}), }),
'options': dict({ 'options': dict({
'chat_model': 'models/gemini-2.0-flash', 'chat_model': 'models/gemini-2.5-flash',
'dangerous_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE', 'dangerous_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',
'harassment_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE', 'harassment_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',
'hate_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE', 'hate_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',

8
tests/components/google_generative_ai_conversation/snapshots/test_init.ambr
View File

@ -11,7 +11,7 @@
File(name='doorbell_snapshot.jpg', display_name=None, mime_type=None, size_bytes=None, create_time=None, expiration_time=None, update_time=None, sha256_hash=None, uri=None, download_uri=None, state=<FileState.ACTIVE: 'ACTIVE'>, source=None, video_metadata=None, error=None), File(name='doorbell_snapshot.jpg', display_name=None, mime_type=None, size_bytes=None, create_time=None, expiration_time=None, update_time=None, sha256_hash=None, uri=None, download_uri=None, state=<FileState.ACTIVE: 'ACTIVE'>, source=None, video_metadata=None, error=None),
File(name='context.txt', display_name=None, mime_type=None, size_bytes=None, create_time=None, expiration_time=None, update_time=None, sha256_hash=None, uri=None, download_uri=None, state=<FileState.PROCESSING: 'PROCESSING'>, source=None, video_metadata=None, error=None), File(name='context.txt', display_name=None, mime_type=None, size_bytes=None, create_time=None, expiration_time=None, update_time=None, sha256_hash=None, uri=None, download_uri=None, state=<FileState.PROCESSING: 'PROCESSING'>, source=None, video_metadata=None, error=None),
]), ]),
'model': 'models/gemini-2.0-flash', 'model': 'models/gemini-2.5-flash',
}), }),
), ),
]) ])
@ -28,7 +28,7 @@
b'some file', b'some file',
b'some file', b'some file',
]), ]),
'model': 'models/gemini-2.0-flash', 'model': 'models/gemini-2.5-flash',
}), }),
), ),
]) ])
@ -43,7 +43,7 @@
'contents': list([ 'contents': list([
'Write an opening speech for a Home Assistant release party', 'Write an opening speech for a Home Assistant release party',
]), ]),
'model': 'models/gemini-2.0-flash', 'model': 'models/gemini-2.5-flash',
}), }),
), ),
]) ])
@ -58,7 +58,7 @@
'contents': list([ 'contents': list([
'Write an opening speech for a Home Assistant release party', 'Write an opening speech for a Home Assistant release party',
]), ]),
'model': 'models/gemini-2.0-flash', 'model': 'models/gemini-2.5-flash',
}), }),
), ),
]) ])

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

@ -41,6 +41,12 @@ 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(
display_name="Gemini 2.5 Flash",
supported_actions=["generateContent"],
)
model_25_flash.name = "models/gemini-2.5-flash"
model_20_flash = Mock( model_20_flash = Mock(
display_name="Gemini 2.0 Flash", display_name="Gemini 2.0 Flash",
supported_actions=["generateContent"], supported_actions=["generateContent"],
@ -59,17 +65,11 @@ def get_models_pager():
) )
model_15_pro.name = "models/gemini-1.5-pro-latest" model_15_pro.name = "models/gemini-1.5-pro-latest"
model_10_pro = Mock(
display_name="Gemini 1.0 Pro",
supported_actions=["generateContent"],
)
model_10_pro.name = "models/gemini-pro"
async def models_pager(): async def models_pager():
yield model_25_flash
yield model_20_flash yield model_20_flash
yield model_15_flash yield model_15_flash
yield model_15_pro yield model_15_pro
yield model_10_pro
return models_pager() return models_pager()