Browse Source

feat: small change to support ollama pull QOL

Use regex replace during trim "sanitize" to support `ollama run ` or `ollama pull ` syntax.
Self Denial 1 year ago
parent
commit
922628c1ee
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/lib/components/chat/Settings/Models.svelte

+ 1 - 1
src/lib/components/chat/Settings/Models.svelte

@@ -139,7 +139,7 @@
 	};
 	};
 
 
 	const pullModelHandler = async () => {
 	const pullModelHandler = async () => {
-		const sanitizedModelTag = modelTag.trim();
+		const sanitizedModelTag = modelTag.trim().replace(/^ollama\s+(run|pull)\s+/, '');
 		if (modelDownloadStatus[sanitizedModelTag]) {
 		if (modelDownloadStatus[sanitizedModelTag]) {
 			toast.error(
 			toast.error(
 				$i18n.t(`Model '{{modelTag}}' is already in queue for downloading.`, {
 				$i18n.t(`Model '{{modelTag}}' is already in queue for downloading.`, {