fix saving config_flow options

ej52 · Nov 10, 2023 · 22071d2 · 22071d2
1 parent 66fd604
commit 22071d2
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/custom_components/ollama_conversation/config_flow.py b/custom_components/ollama_conversation/config_flow.py
@@ -139,6 +139,7 @@ class OllamaOptionsFlow(config_entries.OptionsFlow):
     def __init__(self, config_entry: config_entries.ConfigEntry) -> None:
         """Initialize options flow."""
         self.config_entry = config_entry
+        self.options = dict(config_entry.options)
 
     async def async_step_init(
         self, user_input: dict[str, Any] | None = None
@@ -154,8 +155,8 @@ async def async_step_prompt_system(
     ) -> FlowResult:
         """Manage Prompt Templates."""
         if user_input is not None:
-            user_input.update(self.config_entry.options)
-            return self.async_create_entry(title="", data=user_input)
+            self.options.update(user_input)
+            return self.async_create_entry(title="", data=self.options)
 
         schema = ollama_schema_prompt_system(self.config_entry.options)
         return self.async_show_form(
@@ -168,8 +169,8 @@ async def async_step_model_config(
     ) -> FlowResult:
         """Manage Model Settings."""
         if user_input is not None:
-            user_input.update(self.config_entry.options)
-            return self.async_create_entry(title="", data=user_input)
+            self.options.update(user_input)
+            return self.async_create_entry(title="", data=self.options)
 
         try:
             client = OllamaApiClient(