From 6732852ce6c1a2a445bc001b122c60cf12e0278b Mon Sep 17 00:00:00 2001
From: Timothy Jaeryang Baek <tim@openwebui.com>
Date: Mon, 27 Jul 2026 01:05:52 -0400
Subject: [PATCH] refac

---
 .../workspace/Models/Capabilities.svelte      | 22 +++++++------------
 src/lib/constants.ts                          |  1 +
 2 files changed, 9 insertions(+), 14 deletions(-)

diff --git a/src/lib/components/workspace/Models/Capabilities.svelte b/src/lib/components/workspace/Models/Capabilities.svelte
index 9808aa1123..825d65b39b 100644
--- a/src/lib/components/workspace/Models/Capabilities.svelte
+++ b/src/lib/components/workspace/Models/Capabilities.svelte
@@ -51,6 +51,10 @@
 			label: $i18n.t('Status Updates'),
 			description: $i18n.t('Displays status updates (e.g., web search progress) in the response')
 		},
+		memory: {
+			label: $i18n.t('Memory'),
+			description: $i18n.t('Inject stored memories into conversation context')
+		},
 		builtin_tools: {
 			label: $i18n.t('Builtin Tools'),
 			description: $i18n.t(
@@ -59,22 +63,12 @@
 		}
 	};
 
-	export let capabilities: {
-		file_context?: boolean;
-		vision?: boolean;
-		file_upload?: boolean;
-		web_search?: boolean;
-		image_generation?: boolean;
-		code_interpreter?: boolean;
-		terminal?: boolean;
-		usage?: boolean;
-		citations?: boolean;
-		status_updates?: boolean;
-		builtin_tools?: boolean;
-	} = {};
+	type Capability = keyof typeof capabilityLabels;
+
+	export let capabilities: Partial<Record<Capability, boolean>> = {};
 
 	// Hide file_context when file_upload is disabled
-	$: visibleCapabilities = Object.keys(capabilityLabels).filter((cap) => {
+	$: visibleCapabilities = (Object.keys(capabilityLabels) as Capability[]).filter((cap) => {
 		if (cap === 'file_context' && !capabilities.file_upload) {
 			return false;
 		}
diff --git a/src/lib/constants.ts b/src/lib/constants.ts
index b25ec0f848..aacaa9fbca 100644
--- a/src/lib/constants.ts
+++ b/src/lib/constants.ts
@@ -108,6 +108,7 @@ export const DEFAULT_CAPABILITIES = {
 	citations: true,
 	status_updates: true,
 	usage: undefined,
+	memory: true,
 	builtin_tools: true
 };