@pasko70/pibo 1.9.11 → 1.9.13

package/dist/apps/chat/data/project-service.js

@@ -26,8 +26,8 @@ export class ChatProjectService {
         const folder = resolve(input.projectFolder ?? piboHomePath("projects/workspace"));
         mkdirSync(folder, { recursive: true });
         const now = new Date().toISOString();
-        const legacy = this.db.prepare(`SELECT * FROM projects
-			WHERE project_folder = ?
+        const legacy = this.db.prepare(`SELECT * FROM projects
+			WHERE project_folder = ?
 				AND (id LIKE 'personal_%' OR json_extract(metadata_json, '$.personal') IS 1 OR json_extract(metadata_json, '$.default') IS 1)`)
             .get(folder);
         if (legacy) {
@@ -49,7 +49,7 @@ export class ChatProjectService {
             }
             return this.requireProject(id);
         }
-        this.db.prepare(`INSERT INTO projects (id, name, description, project_folder, configuration_status, metadata_json, created_at, updated_at)
+        this.db.prepare(`INSERT INTO projects (id, name, description, project_folder, configuration_status, metadata_json, created_at, updated_at)
 			VALUES (?, ?, ?, ?, ?, ?, ?, ?)`).run(id, "Project Manager", null, folder, "configured", JSON.stringify({ default: true }), now, now);
         return this.requireProject(id);
     }
@@ -75,7 +75,7 @@ export class ChatProjectService {
         this.ensureProjectFolderUsable(projectFolder);
         const now = new Date().toISOString();
         const id = `prj_${randomUUID()}`;
-        this.db.prepare(`INSERT INTO projects (id, name, description, project_folder, configuration_status, metadata_json, created_at, updated_at)
+        this.db.prepare(`INSERT INTO projects (id, name, description, project_folder, configuration_status, metadata_json, created_at, updated_at)
 			VALUES (?, ?, ?, ?, ?, ?, ?, ?)`).run(id, name, normalizeOptionalString(input.description) ?? null, projectFolder, "configured", "{}", now, now);
         return this.requireProject(id);
     }
@@ -133,8 +133,8 @@ export class ChatProjectService {
                 configuration,
             });
         }
-        this.db.prepare(`INSERT INTO project_sessions (project_id, pibo_session_id, kind, workflow_id, workflow_version, workflow_run_id, parent_main_session_id, title, state, configuration_json, archived, created_at, updated_at)
-			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?)
+        this.db.prepare(`INSERT INTO project_sessions (project_id, pibo_session_id, kind, workflow_id, workflow_version, workflow_run_id, parent_main_session_id, title, state, configuration_json, archived, created_at, updated_at)
+			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 0, ?, ?)
 			ON CONFLICT(pibo_session_id) DO UPDATE SET project_id = excluded.project_id, kind = excluded.kind, workflow_id = excluded.workflow_id, workflow_version = excluded.workflow_version, workflow_run_id = excluded.workflow_run_id, parent_main_session_id = excluded.parent_main_session_id, title = excluded.title, state = excluded.state, configuration_json = excluded.configuration_json, updated_at = excluded.updated_at`)
             .run(input.projectId, input.piboSessionId, kind, workflowId, workflowVersion ?? null, workflowRunId ?? null, parentMainSessionId ?? null, title ?? null, state, serializeProjectSessionConfiguration(configuration), now, now);
         if (kind === "main") {
@@ -166,7 +166,7 @@ export class ChatProjectService {
         const existingForSession = this.getWorkflowSessionSnapshotForSession(snapshot.piboSessionId);
         if (existingForSession)
             throw new Error(`Project workflow session '${snapshot.piboSessionId}' already has a configuration snapshot`);
-        this.db.prepare(`INSERT INTO project_workflow_session_snapshots (id, schema_version, project_id, pibo_session_id, workflow_id, workflow_version, base_definition_hash, effective_definition_hash, snapshot_json, created_at)
+        this.db.prepare(`INSERT INTO project_workflow_session_snapshots (id, schema_version, project_id, pibo_session_id, workflow_id, workflow_version, base_definition_hash, effective_definition_hash, snapshot_json, created_at)
 			VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(snapshot.id, snapshot.schemaVersion, snapshot.projectId, snapshot.piboSessionId, snapshot.workflow.id, snapshot.workflow.version, snapshot.workflow.baseDefinitionHash, snapshot.workflow.effectiveDefinitionHash, JSON.stringify(snapshot), snapshot.createdAt);
         return this.getWorkflowSessionSnapshot(snapshot.id);
     }
@@ -203,36 +203,36 @@ export class ChatProjectService {
         return rows.map(projectWorkflowRunFromRow);
     }
     saveProjectWorkflowWaitToken(token) {
-        this.db.prepare(`INSERT INTO project_workflow_wait_tokens (
-			id,
-			project_id,
-			pibo_session_id,
-			workflow_run_id,
-			node_attempt_id,
-			human_node_id,
-			actions_json,
-			prompt,
-			schema_json,
-			status,
-			resume_payload_json,
-			resume_payload_present,
-			expires_at,
-			created_at,
-			resolved_at
-		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-		ON CONFLICT(id) DO UPDATE SET
-			project_id = excluded.project_id,
-			pibo_session_id = excluded.pibo_session_id,
-			workflow_run_id = excluded.workflow_run_id,
-			node_attempt_id = excluded.node_attempt_id,
-			human_node_id = excluded.human_node_id,
-			actions_json = excluded.actions_json,
-			prompt = excluded.prompt,
-			schema_json = excluded.schema_json,
-			status = excluded.status,
-			resume_payload_json = excluded.resume_payload_json,
-			resume_payload_present = excluded.resume_payload_present,
-			expires_at = excluded.expires_at,
+        this.db.prepare(`INSERT INTO project_workflow_wait_tokens (
+			id,
+			project_id,
+			pibo_session_id,
+			workflow_run_id,
+			node_attempt_id,
+			human_node_id,
+			actions_json,
+			prompt,
+			schema_json,
+			status,
+			resume_payload_json,
+			resume_payload_present,
+			expires_at,
+			created_at,
+			resolved_at
+		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+		ON CONFLICT(id) DO UPDATE SET
+			project_id = excluded.project_id,
+			pibo_session_id = excluded.pibo_session_id,
+			workflow_run_id = excluded.workflow_run_id,
+			node_attempt_id = excluded.node_attempt_id,
+			human_node_id = excluded.human_node_id,
+			actions_json = excluded.actions_json,
+			prompt = excluded.prompt,
+			schema_json = excluded.schema_json,
+			status = excluded.status,
+			resume_payload_json = excluded.resume_payload_json,
+			resume_payload_present = excluded.resume_payload_present,
+			expires_at = excluded.expires_at,
 			resolved_at = excluded.resolved_at`).run(token.id, token.projectId, token.piboSessionId, token.workflowRunId, token.nodeAttemptId ?? null, token.humanNodeId ?? null, JSON.stringify(token.actions), token.prompt, token.schema ? JSON.stringify(token.schema) : null, token.status, token.resumePayload === undefined ? null : JSON.stringify(token.resumePayload), token.resumePayload === undefined ? 0 : 1, token.expiresAt ?? null, token.createdAt, token.resolvedAt ?? null);
         if (token.status === "pending") {
             const now = new Date().toISOString();
@@ -331,18 +331,18 @@ export class ChatProjectService {
                 ...(input.payload !== undefined ? { payload: input.payload } : {}),
                 createdAt: actedAt,
             };
-            this.db.prepare(`INSERT INTO project_workflow_human_actions (
-				id,
-				project_id,
-				pibo_session_id,
-				workflow_run_id,
-				wait_token_id,
-				action_id,
-				kind,
-				actor_json,
-				payload_json,
-				payload_present,
-				created_at
+            this.db.prepare(`INSERT INTO project_workflow_human_actions (
+				id,
+				project_id,
+				pibo_session_id,
+				workflow_run_id,
+				wait_token_id,
+				action_id,
+				kind,
+				actor_json,
+				payload_json,
+				payload_present,
+				created_at
 			) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`)
                 .run(action.id, action.projectId, action.piboSessionId, action.workflowRunId, action.waitTokenId, action.actionId ?? null, action.kind, action.actor ? JSON.stringify(action.actor) : null, action.payload === undefined ? null : JSON.stringify(action.payload), action.payload === undefined ? 0 : 1, action.createdAt);
             this.db.prepare("UPDATE project_workflow_wait_tokens SET status = ?, resume_payload_json = ?, resume_payload_present = ?, resolved_at = ? WHERE id = ?")
@@ -451,128 +451,128 @@ export class ChatProjectService {
         return run;
     }
     insertProjectWorkflowRun(run) {
-        this.db.prepare(`INSERT INTO project_workflow_runs (
-			id,
-			project_id,
-			pibo_session_id,
-			workflow_id,
-			workflow_version,
-			snapshot_id,
-			effective_definition_hash,
-			status,
-			current_json,
-			input_json,
-			validation_json,
-			created_at,
-			updated_at,
-			completed_at,
-			failed_at,
-			cancelled_at
+        this.db.prepare(`INSERT INTO project_workflow_runs (
+			id,
+			project_id,
+			pibo_session_id,
+			workflow_id,
+			workflow_version,
+			snapshot_id,
+			effective_definition_hash,
+			status,
+			current_json,
+			input_json,
+			validation_json,
+			created_at,
+			updated_at,
+			completed_at,
+			failed_at,
+			cancelled_at
 		) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(run.id, run.projectId, run.piboSessionId, run.workflowId, run.workflowVersion, run.snapshotId, run.effectiveDefinitionHash, run.status, JSON.stringify(run.current), JSON.stringify(run.inputValues), run.validation ? JSON.stringify(run.validation) : null, run.createdAt, run.updatedAt, run.completedAt ?? null, run.failedAt ?? null, run.cancelledAt ?? null);
     }
     applySchema() {
-        this.db.exec(`
-			CREATE TABLE IF NOT EXISTS projects (
-				id TEXT PRIMARY KEY,
-				name TEXT NOT NULL,
-				description TEXT,
-				project_folder TEXT NOT NULL,
-				configuration_status TEXT NOT NULL DEFAULT 'configured',
-				current_main_session_id TEXT,
-				archived_at TEXT,
-				metadata_json TEXT NOT NULL DEFAULT '{}',
-				created_at TEXT NOT NULL,
-				updated_at TEXT NOT NULL
-			);
-			CREATE UNIQUE INDEX IF NOT EXISTS projects_name_unique ON projects (lower(name));
-			CREATE UNIQUE INDEX IF NOT EXISTS projects_folder_unique ON projects (project_folder);
-			CREATE TABLE IF NOT EXISTS project_sessions (
-				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
-				pibo_session_id TEXT PRIMARY KEY,
-				kind TEXT NOT NULL DEFAULT 'main',
-				workflow_id TEXT NOT NULL DEFAULT 'simple-chat',
-				workflow_version TEXT,
-				workflow_run_id TEXT,
-				parent_main_session_id TEXT,
-				title TEXT,
-				state TEXT,
-				configuration_json TEXT,
-				retry_count INTEGER,
-				max_retries INTEGER,
-				archived INTEGER NOT NULL DEFAULT 0,
-				created_at TEXT NOT NULL,
-				updated_at TEXT NOT NULL
-			);
-			CREATE INDEX IF NOT EXISTS project_sessions_project_id_idx ON project_sessions(project_id, archived, created_at);
-			CREATE TABLE IF NOT EXISTS project_workflow_session_snapshots (
-				id TEXT PRIMARY KEY,
-				schema_version INTEGER NOT NULL,
-				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
-				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
-				workflow_id TEXT NOT NULL,
-				workflow_version TEXT NOT NULL,
-				base_definition_hash TEXT NOT NULL,
-				effective_definition_hash TEXT NOT NULL,
-				snapshot_json TEXT NOT NULL,
-				created_at TEXT NOT NULL,
-				UNIQUE(pibo_session_id)
-			);
-			CREATE INDEX IF NOT EXISTS project_workflow_session_snapshots_project_idx ON project_workflow_session_snapshots(project_id, created_at);
-			CREATE TABLE IF NOT EXISTS project_workflow_runs (
-				id TEXT PRIMARY KEY,
-				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
-				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
-				workflow_id TEXT NOT NULL,
-				workflow_version TEXT NOT NULL,
-				snapshot_id TEXT NOT NULL REFERENCES project_workflow_session_snapshots(id) ON DELETE RESTRICT,
-				effective_definition_hash TEXT NOT NULL,
-				status TEXT NOT NULL,
-				current_json TEXT NOT NULL,
-				input_json TEXT NOT NULL,
-				validation_json TEXT,
-				created_at TEXT NOT NULL,
-				updated_at TEXT NOT NULL,
-				completed_at TEXT,
-				failed_at TEXT,
-				cancelled_at TEXT,
-				UNIQUE(pibo_session_id)
-			);
-			CREATE INDEX IF NOT EXISTS project_workflow_runs_project_idx ON project_workflow_runs(project_id, created_at);
-			CREATE INDEX IF NOT EXISTS project_workflow_runs_session_idx ON project_workflow_runs(pibo_session_id);
-			CREATE TABLE IF NOT EXISTS project_workflow_wait_tokens (
-				id TEXT PRIMARY KEY,
-				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
-				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
-				workflow_run_id TEXT NOT NULL REFERENCES project_workflow_runs(id) ON DELETE CASCADE,
-				node_attempt_id TEXT,
-				human_node_id TEXT,
-				actions_json TEXT NOT NULL,
-				prompt TEXT NOT NULL,
-				schema_json TEXT,
-				status TEXT NOT NULL,
-				resume_payload_json TEXT,
-				resume_payload_present INTEGER NOT NULL DEFAULT 0,
-				expires_at TEXT,
-				created_at TEXT NOT NULL,
-				resolved_at TEXT
-			);
-			CREATE INDEX IF NOT EXISTS project_workflow_wait_tokens_run_idx ON project_workflow_wait_tokens(workflow_run_id, status, created_at);
-			CREATE INDEX IF NOT EXISTS project_workflow_wait_tokens_session_idx ON project_workflow_wait_tokens(project_id, pibo_session_id, status, created_at);
-			CREATE TABLE IF NOT EXISTS project_workflow_human_actions (
-				id TEXT PRIMARY KEY,
-				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
-				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
-				workflow_run_id TEXT NOT NULL REFERENCES project_workflow_runs(id) ON DELETE CASCADE,
-				wait_token_id TEXT NOT NULL REFERENCES project_workflow_wait_tokens(id) ON DELETE CASCADE,
-				action_id TEXT,
-				kind TEXT NOT NULL,
-				actor_json TEXT,
-				payload_json TEXT,
-				payload_present INTEGER NOT NULL DEFAULT 0,
-				created_at TEXT NOT NULL
-			);
-			CREATE INDEX IF NOT EXISTS project_workflow_human_actions_run_idx ON project_workflow_human_actions(workflow_run_id, created_at);
-			CREATE INDEX IF NOT EXISTS project_workflow_human_actions_wait_token_idx ON project_workflow_human_actions(wait_token_id, created_at);
+        this.db.exec(`
+			CREATE TABLE IF NOT EXISTS projects (
+				id TEXT PRIMARY KEY,
+				name TEXT NOT NULL,
+				description TEXT,
+				project_folder TEXT NOT NULL,
+				configuration_status TEXT NOT NULL DEFAULT 'configured',
+				current_main_session_id TEXT,
+				archived_at TEXT,
+				metadata_json TEXT NOT NULL DEFAULT '{}',
+				created_at TEXT NOT NULL,
+				updated_at TEXT NOT NULL
+			);
+			CREATE UNIQUE INDEX IF NOT EXISTS projects_name_unique ON projects (lower(name));
+			CREATE UNIQUE INDEX IF NOT EXISTS projects_folder_unique ON projects (project_folder);
+			CREATE TABLE IF NOT EXISTS project_sessions (
+				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+				pibo_session_id TEXT PRIMARY KEY,
+				kind TEXT NOT NULL DEFAULT 'main',
+				workflow_id TEXT NOT NULL DEFAULT 'simple-chat',
+				workflow_version TEXT,
+				workflow_run_id TEXT,
+				parent_main_session_id TEXT,
+				title TEXT,
+				state TEXT,
+				configuration_json TEXT,
+				retry_count INTEGER,
+				max_retries INTEGER,
+				archived INTEGER NOT NULL DEFAULT 0,
+				created_at TEXT NOT NULL,
+				updated_at TEXT NOT NULL
+			);
+			CREATE INDEX IF NOT EXISTS project_sessions_project_id_idx ON project_sessions(project_id, archived, created_at);
+			CREATE TABLE IF NOT EXISTS project_workflow_session_snapshots (
+				id TEXT PRIMARY KEY,
+				schema_version INTEGER NOT NULL,
+				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
+				workflow_id TEXT NOT NULL,
+				workflow_version TEXT NOT NULL,
+				base_definition_hash TEXT NOT NULL,
+				effective_definition_hash TEXT NOT NULL,
+				snapshot_json TEXT NOT NULL,
+				created_at TEXT NOT NULL,
+				UNIQUE(pibo_session_id)
+			);
+			CREATE INDEX IF NOT EXISTS project_workflow_session_snapshots_project_idx ON project_workflow_session_snapshots(project_id, created_at);
+			CREATE TABLE IF NOT EXISTS project_workflow_runs (
+				id TEXT PRIMARY KEY,
+				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
+				workflow_id TEXT NOT NULL,
+				workflow_version TEXT NOT NULL,
+				snapshot_id TEXT NOT NULL REFERENCES project_workflow_session_snapshots(id) ON DELETE RESTRICT,
+				effective_definition_hash TEXT NOT NULL,
+				status TEXT NOT NULL,
+				current_json TEXT NOT NULL,
+				input_json TEXT NOT NULL,
+				validation_json TEXT,
+				created_at TEXT NOT NULL,
+				updated_at TEXT NOT NULL,
+				completed_at TEXT,
+				failed_at TEXT,
+				cancelled_at TEXT,
+				UNIQUE(pibo_session_id)
+			);
+			CREATE INDEX IF NOT EXISTS project_workflow_runs_project_idx ON project_workflow_runs(project_id, created_at);
+			CREATE INDEX IF NOT EXISTS project_workflow_runs_session_idx ON project_workflow_runs(pibo_session_id);
+			CREATE TABLE IF NOT EXISTS project_workflow_wait_tokens (
+				id TEXT PRIMARY KEY,
+				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
+				workflow_run_id TEXT NOT NULL REFERENCES project_workflow_runs(id) ON DELETE CASCADE,
+				node_attempt_id TEXT,
+				human_node_id TEXT,
+				actions_json TEXT NOT NULL,
+				prompt TEXT NOT NULL,
+				schema_json TEXT,
+				status TEXT NOT NULL,
+				resume_payload_json TEXT,
+				resume_payload_present INTEGER NOT NULL DEFAULT 0,
+				expires_at TEXT,
+				created_at TEXT NOT NULL,
+				resolved_at TEXT
+			);
+			CREATE INDEX IF NOT EXISTS project_workflow_wait_tokens_run_idx ON project_workflow_wait_tokens(workflow_run_id, status, created_at);
+			CREATE INDEX IF NOT EXISTS project_workflow_wait_tokens_session_idx ON project_workflow_wait_tokens(project_id, pibo_session_id, status, created_at);
+			CREATE TABLE IF NOT EXISTS project_workflow_human_actions (
+				id TEXT PRIMARY KEY,
+				project_id TEXT NOT NULL REFERENCES projects(id) ON DELETE CASCADE,
+				pibo_session_id TEXT NOT NULL REFERENCES project_sessions(pibo_session_id) ON DELETE CASCADE,
+				workflow_run_id TEXT NOT NULL REFERENCES project_workflow_runs(id) ON DELETE CASCADE,
+				wait_token_id TEXT NOT NULL REFERENCES project_workflow_wait_tokens(id) ON DELETE CASCADE,
+				action_id TEXT,
+				kind TEXT NOT NULL,
+				actor_json TEXT,
+				payload_json TEXT,
+				payload_present INTEGER NOT NULL DEFAULT 0,
+				created_at TEXT NOT NULL
+			);
+			CREATE INDEX IF NOT EXISTS project_workflow_human_actions_run_idx ON project_workflow_human_actions(workflow_run_id, created_at);
+			CREATE INDEX IF NOT EXISTS project_workflow_human_actions_wait_token_idx ON project_workflow_human_actions(wait_token_id, created_at);
 		`);
         this.ensureProjectSessionWorkflowVersionColumn();
         this.ensureProjectSessionConfigurationColumn();

package/dist/apps/chat/data/read-state-service.js

@@ -5,13 +5,13 @@ export class ChatReadStateService {
     }
     markSessionRead(piboSessionId, lastReadStreamId) {
         const now = new Date().toISOString();
-        this.store.db.prepare(`
-			INSERT INTO app_session_read_state (session_id, last_read_stream_id, last_read_at, updated_at)
-			VALUES (?, ?, ?, ?)
-			ON CONFLICT(session_id) DO UPDATE SET
-				last_read_stream_id = MAX(app_session_read_state.last_read_stream_id, excluded.last_read_stream_id),
-				last_read_at = excluded.last_read_at,
-				updated_at = excluded.updated_at
+        this.store.db.prepare(`
+			INSERT INTO app_session_read_state (session_id, last_read_stream_id, last_read_at, updated_at)
+			VALUES (?, ?, ?, ?)
+			ON CONFLICT(session_id) DO UPDATE SET
+				last_read_stream_id = MAX(app_session_read_state.last_read_stream_id, excluded.last_read_stream_id),
+				last_read_at = excluded.last_read_at,
+				updated_at = excluded.updated_at
 		`).run(piboSessionId, lastReadStreamId, now, now);
     }
     countUnreadMessagesBySession(input) {
@@ -21,17 +21,17 @@ export class ChatReadStateService {
         for (let offset = 0; offset < input.piboSessionIds.length; offset += 400) {
             const ids = [...new Set(input.piboSessionIds)].slice(offset, offset + 400);
             const placeholders = ids.map(() => "?").join(", ");
-            const rows = this.store.db.prepare(`
-				SELECT e.session_id, COUNT(*) AS count
-				FROM event_log e
-				LEFT JOIN app_session_read_state reads ON reads.session_id = e.session_id
-				WHERE e.session_id IN (${placeholders})
-					AND e.stream_id > COALESCE(reads.last_read_stream_id, 0)
-					AND (
-						(e.retention_class = 'chat_message' AND e.type IN ('user.message.accepted', 'assistant_message'))
-						OR e.type = 'session_error'
-					)
-				GROUP BY e.session_id
+            const rows = this.store.db.prepare(`
+				SELECT e.session_id, COUNT(*) AS count
+				FROM event_log e
+				LEFT JOIN app_session_read_state reads ON reads.session_id = e.session_id
+				WHERE e.session_id IN (${placeholders})
+					AND e.stream_id > COALESCE(reads.last_read_stream_id, 0)
+					AND (
+						(e.retention_class = 'chat_message' AND e.type IN ('user.message.accepted', 'assistant_message'))
+						OR e.type = 'session_error'
+					)
+				GROUP BY e.session_id
 			`).all(...ids);
             for (const row of rows)
                 if (Number(row.count) > 0)

package/dist/apps/chat/data/session-query-service.js

@@ -58,31 +58,31 @@ export class ChatSessionQueryService {
         const title = session.title || "Untitled Session";
         const metadataJson = JSON.stringify(session.metadata ?? {});
         const activeModelJson = session.activeModel ? JSON.stringify(session.activeModel) : null;
-        const row = this.store.db.prepare(`
-			SELECT
-				s.pi_session_id,
-				s.room_id,
-				s.root_session_id,
-				s.parent_id,
-				s.origin_id,
-				s.channel,
-				s.kind,
-				s.profile,
-				s.active_model_json,
-				s.workspace,
-				s.title,
-				s.status,
-				s.metadata_json,
-				s.created_at,
-				s.updated_at,
-				s.last_activity_at,
-				n.status AS navigation_status,
-				n.title AS navigation_title,
-				n.sort_key AS navigation_sort_key,
-				n.updated_at AS navigation_updated_at
-			FROM sessions s
-			LEFT JOIN session_navigation n ON n.session_id = s.id
-			WHERE s.id = ? AND s.deleted_at IS NULL
+        const row = this.store.db.prepare(`
+			SELECT
+				s.pi_session_id,
+				s.room_id,
+				s.root_session_id,
+				s.parent_id,
+				s.origin_id,
+				s.channel,
+				s.kind,
+				s.profile,
+				s.active_model_json,
+				s.workspace,
+				s.title,
+				s.status,
+				s.metadata_json,
+				s.created_at,
+				s.updated_at,
+				s.last_activity_at,
+				n.status AS navigation_status,
+				n.title AS navigation_title,
+				n.sort_key AS navigation_sort_key,
+				n.updated_at AS navigation_updated_at
+			FROM sessions s
+			LEFT JOIN session_navigation n ON n.session_id = s.id
+			WHERE s.id = ? AND s.deleted_at IS NULL
 		`).get(session.id);
         if (!row)
             return false;

package/dist/apps/chat/data/timeline-query-service.js

@@ -1,3 +1,4 @@
+import { messageTurnTimingsFromEvents } from "../../../shared/trace-event-projection.js";
 import { storedChatEventFromV2Row, storedPiboEventFromV2Row } from "./chat-data-mappers.js";
 export class ChatTimelineQueryService {
     store;
@@ -30,6 +31,16 @@ export class ChatTimelineQueryService {
     listAllSessionEvents(piboSessionId) {
         return this.listTraceEvents({ piboSessionId, limit: 10000, includeLive: true });
     }
+    listMessageTurnTimings(piboSessionId) {
+        const rows = this.store.db.prepare(`
+			SELECT * FROM event_log
+			WHERE session_id = ?
+				AND type IN ('message_started', 'message_finished')
+			ORDER BY session_sequence ASC, stream_id ASC
+		`).all(piboSessionId);
+        const events = rows.map(storedPiboEventFromV2Row).filter((event) => event !== undefined);
+        return messageTurnTimingsFromEvents(events);
+    }
     listTraceEvents(input) {
         const piboSessionId = typeof input === "string" ? input : input.piboSessionId;
         const limit = Math.max(1, Math.min((typeof input === "string" ? undefined : input.limit) ?? 2000, 10000));
@@ -48,14 +59,14 @@ export class ChatTimelineQueryService {
             clauses.push("session_sequence <= ?");
             values.push(beforeOrAtSequence);
         }
-        const rows = this.store.db.prepare(`
-			SELECT * FROM (
-				SELECT * FROM event_log
-				WHERE ${clauses.join(" AND ")}
-				ORDER BY session_sequence DESC, stream_id DESC
-				LIMIT ?
-			)
-			ORDER BY session_sequence ASC, stream_id ASC
+        const rows = this.store.db.prepare(`
+			SELECT * FROM (
+				SELECT * FROM event_log
+				WHERE ${clauses.join(" AND ")}
+				ORDER BY session_sequence DESC, stream_id DESC
+				LIMIT ?
+			)
+			ORDER BY session_sequence ASC, stream_id ASC
 		`).all(...values, limit);
         return rows.map(storedPiboEventFromV2Row).filter((event) => event !== undefined);
     }
@@ -75,12 +86,12 @@ export class ChatTimelineQueryService {
         return rows.map((row) => ({ eventType: row.type, count: Number(row.count) }));
     }
     getLatestEventSequence(piboSessionId) {
-        const row = this.store.db.prepare(`
-			SELECT session_sequence AS latest_sequence
-			FROM event_log
-			WHERE session_id = ?
-			ORDER BY session_sequence DESC, stream_id DESC
-			LIMIT 1
+        const row = this.store.db.prepare(`
+			SELECT session_sequence AS latest_sequence
+			FROM event_log
+			WHERE session_id = ?
+			ORDER BY session_sequence DESC, stream_id DESC
+			LIMIT 1
 		`).get(piboSessionId);
         return row?.latest_sequence ?? 0;
     }