diff --git a/custom_components/groq_cloud_api/config_flow.py b/custom_components/groq_cloud_api/config_flow.py index 8b3304a..3a3f85f 100644 --- a/custom_components/groq_cloud_api/config_flow.py +++ b/custom_components/groq_cloud_api/config_flow.py @@ -71,6 +71,9 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]): LOGGER.debug("User validation got: %s", data) data[CONF_API_KEY] = obscured_api_key + if data.get(CONF_TEMPERATURE) is None: + data[CONF_CHAT_MODEL] = RECOMMENDED_CHAT_MODEL + response = await asyncio.to_thread( requests.get, url="https://api.groq.com/openai/v1/models", @@ -92,7 +95,7 @@ async def validate_input(hass: HomeAssistant, data: dict[str, Any]): raise UnknownError for model in response.json().get("data", []): - if model.get("id") == data.get(CONF_MODEL): + if model.get("id") == data.get(CONF_CHAT_MODEL): break if model == response.json().get("data")[-1]: raise ModelNotFound diff --git a/custom_components/groq_cloud_api/const.py b/custom_components/groq_cloud_api/const.py index 1662ace..822ea43 100644 --- a/custom_components/groq_cloud_api/const.py +++ b/custom_components/groq_cloud_api/const.py @@ -10,7 +10,7 @@ DEFAULT_NAME = "Groq Cloud API" CONF_RECOMMENDED = "recommended" CONF_PROMPT = "prompt" CONF_CHAT_MODEL = "chat_model" -RECOMMENDED_CHAT_MODEL = "llama3-groq-70b-8192-tool-use-preview" +RECOMMENDED_CHAT_MODEL = "llama3-groq-8b-8192-tool-use-preview" CONF_MAX_TOKENS = "max_tokens" RECOMMENDED_MAX_TOKENS = 150 CONF_TOP_P = "top_p" diff --git a/custom_components/groq_cloud_api/manifest.json b/custom_components/groq_cloud_api/manifest.json index 754f80b..feb6a11 100644 --- a/custom_components/groq_cloud_api/manifest.json +++ b/custom_components/groq_cloud_api/manifest.json @@ -1,7 +1,7 @@ { "domain": "groq_cloud_api", "name": "Groq Cloud API", - "version": "0.0.5", + "version": "0.0.6", "codeowners": [ "@HunorLaczko" ],