@neta-art/cohub 1.9.0 → 1.10.1

package/dist/session-generation-stream.js

@@ -1,514 +0,0 @@
-import { ensureRealtimeConnected } from "./realtime.js";
-import { SessionPatchReducer, } from "./session-patch-reducer.js";
-const isRecord = (value) => Boolean(value && typeof value === "object" && !Array.isArray(value));
-const isContentBlockArray = (value) => Array.isArray(value) &&
-    value.every((item) => isRecord(item) && typeof item.type === "string");
-const stringField = (record, key) => {
-    const value = record[key];
-    return typeof value === "string" ? value : null;
-};
-const numberField = (record, key) => {
-    const value = record[key];
-    return typeof value === "number" && Number.isFinite(value) ? value : null;
-};
-const getMessageKind = (message) => {
-    const kind = message.meta?.messageKind;
-    return typeof kind === "string" ? kind : null;
-};
-export function parseAssistantMessageCommit(message) {
-    if (message.role !== "assistant") {
-        return { kind: "ignored", message, isFinal: false };
-    }
-    const kind = getMessageKind(message);
-    if (kind === "assistant_intermediate") {
-        return { kind: "intermediate", message, isFinal: false };
-    }
-    if (kind === "assistant_final") {
-        return { kind: "final", message, isFinal: true };
-    }
-    if (kind === "assistant_error") {
-        return { kind: "error", message, isFinal: true };
-    }
-    return { kind: "ignored", message, isFinal: false };
-}
-function cloneContentBlock(block) {
-    return structuredClone(block);
-}
-function getStreamIndex(block) {
-    const value = block._meta?.streamIndex;
-    return typeof value === "number" && Number.isFinite(value) ? value : null;
-}
-function findMergeTargetIndex(result, block) {
-    const streamIndex = getStreamIndex(block);
-    if (streamIndex != null) {
-        return result.findIndex((existing) => existing.type === block.type && getStreamIndex(existing) === streamIndex);
-    }
-    if (block.type === "tool_use") {
-        return result.findIndex((existing) => existing.type === "tool_use" && existing.id === block.id);
-    }
-    if (block.type === "tool_result") {
-        return result.findIndex((existing) => existing.type === "tool_result" &&
-            existing.tool_use_id === block.tool_use_id);
-    }
-    return -1;
-}
-function mergeStreamingDeltaBlocks(existing, delta) {
-    if (delta.length === 0)
-        return existing;
-    const result = existing.map(cloneContentBlock);
-    for (const block of delta) {
-        const targetIndex = findMergeTargetIndex(result, block);
-        if (targetIndex === -1) {
-            result.push(cloneContentBlock(block));
-            continue;
-        }
-        const target = result[targetIndex];
-        if (block.type === "text" && target?.type === "text") {
-            target.text += block.text;
-            continue;
-        }
-        if (block.type === "thinking" && target?.type === "thinking") {
-            target.thinking += block.thinking;
-            if (block.signature)
-                target.signature = block.signature;
-            if (block._meta)
-                target._meta = { ...(target._meta ?? {}), ...block._meta };
-            continue;
-        }
-        result[targetIndex] = Object.assign(target ?? {}, cloneContentBlock(block));
-    }
-    return result;
-}
-function parseSnapshotMessage(value) {
-    if (!isRecord(value) || !isContentBlockArray(value.content))
-        return null;
-    return {
-        messageId: stringField(value, "messageId"),
-        messageOrdinal: numberField(value, "messageOrdinal"),
-        content: value.content,
-        ...(typeof value.id === "string" ? { id: value.id } : {}),
-        ...(typeof value.sessionId === "string" ? { sessionId: value.sessionId } : {}),
-        ...(value.role === "user" ||
-            value.role === "assistant" ||
-            value.role === "system"
-            ? { role: value.role }
-            : {}),
-        ...(typeof value.text === "string" ? { text: value.text } : {}),
-        ...(typeof value.provider === "string" ? { provider: value.provider } : {}),
-        ...(typeof value.model === "string" ? { model: value.model } : {}),
-        ...(typeof value.stopReason === "string"
-            ? { stopReason: value.stopReason }
-            : {}),
-        ...(typeof value.errorMessage === "string"
-            ? { errorMessage: value.errorMessage }
-            : {}),
-        ...(isRecord(value.usage) ? { usage: value.usage } : {}),
-        ...(typeof value.toolCallsObjectKey === "string"
-            ? { toolCallsObjectKey: value.toolCallsObjectKey }
-            : {}),
-        ...(isRecord(value.meta) ? { meta: value.meta } : {}),
-        ...(typeof value.createdAt === "string" ? { createdAt: value.createdAt } : {}),
-    };
-}
-function messageRecordToIntermediate(message) {
-    if (!isContentBlockArray(message.content) || message.content.length === 0) {
-        return null;
-    }
-    const meta = isRecord(message.meta) ? message.meta : {};
-    return {
-        id: message.id,
-        sessionId: message.sessionId,
-        role: message.role,
-        messageId: typeof meta.streamMessageId === "string"
-            ? meta.streamMessageId
-            : message.id ?? null,
-        messageOrdinal: typeof meta.messageOrdinal === "number" ? meta.messageOrdinal : null,
-        content: message.content,
-        text: message.text,
-        provider: message.provider,
-        model: message.model,
-        stopReason: message.stopReason,
-        errorMessage: message.errorMessage,
-        usage: message.usage,
-        meta: message.meta,
-        createdAt: message.createdAt,
-    };
-}
-function resolveStreamMessageId(input) {
-    if (input.messageId?.trim())
-        return input.messageId.trim();
-    if (input.messageOrdinal == null)
-        return null;
-    if (input.turnId?.trim()) {
-        return `turn:${input.turnId.trim()}:assistant:${input.messageOrdinal}`;
-    }
-    return `session:${input.sessionId}:assistant:${input.messageOrdinal}:${input.anchorUserMessageId ?? "unknown"}`;
-}
-function getTurnIdFromMessage(message) {
-    const turnId = message.meta?.turnId;
-    return typeof turnId === "string" ? turnId : null;
-}
-export class SessionGenerationStreamClient {
-    websocketClient;
-    spaceId;
-    sessionId;
-    reducer = new SessionPatchReducer();
-    messageId = null;
-    messageOrdinal = null;
-    intermediateMessages = [];
-    progressState = null;
-    constructor(websocketClient, spaceId, sessionId) {
-        this.websocketClient = websocketClient;
-        this.spaceId = spaceId;
-        this.sessionId = sessionId;
-    }
-    subscribe(handlers) {
-        if (!this.websocketClient) {
-            throw new Error("realtime transport is not configured for this client");
-        }
-        ensureRealtimeConnected(this.websocketClient);
-        const stream = new SessionGenerationStreamClient(this.websocketClient, this.spaceId, this.sessionId);
-        const unsubscribe = this.websocketClient.on("event", (event) => {
-            if (event.spaceId !== this.spaceId || event.sessionId !== this.sessionId) {
-                return;
-            }
-            stream.handleEvent(event, handlers);
-        });
-        return () => unsubscribe();
-    }
-    emit(handlers, event) {
-        handlers.event?.(event);
-        if (event.type === "state")
-            handlers.state?.(event);
-        if (event.type === "commit")
-            handlers.commit?.(event);
-        if (event.type === "finalized")
-            handlers.finalized?.(event);
-        if (event.type === "turn_updated")
-            handlers.turnUpdated?.(event);
-        if (event.type === "error")
-            handlers.error?.(event);
-        if (event.type === "out_of_sync")
-            handlers.outOfSync?.(event);
-    }
-    resetCurrentMessage() {
-        this.messageId = null;
-        this.messageOrdinal = null;
-        this.progressState = null;
-    }
-    appendCurrentMessage(state) {
-        if (state.contentBlocks.length === 0)
-            return;
-        this.addIntermediateMessage({
-            messageId: this.messageId,
-            messageOrdinal: this.messageOrdinal,
-            content: state.contentBlocks,
-        });
-    }
-    addIntermediateMessage(message) {
-        const index = this.intermediateMessages.findIndex((existing) => {
-            if (message.messageId && existing.messageId) {
-                return existing.messageId === message.messageId;
-            }
-            return (message.messageOrdinal !== null &&
-                existing.messageOrdinal === message.messageOrdinal);
-        });
-        if (index < 0) {
-            this.intermediateMessages = [...this.intermediateMessages, message];
-            return;
-        }
-        this.intermediateMessages = this.intermediateMessages.map((existing, i) => i === index ? { ...existing, ...message } : existing);
-    }
-    handleAppliedState(handlers, source, result, rawEvent, messageId, messageOrdinal) {
-        if (!result.applied) {
-            this.emit(handlers, {
-                type: "out_of_sync",
-                source: source === "progress" ? "patch" : source,
-                reason: result.reason,
-                state: result.state,
-                rawEvent,
-            });
-            return;
-        }
-        this.progressState = result.state;
-        this.messageId = messageId;
-        this.messageOrdinal = messageOrdinal;
-        this.emit(handlers, {
-            type: "state",
-            source,
-            state: result.state,
-            messageId,
-            messageOrdinal,
-            intermediateMessages: [...this.intermediateMessages],
-            rawEvent,
-        });
-    }
-    prepareMessageBoundary(input) {
-        const current = this.progressState ??
-            this.reducer.get({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-            });
-        const nextMessageId = resolveStreamMessageId({
-            sessionId: this.sessionId,
-            turnId: input.turnId,
-            anchorUserMessageId: input.anchorUserMessageId,
-            messageId: input.messageId,
-            messageOrdinal: input.messageOrdinal,
-        });
-        const differentTurn = Boolean(current.turnId && input.turnId && current.turnId !== input.turnId);
-        const messageChanged = Boolean(nextMessageId &&
-            current.contentBlocks.length > 0 &&
-            this.messageId &&
-            nextMessageId !== this.messageId);
-        if (differentTurn) {
-            this.intermediateMessages = [];
-            this.resetCurrentMessage();
-            this.reducer.start({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: input.turnId,
-            });
-        }
-        else if (messageChanged) {
-            this.appendCurrentMessage(current);
-            this.resetCurrentMessage();
-            this.reducer.start({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: input.turnId ?? current.turnId,
-            });
-        }
-        return nextMessageId;
-    }
-    handleSnapshot(event, handlers) {
-        const payload = event.payload;
-        const current = isRecord(payload.current) ? payload.current : null;
-        const content = current ? current.content : null;
-        const seq = typeof payload.seq === "number" ? payload.seq : null;
-        if (!current || !isContentBlockArray(content) || seq === null) {
-            this.emit(handlers, {
-                type: "out_of_sync",
-                source: "snapshot",
-                reason: "invalid",
-                state: this.reducer.get({ spaceId: this.spaceId, sessionId: this.sessionId }),
-                rawEvent: event,
-            });
-            return;
-        }
-        const turnId = typeof payload.turnId === "string" ? payload.turnId : null;
-        const anchorUserMessageId = typeof payload.anchorUserMessageId === "string"
-            ? payload.anchorUserMessageId
-            : null;
-        const messageOrdinal = numberField(current, "messageOrdinal");
-        const messageId = this.prepareMessageBoundary({
-            turnId,
-            messageId: stringField(current, "messageId"),
-            messageOrdinal,
-            anchorUserMessageId,
-        });
-        this.intermediateMessages = Array.isArray(payload.intermediateMessages)
-            ? payload.intermediateMessages
-                .map(parseSnapshotMessage)
-                .filter((message) => message !== null)
-            : [];
-        const result = this.reducer.applySnapshot({
-            spaceId: this.spaceId,
-            sessionId: this.sessionId,
-            turnId,
-            seq,
-            contentBlocks: content,
-            anchorUserMessageId,
-            appendPath: stringField(current, "appendPath"),
-        });
-        this.handleAppliedState(handlers, "snapshot", result, event, messageId, messageOrdinal);
-    }
-    handlePatch(event, handlers) {
-        const payload = event.payload;
-        const seq = typeof payload.seq === "number" ? payload.seq : null;
-        const baseSeq = typeof payload.baseSeq === "number" ? payload.baseSeq : null;
-        if (seq === null || baseSeq === null || !Array.isArray(payload.ops)) {
-            this.emit(handlers, {
-                type: "out_of_sync",
-                source: "patch",
-                reason: "invalid",
-                state: this.reducer.get({ spaceId: this.spaceId, sessionId: this.sessionId }),
-                rawEvent: event,
-            });
-            return;
-        }
-        const turnId = typeof payload.turnId === "string" ? payload.turnId : null;
-        const anchorUserMessageId = typeof payload.anchorUserMessageId === "string"
-            ? payload.anchorUserMessageId
-            : null;
-        const messageOrdinal = typeof payload.messageOrdinal === "number" ? payload.messageOrdinal : null;
-        const messageId = this.prepareMessageBoundary({
-            turnId,
-            messageId: typeof payload.messageId === "string" ? payload.messageId : null,
-            messageOrdinal,
-            anchorUserMessageId,
-        });
-        const input = {
-            spaceId: this.spaceId,
-            sessionId: this.sessionId,
-            turnId,
-            seq,
-            baseSeq,
-            ops: payload.ops,
-            anchorUserMessageId,
-        };
-        const result = this.reducer.applyPatch(input);
-        this.handleAppliedState(handlers, "patch", result, event, messageId, messageOrdinal);
-    }
-    handleProgress(event, handlers) {
-        const payload = event.payload;
-        if (!isContentBlockArray(payload.content))
-            return;
-        const current = this.progressState ??
-            this.reducer.get({ spaceId: this.spaceId, sessionId: this.sessionId });
-        const turnId = typeof payload.turnId === "string" ? payload.turnId : current.turnId;
-        const anchorUserMessageId = typeof payload.anchorUserMessageId === "string"
-            ? payload.anchorUserMessageId
-            : current.anchorUserMessageId;
-        const messageOrdinal = typeof payload.messageOrdinal === "number"
-            ? payload.messageOrdinal
-            : this.messageOrdinal;
-        const messageId = this.prepareMessageBoundary({
-            turnId,
-            messageId: typeof payload.messageId === "string" ? payload.messageId : this.messageId,
-            messageOrdinal,
-            anchorUserMessageId,
-        });
-        const base = this.progressState?.turnId === turnId ? this.progressState : current;
-        const state = {
-            ...base,
-            spaceId: this.spaceId,
-            sessionId: this.sessionId,
-            status: "streaming",
-            contentBlocks: mergeStreamingDeltaBlocks(base.contentBlocks, payload.content),
-            anchorUserMessageId,
-            turnId,
-        };
-        this.progressState = state;
-        this.messageId = messageId;
-        this.messageOrdinal = messageOrdinal;
-        this.emit(handlers, {
-            type: "state",
-            source: "progress",
-            state,
-            messageId,
-            messageOrdinal,
-            intermediateMessages: [...this.intermediateMessages],
-            rawEvent: event,
-        });
-    }
-    handlePersisted(event, handlers) {
-        const message = event.payload.message;
-        if (!isRecord(message))
-            return;
-        const commit = parseAssistantMessageCommit(message);
-        if (commit.kind === "intermediate") {
-            const intermediate = messageRecordToIntermediate(commit.message);
-            if (intermediate) {
-                this.addIntermediateMessage(intermediate);
-            }
-            this.reducer.reset({ spaceId: this.spaceId, sessionId: this.sessionId });
-            this.resetCurrentMessage();
-        }
-        if (commit.kind === "final") {
-            this.reducer.complete({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: getTurnIdFromMessage(commit.message),
-            });
-            this.resetCurrentMessage();
-        }
-        if (commit.kind === "error") {
-            this.reducer.fail({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: getTurnIdFromMessage(commit.message),
-            });
-            this.resetCurrentMessage();
-        }
-        this.emit(handlers, {
-            type: "commit",
-            commit,
-            rawEvent: event,
-        });
-    }
-    handleFinalized(event, handlers) {
-        const turn = event.payload.turn;
-        if (!isRecord(turn))
-            return;
-        const typedTurn = turn;
-        if (typedTurn.status === "interrupted") {
-            this.reducer.interrupt({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: typedTurn.id,
-            });
-        }
-        else {
-            this.reducer.complete({
-                spaceId: this.spaceId,
-                sessionId: this.sessionId,
-                turnId: typedTurn.id,
-            });
-        }
-        this.resetCurrentMessage();
-        this.emit(handlers, {
-            type: "finalized",
-            turn: typedTurn,
-            rawEvent: event,
-        });
-    }
-    handleEvent(event, handlers) {
-        switch (event.type) {
-            case "session.turn.snapshot":
-                this.handleSnapshot(event, handlers);
-                return;
-            case "session.turn.patch":
-                this.handlePatch(event, handlers);
-                return;
-            case "session.turn.progress":
-                this.handleProgress(event, handlers);
-                return;
-            case "session.message.persisted":
-                this.handlePersisted(event, handlers);
-                return;
-            case "session.turn.finalized":
-                this.handleFinalized(event, handlers);
-                return;
-            case "session.turn.updated": {
-                const turn = event.payload.turn;
-                if (!isRecord(turn))
-                    return;
-                this.emit(handlers, {
-                    type: "turn_updated",
-                    turn: turn,
-                    rawEvent: event,
-                });
-                return;
-            }
-            case "session.turn.error": {
-                const message = typeof event.payload.error === "string" && event.payload.error.trim()
-                    ? event.payload.error.trim()
-                    : "Generation failed";
-                this.reducer.fail({ spaceId: this.spaceId, sessionId: this.sessionId });
-                this.resetCurrentMessage();
-                this.emit(handlers, {
-                    type: "error",
-                    message,
-                    rawEvent: event,
-                });
-                return;
-            }
-            default:
-                return;
-        }
-    }
-}
-export function createSessionGenerationStreamClient(input) {
-    return new SessionGenerationStreamClient(input.websocketClient, input.spaceId, input.sessionId);
-}

package/dist/session-patch-reducer.d.ts

@@ -1,61 +0,0 @@
-import type { ContentBlock } from "@neta-art/cohub-protocol/core";
-import type { RealtimePatchOperation, SessionTurnPatchEvent } from "@neta-art/cohub-protocol/realtime";
-export type SessionPatchStatus = "idle" | "pending" | "streaming" | "completed" | "failed" | "interrupted";
-export type SessionPatchState = {
-    spaceId: string | null;
-    sessionId: string;
-    status: SessionPatchStatus;
-    contentBlocks: ContentBlock[];
-    anchorUserMessageId: string | null;
-    patchSeq: number;
-    turnId: string | null;
-    appendPath: string | null;
-};
-export type SessionPatchApplyInput = {
-    spaceId?: string | null;
-    sessionId: string;
-    turnId?: string | null;
-    seq: number;
-    baseSeq: number;
-    ops: RealtimePatchOperation[];
-    anchorUserMessageId?: string | null;
-};
-export type SessionPatchApplyResult = {
-    applied: true;
-    state: SessionPatchState;
-} | {
-    applied: false;
-    reason: "duplicate" | "version_mismatch" | "invalid";
-    state: SessionPatchState;
-};
-export type SessionPatchSnapshotInput = SessionPatchKeyInput & {
-    turnId?: string | null;
-    seq: number;
-    contentBlocks: ContentBlock[];
-    anchorUserMessageId?: string | null;
-    appendPath?: string | null;
-};
-type SessionPatchKeyInput = {
-    spaceId?: string | null;
-    sessionId: string;
-    turnId?: string | null;
-};
-export declare class SessionPatchReducer {
-    private readonly states;
-    private key;
-    get(input: SessionPatchKeyInput): SessionPatchState;
-    start(input: SessionPatchKeyInput): SessionPatchState;
-    replaceTurnId(input: SessionPatchKeyInput & {
-        nextTurnId: string | null;
-    }): SessionPatchState;
-    complete(input: SessionPatchKeyInput): SessionPatchState;
-    fail(input: SessionPatchKeyInput): SessionPatchState;
-    interrupt(input: SessionPatchKeyInput): SessionPatchState;
-    reset(input: SessionPatchKeyInput): void;
-    applySnapshot(input: SessionPatchSnapshotInput): SessionPatchApplyResult;
-    resetAll(): void;
-    applyEvent(event: SessionTurnPatchEvent): SessionPatchApplyResult;
-    applyPatch(input: SessionPatchApplyInput): SessionPatchApplyResult;
-}
-export declare const createSessionPatchReducer: () => SessionPatchReducer;
-export {};