blueprints/api.py

@@ -72,8 +72,7 @@ def generate():
 
         model_value = data.get('model')
         prompt = data.get('prompt')
-        model_lower = model_value.lower()
-        is_chat_model = ("gemini" in model_lower or "gpt" in model_lower) and "image" not in model_lower
+        is_chat_model = "gemini" in model_value.lower() or "gpt" in model_value.lower()
 
         # 3. 处理聊天模型 (同步)
         if is_chat_model: