fix: prevent activity reasoning flash

This commit is contained in:
Bohdan Triapitsyn
2026-05-25 02:07:25 +03:00
parent bb3a51db39
commit f6c76239c5
3 changed files with 142 additions and 97 deletions
@@ -1038,6 +1038,7 @@ const AssistantMessageBody = React.memo(({
const collapsedPreviewCount = 7;
const isLastAssistantInTurn = turnGroupingContext?.isLastAssistantInTurn ?? false;
const hasStopFinish = messageFinish === 'stop';
const effectiveStreamPhase: StreamPhase = hasStopFinish ? 'completed' : streamPhase;
const availableWorktreesByProject = useSessionUIStore((state) => state.availableWorktreesByProject);
const currentProjectRef = React.useMemo(() => {
@@ -1506,7 +1507,7 @@ const AssistantMessageBody = React.memo(({
onToggleTool={onToggleTool}
onShowPopup={onShowPopup}
onContentChange={onContentChange}
streamPhase={streamPhase}
streamPhase={effectiveStreamPhase}
showHeader={true}
animateRows={animateActivityRows}
animatedToolIds={animatedToolIdsLookup}
@@ -1551,7 +1552,7 @@ const AssistantMessageBody = React.memo(({
part={part}
sessionId={sessionId}
messageId={messageId}
streamPhase={streamPhase}
streamPhase={effectiveStreamPhase}
chatRenderMode={chatRenderMode}
onContentChange={onContentChange}
/>
@@ -1585,7 +1586,7 @@ const AssistantMessageBody = React.memo(({
part={part}
sessionId={sessionId}
messageId={messageId}
streamPhase={streamPhase}
streamPhase={effectiveStreamPhase}
chatRenderMode={chatRenderMode}
onContentChange={onContentChange}
/>
@@ -1599,7 +1600,7 @@ const AssistantMessageBody = React.memo(({
key={`reasoning-merged-${messageId}`}
parts={flatReasoningParts}
messageId={messageId}
streamPhase={streamPhase}
streamPhase={effectiveStreamPhase}
onContentChange={onContentChange}
/>
);
@@ -1611,7 +1612,7 @@ const AssistantMessageBody = React.memo(({
key={`reasoning-${messageId}-${i}`}
part={part}
messageId={messageId}
streamPhase={streamPhase}
streamPhase={effectiveStreamPhase}
onContentChange={onContentChange}
/>
);
@@ -1719,7 +1720,7 @@ const AssistantMessageBody = React.memo(({
shouldRenderActivityGroup,
shouldShowStandaloneMessageActions,
shouldShowTool,
streamPhase,
effectiveStreamPhase,
showReasoningTraces,
shouldDeferSortedInlineText,
syntaxTheme,