fix: refine message handling for assistant role in chat and update shell component props

Zhang-Henry · Zhang-Henry · commit 9dcc07963683 · 2026-02-26T14:56:00.000-06:00
diff --git a/src/components/chat/hooks/useChatRealtimeHandlers.ts b/src/components/chat/hooks/useChatRealtimeHandlers.ts
@@ -344,7 +344,7 @@ export function useChatRealtimeHandlers({
           return;
         }
 
-        if (structuredMessageData && Array.isArray(structuredMessageData.content)) {
+        if (structuredMessageData && Array.isArray(structuredMessageData.content) && structuredMessageData.role === 'assistant') {
           const parentToolUseId = rawStructuredData?.parentToolUseId;
 
           structuredMessageData.content.forEach((part: any) => {
@@ -415,7 +415,7 @@ export function useChatRealtimeHandlers({
               ]);
             }
           });
-        } else if (structuredMessageData && typeof structuredMessageData.content === 'string' && structuredMessageData.content.trim()) {
+        } else if (structuredMessageData && structuredMessageData.role === 'assistant' && typeof structuredMessageData.content === 'string' && structuredMessageData.content.trim()) {
           let content = decodeHtmlEntities(structuredMessageData.content);
           content = formatUsageLimitText(content);
           setChatMessages((previous) => [
diff --git a/src/components/main-content/view/MainContent.tsx b/src/components/main-content/view/MainContent.tsx
@@ -139,7 +139,7 @@ function MainContent({
 
           {activeTab === 'shell' && (
             <div className="h-full w-full overflow-hidden">
-              <AnyStandaloneShell project={selectedProject} session={selectedSession} showHeader={false} />
+              <AnyStandaloneShell project={selectedProject} session={selectedSession} isPlainShell={true} showHeader={false} />
             </div>
           )}
 
