@pasko70/pibo 1.1.0 → 1.2.0

package/dist/data/ingest-service.js

@@ -0,0 +1,389 @@
+import { createHash, randomUUID } from "node:crypto";
+import { rootSessionId } from "./session-store.js";
+const INLINE_MESSAGE_PAYLOAD_THRESHOLD_BYTES = 16 * 1024;
+const INLINE_JSON_PAYLOAD_THRESHOLD_BYTES = 16 * 1024;
+export class ChatDataIngestService {
+    store;
+    constructor(store) {
+        this.store = store;
+    }
+    ingestUserMessageAccepted(input) {
+        const idempotencyKey = input.clientTxnId
+            ? `chat:user.accepted:${input.roomId}:${input.actorId}:${input.clientTxnId}`
+            : undefined;
+        const messageId = idempotencyKey ? deterministicId("msg", idempotencyKey) : `msg_${randomUUID()}`;
+        const existingMessage = this.store.messages.getMessage(messageId);
+        if (existingMessage && idempotencyKey) {
+            const existingEvent = this.store.eventLog.findByIdempotencyKey(idempotencyKey);
+            return {
+                streamId: existingEvent?.streamId ?? existingMessage.sourceStreamId ?? 0,
+                messageId: existingMessage.id,
+                duplicate: true,
+            };
+        }
+        return this.store.transaction(() => {
+            const now = input.legacyEvent?.createdAt ?? new Date().toISOString();
+            this.store.sessions.upsertSession({ session: input.session, roomId: input.roomId, firstMessagePreview: input.text, lastActivityAt: now });
+            const payloadRef = this.writeTextPayloadIfLarge(input.text, now, "chat_message");
+            const event = this.store.eventLog.appendEvent({
+                sessionId: input.session.id,
+                sessionSequence: this.nextEventSequence(input.session.id),
+                roomId: input.roomId,
+                topic: "chat",
+                type: "user.message.accepted",
+                source: "user",
+                actorType: "user",
+                actorId: input.actorId,
+                eventId: input.legacyEvent?.eventId,
+                idempotencyKey,
+                retentionClass: "chat_message",
+                payloadRef,
+                previewText: previewText(input.text),
+                attributes: compactObject({
+                    clientTxnId: input.clientTxnId,
+                    legacyStreamId: input.legacyEvent?.streamId,
+                    inlineText: payloadRef ? undefined : input.text,
+                }),
+                createdAt: now,
+                indexedAt: now,
+            });
+            const existingAfterEvent = this.store.messages.getMessage(messageId);
+            if (existingAfterEvent && idempotencyKey) {
+                return { streamId: event.streamId, messageId: existingAfterEvent.id, duplicate: true };
+            }
+            this.store.messages.insertMessage({
+                id: messageId,
+                sessionId: input.session.id,
+                roomId: input.roomId,
+                sequence: this.nextMessageSequence(input.session.id),
+                role: "user",
+                actorId: input.actorId,
+                status: "complete",
+                createdAt: now,
+                completedAt: now,
+                contentPreview: previewText(input.text),
+                contentPayloadRef: payloadRef,
+                sourceStreamId: event.streamId,
+                attributes: compactObject({
+                    clientTxnId: input.clientTxnId,
+                    inlineText: payloadRef ? undefined : input.text,
+                }),
+            });
+            this.upsertNavigation(input.session, input.roomId, previewText(input.text), now, "running");
+            return { streamId: event.streamId, messageId, duplicate: false };
+        });
+    }
+    ingestOutputEvent(input) {
+        const event = input.event;
+        const idempotencyKey = outputIdempotencyKey(event);
+        if (idempotencyKey) {
+            const existing = this.store.eventLog.findByIdempotencyKey(idempotencyKey);
+            if (existing) {
+                return {
+                    streamId: existing.streamId,
+                    duplicate: true,
+                    messageId: messageIdForOutputEvent(event),
+                    observationId: observationIdForOutputEvent(event),
+                };
+            }
+        }
+        return this.store.transaction(() => {
+            const now = input.createdAt ?? new Date().toISOString();
+            if (input.roomId) {
+                this.store.sessions.upsertSession({ session: input.session, roomId: input.roomId, lastActivityAt: now, status: outputSessionStatus(event) });
+            }
+            const payload = payloadForOutputEvent(event);
+            const payloadRef = payload ? this.writePayloadIfLarge(payload.value, payload.contentType, now, retentionClassForOutputEvent(event)) : undefined;
+            const storedEvent = this.store.eventLog.appendEvent({
+                sessionId: input.session.id,
+                sessionSequence: this.nextEventSequence(input.session.id),
+                roomId: input.roomId,
+                topic: "pibo.output",
+                type: event.type,
+                source: "actor",
+                actorType: actorTypeForOutputEvent(event),
+                actorId: input.actorId,
+                turnId: turnIdForOutputEvent(event),
+                eventId: eventIdForOutputEvent(event),
+                toolCallId: "toolCallId" in event ? event.toolCallId : undefined,
+                runId: eventIdForOutputEvent(event),
+                idempotencyKey,
+                retentionClass: retentionClassForOutputEvent(event),
+                payloadRef,
+                previewText: previewTextForOutputEvent(event),
+                attributes: compactObject({
+                    legacyStreamId: input.legacyStreamId,
+                    inlinePayload: payloadRef ? undefined : toPiboJsonValue(payload?.value),
+                    ...attributesForOutputEvent(event),
+                }),
+                createdAt: now,
+                indexedAt: now,
+            });
+            let messageId;
+            if (event.type === "assistant_message") {
+                messageId = messageIdForOutputEvent(event);
+                if (messageId && !this.store.messages.getMessage(messageId)) {
+                    this.store.messages.insertMessage({
+                        id: messageId,
+                        sessionId: input.session.id,
+                        roomId: input.roomId,
+                        sequence: this.nextMessageSequence(input.session.id),
+                        turnId: turnIdForOutputEvent(event),
+                        role: "assistant",
+                        actorId: input.actorId,
+                        status: "complete",
+                        createdAt: now,
+                        completedAt: now,
+                        contentPreview: previewText(event.text),
+                        contentPayloadRef: payloadRef,
+                        sourceStreamId: storedEvent.streamId,
+                        attributes: compactObject({
+                            eventId: event.eventId,
+                            assistantIndex: event.assistantIndex,
+                            contentIndex: event.contentIndex,
+                            inlineText: payloadRef ? undefined : event.text,
+                        }),
+                    });
+                }
+            }
+            const observationId = observationIdForOutputEvent(event);
+            this.store.observations.insertObservation({
+                id: observationId,
+                sessionId: input.session.id,
+                turnId: turnIdForOutputEvent(event),
+                eventStreamId: storedEvent.streamId,
+                kind: observationKindForOutputEvent(event),
+                role: observationRoleForOutputEvent(event),
+                name: observationNameForOutputEvent(event),
+                status: observationStatusForOutputEvent(event),
+                startedAt: now,
+                endedAt: isTerminalOutputEvent(event) ? now : undefined,
+                previewText: previewTextForOutputEvent(event),
+                payloadRef,
+                attributes: compactObject({ eventType: event.type, eventId: eventIdForOutputEvent(event), ...attributesForOutputEvent(event) }),
+            });
+            if (input.roomId && (event.type === "assistant_message" || event.type === "message_finished" || event.type === "session_error")) {
+                const navigationPreview = event.type === "message_finished" ? undefined : previewTextForOutputEvent(event);
+                this.upsertNavigation(input.session, input.roomId, navigationPreview, now, outputSessionStatus(event));
+            }
+            return { streamId: storedEvent.streamId, duplicate: false, messageId, observationId };
+        });
+    }
+    nextEventSequence(sessionId) {
+        const row = this.store.db.prepare("SELECT COALESCE(MAX(session_sequence), 0) + 1 AS next_sequence FROM event_log WHERE session_id = ?").get(sessionId);
+        return row.next_sequence;
+    }
+    nextMessageSequence(sessionId) {
+        const row = this.store.db.prepare("SELECT COALESCE(MAX(sequence), 0) + 1 AS next_sequence FROM chat_messages WHERE session_id = ?").get(sessionId);
+        return row.next_sequence;
+    }
+    writeTextPayloadIfLarge(text, createdAt, retentionClass) {
+        if (Buffer.byteLength(text, "utf8") <= INLINE_MESSAGE_PAYLOAD_THRESHOLD_BYTES)
+            return undefined;
+        return this.store.payloads.writePayload({
+            value: text,
+            contentType: "text/plain; charset=utf-8",
+            retentionClass,
+            createdAt,
+        }).id;
+    }
+    writePayloadIfLarge(value, contentType, createdAt, retentionClass) {
+        const bytes = Buffer.byteLength(typeof value === "string" ? value : JSON.stringify(value), "utf8");
+        const threshold = typeof value === "string" ? INLINE_MESSAGE_PAYLOAD_THRESHOLD_BYTES : INLINE_JSON_PAYLOAD_THRESHOLD_BYTES;
+        if (bytes <= threshold)
+            return undefined;
+        return this.store.payloads.writePayload({ value, contentType, retentionClass, createdAt }).id;
+    }
+    upsertNavigation(session, roomId, lastMessagePreview, now, status) {
+        this.store.navigation.upsertSession({
+            roomId,
+            sessionId: session.id,
+            rootSessionId: rootSessionId(session),
+            parentId: session.parentId,
+            originId: session.originId,
+            title: session.title || lastMessagePreview || "Untitled Session",
+            profile: session.profile,
+            status,
+            lastActivityAt: now,
+            lastMessagePreview,
+            sortKey: now,
+            updatedAt: now,
+        });
+    }
+}
+function deterministicId(prefix, value) {
+    return `${prefix}_${createHash("sha256").update(value).digest("hex").slice(0, 32)}`;
+}
+function hashJson(value) {
+    return createHash("sha256").update(JSON.stringify(value) ?? "null").digest("hex").slice(0, 16);
+}
+function previewText(text) {
+    const normalized = text.replace(/\s+/g, " ").trim();
+    return normalized ? normalized.slice(0, 512) : undefined;
+}
+function compactObject(value) {
+    return Object.fromEntries(Object.entries(value).filter(([, item]) => item !== undefined));
+}
+function outputIdempotencyKey(event) {
+    const base = eventIdForOutputEvent(event) ?? ("toolCallId" in event ? event.toolCallId : undefined);
+    if (!base)
+        return undefined;
+    return `pibo.output:${event.piboSessionId}:${event.type}:${base}:${outputPartKey(event)}`;
+}
+function outputPartKey(event) {
+    if (event.type === "tool_call")
+        return `${event.toolCallId}:${event.argsComplete ? "complete" : "partial"}:${hashJson(event.args)}`;
+    if ("toolCallId" in event)
+        return event.toolCallId ?? "main";
+    if (event.type === "assistant_message")
+        return String(event.assistantIndex ?? event.contentIndex ?? 0);
+    if (event.type === "assistant_delta")
+        return String(event.assistantIndex ?? event.contentIndex ?? 0);
+    if (event.type === "thinking_started" || event.type === "thinking_delta" || event.type === "thinking_finished")
+        return String(event.thinkingIndex ?? event.contentIndex ?? 0);
+    if (event.type === "execution_result")
+        return event.action;
+    return "main";
+}
+function messageIdForOutputEvent(event) {
+    if (event.type !== "assistant_message")
+        return undefined;
+    return deterministicId("msg", outputIdempotencyKey(event) ?? JSON.stringify(event));
+}
+function observationIdForOutputEvent(event) {
+    return deterministicId("obs", outputIdempotencyKey(event) ?? JSON.stringify(event));
+}
+function turnIdForOutputEvent(event) {
+    return eventIdForOutputEvent(event) ?? ("toolCallId" in event ? event.toolCallId : undefined);
+}
+function eventIdForOutputEvent(event) {
+    return "eventId" in event ? event.eventId : undefined;
+}
+function retentionClassForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "message_started" || event.type === "message_finished")
+        return "chat_message";
+    if (event.type === "assistant_delta" || event.type === "thinking_delta" || event.type === "tool_execution_updated")
+        return "live_delta";
+    return "trace_event";
+}
+function actorTypeForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "assistant_delta" || event.type.startsWith("thinking_"))
+        return "assistant";
+    if (event.type.startsWith("tool_") || event.type === "subagent_session")
+        return "agent";
+    if (event.type === "session_error")
+        return "system";
+    return "agent";
+}
+function payloadForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "assistant_delta" || event.type === "thinking_delta" || event.type === "thinking_finished") {
+        return { value: event.text ?? "", contentType: "text/plain; charset=utf-8" };
+    }
+    if (event.type === "tool_call" || event.type === "tool_execution_started")
+        return { value: toPiboJsonValueOrNull(event.args), contentType: "application/json" };
+    if (event.type === "tool_execution_updated")
+        return { value: toPiboJsonValueOrNull(event.partialResult), contentType: "application/json" };
+    if (event.type === "tool_execution_finished")
+        return { value: toPiboJsonValueOrNull(event.result), contentType: "application/json" };
+    if (event.type === "execution_result")
+        return { value: toPiboJsonValueOrNull(event.result), contentType: "application/json" };
+    if (event.type === "pi_event")
+        return { value: toPiboJsonValueOrNull(event.event), contentType: "application/json" };
+    if (event.type === "compaction_end" && event.result !== undefined)
+        return { value: toPiboJsonValueOrNull(event.result), contentType: "application/json" };
+    return undefined;
+}
+function previewTextForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "assistant_delta" || event.type === "thinking_delta" || event.type === "thinking_finished")
+        return previewText(event.text ?? "");
+    if (event.type === "message_queued" || event.type === "message_started")
+        return previewText(event.text);
+    if (event.type === "tool_call" || event.type === "tool_execution_started" || event.type === "tool_execution_updated" || event.type === "tool_execution_finished")
+        return event.toolName;
+    if (event.type === "subagent_session")
+        return `${event.subagentName} via ${event.toolName}`;
+    if (event.type === "execution_result")
+        return event.action;
+    if (event.type === "session_error")
+        return previewText(event.error);
+    if (event.type === "compaction_start" || event.type === "compaction_end")
+        return event.reason;
+    return event.type;
+}
+function attributesForOutputEvent(event) {
+    if (event.type === "message_queued")
+        return { inlineText: event.text, source: event.source, queuedMessages: event.queuedMessages };
+    if (event.type === "assistant_message" || event.type === "assistant_delta")
+        return { assistantIndex: event.assistantIndex, contentIndex: event.contentIndex };
+    if (event.type === "thinking_started" || event.type === "thinking_delta" || event.type === "thinking_finished")
+        return { thinkingIndex: event.thinkingIndex, contentIndex: event.contentIndex };
+    if (event.type === "tool_call")
+        return { toolCallId: event.toolCallId, toolName: event.toolName, argsComplete: event.argsComplete };
+    if (event.type === "tool_execution_started" || event.type === "tool_execution_updated" || event.type === "tool_execution_finished")
+        return { toolCallId: event.toolCallId, toolName: event.toolName, isError: "isError" in event ? event.isError : undefined };
+    if (event.type === "subagent_session")
+        return { toolCallId: event.toolCallId, toolName: event.toolName, subagentName: event.subagentName, childPiboSessionId: event.childPiboSessionId, threadKey: event.threadKey };
+    if (event.type === "execution_result")
+        return { action: event.action };
+    if (event.type === "session_error")
+        return { error: event.error, ...(event.errorDetails ? { errorDetails: event.errorDetails } : {}) };
+    if (event.type === "compaction_start" || event.type === "compaction_end")
+        return { reason: event.reason, aborted: "aborted" in event ? event.aborted : undefined, errorMessage: "errorMessage" in event ? event.errorMessage : undefined };
+    return {};
+}
+function observationKindForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "assistant_delta")
+        return "message";
+    if (event.type.startsWith("thinking_"))
+        return "thinking";
+    if (event.type.startsWith("tool_") || event.type === "subagent_session")
+        return "tool";
+    if (event.type === "execution_result")
+        return "execution";
+    if (event.type === "session_error")
+        return "error";
+    if (event.type.startsWith("compaction_"))
+        return "compaction";
+    return "event";
+}
+function observationRoleForOutputEvent(event) {
+    if (event.type === "assistant_message" || event.type === "assistant_delta" || event.type.startsWith("thinking_"))
+        return "assistant";
+    if (event.type.startsWith("tool_"))
+        return "tool";
+    return undefined;
+}
+function observationNameForOutputEvent(event) {
+    if ("toolName" in event)
+        return event.toolName;
+    if (event.type === "execution_result")
+        return event.action;
+    return event.type;
+}
+function observationStatusForOutputEvent(event) {
+    if (event.type.endsWith("_started") || event.type === "tool_call" || event.type === "compaction_start")
+        return "running";
+    if (event.type === "session_error")
+        return "error";
+    if (event.type === "tool_execution_finished" && event.isError)
+        return "error";
+    return "completed";
+}
+function isTerminalOutputEvent(event) {
+    return !event.type.endsWith("_started") && event.type !== "tool_call" && event.type !== "assistant_delta" && event.type !== "thinking_delta" && event.type !== "tool_execution_updated";
+}
+function outputSessionStatus(event) {
+    if (event.type === "session_error")
+        return "error";
+    if (event.type === "message_finished")
+        return "idle";
+    return "running";
+}
+function toPiboJsonValue(value) {
+    if (value === undefined)
+        return undefined;
+    return JSON.parse(JSON.stringify(value));
+}
+function toPiboJsonValueOrNull(value) {
+    return toPiboJsonValue(value) ?? null;
+}

package/dist/data/message-store.js

@@ -0,0 +1,62 @@
+export class MessageStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    insertMessage(input) {
+        this.db.prepare(`
+			INSERT INTO chat_messages (
+				id,
+				session_id,
+				room_id,
+				sequence,
+				turn_id,
+				role,
+				actor_id,
+				status,
+				created_at,
+				completed_at,
+				content_preview,
+				content_payload_ref,
+				source_stream_id,
+				input_tokens,
+				output_tokens,
+				cost_usd,
+				attributes_json
+			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+		`).run(input.id, input.sessionId, input.roomId ?? null, input.sequence, input.turnId ?? null, input.role, input.actorId ?? null, input.status, input.createdAt, input.completedAt ?? null, input.contentPreview ?? null, input.contentPayloadRef ?? null, input.sourceStreamId ?? null, input.inputTokens ?? null, input.outputTokens ?? null, input.costUsd ?? null, JSON.stringify(input.attributes ?? {}));
+        const stored = this.getMessage(input.id);
+        if (!stored)
+            throw new Error(`Failed to insert chat message \"${input.id}\"`);
+        return stored;
+    }
+    getMessage(id) {
+        const row = this.db.prepare("SELECT * FROM chat_messages WHERE id = ?").get(id);
+        return row ? messageFromRow(row) : undefined;
+    }
+    listMessages(sessionId) {
+        const rows = this.db.prepare("SELECT * FROM chat_messages WHERE session_id = ? ORDER BY sequence ASC").all(sessionId);
+        return rows.map(messageFromRow);
+    }
+}
+function messageFromRow(row) {
+    return {
+        id: row.id,
+        sessionId: row.session_id,
+        roomId: row.room_id ?? undefined,
+        sequence: row.sequence,
+        turnId: row.turn_id ?? undefined,
+        role: row.role,
+        actorId: row.actor_id ?? undefined,
+        status: row.status,
+        createdAt: row.created_at,
+        completedAt: row.completed_at ?? undefined,
+        contentPreview: row.content_preview ?? undefined,
+        contentPayloadRef: row.content_payload_ref ?? undefined,
+        sourceStreamId: row.source_stream_id ?? undefined,
+        inputTokens: row.input_tokens ?? undefined,
+        outputTokens: row.output_tokens ?? undefined,
+        costUsd: row.cost_usd ?? undefined,
+        attributes: JSON.parse(row.attributes_json),
+    };
+}

package/dist/data/navigation-store.js

@@ -0,0 +1,75 @@
+export class NavigationStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    upsertSession(input) {
+        const insertColumns = [
+            "room_id", "session_id", "root_session_id", "parent_id", "origin_id", "title", "profile", "status", "archived_at", "last_activity_at", "last_message_preview", "child_count", "sort_key", "updated_at",
+        ];
+        const assignments = [
+            "room_id = excluded.room_id",
+            "root_session_id = excluded.root_session_id",
+            "parent_id = excluded.parent_id",
+            "origin_id = excluded.origin_id",
+            "title = excluded.title",
+            "profile = excluded.profile",
+            "status = excluded.status",
+            "archived_at = excluded.archived_at",
+            "last_activity_at = excluded.last_activity_at",
+            "last_message_preview = COALESCE(excluded.last_message_preview, session_navigation.last_message_preview)",
+            "child_count = excluded.child_count",
+            "sort_key = excluded.sort_key",
+            "updated_at = excluded.updated_at",
+        ];
+        this.db.prepare(`
+			INSERT INTO session_navigation (${insertColumns.join(", ")})
+			VALUES (${insertColumns.map(() => "?").join(", ")})
+			ON CONFLICT(session_id) DO UPDATE SET ${assignments.join(", ")}
+		`).run(input.roomId ?? null, input.sessionId, input.rootSessionId ?? null, input.parentId ?? null, input.originId ?? null, input.title, input.profile, input.status, input.archivedAt ?? null, input.lastActivityAt, input.lastMessagePreview ?? null, input.childCount ?? 0, input.sortKey, input.updatedAt);
+        const stored = this.getSession(input.sessionId);
+        if (!stored)
+            throw new Error(`Failed to upsert session navigation \"${input.sessionId}\"`);
+        return stored;
+    }
+    getSession(sessionId) {
+        const row = this.db.prepare("SELECT * FROM session_navigation WHERE session_id = ?").get(sessionId);
+        return row ? sessionNavigationFromRow(row) : undefined;
+    }
+    listSessions(input = {}) {
+        const clauses = [];
+        const values = [];
+        if (input.roomId !== undefined) {
+            clauses.push("room_id = ?");
+            values.push(input.roomId);
+        }
+        if (!input.includeArchived)
+            clauses.push("archived_at IS NULL");
+        const limit = Math.max(1, Math.min(input.limit ?? 100, 1000));
+        const rows = this.db.prepare(`
+			SELECT * FROM session_navigation
+			${clauses.length ? `WHERE ${clauses.join(" AND ")}` : ""}
+			ORDER BY sort_key DESC, updated_at DESC
+			LIMIT ?
+		`).all(...values, limit);
+        return rows.map(sessionNavigationFromRow);
+    }
+}
+function sessionNavigationFromRow(row) {
+    return {
+        roomId: row.room_id ?? undefined,
+        sessionId: row.session_id,
+        rootSessionId: row.root_session_id ?? undefined,
+        parentId: row.parent_id ?? undefined,
+        originId: row.origin_id ?? undefined,
+        title: row.title,
+        profile: row.profile,
+        status: row.status,
+        archivedAt: row.archived_at ?? undefined,
+        lastActivityAt: row.last_activity_at,
+        lastMessagePreview: row.last_message_preview ?? undefined,
+        childCount: row.child_count,
+        sortKey: row.sort_key,
+        updatedAt: row.updated_at,
+    };
+}

package/dist/data/observation-store.js

@@ -0,0 +1,46 @@
+import { randomUUID } from "node:crypto";
+export class ObservationStore {
+    db;
+    constructor(db) {
+        this.db = db;
+    }
+    insertObservation(input) {
+        const sequence = input.sequence ?? this.nextSequence(input.sessionId);
+        const startedAt = input.startedAt ?? new Date().toISOString();
+        const id = input.id ?? `obs_${randomUUID()}`;
+        this.db.prepare(`INSERT OR IGNORE INTO observations (
+			id, session_id, sequence, trace_id, span_id, parent_span_id, parent_observation_id, turn_id, event_stream_id,
+			kind, role, name, status, started_at, ended_at, latency_ms, model_provider, model_id, input_tokens, output_tokens,
+			cost_usd, preview_text, payload_ref, attributes_json
+		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(id, input.sessionId, sequence, input.traceId ?? null, input.spanId ?? null, input.parentSpanId ?? null, input.parentObservationId ?? null, input.turnId ?? null, input.eventStreamId ?? null, input.kind, input.role ?? null, input.name ?? null, input.status ?? "completed", startedAt, input.endedAt ?? null, input.latencyMs ?? null, input.modelProvider ?? null, input.modelId ?? null, input.inputTokens ?? null, input.outputTokens ?? null, input.costUsd ?? null, input.previewText ?? null, input.payloadRef ?? null, JSON.stringify(input.attributes ?? {}));
+        return this.getObservation(id) ?? this.listObservations(input.sessionId).find((row) => row.sequence === sequence);
+    }
+    appendObservation(input) {
+        return this.insertObservation(input);
+    }
+    getObservation(id) {
+        const row = this.db.prepare("SELECT * FROM observations WHERE id = ?").get(id);
+        return row ? mapRow(row) : undefined;
+    }
+    listObservations(sessionId, limit = 200) {
+        return this.db.prepare("SELECT * FROM observations WHERE session_id = ? ORDER BY sequence ASC LIMIT ?").all(sessionId, limit).map(mapRow);
+    }
+    listSession(sessionId, limit = 200) {
+        return this.listObservations(sessionId, limit);
+    }
+    nextSequence(sessionId) {
+        const row = this.db.prepare("SELECT COALESCE(MAX(sequence), 0) + 1 AS next_sequence FROM observations WHERE session_id = ?").get(sessionId);
+        return row.next_sequence;
+    }
+}
+function mapRow(row) {
+    return {
+        id: row.id, sessionId: row.session_id, sequence: row.sequence, traceId: row.trace_id ?? undefined, spanId: row.span_id ?? undefined,
+        parentSpanId: row.parent_span_id ?? undefined, parentObservationId: row.parent_observation_id ?? undefined, turnId: row.turn_id ?? undefined,
+        eventStreamId: row.event_stream_id ?? undefined, kind: row.kind, role: row.role ?? undefined, name: row.name ?? undefined,
+        status: row.status, startedAt: row.started_at, endedAt: row.ended_at ?? undefined, latencyMs: row.latency_ms ?? undefined,
+        modelProvider: row.model_provider ?? undefined, modelId: row.model_id ?? undefined, inputTokens: row.input_tokens ?? undefined,
+        outputTokens: row.output_tokens ?? undefined, costUsd: row.cost_usd ?? undefined, previewText: row.preview_text ?? undefined,
+        payloadRef: row.payload_ref ?? undefined, attributes: JSON.parse(row.attributes_json),
+    };
+}