From dd5e500e011b1581da63c6dd5649c654e8373dd7 Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Tue, 19 Nov 2024 14:23:19 -0800
Subject: [PATCH] enh: reintroduce update all ollama models feature

---
 .../Connections/ManageOllamaModal.svelte      | 39 ++++++++++++++++++-
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/src/lib/components/admin/Settings/Connections/ManageOllamaModal.svelte b/src/lib/components/admin/Settings/Connections/ManageOllamaModal.svelte
index 42e436c15..220214ed1 100644
--- a/src/lib/components/admin/Settings/Connections/ManageOllamaModal.svelte
+++ b/src/lib/components/admin/Settings/Connections/ManageOllamaModal.svelte
@@ -527,8 +527,37 @@
 <Modal size="sm" bind:show>
 	<div>
 		<div class=" flex justify-between dark:text-gray-100 px-5 pt-4 pb-2">
-			<div class=" text-lg font-medium self-center font-primary">
-				{$i18n.t('Manage Ollama')}
+			<div
+				class="flex w-full justify-between items-center text-lg font-medium self-center font-primary"
+			>
+				<div class=" flex-shrink-0">
+					{$i18n.t('Manage Ollama')}
+				</div>
+
+				<div>
+					<Tooltip content="Update All Models" placement="top">
+						<button
+							class="p-2.5 flex gap-2 items-center bg-transparent rounded-lg transition"
+							on:click={() => {
+								updateModelsHandler();
+							}}
+						>
+							<svg
+								xmlns="http://www.w3.org/2000/svg"
+								viewBox="0 0 16 16"
+								fill="currentColor"
+								class="w-4 h-4"
+							>
+								<path
+									d="M7 1a.75.75 0 0 1 .75.75V6h-1.5V1.75A.75.75 0 0 1 7 1ZM6.25 6v2.94L5.03 7.72a.75.75 0 0 0-1.06 1.06l2.5 2.5a.75.75 0 0 0 1.06 0l2.5-2.5a.75.75 0 1 0-1.06-1.06L7.75 8.94V6H10a2 2 0 0 1 2 2v3a2 2 0 0 1-2 2H4a2 2 0 0 1-2-2V8a2 2 0 0 1 2-2h2.25Z"
+								/>
+								<path
+									d="M4.268 14A2 2 0 0 0 6 15h6a2 2 0 0 0 2-2v-3a2 2 0 0 0-1-1.732V11a3 3 0 0 1-3 3H4.268Z"
+								/>
+							</svg>
+						</button>
+					</Tooltip>
+				</div>
 			</div>
 			<button
 				class="self-center"
@@ -554,6 +583,12 @@
 				<div class=" flex flex-col w-full">
 					<div>
 						<div class="space-y-2">
+							{#if updateModelId}
+								<div class="text-xs">
+									Updating "{updateModelId}" {updateProgress ? `(${updateProgress}%)` : ''}
+								</div>
+							{/if}
+
 							<div>
 								<div class=" mb-2 text-sm font-medium">
 									{$i18n.t('Pull a model from Ollama.com')}