Merge pull request #204 from haesleinhuepf/persistent_endpoint_config

Persistent endpoint config
haesleinhuepf · Sep 3, 2024 · a0ccd9b · a0ccd9b
2 parents 0fec557 + eba27a0
commit a0ccd9b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/bia_bob/_machinery.py b/src/bia_bob/_machinery.py
@@ -207,6 +207,7 @@ def init_assistant(model=None, auto_execute:bool = False, variables:dict=None, e
     config = {
         "model": DEFAULT_MODEL,
         "vision_model": DEFAULT_VISION_MODEL,
+        "endpoint": None,
     }
 
     # load config from disk
@@ -222,10 +223,13 @@ def init_assistant(model=None, auto_execute:bool = False, variables:dict=None, e
         model = config["model"]
     if vision_model is None:
         vision_model = config["vision_model"]
+    if endpoint is None:
+        endpoint = config["endpoint"]
 
     # store config to disk
     config["model"] = model
     config["vision_model"] = vision_model
+    config["endpoint"] = endpoint
     with open(config_filename, 'w') as file:
         yaml.dump(config, file, default_flow_style=False)