From 74a4f642fd891a889f3395c131acd2af384c99da Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Mon, 24 Jun 2024 10:47:40 -0700
Subject: [PATCH] refac: valves save

---
 src/lib/components/workspace/Functions.svelte        | 12 ++++++++++--
 .../components/workspace/common/ValvesModal.svelte   |  1 +
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/src/lib/components/workspace/Functions.svelte b/src/lib/components/workspace/Functions.svelte
index dbb050b0d..faad68214 100644
--- a/src/lib/components/workspace/Functions.svelte
+++ b/src/lib/components/workspace/Functions.svelte
@@ -4,7 +4,7 @@
 	const { saveAs } = fileSaver;
 
 	import { WEBUI_NAME, functions, models } from '$lib/stores';
-	import { onMount, getContext } from 'svelte';
+	import { onMount, getContext, tick } from 'svelte';
 	import { createNewPrompt, deletePromptByCommand, getPrompts } from '$lib/apis/prompts';
 
 	import { goto } from '$app/navigation';
@@ -387,7 +387,15 @@
 </div>
 
 <ManifestModal bind:show={showManifestModal} manifest={selectedFunction?.meta?.manifest ?? {}} />
-<ValvesModal bind:show={showValvesModal} type="function" id={selectedFunction?.id ?? null} />
+<ValvesModal
+	bind:show={showValvesModal}
+	type="function"
+	id={selectedFunction?.id ?? null}
+	on:save={async () => {
+		await tick();
+		models.set(await getModels(localStorage.token));
+	}}
+/>
 
 <ConfirmDialog
 	bind:show={showConfirm}
diff --git a/src/lib/components/workspace/common/ValvesModal.svelte b/src/lib/components/workspace/common/ValvesModal.svelte
index 0bf2b8f19..e4bdfb567 100644
--- a/src/lib/components/workspace/common/ValvesModal.svelte
+++ b/src/lib/components/workspace/common/ValvesModal.svelte
@@ -52,6 +52,7 @@
 
 			if (res) {
 				toast.success('Valves updated successfully');
+				dispatch('save');
 			}
 		}