Update components/ai/WorkAssistantPanel.tsx

components/ai/WorkAssistantPanel.tsx

@@ -241,7 +241,8 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                 text: '', 
                 thought: '', 
                 timestamp: Date.now(),
-                images: base64Images 
+                images: base64Images,
+                isSearching: enableSearch 
             };
 
             setMessages(prev => [...prev, newUserMsg, newAiMsg]);
@@ -320,17 +321,22 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                                     setIsThinkingExpanded(prev => ({ ...prev, [aiMsgId]: false }));
                                 }
                                 aiTextAccumulated += data.content;
-                                setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: aiTextAccumulated } : m));
+                                setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: aiTextAccumulated, isSearching: false } : m));
+                            } else if (data.type === 'search') {
+                                // Enable search visual
+                                setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, isSearching: true } : m));
                             } else if (data.type === 'error') {
-                                setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: `⚠️ 错误: ${data.message}` } : m));
+                                setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: `⚠️ 错误: ${data.message}`, isSearching: false } : m));
                             }
                         } catch (e) {}
                     }
                 }
             }
+            // Cleanup
+            setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, isSearching: false } : m));
 
         } catch (error: any) {
-            setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: `抱歉，处理失败: ${error.message}` } : m));
+            setMessages(prev => prev.map(m => m.id === aiMsgId ? { ...m, text: `抱歉，处理失败: ${error.message}`, isSearching: false } : m));
         } finally {
             setIsProcessing(false);
         }
@@ -504,6 +510,14 @@ export const WorkAssistantPanel: React.FC<WorkAssistantPanelProps> = ({ currentU
                                     </div>
                                 )}
 
+                                {/* Search Status Bubble */}
+                                {msg.role === 'model' && msg.isSearching && (
+                                    <div className="flex items-center gap-2 bg-blue-50 text-blue-600 px-3 py-2 rounded-xl mb-2 text-xs border border-blue-100 animate-pulse w-fit">
+                                        <Globe size={14} className="animate-spin"/>
+                                        <span>正在联网搜索相关信息...</span>
+                                    </div>
+                                )}
+
                                 <div className={`p-5 rounded-2xl shadow-sm text-sm overflow-hidden relative group w-full ${msg.role === 'user' ? 'bg-blue-600 text-white rounded-tr-none' : 'bg-white text-gray-800 border border-gray-100 rounded-tl-none'}`}>
                                     {/* User Image Preview Grid */}
                                     {msg.role === 'user' && sourceImages.length > 0 && (