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

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

Browse Source
This commit is contained in:
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_RECOMMENDED = "recommended"
CONF_CHAT_MODEL = "chat_model" 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" CONF_TEMPERATURE = "temperature"
RECOMMENDED_TEMPERATURE = 1.0 RECOMMENDED_TEMPERATURE = 1.0
CONF_TOP_P = "top_p" 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_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE', 'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -296,7 +296,7 @@
'top_k': 64, 'top_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE', 'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -348,7 +348,7 @@
'top_k': 64, 'top_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE', 'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -401,7 +401,7 @@
'top_k': 64, 'top_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE', 'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -454,7 +454,7 @@
'top_k': 64, 'top_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE', 'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -565,7 +565,7 @@
'top_k': 64, 'top_k': 64,
'top_p': 0.95, 'top_p': 0.95,
}), }),
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
'safety_settings': dict({ 'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE', 'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': '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**', 'api_key': '**REDACTED**',
}), }),
'options': dict({ 'options': dict({
'chat_model': 'models/gemini-1.5-flash-latest', 'chat_model': 'models/gemini-2.0-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',

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

@ -6,7 +6,7 @@
tuple( tuple(
), ),
dict({ dict({
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
}), }),
), ),
tuple( tuple(
@ -32,7 +32,7 @@
tuple( tuple(
), ),
dict({ dict({
'model_name': 'models/gemini-1.5-flash-latest', 'model_name': 'models/gemini-2.0-flash',
}), }),
), ),
tuple( 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 @pytest.fixture
def mock_models(): def mock_models():
"""Mock the model list API.""" """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( model_15_flash = Mock(
display_name="Gemini 1.5 Flash", display_name="Gemini 1.5 Flash",
supported_generation_methods=["generateContent"], supported_generation_methods=["generateContent"],
@ -58,7 +64,7 @@ def mock_models():
model_10_pro.name = "models/gemini-pro" model_10_pro.name = "models/gemini-pro"
with patch( with patch(
"homeassistant.components.google_generative_ai_conversation.config_flow.genai.list_models", "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 yield