067fc46170
PR 2 из 2 (#381, фаза 1.5 #184). Все вкладки теперь на ОДНОМ транспорте: любая подключается к рану через GET-attach (реплей кадров + живой хвост, реестр из PR 1). Наблюдатель — обычный стример; Stop, точки, инвалидации работают штатно. Двухпутёвый поллинг снапшотов и вся latch-механика F4/F5/F7 удалены. - utils/resume-helpers.ts (новый): isStreamingTail / isSettledAssistantTail / seedRows / mergeById (дословный перенос mergeObservedMessage из run-polling). - components/chat-thread.tsx: resume-машинерия (гейтинг по не-settled хвосту, strip streaming-хвоста + attach ?expect=live&anchor=<row id>, транспорт prepareReconnectToStreamRequest+fetch, 204-обработчик из 4 частей, reconcile+degraded-merge, recovery с АСИММЕТРИЕЙ arm-vs-restore — при isDisconnect с видимым контентом только arm, без restore-клоббера живого стрима (инв. 9), строгий порядок onFinish с ранним return до обеих веток отправки (инв. 7), «Send now» скрыт на resumed-ходе, Stop абортит attach). - components/ai-chat-window.tsx: degraded-poll фолбэк вместо латчей — тупой таймер (2500ms, 10-мин кап, без проверок ошибок/хвоста; переживает рестарт сервера), гасится тредом через onResumeFallback(false). - Удалено: run-polling.ts(+test), useAiChatRunQuery/AI_CHAT_RUN_RQ_KEY, getAiChatRun (stopRun оставлен), IAiChatRun/IAiChatRunResponse, латчи stoppingRun/localStreaming/observedRow/onStreamingChange, F7-эффект, observer-merge. Серверный POST /ai-chat/run не тронут. Проверка: tsc (мои файлы чисты), vitest src/features/ai-chat 34 файла/301 тест зелёные, grep-guard по удалённым символам пуст. Отдельное внутреннее ревью на инварианты 7/8/9 + 204-null-safety — чисто. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
63 lines
2.4 KiB
TypeScript
63 lines
2.4 KiB
TypeScript
import type { UIMessage } from "@ai-sdk/react";
|
|
import type { IAiChatMessageRow } from "@/features/ai-chat/types/ai-chat.types.ts";
|
|
|
|
/**
|
|
* Pure decisions for the resumable-SSE resume machinery (#184 phase 1.5). A tab
|
|
* that reopens a chat whose agent run is still going attaches to the server's
|
|
* run-stream registry (replay + live tail) instead of polling snapshots; these
|
|
* small predicates decide WHICH tail is safe to resume and how to seed the store,
|
|
* extracted so they can be unit-tested in isolation.
|
|
*/
|
|
|
|
/**
|
|
* A STREAMING tail: the last persisted row is an assistant row still marked
|
|
* `status === 'streaming'`. Such a tail is stripped from the seed and rebuilt by
|
|
* the replay (`expect=live`), since the SDK's `text-start` always pushes a new
|
|
* part and replaying over a seeded in-progress row would duplicate its text.
|
|
*/
|
|
export function isStreamingTail(rows: IAiChatMessageRow[]): boolean {
|
|
const tail = rows[rows.length - 1];
|
|
return !!tail && tail.role === "assistant" && tail.status === "streaming";
|
|
}
|
|
|
|
/**
|
|
* A SETTLED assistant tail: the last row is an assistant row whose status is
|
|
* anything OTHER than 'streaming'. A settled assistant tail must NEVER resume —
|
|
* replaying a finished run into a store that already holds its message duplicates
|
|
* parts (`text-start` always pushes a new part).
|
|
*/
|
|
export function isSettledAssistantTail(rows: IAiChatMessageRow[]): boolean {
|
|
const tail = rows[rows.length - 1];
|
|
return !!tail && tail.role === "assistant" && tail.status !== "streaming";
|
|
}
|
|
|
|
/**
|
|
* Seed rows for `useChat`: return the rows unchanged, or without the last row when
|
|
* `strip` is set (the streaming tail is stripped so the live replay rebuilds it
|
|
* without duplicating parts).
|
|
*/
|
|
export function seedRows(
|
|
rows: IAiChatMessageRow[],
|
|
strip: boolean,
|
|
): IAiChatMessageRow[] {
|
|
return strip ? rows.slice(0, -1) : rows;
|
|
}
|
|
|
|
/**
|
|
* Merge an assistant message into the rendered list by id: replace the message
|
|
* with the same id in place (the in-progress assistant row is already seeded from
|
|
* history, so per-step growth replaces it), or append it when absent. Returns a
|
|
* new array; the input is never mutated.
|
|
*/
|
|
export function mergeById(
|
|
messages: UIMessage[],
|
|
incoming: UIMessage | null | undefined,
|
|
): UIMessage[] {
|
|
if (!incoming) return messages;
|
|
const idx = messages.findIndex((m) => m.id === incoming.id);
|
|
if (idx === -1) return [...messages, incoming];
|
|
const next = messages.slice();
|
|
next[idx] = incoming;
|
|
return next;
|
|
}
|