File tree Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Expand file tree Collapse file tree 1 file changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -651,8 +651,8 @@ def download_model(model_type: str) -> str:
651651
652652 except Exception as e :
653653 return f"❌ Error downloading model: { str (e )} "
654-
655- elif "Ollama" in model_type :
654+ # all ollama models
655+ else :
656656 # Extract model name from model_type
657657 # Remove the 'Ollama - ' prefix and any leading/trailing whitespace
658658 model_name = model_type .replace ("Ollama - " , "" ).strip ()
@@ -703,8 +703,6 @@ def download_model(model_type: str) -> str:
703703 return "❌ Error: Could not connect to Ollama. Please make sure Ollama is installed and running."
704704 except Exception as e :
705705 return f"❌ Error pulling Ollama model: { str (e )} "
706- else :
707- return "❌ Error: Unknown model type"
708706
709707 except Exception as e :
710708 return f"❌ Error: { str (e )} "
You can’t perform that action at this time.
0 commit comments