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:
651
651
652
652
except Exception as e :
653
653
return f"❌ Error downloading model: { str (e )} "
654
-
655
- elif "Ollama" in model_type :
654
+ # all ollama models
655
+ else :
656
656
# Extract model name from model_type
657
657
# Remove the 'Ollama - ' prefix and any leading/trailing whitespace
658
658
model_name = model_type .replace ("Ollama - " , "" ).strip ()
@@ -703,8 +703,6 @@ def download_model(model_type: str) -> str:
703
703
return "❌ Error: Could not connect to Ollama. Please make sure Ollama is installed and running."
704
704
except Exception as e :
705
705
return f"❌ Error pulling Ollama model: { str (e )} "
706
- else :
707
- return "❌ Error: Unknown model type"
708
706
709
707
except Exception as e :
710
708
return f"❌ Error: { str (e )} "
You can’t perform that action at this time.
0 commit comments