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

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

pull/118342/head
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_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"
RECOMMENDED_TEMPERATURE = 1.0
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_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-1.5-pro-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -64,7 +64,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-1.5-pro-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -128,7 +128,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-1.5-pro-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -184,7 +184,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-1.5-pro-latest',
'safety_settings': dict({
'DANGEROUS': 'BLOCK_MEDIUM_AND_ABOVE',
'HARASSMENT': 'BLOCK_MEDIUM_AND_ABOVE',
@ -240,7 +240,7 @@
'top_k': 64,
'top_p': 0.95,
}),
'model_name': 'models/gemini-1.5-flash-latest',
'model_name': 'models/gemini-1.5-pro-latest',
'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-1.5-pro-latest',
'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-1.5-pro-latest',
'dangerous_block_threshold': 'BLOCK_MEDIUM_AND_ABOVE',
'harassment_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_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(
display_name="Gemini 1.0 Pro",
supported_generation_methods=["generateContent"],
@ -52,7 +58,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_10_pro]),
return_value=iter([model_15_flash, model_15_pro, model_10_pro]),
):
yield