From 922628c1ee21cb1ae601aa1eca43bdbc3be156d6 Mon Sep 17 00:00:00 2001
From: Self Denial <selfdenial@pm.me>
Date: Sat, 13 Apr 2024 03:04:11 -0600
Subject: [PATCH] feat: small change to support ollama pull QOL

Use regex replace during trim "sanitize" to support `ollama run ` or `ollama pull ` syntax.
---
 src/lib/components/chat/Settings/Models.svelte | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/lib/components/chat/Settings/Models.svelte b/src/lib/components/chat/Settings/Models.svelte
index baa2f6c4..15b05402 100644
--- a/src/lib/components/chat/Settings/Models.svelte
+++ b/src/lib/components/chat/Settings/Models.svelte
@@ -139,7 +139,7 @@
 	};
 
 	const pullModelHandler = async () => {
-		const sanitizedModelTag = modelTag.trim();
+		const sanitizedModelTag = modelTag.trim().replace(/^ollama\s+(run|pull)\s+/, '');
 		if (modelDownloadStatus[sanitizedModelTag]) {
 			toast.error(
 				$i18n.t(`Model '{{modelTag}}' is already in queue for downloading.`, {