Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update default Google model to Gemini Flash 2.0 (#137505)

pull/137508/head
Paulus Schoutsen 2025-02-05 23:27:13 -05:00 committed by GitHub
parent 39b46baeaa
commit 3b871afcc4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 17 additions and 11 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

@ -8,7 +8,7 @@ CONF_PROMPT = "prompt"
CONF_RECOMMENDED = "recommended"
CONF_CHAT_MODEL = "chat_model"
RECOMMENDED_CHAT_MODEL = "models/gemini-1.5-flash-latest"
RECOMMENDED_CHAT_MODEL = "models/gemini-2.0-flash"
CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0
CONF_TOP_P = "top_p"

12
tests/components/google_generative_ai_conversation/snapshots/test_conversation.ambr
View File

@ -244,7 +244,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -296,7 +296,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -348,7 +348,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -401,7 +401,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -454,7 +454,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -565,7 +565,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',

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

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

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

@ -6,7 +6,7 @@
tuple(
),
dict({
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
}),
),
tuple(
@ -32,7 +32,7 @@
tuple(
),
dict({
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-2.0-flash',
}),
),
tuple(

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

@ -39,6 +39,12 @@ from tests.common import MockConfigEntry
@pytest.fixture
def mock_models():
"""Mock the model list API."""
model_20_flash = Mock(
display_name="Gemini 2.0 Flash",
supported_generation_methods=["generateContent"],
)
model_20_flash.name = "models/gemini-2.0-flash"
model_15_flash = Mock(
display_name="Gemini 1.5 Flash",
supported_generation_methods=["generateContent"],
@ -58,7 +64,7 @@ def mock_models():
model_10_pro.name = "models/gemini-pro"
with patch(
"homeassistant.components.google_generative_ai_conversation.config_flow.genai.list_models",
return_value=iter([model_15_flash, model_15_pro, model_10_pro]),
return_value=iter([model_20_flash, model_15_flash, model_15_pro, model_10_pro]),
):
yield