From a10bfe32cdfac742dae35aa38f81444619e5c4c2 Mon Sep 17 00:00:00 2001
From: Anish Sarkar <104695310+AnishSarkar22@users.noreply.github.com>
Date: Sun, 28 Dec 2025 21:29:28 +0530
Subject: [PATCH] feat: Adjust document mention picker max height and refine
 thread header shimmer display logic.

---
 surfsense_web/components/assistant-ui/thread.tsx              | 2 +-
 surfsense_web/components/new-chat/document-mention-picker.tsx | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/surfsense_web/components/assistant-ui/thread.tsx b/surfsense_web/components/assistant-ui/thread.tsx
index 9ec7386c3..82863f06d 100644
--- a/surfsense_web/components/assistant-ui/thread.tsx
+++ b/surfsense_web/components/assistant-ui/thread.tsx
@@ -155,7 +155,7 @@ const ThinkingStepsDisplay: FC<{ steps: ThinkingStep[]; isThreadRunning?: boolea
 					)}
 				>
 					{/* Header text with shimmer if processing or has in-progress step */}
-					{isProcessing || inProgressStep ? (
+					{isProcessing && inProgressStep ? (
 						<TextShimmerLoader text={getHeaderText()} size="sm" />
 					) : (
 						<span>{getHeaderText()}</span>
diff --git a/surfsense_web/components/new-chat/document-mention-picker.tsx b/surfsense_web/components/new-chat/document-mention-picker.tsx
index d46955324..f250025c4 100644
--- a/surfsense_web/components/new-chat/document-mention-picker.tsx
+++ b/surfsense_web/components/new-chat/document-mention-picker.tsx
@@ -184,8 +184,8 @@ export const DocumentMentionPicker = forwardRef<
 			role="listbox"
 			tabIndex={-1}
 		>
-			{/* Document List */}
-			<div className="max-h-[280px] overflow-y-auto">
+			{/* Document List - Shows max 3 items on mobile, 5 items on desktop */}
+		<div className="max-h-[108px] sm:max-h-[180px] overflow-y-auto">
 				{actualLoading ? (
 					<div className="flex items-center justify-center py-4">
 						<div className="animate-spin h-5 w-5 border-2 border-primary border-t-transparent rounded-full" />