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

Update the recommended model for Google Gen AI (#118323)

Browse Source
This commit is contained in:
Paulus Schoutsen 2024-05-28 21:21:28 -04:00 committed by GitHub
parent 8d7dff0228
commit 0cf574dc42
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 15 additions and 9 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-1.5-pro-latest"
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

@ -12,7 +12,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-1.5-pro-latest',
'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',
@ -64,7 +64,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-1.5-pro-latest',
'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',
@ -128,7 +128,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-1.5-pro-latest',
'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',
@ -184,7 +184,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-1.5-pro-latest',
'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',
@ -240,7 +240,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-1.5-pro-latest',
'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-1.5-pro-latest',
'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-1.5-pro-latest',
'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/test_config_flow.py
View File

@ -45,6 +45,12 @@ def mock_models():
) )
model_15_flash.name = "models/gemini-1.5-flash-latest" model_15_flash.name = "models/gemini-1.5-flash-latest"
model_15_pro = Mock(
display_name="Gemini 1.5 Pro",
supported_generation_methods=["generateContent"],
)
model_15_pro.name = "models/gemini-1.5-pro-latest"
model_10_pro = Mock( model_10_pro = Mock(
display_name="Gemini 1.0 Pro", display_name="Gemini 1.0 Pro",
supported_generation_methods=["generateContent"], supported_generation_methods=["generateContent"],
@ -52,7 +58,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_10_pro]), return_value=iter([model_15_flash, model_15_pro, model_10_pro]),
): ):
yield yield