From 738ba6cc3a4c328cfc7ade1529486a4f2631edf9 Mon Sep 17 00:00:00 2001
From: Gemini AI <gemini@google.com>
Date: Mon, 29 Dec 2025 02:15:31 +0400
Subject: [PATCH] fix: preserve manually selected agent in AI Assist during
 streaming

---
 components/AIAssist.tsx | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/components/AIAssist.tsx b/components/AIAssist.tsx
index 0866ddf..797edb1 100644
--- a/components/AIAssist.tsx
+++ b/components/AIAssist.tsx
@@ -157,8 +157,8 @@ const ThinkingIndicator = () => (
 
 // --- Helper Functions ---
 
-function parseStreamingContent(text: string) {
-    let agent = "general";
+function parseStreamingContent(text: string, currentAgent: string) {
+    let agent = currentAgent;
     let preview: PreviewData | null = null;
     let chatDisplay = text.trim();
     let status: string | null = null;
@@ -351,7 +351,7 @@ export default function AIAssist() {
                     currentAgent,
                     onChunk: (chunk) => {
                         accumulated += chunk;
-                        const { chatDisplay, preview, agent, status: streamStatus } = parseStreamingContent(accumulated);
+                        const { chatDisplay, preview, agent, status: streamStatus } = parseStreamingContent(accumulated, currentAgent);
 
                         if (streamStatus) setStatus(streamStatus);