AtHeartEngineering/ai_automation_suggester
1
0
Fork 0
mirror of https://github.com/AtHeartEngineering/ai_automation_suggester.git synced 2026-01-09 23:28:03 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #105 from RmG152/fix-config_flow

Browse Source 
Fix config flow
This commit is contained in:
Graham Hosking
2025-05-24 06:25:59 +01:00
committed by GitHub
parent 7b4f6fe1ca 6cfa28c49f
commit 597b7436a0
1 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
custom_components/ai_automation_suggester/config_flow.py
View File

@@ -211,7 +211,11 @@ class AIAutomationConfigFlow(config_entries.ConfigFlow, domain=DOMAIN):
self.validator = ProviderValidator(self.hass)
err = await validate_fn(user_input)
if err is None:
self.data.update(user_input)
self.data.update({
**user_input,
CONF_MAX_INPUT_TOKENS: user_input.get(CONF_MAX_INPUT_TOKENS, DEFAULT_MAX_INPUT_TOKENS),
CONF_MAX_OUTPUT_TOKENS: user_input.get(CONF_MAX_OUTPUT_TOKENS, DEFAULT_MAX_OUTPUT_TOKENS),
})
return self.async_create_entry(title=title, data=self.data)
errors["base"] = "api_error"
placeholders["error_message"] = err
@@ -414,17 +418,41 @@ class AIAutomationOptionsFlowHandler(config_entries.OptionsFlow):
async def async_step_init(self, user_input=None):
if user_input:
return self.async_create_entry(title="", data=user_input)
new_data = {
**self.config_entry.options,
**user_input,
CONF_MAX_INPUT_TOKENS: user_input.get(
CONF_MAX_INPUT_TOKENS,
self.config_entry.options.get(
CONF_MAX_INPUT_TOKENS,
self.config_entry.data.get(CONF_MAX_INPUT_TOKENS, DEFAULT_MAX_INPUT_TOKENS)
)
),
CONF_MAX_OUTPUT_TOKENS: user_input.get(
CONF_MAX_OUTPUT_TOKENS,
self.config_entry.options.get(
CONF_MAX_OUTPUT_TOKENS,
self.config_entry.data.get(CONF_MAX_OUTPUT_TOKENS, DEFAULT_MAX_OUTPUT_TOKENS)
)
),
}
return self.async_create_entry(title="", data=new_data)
provider = self.config_entry.data.get(CONF_PROVIDER)
schema: Dict[Any, Any] = {
vol.Optional(
CONF_MAX_INPUT_TOKENS,
default=self.config_entry.data.get(CONF_MAX_INPUT_TOKENS, DEFAULT_MAX_INPUT_TOKENS),
default=self.config_entry.options.get(
CONF_MAX_INPUT_TOKENS,
self.config_entry.data.get(CONF_MAX_INPUT_TOKENS, DEFAULT_MAX_INPUT_TOKENS)
)
): vol.All(vol.Coerce(int), vol.Range(min=100)),
vol.Optional(
CONF_MAX_OUTPUT_TOKENS,
default=self.config_entry.data.get(CONF_MAX_OUTPUT_TOKENS, DEFAULT_MAX_OUTPUT_TOKENS),
default=self.config_entry.options.get(
CONF_MAX_OUTPUT_TOKENS,
self.config_entry.data.get(CONF_MAX_OUTPUT_TOKENS, DEFAULT_MAX_OUTPUT_TOKENS)
)
): vol.All(vol.Coerce(int), vol.Range(min=100)),
}