Robot/ha-anthropic-custom
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: pass setup model into initial subentries; fall back to entry.data model in entity

Browse Source
This commit is contained in:
Robot committed 2026-02-26 17:44:56 +00:00
1 parent 1492d4b503
commit 34c0903e17
2 files changed
+13 -4
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

custom_components/anthropic_custom/config_flow.py
+11 -2
View File
@@ -200,19 +200,28 @@ class AnthropicConfigFlow(ConfigFlow, domain=DOMAIN):
return self.async_update_reload_and_abort(
self._get_reauth_entry(), data_updates=user_input
)
# Build initial subentry data with the model from setup
conversation_data = {
**DEFAULT_CONVERSATION_OPTIONS,
CONF_CHAT_MODEL: user_input.get(CONF_CHAT_MODEL, DEFAULT[CONF_CHAT_MODEL]),
}
ai_task_data = {
**DEFAULT_AI_TASK_OPTIONS,
CONF_CHAT_MODEL: user_input.get(CONF_CHAT_MODEL, DEFAULT[CONF_CHAT_MODEL]),
}
return self.async_create_entry(
title="Claude",
data=user_input,
subentries=[
{
"subentry_type": "conversation",
"data": DEFAULT_CONVERSATION_OPTIONS,
"data": conversation_data,
"title": DEFAULT_CONVERSATION_NAME,
"unique_id": None,
},
{
"subentry_type": "ai_task_data",
"data": DEFAULT_AI_TASK_OPTIONS,
"data": ai_task_data,
"title": DEFAULT_AI_TASK_NAME,
"unique_id": None,
},
custom_components/anthropic_custom/entity.py
+2 -2
View File
@@ -650,7 +650,7 @@ class AnthropicBaseLLMEntity(Entity):
identifiers={(DOMAIN, subentry.subentry_id)},
name=subentry.title,
manufacturer="Anthropic",
model=subentry.data.get(CONF_CHAT_MODEL, DEFAULT[CONF_CHAT_MODEL]),
model=subentry.data.get(CONF_CHAT_MODEL) or entry.data.get(CONF_CHAT_MODEL) or DEFAULT[CONF_CHAT_MODEL],
entry_type=dr.DeviceEntryType.SERVICE,
)
@@ -679,7 +679,7 @@ class AnthropicBaseLLMEntity(Entity):
messages, container_id = _convert_content(chat_log.content[1:])
model = options.get(CONF_CHAT_MODEL, DEFAULT[CONF_CHAT_MODEL])
model = options.get(CONF_CHAT_MODEL) or self.entry.data.get(CONF_CHAT_MODEL) or DEFAULT[CONF_CHAT_MODEL]
model_args = MessageCreateParamsStreaming(
model=model,