diff --git a/surfsense_web/components/assistant-ui/assistant-message.tsx b/surfsense_web/components/assistant-ui/assistant-message.tsx index d0cada0bd..5567cfca8 100644 --- a/surfsense_web/components/assistant-ui/assistant-message.tsx +++ b/surfsense_web/components/assistant-ui/assistant-message.tsx @@ -465,16 +465,8 @@ const AssistantActionBar: FC = () => { const isLast = useAuiState((s) => s.message.isLast); const aui = useAui(); const api = useElectronAPI(); - const [quickAskMode, setQuickAskMode] = useState(""); - useEffect(() => { - if (!isLast || !api?.getQuickAskMode) return; - api.getQuickAskMode().then((mode) => { - if (mode) setQuickAskMode(mode); - }); - }, [isLast, api]); - - const isTransform = isLast && !!api?.replaceText && quickAskMode === "transform"; + const isQuickAssist = !!api?.replaceText && !!api?.getQuickAskMode; return ( { className="aui-assistant-action-bar-root -ml-1 col-start-3 row-start-2 flex gap-1 text-muted-foreground md:data-floating:absolute md:data-floating:rounded-md md:data-floating:p-1 [&>button]:opacity-100 md:[&>button]:opacity-[var(--aui-button-opacity,1)]" > - + message.isCopied}> @@ -494,29 +486,27 @@ const AssistantActionBar: FC = () => { - + {isLast && ( - + )} - {isTransform && ( - + + )} );