diff --git a/src/lib/components/workspace/Functions.svelte b/src/lib/components/workspace/Functions.svelte
index 8f0139009..35e308220 100644
--- a/src/lib/components/workspace/Functions.svelte
+++ b/src/lib/components/workspace/Functions.svelte
@@ -3,7 +3,7 @@
 	import fileSaver from 'file-saver';
 	const { saveAs } = fileSaver;
 
-	import { WEBUI_NAME, functions } from '$lib/stores';
+	import { WEBUI_NAME, functions, models } from '$lib/stores';
 	import { onMount, getContext } from 'svelte';
 	import { createNewPrompt, deletePromptByCommand, getPrompts } from '$lib/apis/prompts';
 
@@ -19,6 +19,7 @@
 	import ArrowDownTray from '../icons/ArrowDownTray.svelte';
 	import Tooltip from '../common/Tooltip.svelte';
 	import ConfirmDialog from '../common/ConfirmDialog.svelte';
+	import { getModels } from '$lib/apis';
 
 	const i18n = getContext('i18n');
 
@@ -226,7 +227,9 @@
 
 							if (res) {
 								toast.success('Function deleted successfully');
+
 								functions.set(await getFunctions(localStorage.token));
+								models.set(await getModels(localStorage.token));
 							}
 						}}
 					>
@@ -349,6 +352,7 @@
 
 			toast.success('Functions imported successfully');
 			functions.set(await getFunctions(localStorage.token));
+			models.set(await getModels(localStorage.token));
 		};
 
 		reader.readAsText(importFiles[0]);
diff --git a/src/routes/(app)/workspace/functions/create/+page.svelte b/src/routes/(app)/workspace/functions/create/+page.svelte
index 5faf3d50b..0f73cf94e 100644
--- a/src/routes/(app)/workspace/functions/create/+page.svelte
+++ b/src/routes/(app)/workspace/functions/create/+page.svelte
@@ -3,9 +3,10 @@
 	import { onMount } from 'svelte';
 	import { goto } from '$app/navigation';
 
-	import { functions } from '$lib/stores';
+	import { functions, models } from '$lib/stores';
 	import { createNewFunction, getFunctions } from '$lib/apis/functions';
 	import FunctionEditor from '$lib/components/workspace/Functions/FunctionEditor.svelte';
+	import { getModels } from '$lib/apis';
 
 	let mounted = false;
 	let clone = false;
@@ -26,6 +27,8 @@
 		if (res) {
 			toast.success('Function created successfully');
 			functions.set(await getFunctions(localStorage.token));
+			models.set(await getModels(localStorage.token));
+
 			await goto('/workspace/functions');
 		}
 	};
diff --git a/src/routes/(app)/workspace/functions/edit/+page.svelte b/src/routes/(app)/workspace/functions/edit/+page.svelte
index a61dda142..21fc5acb6 100644
--- a/src/routes/(app)/workspace/functions/edit/+page.svelte
+++ b/src/routes/(app)/workspace/functions/edit/+page.svelte
@@ -4,11 +4,12 @@
 
 	import { goto } from '$app/navigation';
 	import { page } from '$app/stores';
-	import { functions } from '$lib/stores';
+	import { functions, models } from '$lib/stores';
 	import { updateFunctionById, getFunctions, getFunctionById } from '$lib/apis/functions';
 
 	import FunctionEditor from '$lib/components/workspace/Functions/FunctionEditor.svelte';
 	import Spinner from '$lib/components/common/Spinner.svelte';
+	import { getModels } from '$lib/apis';
 
 	let func = null;
 
@@ -27,6 +28,7 @@
 		if (res) {
 			toast.success('Function updated successfully');
 			functions.set(await getFunctions(localStorage.token));
+			models.set(await getModels(localStorage.token));
 		}
 	};