@xopcai/xopc 0.0.285 → 0.0.291

package/dist/src/tasks/task-run-dispatcher.js

@@ -1,10 +1,11 @@
 import { createLogger } from "../utils/logger/index.js";
 import { init_logger } from "../utils/logger.js";
-import { TaskRunRepository } from "./task-run-repository.js";
+import { TaskRunRepository, init_task_run_repository } from "./task-run-repository.js";
 import { TaskRepository } from "./task-repository.js";
 import { TaskApplicationService } from "./task-application-service.js";
 //#region src/tasks/task-run-dispatcher.ts
 init_logger();
+init_task_run_repository();
 const log = createLogger("TaskRunDispatcher");
 var TaskRunDispatcher = class {
 	#runs = new TaskRunRepository();
@@ -42,8 +43,8 @@ var TaskRunDispatcher = class {
 						}
 					}) : run)) continue;
 					const agentId = typeof run.executorRef.agentId === "string" ? run.executorRef.agentId : void 0;
-					const sessionKey = await this.deps.ensureSession(task.id, run.id, agentId);
-					await this.deps.runAgent(run.id, sessionKey, task.contract?.objective ?? task.title);
+					const conversationId = await this.deps.ensureSession(task.id, run.id, agentId);
+					await this.deps.runAgent(run.id, conversationId, task.contract?.objective ?? task.title);
 				} catch (error) {
 					const current = this.#runs.get(run.id);
 					if (current && [

package/dist/src/tasks/task-run-repository.d.ts

@@ -11,7 +11,7 @@ export interface TaskRunCreateInput {
     idempotencyKey: string;
     contractVersion: number;
     scheduledAt?: number;
-    sessionKey?: string;
+    conversationId?: string;
     retryPolicy?: Record<string, unknown>;
     retryOfRunId?: string;
     now?: number;
@@ -35,7 +35,7 @@ export declare class TaskRunRepository {
         expectedVersion: number;
         contextSnapshotId: string;
         policySnapshot: Record<string, unknown>;
-        sessionKey?: string;
+        conversationId?: string;
         timeoutAt?: number;
         actor?: ActorRef;
         now?: number;

package/dist/src/tasks/task-run-repository.js

@@ -1,13 +1,8 @@
-import { getSqliteDatabase, runSqliteWriteTransaction } from "../storage/sqlite/transaction.js";
-import { enqueueTaskAttentionRequiredEvent } from "./task-change-events.js";
+import { __esmMin } from "../../_virtual/_rolldown/runtime.js";
+import { getSqliteDatabase, init_transaction, runSqliteWriteTransaction } from "../storage/sqlite/transaction.js";
+import { enqueueTaskAttentionRequiredEvent, init_task_change_events } from "./task-change-events.js";
 import { randomUUID } from "node:crypto";
 //#region src/tasks/task-run-repository.ts
-const ACTIVE_RUN_STATUSES = [
-	"queued",
-	"running",
-	"waiting",
-	"verifying"
-];
 function parseJson(value) {
 	return JSON.parse(value);
 }
@@ -28,7 +23,7 @@ function fromRow(row) {
 		contractVersion: row.contract_version,
 		...row.context_snapshot_id ? { contextSnapshotId: row.context_snapshot_id } : {},
 		...row.policy_snapshot_json ? { policySnapshot: parseJson(row.policy_snapshot_json) } : {},
-		...row.session_key ? { sessionKey: row.session_key } : {},
+		...row.conversation_id ? { conversationId: row.conversation_id } : {},
 		queuedAt: row.queued_at,
 		...row.scheduled_at === null ? {} : { scheduledAt: row.scheduled_at },
 		...row.started_at === null ? {} : { startedAt: row.started_at },
@@ -78,58 +73,68 @@ function receiptFromRow(row) {
 		finalizedAt: row.finalized_at
 	};
 }
-var TaskRunRepository = class {
-	create(input) {
-		const id = input.id ?? randomUUID();
-		const rootRunId = input.parentRunId ? this.require(input.parentRunId).rootRunId : id;
-		const now = input.now ?? Date.now();
-		return runSqliteWriteTransaction((db) => {
-			const attemptRow = db.prepare(`SELECT COALESCE(MAX(attempt), 0) + 1 AS attempt FROM task_runs
+var ACTIVE_RUN_STATUSES, TaskRunRepository;
+var init_task_run_repository = __esmMin((() => {
+	init_transaction();
+	init_task_change_events();
+	ACTIVE_RUN_STATUSES = [
+		"queued",
+		"running",
+		"waiting",
+		"verifying"
+	];
+	TaskRunRepository = class {
+		create(input) {
+			const id = input.id ?? randomUUID();
+			const rootRunId = input.parentRunId ? this.require(input.parentRunId).rootRunId : id;
+			const now = input.now ?? Date.now();
+			return runSqliteWriteTransaction((db) => {
+				const attemptRow = db.prepare(`SELECT COALESCE(MAX(attempt), 0) + 1 AS attempt FROM task_runs
          WHERE task_id = ? AND parent_run_id IS NULL`).get(input.taskId);
-			db.prepare(`INSERT INTO task_runs (
+				db.prepare(`INSERT INTO task_runs (
           run_id, task_id, root_run_id, parent_run_id, attempt, status,
           executor_kind, executor_ref_json, trigger_json, correlation_id,
-          causation_id, idempotency_key, contract_version, session_key,
+          causation_id, idempotency_key, contract_version, conversation_id,
           queued_at, scheduled_at, retry_policy_json, retry_of_run_id, version
-        ) VALUES (?, ?, ?, ?, ?, 'queued', ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 1)`).run(id, input.taskId, rootRunId, input.parentRunId ?? null, input.parentRunId ? 1 : attemptRow.attempt, input.executorKind, JSON.stringify(input.executorRef), JSON.stringify(input.trigger), input.correlationId, input.causationId ?? null, input.idempotencyKey, input.contractVersion, input.sessionKey ?? null, now, input.scheduledAt ?? null, JSON.stringify(input.retryPolicy ?? {}), input.retryOfRunId ?? null);
-			this.appendEvent(db, id, "task_run.created", {
-				executorKind: input.executorKind,
-				scheduledAt: input.scheduledAt
-			}, { kind: "system" }, now);
-			return this.require(id);
-		});
-	}
-	get(id) {
-		const row = getSqliteDatabase().prepare("SELECT * FROM task_runs WHERE run_id = ?").get(id);
-		return row ? fromRow(row) : void 0;
-	}
-	require(id) {
-		const run = this.get(id);
-		if (!run) throw new Error(`TaskRun not found: ${id}`);
-		return run;
-	}
-	getByIdempotencyKey(key) {
-		const row = getSqliteDatabase().prepare("SELECT * FROM task_runs WHERE idempotency_key = ?").get(key);
-		return row ? fromRow(row) : void 0;
-	}
-	getActiveRoot(taskId) {
-		const row = getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? AND parent_run_id IS NULL
+        ) VALUES (?, ?, ?, ?, ?, 'queued', ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 1)`).run(id, input.taskId, rootRunId, input.parentRunId ?? null, input.parentRunId ? 1 : attemptRow.attempt, input.executorKind, JSON.stringify(input.executorRef), JSON.stringify(input.trigger), input.correlationId, input.causationId ?? null, input.idempotencyKey, input.contractVersion, input.conversationId ?? null, now, input.scheduledAt ?? null, JSON.stringify(input.retryPolicy ?? {}), input.retryOfRunId ?? null);
+				this.appendEvent(db, id, "task_run.created", {
+					executorKind: input.executorKind,
+					scheduledAt: input.scheduledAt
+				}, { kind: "system" }, now);
+				return this.require(id);
+			});
+		}
+		get(id) {
+			const row = getSqliteDatabase().prepare("SELECT * FROM task_runs WHERE run_id = ?").get(id);
+			return row ? fromRow(row) : void 0;
+		}
+		require(id) {
+			const run = this.get(id);
+			if (!run) throw new Error(`TaskRun not found: ${id}`);
+			return run;
+		}
+		getByIdempotencyKey(key) {
+			const row = getSqliteDatabase().prepare("SELECT * FROM task_runs WHERE idempotency_key = ?").get(key);
+			return row ? fromRow(row) : void 0;
+		}
+		getActiveRoot(taskId) {
+			const row = getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? AND parent_run_id IS NULL
        AND status IN ('queued', 'running', 'waiting', 'verifying')
        ORDER BY queued_at DESC LIMIT 1`).get(taskId);
-		return row ? fromRow(row) : void 0;
-	}
-	getLatestRoot(taskId) {
-		const row = getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? AND parent_run_id IS NULL
+			return row ? fromRow(row) : void 0;
+		}
+		getLatestRoot(taskId) {
+			const row = getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? AND parent_run_id IS NULL
        ORDER BY queued_at DESC LIMIT 1`).get(taskId);
-		return row ? fromRow(row) : void 0;
-	}
-	listByTask(taskId, limit = 100) {
-		return getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? ORDER BY queued_at DESC LIMIT ?`).all(taskId, Math.max(1, Math.min(500, Math.floor(limit)))).map(fromRow);
-	}
-	claimNext(input) {
-		const now = input.now ?? Date.now();
-		return runSqliteWriteTransaction((db) => {
-			const row = db.prepare(`SELECT * FROM task_runs
+			return row ? fromRow(row) : void 0;
+		}
+		listByTask(taskId, limit = 100) {
+			return getSqliteDatabase().prepare(`SELECT * FROM task_runs WHERE task_id = ? ORDER BY queued_at DESC LIMIT ?`).all(taskId, Math.max(1, Math.min(500, Math.floor(limit)))).map(fromRow);
+		}
+		claimNext(input) {
+			const now = input.now ?? Date.now();
+			return runSqliteWriteTransaction((db) => {
+				const row = db.prepare(`SELECT * FROM task_runs
          WHERE (
            status = 'queued'
            OR (status = 'waiting' AND NOT EXISTS (
@@ -144,186 +149,188 @@ var TaskRunRepository = class {
            AND COALESCE(scheduled_at, 0) <= ?
            AND (lease_expires_at IS NULL OR lease_expires_at <= ?)
          ORDER BY COALESCE(scheduled_at, queued_at), queued_at LIMIT 1`).get(input.executorKind ?? null, input.executorKind ?? null, now, now);
-			if (!row) return void 0;
-			return db.prepare(`UPDATE task_runs SET lease_owner = ?, lease_expires_at = ?, version = version + 1
+				if (!row) return void 0;
+				return db.prepare(`UPDATE task_runs SET lease_owner = ?, lease_expires_at = ?, version = version + 1
          WHERE run_id = ? AND version = ?
            AND (lease_expires_at IS NULL OR lease_expires_at <= ?)`).run(input.owner, now + input.leaseMs, row.run_id, row.version, now).changes === 0 ? void 0 : this.get(row.run_id);
-		});
-	}
-	start(input) {
-		const now = input.now ?? Date.now();
-		return this.transition({
-			runId: input.runId,
-			expectedVersion: input.expectedVersion,
-			from: ["queued"],
-			to: "running",
-			fields: {
-				context_snapshot_id: input.contextSnapshotId,
-				policy_snapshot_json: JSON.stringify(input.policySnapshot),
-				session_key: input.sessionKey ?? null,
-				started_at: now,
-				heartbeat_at: now,
-				timeout_at: input.timeoutAt ?? null
-			},
-			actor: input.actor ?? { kind: "system" },
-			now
-		});
-	}
-	setStatus(input) {
-		return this.transition({
-			...input,
-			fields: input.to === "running" ? { heartbeat_at: input.now ?? Date.now() } : input.to === "waiting" ? {
-				lease_owner: null,
-				lease_expires_at: null
-			} : {},
-			actor: input.actor ?? { kind: "system" },
-			now: input.now ?? Date.now()
-		});
-	}
-	heartbeat(input) {
-		const now = input.now ?? Date.now();
-		return getSqliteDatabase().prepare(`UPDATE task_runs SET heartbeat_at = ?, lease_expires_at = ?, version = version + 1
+			});
+		}
+		start(input) {
+			const now = input.now ?? Date.now();
+			return this.transition({
+				runId: input.runId,
+				expectedVersion: input.expectedVersion,
+				from: ["queued"],
+				to: "running",
+				fields: {
+					context_snapshot_id: input.contextSnapshotId,
+					policy_snapshot_json: JSON.stringify(input.policySnapshot),
+					conversation_id: input.conversationId ?? null,
+					started_at: now,
+					heartbeat_at: now,
+					timeout_at: input.timeoutAt ?? null
+				},
+				actor: input.actor ?? { kind: "system" },
+				now
+			});
+		}
+		setStatus(input) {
+			return this.transition({
+				...input,
+				fields: input.to === "running" ? { heartbeat_at: input.now ?? Date.now() } : input.to === "waiting" ? {
+					lease_owner: null,
+					lease_expires_at: null
+				} : {},
+				actor: input.actor ?? { kind: "system" },
+				now: input.now ?? Date.now()
+			});
+		}
+		heartbeat(input) {
+			const now = input.now ?? Date.now();
+			return getSqliteDatabase().prepare(`UPDATE task_runs SET heartbeat_at = ?, lease_expires_at = ?, version = version + 1
        WHERE run_id = ? AND lease_owner = ?
          AND status IN ('running', 'waiting', 'verifying')`).run(now, now + input.leaseMs, input.runId, input.owner).changes === 0 ? void 0 : this.get(input.runId);
-	}
-	finalize(input) {
-		const now = input.now ?? Date.now();
-		return runSqliteWriteTransaction((db) => {
-			const run = this.get(input.runId);
-			if (!run || run.version !== input.expectedVersion || !ACTIVE_RUN_STATUSES.includes(run.status)) return;
-			if (db.prepare(`UPDATE task_runs SET status = ?, completed_at = ?, terminal_code = ?,
+		}
+		finalize(input) {
+			const now = input.now ?? Date.now();
+			return runSqliteWriteTransaction((db) => {
+				const run = this.get(input.runId);
+				if (!run || run.version !== input.expectedVersion || !ACTIVE_RUN_STATUSES.includes(run.status)) return;
+				if (db.prepare(`UPDATE task_runs SET status = ?, completed_at = ?, terminal_code = ?,
          terminal_message = ?, lease_owner = NULL, lease_expires_at = NULL,
          version = version + 1 WHERE run_id = ? AND version = ?`).run(input.receipt.status, now, input.terminalCode ?? null, input.terminalMessage ?? null, input.runId, input.expectedVersion).changes === 0) return void 0;
-			db.prepare(`INSERT INTO task_run_receipts (
+				db.prepare(`INSERT INTO task_run_receipts (
           run_id, status, summary, changes_json, evidence_json, verification_json,
           remaining_work_json, next_action, needs_user, completion_verdict,
           failure_json, judgment_json, context_trace_id, finalized_at
         ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`).run(input.runId, input.receipt.status, input.receipt.summary, JSON.stringify(input.receipt.changes), JSON.stringify(input.receipt.evidence), JSON.stringify(input.receipt.verification), JSON.stringify(input.receipt.remainingWork), input.receipt.nextAction ?? null, input.receipt.needsUser ? 1 : 0, input.receipt.completionVerdict ?? null, input.receipt.failure ? JSON.stringify(input.receipt.failure) : null, input.receipt.judgment ? JSON.stringify(input.receipt.judgment) : null, input.receipt.contextTraceId ?? null, now);
-			this.appendEvent(db, input.runId, `task_run.${input.receipt.status}`, { completionVerdict: input.receipt.completionVerdict }, input.actor ?? { kind: "system" }, now);
-			return this.get(input.runId);
-		});
-	}
-	getReceipt(runId) {
-		const row = getSqliteDatabase().prepare("SELECT * FROM task_run_receipts WHERE run_id = ?").get(runId);
-		return row ? receiptFromRow(row) : void 0;
-	}
-	listReceipts(taskId, limit = 100) {
-		return getSqliteDatabase().prepare(`SELECT receipt.* FROM task_run_receipts receipt
+				this.appendEvent(db, input.runId, `task_run.${input.receipt.status}`, { completionVerdict: input.receipt.completionVerdict }, input.actor ?? { kind: "system" }, now);
+				return this.get(input.runId);
+			});
+		}
+		getReceipt(runId) {
+			const row = getSqliteDatabase().prepare("SELECT * FROM task_run_receipts WHERE run_id = ?").get(runId);
+			return row ? receiptFromRow(row) : void 0;
+		}
+		listReceipts(taskId, limit = 100) {
+			return getSqliteDatabase().prepare(`SELECT receipt.* FROM task_run_receipts receipt
        JOIN task_runs run ON run.run_id = receipt.run_id
        WHERE run.task_id = ? AND run.parent_run_id IS NULL
        ORDER BY receipt.finalized_at DESC, run.queued_at DESC LIMIT ?`).all(taskId, Math.max(1, Math.min(500, Math.floor(limit)))).map(receiptFromRow);
-	}
-	recordFeedback(input) {
-		this.require(input.runId);
-		const id = randomUUID();
-		const createdAt = input.now ?? Date.now();
-		getSqliteDatabase().prepare(`INSERT INTO task_run_feedback (
+		}
+		recordFeedback(input) {
+			this.require(input.runId);
+			const id = randomUUID();
+			const createdAt = input.now ?? Date.now();
+			getSqliteDatabase().prepare(`INSERT INTO task_run_feedback (
         feedback_id, run_id, rating, reason, needs_correction, created_at
       ) VALUES (?, ?, ?, ?, ?, ?)`).run(id, input.runId, input.rating, input.reason?.trim() || null, input.rating === "not_helpful" ? 1 : 0, createdAt);
-		return {
-			id,
-			runId: input.runId,
-			rating: input.rating,
-			...input.reason?.trim() ? { reason: input.reason.trim() } : {},
-			createdAt
-		};
-	}
-	createWait(input) {
-		const id = randomUUID();
-		const now = input.now ?? Date.now();
-		runSqliteWriteTransaction((db) => {
-			db.prepare(`INSERT INTO task_waits (
+			return {
+				id,
+				runId: input.runId,
+				rating: input.rating,
+				...input.reason?.trim() ? { reason: input.reason.trim() } : {},
+				createdAt
+			};
+		}
+		createWait(input) {
+			const id = randomUUID();
+			const now = input.now ?? Date.now();
+			runSqliteWriteTransaction((db) => {
+				db.prepare(`INSERT INTO task_waits (
           wait_id, task_id, task_run_id, kind, status, reason,
           condition_json, resume_at, created_at
         ) VALUES (?, ?, ?, ?, 'active', ?, ?, ?, ?)`).run(id, input.taskId, input.taskRunId ?? null, input.kind, input.reason.trim(), JSON.stringify(input.condition ?? {}), input.resumeAt ?? null, now);
-			if (input.taskRunId) this.appendEvent(db, input.taskRunId, "task_run.wait_created", {
-				waitId: id,
-				kind: input.kind,
-				reason: input.reason.trim(),
-				resumeAt: input.resumeAt
-			}, input.actor ?? { kind: "system" }, now);
-			if (input.kind === "user_input" || input.kind === "approval") {
-				const task = db.prepare("SELECT title, project_id FROM tasks WHERE task_id = ?").get(input.taskId);
-				if (task) enqueueTaskAttentionRequiredEvent(db, {
-					taskId: input.taskId,
-					taskTitle: task.title,
-					...task.project_id ? { projectId: task.project_id } : {},
-					reason: input.kind,
-					detail: input.reason.trim(),
-					correlationId: id,
-					occurredAt: now
-				});
-			}
-		});
-		return this.requireWait(id);
-	}
-	getWait(waitId) {
-		const row = getSqliteDatabase().prepare("SELECT * FROM task_waits WHERE wait_id = ?").get(waitId);
-		return row ? waitFromRow(row) : void 0;
-	}
-	requireWait(waitId) {
-		const wait = this.getWait(waitId);
-		if (!wait) throw new Error(`TaskWait not found: ${waitId}`);
-		return wait;
-	}
-	listActiveWaits(taskId) {
-		return getSqliteDatabase().prepare(`SELECT * FROM task_waits WHERE task_id = ? AND status = 'active'
+				if (input.taskRunId) this.appendEvent(db, input.taskRunId, "task_run.wait_created", {
+					waitId: id,
+					kind: input.kind,
+					reason: input.reason.trim(),
+					resumeAt: input.resumeAt
+				}, input.actor ?? { kind: "system" }, now);
+				if (input.kind === "user_input" || input.kind === "approval") {
+					const task = db.prepare("SELECT title, project_id FROM tasks WHERE task_id = ?").get(input.taskId);
+					if (task) enqueueTaskAttentionRequiredEvent(db, {
+						taskId: input.taskId,
+						taskTitle: task.title,
+						...task.project_id ? { projectId: task.project_id } : {},
+						reason: input.kind,
+						detail: input.reason.trim(),
+						correlationId: id,
+						occurredAt: now
+					});
+				}
+			});
+			return this.requireWait(id);
+		}
+		getWait(waitId) {
+			const row = getSqliteDatabase().prepare("SELECT * FROM task_waits WHERE wait_id = ?").get(waitId);
+			return row ? waitFromRow(row) : void 0;
+		}
+		requireWait(waitId) {
+			const wait = this.getWait(waitId);
+			if (!wait) throw new Error(`TaskWait not found: ${waitId}`);
+			return wait;
+		}
+		listActiveWaits(taskId) {
+			return getSqliteDatabase().prepare(`SELECT * FROM task_waits WHERE task_id = ? AND status = 'active'
        ORDER BY created_at ASC`).all(taskId).map(waitFromRow);
-	}
-	resolveWait(input) {
-		const now = input.now ?? Date.now();
-		return runSqliteWriteTransaction((db) => {
-			const wait = this.getWait(input.waitId);
-			if (!wait || wait.status !== "active") return void 0;
-			if (db.prepare(`UPDATE task_waits SET status = 'resolved', resolved_by_json = ?,
+		}
+		resolveWait(input) {
+			const now = input.now ?? Date.now();
+			return runSqliteWriteTransaction((db) => {
+				const wait = this.getWait(input.waitId);
+				if (!wait || wait.status !== "active") return void 0;
+				if (db.prepare(`UPDATE task_waits SET status = 'resolved', resolved_by_json = ?,
          resolution_json = ?, resolved_at = ? WHERE wait_id = ? AND status = 'active'`).run(JSON.stringify(input.actor), input.resolution === void 0 ? null : JSON.stringify(input.resolution), now, input.waitId).changes === 0) return void 0;
-			if (wait.taskRunId) this.appendEvent(db, wait.taskRunId, "task_run.wait_resolved", {
-				waitId: wait.id,
-				resolution: input.resolution
-			}, input.actor, now);
-			return this.getWait(input.waitId);
-		});
-	}
-	listEvents(runId) {
-		return getSqliteDatabase().prepare(`SELECT event_id, run_id, sequence, type, payload_json, actor_json, occurred_at
+				if (wait.taskRunId) this.appendEvent(db, wait.taskRunId, "task_run.wait_resolved", {
+					waitId: wait.id,
+					resolution: input.resolution
+				}, input.actor, now);
+				return this.getWait(input.waitId);
+			});
+		}
+		listEvents(runId) {
+			return getSqliteDatabase().prepare(`SELECT event_id, run_id, sequence, type, payload_json, actor_json, occurred_at
        FROM task_run_events WHERE run_id = ? ORDER BY sequence`).all(runId).map((row) => ({
-			id: row.event_id,
-			runId: row.run_id,
-			sequence: row.sequence,
-			type: row.type,
-			payload: parseJson(row.payload_json),
-			actor: parseJson(row.actor_json),
-			occurredAt: row.occurred_at
-		}));
-	}
-	transition(input) {
-		const allowedColumns = new Set([
-			"context_snapshot_id",
-			"policy_snapshot_json",
-			"session_key",
-			"started_at",
-			"heartbeat_at",
-			"timeout_at",
-			"lease_owner",
-			"lease_expires_at"
-		]);
-		const entries = Object.entries(input.fields);
-		for (const [column] of entries) if (!allowedColumns.has(column)) throw new Error(`Unsupported TaskRun field: ${column}`);
-		return runSqliteWriteTransaction((db) => {
-			const placeholders = input.from.map(() => "?").join(", ");
-			const assignments = entries.map(([column]) => `${column} = ?`);
-			if (db.prepare(`UPDATE task_runs SET status = ?, ${assignments.length ? `${assignments.join(", ")}, ` : ""}
+				id: row.event_id,
+				runId: row.run_id,
+				sequence: row.sequence,
+				type: row.type,
+				payload: parseJson(row.payload_json),
+				actor: parseJson(row.actor_json),
+				occurredAt: row.occurred_at
+			}));
+		}
+		transition(input) {
+			const allowedColumns = new Set([
+				"context_snapshot_id",
+				"policy_snapshot_json",
+				"conversation_id",
+				"started_at",
+				"heartbeat_at",
+				"timeout_at",
+				"lease_owner",
+				"lease_expires_at"
+			]);
+			const entries = Object.entries(input.fields);
+			for (const [column] of entries) if (!allowedColumns.has(column)) throw new Error(`Unsupported TaskRun field: ${column}`);
+			return runSqliteWriteTransaction((db) => {
+				const placeholders = input.from.map(() => "?").join(", ");
+				const assignments = entries.map(([column]) => `${column} = ?`);
+				if (db.prepare(`UPDATE task_runs SET status = ?, ${assignments.length ? `${assignments.join(", ")}, ` : ""}
          version = version + 1 WHERE run_id = ? AND version = ? AND status IN (${placeholders})`).run(input.to, ...entries.map(([, value]) => value), input.runId, input.expectedVersion, ...input.from).changes === 0) return void 0;
-			this.appendEvent(db, input.runId, `task_run.${input.to}`, {}, input.actor, input.now);
-			return this.get(input.runId);
-		});
-	}
-	appendEvent(db, runId, type, payload, actor, occurredAt) {
-		const row = db.prepare("SELECT COALESCE(MAX(sequence), 0) + 1 AS sequence FROM task_run_events WHERE run_id = ?").get(runId);
-		db.prepare(`INSERT INTO task_run_events (
+				this.appendEvent(db, input.runId, `task_run.${input.to}`, {}, input.actor, input.now);
+				return this.get(input.runId);
+			});
+		}
+		appendEvent(db, runId, type, payload, actor, occurredAt) {
+			const row = db.prepare("SELECT COALESCE(MAX(sequence), 0) + 1 AS sequence FROM task_run_events WHERE run_id = ?").get(runId);
+			db.prepare(`INSERT INTO task_run_events (
         event_id, run_id, sequence, type, payload_json, actor_json, occurred_at
       ) VALUES (?, ?, ?, ?, ?, ?, ?)`).run(randomUUID(), runId, row.sequence, type, JSON.stringify(payload), JSON.stringify(actor), occurredAt);
-	}
-};
+		}
+	};
+}));
 //#endregion
-export { TaskRunRepository };
+init_task_run_repository();
+export { TaskRunRepository, init_task_run_repository };

package/dist/src/tasks/task-signal-service.js

@@ -1,10 +1,14 @@
-import { getSqliteDatabase } from "../storage/sqlite/transaction.js";
+import { getSqliteDatabase, init_transaction } from "../storage/sqlite/transaction.js";
 import { TaskExecutorSelectionSchema } from "../../packages/gateway-contract/src/tasks.js";
-import { TaskRunRepository } from "./task-run-repository.js";
+import { init_src } from "../../packages/gateway-contract/src/index.js";
+import { TaskRunRepository, init_task_run_repository } from "./task-run-repository.js";
 import { TaskRepository } from "./task-repository.js";
 import { TaskApplicationService } from "./task-application-service.js";
 import { randomUUID } from "node:crypto";
 //#region src/tasks/task-signal-service.ts
+init_src();
+init_transaction();
+init_task_run_repository();
 var TaskSignalService = class {
 	#runs = new TaskRunRepository();
 	#tasks = new TaskRepository();

package/dist/src/tasks/task-workflow-coordinator.js

@@ -1,6 +1,7 @@
-import { TaskRunRepository } from "./task-run-repository.js";
+import { TaskRunRepository, init_task_run_repository } from "./task-run-repository.js";
 import { TaskApplicationService } from "./task-application-service.js";
 //#region src/tasks/task-workflow-coordinator.ts
+init_task_run_repository();
 var TaskWorkflowCoordinator = class {
 	#runs = new TaskRunRepository();
 	#application = new TaskApplicationService();