[Fix] image prompt filtered models incorrectly

amidabuddha · Feb 1, 2025 · d81e7ad · d81e7ad
1 parent 81db60a
commit d81e7ad
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/console_gpt/menus/command_handler.py b/console_gpt/menus/command_handler.py
@@ -95,7 +95,7 @@ def command_handler(model_title, model_name, user_input, conversation, cached) -
                 return user_input
             return "continue"
         case "image":
-            if model_title.lower().startswith("mistral", "o1-mini", "o1-preview"):
+            if model_title.lower().startswith(('mistral', 'o1-mini', 'o1-preview')):
                 custom_print(
                     "error",
                     f"Cannot upload images unless you're using vision supported model. Current model: {model_name}!",