From 86a24cc3b901953bf0df26325f640a6a19ac567e Mon Sep 17 00:00:00 2001 From: Paulus Schoutsen Date: Fri, 24 May 2024 23:55:11 -0400 Subject: [PATCH] Fix default Google AI prompt on initial setup (#118078) --- .../google_generative_ai_conversation/config_flow.py | 2 +- .../google_generative_ai_conversation/test_config_flow.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/google_generative_ai_conversation/config_flow.py b/homeassistant/components/google_generative_ai_conversation/config_flow.py index 4fff5bff655..50b626f553c 100644 --- a/homeassistant/components/google_generative_ai_conversation/config_flow.py +++ b/homeassistant/components/google_generative_ai_conversation/config_flow.py @@ -63,7 +63,7 @@ STEP_USER_DATA_SCHEMA = vol.Schema( RECOMMENDED_OPTIONS = { CONF_RECOMMENDED: True, CONF_LLM_HASS_API: llm.LLM_API_ASSIST, - CONF_PROMPT: "", + CONF_PROMPT: DEFAULT_PROMPT, } diff --git a/tests/components/google_generative_ai_conversation/test_config_flow.py b/tests/components/google_generative_ai_conversation/test_config_flow.py index 6426386243c..55350325eee 100644 --- a/tests/components/google_generative_ai_conversation/test_config_flow.py +++ b/tests/components/google_generative_ai_conversation/test_config_flow.py @@ -19,6 +19,7 @@ from homeassistant.components.google_generative_ai_conversation.const import ( CONF_TEMPERATURE, CONF_TOP_K, CONF_TOP_P, + DEFAULT_PROMPT, DOMAIN, RECOMMENDED_CHAT_MODEL, RECOMMENDED_HARM_BLOCK_THRESHOLD, @@ -94,7 +95,7 @@ async def test_form(hass: HomeAssistant) -> None: assert result2["options"] == { CONF_RECOMMENDED: True, CONF_LLM_HASS_API: llm.LLM_API_ASSIST, - CONF_PROMPT: "", + CONF_PROMPT: DEFAULT_PROMPT, } assert len(mock_setup_entry.mock_calls) == 1