File tree Expand file tree Collapse file tree 3 files changed +7
-2
lines changed Expand file tree Collapse file tree 3 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -479,6 +479,8 @@ const AIViewInner: React.FC<AIViewProps> = ({
479479 ? aggregator . getStreamingTPS ( activeStreamMessageId )
480480 : undefined
481481 }
482+ interrupting = { interrupting
483+ }
482484 />
483485 ) }
484486 { workspaceState ?. queuedMessage && (
Original file line number Diff line number Diff line change @@ -7,6 +7,7 @@ interface StreamingBarrierProps {
77 cancelText : string ; // e.g., "hit Esc to cancel"
88 tokenCount ?: number ;
99 tps ?: number ;
10+ interrupting ?: boolean ;
1011}
1112
1213export const StreamingBarrier : React . FC < StreamingBarrierProps > = ( {
@@ -15,11 +16,13 @@ export const StreamingBarrier: React.FC<StreamingBarrierProps> = ({
1516 cancelText,
1617 tokenCount,
1718 tps,
19+ interrupting,
1820} ) => {
21+ const color = interrupting ? "var(--color-interrupted)" : "var(--color-assistant-border)" ;
1922 return (
2023 < div className = { `flex items-center justify-between gap-4 ${ className ?? "" } ` } >
2124 < div className = "flex flex-1 items-center gap-2" >
22- < BaseBarrier text = { statusText } color = "var(-- color-assistant-border)" animate />
25+ < BaseBarrier text = { statusText } color = { color } animate />
2326 { tokenCount !== undefined && (
2427 < span className = "text-assistant-border font-mono text-[11px] whitespace-nowrap select-none" >
2528 ~{ tokenCount . toLocaleString ( ) } tokens
Original file line number Diff line number Diff line change @@ -862,12 +862,12 @@ export class StreamManager extends EventEmitter {
862862 case "start-step" :
863863 case "text-start" :
864864 case "finish" :
865+ case "tool-input-end" :
865866 // These events can be logged or handled if needed
866867 break ;
867868
868869 case "finish-step" :
869870 case "text-end" :
870- case "tool-input-end" :
871871 await this . checkSoftCancelStream ( workspaceId , streamInfo ) ;
872872 break ;
873873 }
You can’t perform that action at this time.
0 commit comments