evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Default to gemini-1.5-flash-latest in Google Generative AI ()

Browse source 
Default to flash
This commit is contained in:
tronikos 2024-05-29 05:04:47 -07:00 • committed by GitHub
parent 4056c4c2cc
commit aeee222df4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 8 additions and 8 deletions
homeassistant/components/google_generative_ai_conversation
const.py
tests/components/google_generative_ai_conversation/snapshots
test_conversation.ambrtest_diagnostics.ambr

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-pro-latest" RECOMMENDED_CHAT_MODEL = "models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'model_name': 'models/gemini-1.5-flash-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-pro-latest', 'chat_model': 'models/gemini-1.5-flash-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',