From 151d6a853e3c2fe1d1d44c6ee2453c2b60a5f96c Mon Sep 17 00:00:00 2001
From: CREDO23 <thierrybakera12@gmail.com>
Date: Fri, 27 Mar 2026 19:20:14 +0200
Subject: [PATCH] use sessionStorage for quickAskMode to survive route changes

---
 surfsense_web/app/dashboard/quick-ask/page.tsx              | 3 ++-
 surfsense_web/components/assistant-ui/assistant-message.tsx | 4 +---
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/surfsense_web/app/dashboard/quick-ask/page.tsx b/surfsense_web/app/dashboard/quick-ask/page.tsx
index 0a304b3db..b6194d0af 100644
--- a/surfsense_web/app/dashboard/quick-ask/page.tsx
+++ b/surfsense_web/app/dashboard/quick-ask/page.tsx
@@ -35,8 +35,9 @@ export default function QuickAskPage() {
 	}, []);
 
 	const navigateToChat = (prompt: string, mode: string) => {
+		sessionStorage.setItem("quickAskMode", mode);
 		const encoded = encodeURIComponent(prompt);
-		window.location.href = `/dashboard?quickAskPrompt=${encoded}&quickAskMode=${mode}`;
+		window.location.href = `/dashboard?quickAskPrompt=${encoded}`;
 	};
 
 	const handleAction = (actionId: string) => {
diff --git a/surfsense_web/components/assistant-ui/assistant-message.tsx b/surfsense_web/components/assistant-ui/assistant-message.tsx
index 27ba08395..d9837b224 100644
--- a/surfsense_web/components/assistant-ui/assistant-message.tsx
+++ b/surfsense_web/components/assistant-ui/assistant-message.tsx
@@ -8,7 +8,6 @@ import {
 } from "@assistant-ui/react";
 import { useAtomValue } from "jotai";
 import { CheckIcon, ClipboardPaste, CopyIcon, DownloadIcon, MessageSquare, RefreshCwIcon } from "lucide-react";
-import { useSearchParams } from "next/navigation";
 import type { FC } from "react";
 import { useEffect, useMemo, useRef, useState } from "react";
 import { commentsEnabledAtom, targetCommentIdAtom } from "@/atoms/chat/current-thread.atom";
@@ -275,12 +274,11 @@ export const AssistantMessage: FC = () => {
 const AssistantActionBar: FC = () => {
 	const isLast = useAuiState((s) => s.message.isLast);
 	const aui = useAui();
-	const searchParams = useSearchParams();
 	const isTransform =
 		isLast &&
 		typeof window !== "undefined" &&
 		!!window.electronAPI?.replaceText &&
-		searchParams.get("quickAskMode") === "transform";
+		sessionStorage.getItem("quickAskMode") === "transform";
 
 	return (
 		<ActionBarPrimitive.Root