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

Use gemini-1.5-flash-latest in google_generative_ai_conversation.generate_content (#118594)

Browse source
This commit is contained in:
tronikos 2024-05-31 21:57:14 -07:00 • committed by GitHub
parent 7af469f81e
commit a4612143e6
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
homeassistant/components/google_generative_ai_conversation/__init__.py
View file

@ -66,8 +66,7 @@ async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
}
)
model_name = "gemini-pro-vision" if image_filenames else "gemini-pro"
model = genai.GenerativeModel(model_name=model_name)
model = genai.GenerativeModel(model_name=RECOMMENDED_CHAT_MODEL)
try:
response = await model.generate_content_async(prompt_parts)

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

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