diff --git a/src/lib/components/chat/MessageInput.svelte b/src/lib/components/chat/MessageInput.svelte
index 07f5485936..26cca09a13 100644
--- a/src/lib/components/chat/MessageInput.svelte
+++ b/src/lib/components/chat/MessageInput.svelte
@@ -2358,6 +2358,7 @@
)
? selectedModelIds
: fileUploadCapableModels}
+ {visionCapableModels}
{toolApprovalMode}
{onToolApprovalModeChange}
{screenCaptureHandler}
diff --git a/src/lib/components/chat/MessageInput/InputMenu.svelte b/src/lib/components/chat/MessageInput/InputMenu.svelte
index 1916fe8fcc..a8dedb26bd 100644
--- a/src/lib/components/chat/MessageInput/InputMenu.svelte
+++ b/src/lib/components/chat/MessageInput/InputMenu.svelte
@@ -34,6 +34,7 @@
export let selectedModels: string[] = [];
export let fileUploadCapableModels: string[] = [];
+ export let visionCapableModels: string[] = [];
export let screenCaptureHandler: Function;
export let uploadFilesHandler: Function;
@@ -213,39 +214,41 @@
-
-
-
+
+
+ {/if}