diff --git a/client/src/components/Chat/Input/Files/AttachFileMenu.tsx b/client/src/components/Chat/Input/Files/AttachFileMenu.tsx
index 89259fadf7..33d01887e1 100644
--- a/client/src/components/Chat/Input/Files/AttachFileMenu.tsx
+++ b/client/src/components/Chat/Input/Files/AttachFileMenu.tsx
@@ -110,7 +110,11 @@ const AttachFileMenu = ({
           label: localize('com_ui_upload_provider'),
           onClick: () => {
             setToolResource(EToolResources.direct_attach);
-            onAction(endpoint === EModelEndpoint.google ? 'google_multimodal' : 'multimodal');
+            onAction(
+              (agent?.provider ?? endpoint) === EModelEndpoint.google
+                ? 'google_multimodal'
+                : 'multimodal',
+            );
           },
           icon: <FileImageIcon className="icon-md" />,
         });