Default to gemini-2.5-flash (#147334)

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

View File

@ -9,7 +9,7 @@ CONF_PROMPT = "prompt"
ATTR_MODEL = "model"
CONF_RECOMMENDED = "recommended"
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"
CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0

View File

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

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='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',
]),
'model': 'models/gemini-2.0-flash',
'model': 'models/gemini-2.5-flash',
}),
),
])
@ -43,7 +43,7 @@
'contents': list([
'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([
'Write an opening speech for a Home Assistant release party',
]),
'model': 'models/gemini-2.0-flash',
'model': 'models/gemini-2.5-flash',
}),
),
])

View File

@ -41,6 +41,12 @@ from tests.common import MockConfigEntry
def get_models_pager():
"""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(
display_name="Gemini 2.0 Flash",
supported_actions=["generateContent"],
@ -59,17 +65,11 @@ def get_models_pager():
)
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():
yield model_25_flash
yield model_20_flash
yield model_15_flash
yield model_15_pro
yield model_10_pro
return models_pager()