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

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 Paulus Schoutsen
parent 1afbfd687f
commit 236b19c5b3
2 changed files with 3 additions and 4 deletions
Show Stats 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=RECOMMENDED_CHAT_MODEL)
model = genai.GenerativeModel(model_name=model_name)
try: try:
response = await model.generate_content_async(prompt_parts) 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( tuple(
), ),
dict({ dict({
'model_name': 'gemini-pro-vision', 'model_name': 'models/gemini-1.5-flash-latest',
}), }),
), ),
tuple( tuple(
@ -32,7 +32,7 @@
tuple( tuple(
), ),
dict({ dict({
'model_name': 'gemini-pro', 'model_name': 'models/gemini-1.5-flash-latest',
}), }),
), ),
tuple( tuple(