@pasko70/pibo 3.0.1 → 3.0.2

package/dist/sessions/pibo-data-store.js

@@ -100,6 +100,151 @@ export class PiboDataSessionStore {
 		`).run(updated.piSessionId || null, rootSessionId(updated), updated.parentId ?? null, updated.originId ?? null, updated.profile, updated.activeModel ? JSON.stringify(updated.activeModel) : null, updated.workspace ?? null, updated.title ?? "Untitled Session", JSON.stringify(updated.metadata ?? {}), updated.updatedAt, updated.updatedAt, id);
         return this.get(id);
     }
+    claimOutputRenderSequence(piboSessionId, minimum) {
+        return this.dataStore.transaction(() => {
+            const session = this.db.prepare("SELECT 1 AS present FROM sessions WHERE id = ? AND deleted_at IS NULL").get(piboSessionId);
+            if (!session)
+                return minimum;
+            const row = this.db.prepare("SELECT high_water FROM session_output_render_high_water WHERE pibo_session_id = ?").get(piboSessionId);
+            const current = Math.max(row?.high_water ?? 0, this.durableOutputRenderSequenceHighWater(piboSessionId));
+            const next = Math.max(minimum, current + 1);
+            this.db.prepare(`
+				INSERT INTO session_output_render_high_water (pibo_session_id, high_water, updated_at)
+				VALUES (?, ?, ?)
+				ON CONFLICT(pibo_session_id) DO UPDATE SET
+					high_water = MAX(session_output_render_high_water.high_water, excluded.high_water),
+					updated_at = CASE
+						WHEN excluded.high_water > session_output_render_high_water.high_water THEN excluded.updated_at
+						ELSE session_output_render_high_water.updated_at
+					END
+			`).run(piboSessionId, next, new Date().toISOString());
+            return next;
+        });
+    }
+    durableOutputRenderSequenceHighWater(piboSessionId) {
+        const row = this.db.prepare(`
+			SELECT MAX(MAX(
+				COALESCE(CAST(json_extract(attributes_json, '$.renderSequence') AS INTEGER), 0),
+				COALESCE(session_sequence, 0)
+			)) AS high_water
+			FROM event_log
+			WHERE session_id = ?
+		`).get(piboSessionId);
+        const value = row?.high_water;
+        return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : 0;
+    }
+    observeOutputRenderSequence(piboSessionId, sequence) {
+        this.dataStore.transaction(() => {
+            this.db.prepare(`
+				INSERT INTO session_output_render_high_water (pibo_session_id, high_water, updated_at)
+				SELECT id, ?, ? FROM sessions WHERE id = ? AND deleted_at IS NULL
+				ON CONFLICT(pibo_session_id) DO UPDATE SET
+					high_water = MAX(session_output_render_high_water.high_water, excluded.high_water),
+					updated_at = CASE
+						WHEN excluded.high_water > session_output_render_high_water.high_water THEN excluded.updated_at
+						ELSE session_output_render_high_water.updated_at
+					END
+			`).run(sequence, new Date().toISOString(), piboSessionId);
+        });
+    }
+    claimOutputToolInvocationOrdinal(piboSessionId, eventId, toolCallId) {
+        return this.dataStore.transaction(() => {
+            const durable = this.db.prepare(`
+				SELECT MAX(COALESCE(CAST(json_extract(attributes_json, '$.toolInvocationOrdinal') AS INTEGER), 0)) AS maximum
+				FROM event_log
+				WHERE session_id = ? AND event_id = ? AND tool_call_id = ?
+			`).get(piboSessionId, eventId, toolCallId);
+            const minimumNextOrdinal = typeof durable.maximum === "number" && Number.isSafeInteger(durable.maximum)
+                ? durable.maximum + 1
+                : 0;
+            const row = this.db.prepare(`
+				INSERT INTO session_tool_invocation_counters (
+					pibo_session_id, event_id, tool_call_id, next_ordinal, updated_at
+				) VALUES (?, ?, ?, ? + 1, ?)
+				ON CONFLICT(pibo_session_id, event_id, tool_call_id) DO UPDATE SET
+					next_ordinal = MAX(session_tool_invocation_counters.next_ordinal, ? ) + 1,
+					updated_at = excluded.updated_at
+				RETURNING next_ordinal - 1 AS ordinal
+			`).get(piboSessionId, eventId, toolCallId, minimumNextOrdinal, new Date().toISOString(), minimumNextOrdinal);
+            return row.ordinal;
+        });
+    }
+    observeOutputToolInvocationOrdinal(piboSessionId, eventId, toolCallId, ordinal) {
+        this.dataStore.transaction(() => {
+            this.db.prepare(`
+				INSERT INTO session_tool_invocation_counters (
+					pibo_session_id, event_id, tool_call_id, next_ordinal, updated_at
+				) VALUES (?, ?, ?, ?, ?)
+				ON CONFLICT(pibo_session_id, event_id, tool_call_id) DO UPDATE SET
+					next_ordinal = MAX(session_tool_invocation_counters.next_ordinal, excluded.next_ordinal),
+					updated_at = CASE
+						WHEN excluded.next_ordinal > session_tool_invocation_counters.next_ordinal THEN excluded.updated_at
+						ELSE session_tool_invocation_counters.updated_at
+					END
+			`).run(piboSessionId, eventId, toolCallId, ordinal + 1, new Date().toISOString());
+        });
+    }
+    claimOrAttachOutputToolInvocation(input) {
+        return this.dataStore.transaction(() => {
+            const rows = this.db.prepare(`
+				SELECT invocation_ordinal, call_fingerprint, status, seen_call
+				FROM session_tool_invocations
+				WHERE pibo_session_id = ? AND event_id = ? AND tool_call_id = ?
+				ORDER BY invocation_ordinal DESC
+			`).all(input.piboSessionId, input.eventId, input.toolCallId);
+            const latestOpen = rows.find((row) => row.status === "open");
+            let ordinal;
+            if (input.eventType === "tool_call"
+                && latestOpen
+                && (latestOpen.seen_call === 0 || latestOpen.call_fingerprint === input.callFingerprint)) {
+                ordinal = latestOpen.invocation_ordinal;
+            }
+            else if (input.eventType !== "tool_call" && latestOpen) {
+                ordinal = latestOpen.invocation_ordinal;
+            }
+            else if (input.eventType !== "tool_call" && rows[0]) {
+                // A lifecycle replay after completion belongs to the last proven
+                // invocation. Only a new tool_call may cross a closed boundary.
+                ordinal = rows[0].invocation_ordinal;
+            }
+            else {
+                ordinal = this.claimOutputToolInvocationOrdinal(input.piboSessionId, input.eventId, input.toolCallId);
+            }
+            this.persistToolInvocationTransition(input, ordinal);
+            return ordinal;
+        });
+    }
+    observeOutputToolInvocation(input) {
+        this.dataStore.transaction(() => {
+            this.observeOutputToolInvocationOrdinal(input.piboSessionId, input.eventId, input.toolCallId, input.ordinal);
+            this.persistToolInvocationTransition(input, input.ordinal);
+        });
+    }
+    persistToolInvocationTransition(input, ordinal) {
+        const timestamp = new Date().toISOString();
+        const seenCall = input.eventType === "tool_call" ? 1 : 0;
+        const seenStarted = input.eventType === "tool_execution_started" ? 1 : 0;
+        const seenUpdated = input.eventType === "tool_execution_updated" ? 1 : 0;
+        const seenFinished = input.eventType === "tool_execution_finished" ? 1 : 0;
+        this.db.prepare(`
+			INSERT INTO session_tool_invocations (
+				pibo_session_id, event_id, tool_call_id, invocation_ordinal,
+				call_fingerprint, status, seen_call, seen_started, seen_updated, seen_finished,
+				created_at, updated_at
+			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+			ON CONFLICT(pibo_session_id, event_id, tool_call_id, invocation_ordinal) DO UPDATE SET
+				call_fingerprint = COALESCE(session_tool_invocations.call_fingerprint, excluded.call_fingerprint),
+				status = CASE
+					WHEN session_tool_invocations.status = 'closed' OR excluded.status = 'closed' THEN 'closed'
+					ELSE 'open'
+				END,
+				seen_call = MAX(session_tool_invocations.seen_call, excluded.seen_call),
+				seen_started = MAX(session_tool_invocations.seen_started, excluded.seen_started),
+				seen_updated = MAX(session_tool_invocations.seen_updated, excluded.seen_updated),
+				seen_finished = MAX(session_tool_invocations.seen_finished, excluded.seen_finished),
+				updated_at = excluded.updated_at
+		`).run(input.piboSessionId, input.eventId, input.toolCallId, ordinal, input.callFingerprint ?? null, seenFinished ? "closed" : "open", seenCall, seenStarted, seenUpdated, seenFinished, timestamp, timestamp);
+    }
     delete(id) {
         const result = this.db.prepare("DELETE FROM sessions WHERE id = ?").run(id);
         return Number(result.changes ?? 0) > 0;

package/dist/sessions/store.js

@@ -48,6 +48,8 @@ export class InMemoryPiboSessionStore {
     byId = new Map();
     byPiSessionId = new Map();
     byNativeSession = new Map();
+    outputRenderHighWater = new Map();
+    outputToolInvocationNextOrdinal = new Map();
     get(id) {
         return this.byId.get(id);
     }
@@ -101,6 +103,33 @@ export class InMemoryPiboSessionStore {
         this.set(updated, existing.piSessionId, existing.runtimeBinding);
         return updated;
     }
+    claimOutputRenderSequence(piboSessionId, minimum) {
+        const session = this.byId.get(piboSessionId);
+        if (!session)
+            return minimum;
+        const current = Math.max(this.outputRenderHighWater.get(piboSessionId) ?? 0, outputRenderSequenceHighWater(session.metadata));
+        const next = Math.max(minimum, current + 1);
+        this.outputRenderHighWater.set(piboSessionId, next);
+        return next;
+    }
+    observeOutputRenderSequence(piboSessionId, sequence) {
+        const session = this.byId.get(piboSessionId);
+        if (!session)
+            return;
+        const current = Math.max(this.outputRenderHighWater.get(piboSessionId) ?? 0, outputRenderSequenceHighWater(session.metadata));
+        if (sequence > current)
+            this.outputRenderHighWater.set(piboSessionId, sequence);
+    }
+    claimOutputToolInvocationOrdinal(piboSessionId, eventId, toolCallId) {
+        const key = outputToolInvocationCounterKey(piboSessionId, eventId, toolCallId);
+        const ordinal = this.outputToolInvocationNextOrdinal.get(key) ?? 0;
+        this.outputToolInvocationNextOrdinal.set(key, ordinal + 1);
+        return ordinal;
+    }
+    observeOutputToolInvocationOrdinal(piboSessionId, eventId, toolCallId, ordinal) {
+        const key = outputToolInvocationCounterKey(piboSessionId, eventId, toolCallId);
+        this.outputToolInvocationNextOrdinal.set(key, Math.max(this.outputToolInvocationNextOrdinal.get(key) ?? 0, ordinal + 1));
+    }
     getRuntimeBinding(id) {
         const session = this.byId.get(id);
         if (!session)
@@ -133,6 +162,12 @@ export class InMemoryPiboSessionStore {
         const nativeKey = runtimeBindingNativeKey(existing.runtimeBinding);
         if (nativeKey)
             this.byNativeSession.delete(nativeKey);
+        this.outputRenderHighWater.delete(id);
+        const counterPrefix = `${JSON.stringify([id]).slice(0, -1)},`;
+        for (const key of this.outputToolInvocationNextOrdinal.keys()) {
+            if (key.startsWith(counterPrefix))
+                this.outputToolInvocationNextOrdinal.delete(key);
+        }
         return true;
     }
     find(input) {
@@ -165,6 +200,13 @@ export class InMemoryPiboSessionStore {
         return sessions.sort((left, right) => right.updatedAt.localeCompare(left.updatedAt));
     }
 }
+function outputRenderSequenceHighWater(metadata) {
+    const value = metadata?.outputRenderSequenceHighWater;
+    return typeof value === "number" && Number.isSafeInteger(value) && value >= 0 ? value : 0;
+}
+function outputToolInvocationCounterKey(piboSessionId, eventId, toolCallId) {
+    return JSON.stringify([piboSessionId, eventId, toolCallId]);
+}
 export function matchesFindInput(session, input) {
     if (input.ids && !input.ids.includes(session.id))
         return false;