wuyang/zk-data-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix active run cancel and elapsed restore

Browse Source
This commit is contained in:
wuyang6
2026-05-14 11:24:26 +08:00
parent cccdd44ca2
commit e3569b6901
1 changed files with 29 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
frontend/app/components/assistant-ui/thread.tsx
+29 -17
View File
@@ -856,7 +856,11 @@ const ComposerWorkspaceStatus: FC<{ sessionId: string | null }> = ({
const ComposerAction: FC = () => { const ComposerAction: FC = () => {
const runtimeRunning = useAuiState((s) => s.thread.isRunning); const runtimeRunning = useAuiState((s) => s.thread.isRunning);
const replayedRun = useReplayedRunState(); const replayedRun = useReplayedRunState();
const cancelSessionId = replayedRun.sessionId; const currentSessionId = useCurrentThreadSessionId({
includePending: true,
includeActive: true,
});
const cancelSessionId = replayedRun.sessionId ?? currentSessionId;
return ( return (
<div className="aui-composer-action-wrapper relative flex items-center justify-between"> <div className="aui-composer-action-wrapper relative flex items-center justify-between">
<div className="flex items-center gap-1"> <div className="flex items-center gap-1">
@@ -880,21 +884,22 @@ const ComposerAction: FC = () => {
</ComposerPrimitive.Send> </ComposerPrimitive.Send>
) : null} ) : null}
{runtimeRunning ? ( {runtimeRunning ? (
<ComposerPrimitive.Cancel asChild> <Button
<Button type="button"
type="button" variant="default"
variant="default" size="icon"
size="icon" className="aui-composer-cancel size-8 rounded-full"
className="aui-composer-cancel size-8 rounded-full" aria-label="Stop generating"
aria-label="Stop generating" disabled={!cancelSessionId}
onClick={() => { onClick={() => {
if (!cancelSessionId) return; if (!cancelSessionId) return;
void cancelLatestRun(cancelSessionId).catch(() => undefined); void cancelLatestRun(cancelSessionId, replayedRun.runId).catch(
}} () => undefined,
> );
<SquareIcon className="aui-composer-cancel-icon size-3 fill-current" /> }}
</Button> >
</ComposerPrimitive.Cancel> <SquareIcon className="aui-composer-cancel-icon size-3 fill-current" />
</Button>
) : null} ) : null}
{!runtimeRunning && replayedRun.active ? ( {!runtimeRunning && replayedRun.active ? (
<ReplayedRunCancelButton <ReplayedRunCancelButton
@@ -1897,7 +1902,14 @@ const ActivityChainSummary: FC<{
? Math.max(0, Math.round(value)) ? Math.max(0, Math.round(value))
: null; : null;
}); });
const [elapsedMs, setElapsedMs] = useState<number | null>(storedElapsedMs); const initialStartedAtMs = metadataRunStartedAtMs ?? contentRunStartedAtMs;
const [elapsedMs, setElapsedMs] = useState<number | null>(() => {
if (storedElapsedMs !== null) return storedElapsedMs;
if (initialStartedAtMs !== null) {
return Math.max(0, Date.now() - initialStartedAtMs);
}
return null;
});
const label = useAuiState((s) => { const label = useAuiState((s) => {
const message = s.message; const message = s.message;
const status = const status =