fix model config

letta-ai · Nov 10, 2023 · 2ec6c3d · 2ec6c3d
1 parent 8a2c5ff
commit 2ec6c3d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/memgpt/cli/cli_config.py b/memgpt/cli/cli_config.py
@@ -122,8 +122,9 @@ def configure():
         model_options = []
         if use_openai:
             model_options += ["gpt-4", "gpt-4-1106-preview", "gpt-3.5-turbo-16k"]
+        valid_model = config.model in model_options
         default_model = questionary.select(
-            "Select default model (recommended: gpt-4):", choices=["gpt-4", "gpt-4-1106-preview", "gpt-3.5-turbo-16k"], default=config.model
+            "Select default model (recommended: gpt-4):", choices=model_options, default=config.model if valid_model else model_options[0]
         ).ask()
     else:
         default_model = "local"  # TODO: figure out if this is ok? this is for local endpoint