@xopcai/xopc 0.0.331 → 0.0.333

package/dist/src/storage/sqlite/session-read-repository.js

@@ -22,7 +22,6 @@ var init_session_read_repository = __esmMin((() => {
   s.project_id, s.routing_json, s.custom_data_json,
   s.message_count, s.estimated_tokens, s.compacted_count,
   s.last_flushed_at, s.flush_count,
-  s.thinking_level, s.verbose_level,
   t.cwd AS cwd
 `;
 	SESSION_FROM_JOIN = `

package/dist/src/storage/sqlite/session-repository.d.ts

@@ -21,10 +21,6 @@ export declare function resetSessionRecord(conversationId: string, cwd: string):
 } | null;
 export declare function deleteSessionRecord(conversationId: string): boolean;
 export declare function listSessionsByAgent(agentId: string): SessionMetadata[];
-export declare function getSessionPersistedLevels(conversationId: string): {
-    thinkingLevel: string | null;
-    verboseLevel: string | null;
-} | null;
 export declare function findConversationIdByTranscriptId(transcriptId: string): string | null;
 export declare function getGlobalSessionStats(): ReturnType<typeof buildGlobalSessionStats>;
 export declare function resolveSessionAgentId(conversationId: string): string;

package/dist/src/storage/sqlite/session-repository.js

@@ -4,6 +4,7 @@ import { validateConversationId, validateTranscriptId } from "../../../packages/
 import { init_src } from "../../../packages/gateway-contract/src/index.js";
 import { init_session_instance_repository, readCurrentTranscriptId } from "./session-instance-repository.js";
 import { buildDefaultSessionMetadata, init_session_metadata } from "./session-metadata.js";
+import { init_timestamps, optionalTimestampToMs } from "./timestamps.js";
 import { buildGlobalSessionStats, init_row_mappers, metadataToSessionInsert, sessionRowToMetadata } from "./row-mappers.js";
 import { cancelConnectionObjective, init_connection_wait_repository } from "./connection-wait-repository.js";
 import { init_changes, notifyUserContextChange } from "../../user-context/changes.js";
@@ -21,17 +22,15 @@ function insertSessionAndTranscript(db, conversationId, transcriptId, cwd, metad
       parent_conversation_id, workflow_run_id, workflow_definition_id, workflow_agent_id, workflow_agent_label,
       project_id, routing_json, custom_data_json,
       message_count, estimated_tokens, compacted_count,
-      last_flushed_at, flush_count,
-      thinking_level, verbose_level
+      last_flushed_at, flush_count
     ) VALUES (
       ?, ?, ?, ?, ?,
       ?, ?, ?, ?, ?,
       ?, ?, ?, ?,
       ?, ?, ?, ?, ?,
       ?, ?, ?,
-      ?, ?, ?, ?, ?, ?,
-      ?, ?
-    )`).run(row.conversationId, row.agentId, row.transcriptId, row.status, row.name, row.tagsJson, row.createdAt, row.updatedAt, row.lastAccessedAt, row.sessionStartedAt, row.lastInteractionAt, row.sourceChannel, row.sourceChatId, row.sessionType, row.hiddenFromSessionList, row.parentConversationId, row.workflowRunId, row.workflowDefinitionId, row.workflowAgentId, row.workflowAgentLabel, row.projectId, row.routingJson, row.customDataJson, row.messageCount, row.estimatedTokens, row.compactedCount, row.lastFlushedAt, row.flushCount, row.thinkingLevel, row.verboseLevel);
+      ?, ?, ?, ?, ?, ?
+    )`).run(row.conversationId, row.agentId, row.transcriptId, row.status, row.name, row.tagsJson, row.createdAt, row.updatedAt, row.lastAccessedAt, row.sessionStartedAt, row.lastInteractionAt, row.sourceChannel, row.sourceChatId, row.sessionType, row.hiddenFromSessionList, row.parentConversationId, row.workflowRunId, row.workflowDefinitionId, row.workflowAgentId, row.workflowAgentLabel, row.projectId, row.routingJson, row.customDataJson, row.messageCount, row.estimatedTokens, row.compactedCount, row.lastFlushedAt, row.flushCount);
 	const now = Date.now();
 	db.prepare(`INSERT INTO transcripts (transcript_id, conversation_id, status, created_at, cwd)
      VALUES (?, ?, 'active', ?, ?)`).run(transcriptId, conversationId, now, cwd);
@@ -224,10 +223,8 @@ function patchSessionMetadata(conversationId, updates) {
         estimated_tokens = ?,
         compacted_count = ?,
         last_flushed_at = ?,
-        flush_count = ?,
-        thinking_level = ?,
-        verbose_level = ?
-      WHERE conversation_id = ?`).run(merged.status, merged.name ?? null, JSON.stringify(merged.tags ?? []), Date.parse(merged.updatedAt) || now, Date.parse(merged.lastAccessedAt) || now, merged.sessionStartedAt ? Date.parse(merged.sessionStartedAt) : existing.session_started_at, merged.lastInteractionAt ? Date.parse(merged.lastInteractionAt) : existing.last_interaction_at, merged.sourceChannel, merged.sourceChatId, merged.sessionType, merged.hiddenFromSessionList ? 1 : 0, merged.parentConversationId ?? null, merged.workflowRunId ?? null, merged.workflowDefinitionId ?? null, merged.workflowAgentId ?? null, merged.workflowAgentLabel ?? null, merged.projectId ?? null, merged.routing ? JSON.stringify(merged.routing) : null, merged.customData ? JSON.stringify(merged.customData) : null, merged.messageCount, merged.estimatedTokens, merged.compactedCount, merged.lastFlushedAt ?? null, merged.flushCount ?? 0, existing.thinking_level, existing.verbose_level, conversationId);
+        flush_count = ?
+      WHERE conversation_id = ?`).run(merged.status, merged.name ?? null, JSON.stringify(merged.tags ?? []), Date.parse(merged.updatedAt) || now, Date.parse(merged.lastAccessedAt) || now, merged.sessionStartedAt ? Date.parse(merged.sessionStartedAt) : existing.session_started_at, merged.lastInteractionAt ? Date.parse(merged.lastInteractionAt) : existing.last_interaction_at, merged.sourceChannel, merged.sourceChatId, merged.sessionType, merged.hiddenFromSessionList ? 1 : 0, merged.parentConversationId ?? null, merged.workflowRunId ?? null, merged.workflowDefinitionId ?? null, merged.workflowAgentId ?? null, merged.workflowAgentLabel ?? null, merged.projectId ?? null, merged.routing ? JSON.stringify(merged.routing) : null, merged.customData ? JSON.stringify(merged.customData) : null, merged.messageCount, merged.estimatedTokens, merged.compactedCount, optionalTimestampToMs(merged.lastFlushedAt), merged.flushCount ?? 0, conversationId);
 		const row = readSessionRow(db, conversationId);
 		if (!row) throw new Error(`Session not found after patch: ${conversationId}`);
 		return sessionRowToMetadata(conversationId, row);
@@ -308,14 +305,6 @@ function listSessionsByAgent(agentId) {
        WHERE s.agent_id = ?
        ORDER BY s.updated_at DESC`).all(agentId.toLowerCase()).map((row) => sessionRowToMetadata(row.conversation_id, row));
 }
-function getSessionPersistedLevels(conversationId) {
-	const row = getSqliteDatabase().prepare(`SELECT thinking_level, verbose_level FROM sessions WHERE conversation_id = ?`).get(conversationId);
-	if (!row) return null;
-	return {
-		thinkingLevel: row.thinking_level,
-		verboseLevel: row.verbose_level
-	};
-}
 function findConversationIdByTranscriptId(transcriptId) {
 	return getSqliteDatabase().prepare(`SELECT conversation_id FROM sessions WHERE active_transcript_id = ?`).get(transcriptId)?.conversation_id ?? null;
 }
@@ -335,6 +324,7 @@ var init_session_repository = __esmMin((() => {
 	init_session_metadata();
 	init_session_instance_repository();
 	init_row_mappers();
+	init_timestamps();
 	init_transaction();
 	init_session_identity_sql();
 	init_fts();
@@ -342,4 +332,4 @@ var init_session_repository = __esmMin((() => {
 }));
 //#endregion
 init_session_repository();
-export { deleteSessionRecord, ensureSessionInTransaction, ensureSessionRecord, findConversationIdByTranscriptId, getCurrentTranscriptId, getGlobalSessionStats, getSessionMetadata, getSessionPersistedLevels, incrementSessionStatsOnAppend, init_session_repository, listSessionMetadata, listSessionsByAgent, patchSessionMetadata, readCurrentTranscriptId, resetSessionRecord, resolveSessionAgentId, updateSessionStats };
+export { deleteSessionRecord, ensureSessionInTransaction, ensureSessionRecord, findConversationIdByTranscriptId, getCurrentTranscriptId, getGlobalSessionStats, getSessionMetadata, incrementSessionStatsOnAppend, init_session_repository, listSessionMetadata, listSessionsByAgent, patchSessionMetadata, readCurrentTranscriptId, resetSessionRecord, resolveSessionAgentId, updateSessionStats };

package/dist/src/storage/sqlite/timestamps.d.ts

@@ -0,0 +1,4 @@
+export declare function timestampToMs(value: string): number;
+export declare function optionalTimestampToMs(value: string | undefined): number | null;
+export declare function timestampToIso(value: number): string;
+export declare function optionalTimestampToIso(value: number | null): string | undefined;

package/dist/src/storage/sqlite/timestamps.js

@@ -0,0 +1,20 @@
+import { __esmMin } from "../../../_virtual/_rolldown/runtime.js";
+//#region src/storage/sqlite/timestamps.ts
+function timestampToMs(value) {
+	const timestamp = Date.parse(value);
+	if (!Number.isFinite(timestamp)) throw new Error(`Invalid ISO timestamp: ${value}`);
+	return timestamp;
+}
+function optionalTimestampToMs(value) {
+	return value === void 0 ? null : timestampToMs(value);
+}
+function timestampToIso(value) {
+	return new Date(value).toISOString();
+}
+function optionalTimestampToIso(value) {
+	return value === null ? void 0 : timestampToIso(value);
+}
+var init_timestamps = __esmMin((() => {}));
+//#endregion
+init_timestamps();
+export { init_timestamps, optionalTimestampToIso, optionalTimestampToMs, timestampToIso, timestampToMs };

package/dist/src/storage/sqlite/transcript-repository.d.ts

@@ -52,6 +52,7 @@ export declare function paginateTranscriptMessages(conversationId: string, optio
     total: number;
     startSeq: number;
     endSeq: number;
+    revision: number;
 };
 export declare function listCompactionBoundaries(conversationId: string): CompactionBoundarySummary[];
 export declare function restoreBeforeCompactionBoundary(conversationId: string, compactionId: string): void;

package/dist/src/storage/sqlite/transcript-repository.js

@@ -50,24 +50,27 @@ function withSessionInputContextMetadata(db, conversationId, row) {
 	const turnId = typeof message.turnId === "string" ? message.turnId.trim() : "";
 	if (!turnId) return row;
 	const metadata = message.metadata && typeof message.metadata === "object" && !Array.isArray(message.metadata) ? message.metadata : {};
-	if (Array.isArray(metadata.sourceContexts) && metadata.sourceContexts.length > 0) return row;
-	const input = db.prepare(`SELECT context_refs_json FROM session_inputs
-     WHERE conversation_id = ? AND run_id = ? AND context_refs_json IS NOT NULL
+	const input = db.prepare(`SELECT client_message_id, context_refs_json FROM session_inputs
+     WHERE conversation_id = ? AND run_id = ?
      LIMIT 1`).get(conversationId, turnId);
 	if (!input) return row;
+	let sourceContexts = metadata.sourceContexts;
 	try {
-		const sourceContexts = JSON.parse(input.context_refs_json);
-		if (!Array.isArray(sourceContexts) || sourceContexts.length === 0) return row;
-		return {
-			...message,
-			metadata: {
-				...metadata,
-				sourceContexts
-			}
-		};
+		if ((!Array.isArray(sourceContexts) || sourceContexts.length === 0) && input.context_refs_json) {
+			const parsed = JSON.parse(input.context_refs_json);
+			if (Array.isArray(parsed) && parsed.length > 0) sourceContexts = parsed;
+		}
 	} catch {
-		return row;
+		sourceContexts = metadata.sourceContexts;
 	}
+	return {
+		...message,
+		metadata: {
+			...metadata,
+			clientMessageId: input.client_message_id,
+			...Array.isArray(sourceContexts) && sourceContexts.length > 0 ? { sourceContexts } : {}
+		}
+	};
 }
 function appendTranscriptEntry(conversationId, row, opts) {
 	if (isRuntimeOnlyTranscriptMessage(row)) throw new Error("Runtime-only messages cannot be persisted in a session transcript");
@@ -281,7 +284,8 @@ function paginateTranscriptMessages(conversationId, options = {}) {
 		messages: [],
 		total: 0,
 		startSeq: 0,
-		endSeq: 0
+		endSeq: 0,
+		revision: 0
 	};
 	const db = getSqliteDatabase();
 	const kinds = options.includeContext ? ["message", "context"] : ["message"];
@@ -293,6 +297,9 @@ function paginateTranscriptMessages(conversationId, options = {}) {
        FROM transcript_entries e
        JOIN transcripts t ON t.transcript_id = e.transcript_id
        WHERE ${transcriptWhere} AND e.entry_kind IN (${kindPlaceholders})`).get(transcriptArg, ...kinds).total;
+	const revisionRow = db.prepare(`SELECT COALESCE(MAX(seq), 0) AS revision
+     FROM transcript_entries
+     WHERE transcript_id = ? AND entry_kind IN (${kindPlaceholders})`).get(transcriptId, ...kinds);
 	const limit = Math.min(200, Math.max(1, Math.trunc(options.limit ?? 50)));
 	const offset = Math.max(0, Math.trunc(options.offset ?? 0));
 	let rows;
@@ -324,7 +331,8 @@ function paginateTranscriptMessages(conversationId, options = {}) {
 		messages: buildSessionContextForLlm(storedRows),
 		total,
 		startSeq: rows[0]?.seq ?? 0,
-		endSeq: rows[rows.length - 1]?.seq ?? 0
+		endSeq: rows[rows.length - 1]?.seq ?? 0,
+		revision: revisionRow.revision
 	};
 }
 function listCompactionBoundaries(conversationId) {

package/dist/src/storage/sqlite/user-trust-repository.js

@@ -1,12 +1,13 @@
 import { __esmMin } from "../../../_virtual/_rolldown/runtime.js";
 import { getSqliteDatabase, init_transaction, runSqliteWriteTransaction } from "./transaction.js";
+import { init_timestamps, timestampToIso } from "./timestamps.js";
 import { DEFAULT_USER_TRUST_LEVEL, init_trust_policy, isUserTrustLevel } from "../../user-context/trust-policy.js";
 //#region src/storage/sqlite/user-trust-repository.ts
 function fromRow(row) {
 	return {
 		principalId: row.principal_id,
 		defaultActionLevel: isUserTrustLevel(row.default_action_level) ? row.default_action_level : DEFAULT_USER_TRUST_LEVEL,
-		updatedAt: row.updated_at
+		updatedAt: timestampToIso(row.updated_at)
 	};
 }
 function getUserTrustPolicy(principalId = "local-owner") {
@@ -17,7 +18,7 @@ function getUserTrustPolicy(principalId = "local-owner") {
 	};
 }
 function setUserTrustPolicy(defaultActionLevel, principalId = "local-owner") {
-	const updatedAt = (/* @__PURE__ */ new Date()).toISOString();
+	const updatedAt = Date.now();
 	return runSqliteWriteTransaction((db) => {
 		db.prepare(`
       INSERT INTO user_trust_policies (principal_id, default_action_level, updated_at)
@@ -31,6 +32,7 @@ function setUserTrustPolicy(defaultActionLevel, principalId = "local-owner") {
 }
 var init_user_trust_repository = __esmMin((() => {
 	init_trust_policy();
+	init_timestamps();
 	init_transaction();
 }));
 //#endregion

package/dist/src/tasks/capabilities/read.js

@@ -2,10 +2,10 @@ import { ProductReadContracts } from "../../../packages/gateway-contract/src/cap
 import { init_src } from "../../../packages/gateway-contract/src/index.js";
 import { TaskRepository, init_task_repository } from "../task-repository.js";
 import { TaskRunRepository, init_task_run_repository } from "../task-run-repository.js";
-import { CapabilityError } from "../../capabilities/runtime/errors.js";
-import { defineReadCapability } from "../../capabilities/runtime/dispatcher.js";
 import { TaskConversationRepository } from "../task-conversation-repository.js";
 import { TaskReadModelProjector } from "../task-read-model-projector.js";
+import { CapabilityError } from "../../capabilities/runtime/errors.js";
+import { defineReadCapability } from "../../capabilities/runtime/dispatcher.js";
 import { TaskContextRepository } from "../task-context-repository.js";
 import { TaskDependencyService } from "../task-dependency-service.js";
 import { TaskValueMetricsService } from "../task-value-metrics-service.js";

package/dist/src/tasks/capabilities/write.d.ts

@@ -101,8 +101,8 @@ export declare const TaskMutationOutputSchema: z.ZodDiscriminatedUnion<[z.ZodObj
     ok: z.ZodLiteral<false>;
     reason: z.ZodEnum<{
         blocked: "blocked";
-        not_found: "not_found";
         conflict: "conflict";
+        not_found: "not_found";
         invalid_transition: "invalid_transition";
     }>;
     model: z.ZodOptional<z.ZodObject<{

package/dist/src/tasks/home-query-service.js

@@ -37,6 +37,7 @@ function actionLabel(automation) {
 	if (automation.action.kind === "workflow") return `workflow:${automation.action.workflowId}`;
 	if (automation.action.kind === "browser_automation") return `browser-automation:${automation.action.automationId}`;
 	if (automation.action.kind === "task_command") return `task:${automation.action.command.type}`;
+	if (automation.action.kind === "system") return `system:${automation.action.capability}`;
 	return automation.action.agentId ? `agent:${automation.action.agentId}` : "agent";
 }
 function toHomeAutomation(automation) {
@@ -50,7 +51,7 @@ function toHomeAutomation(automation) {
 	};
 }
 function effectiveAutomationTimeoutSeconds(run, automation) {
-	return ("timeoutSeconds" in run.actionSnapshot ? run.actionSnapshot.timeoutSeconds : void 0) ?? automation?.reliability?.timeoutSeconds ?? 1800;
+	return ("timeoutSeconds" in run.actionSnapshot ? run.actionSnapshot.timeoutSeconds : void 0) ?? automation?.reliability?.executionTimeoutSeconds ?? 1800;
 }
 function attentionDetail(kind, status, locale, timeoutSeconds) {
 	const isChinese = locale?.toLowerCase().startsWith("zh") ?? false;
@@ -326,7 +327,7 @@ var HomeQueryService = class {
 		const projectsById = new Map(projects.items.map((project) => [project.id, project]));
 		const activeWorkflowRuns = workflowRuns.filter((run) => run.status === "queued" || run.status === "running").map(toHomeWorkflowRun);
 		const failedWorkflowRuns = workflowRuns.filter((run) => run.status === "failed" || run.status === "timeout").slice(0, 5).map(toHomeWorkflowRun);
-		const upcomingAutomations = automations.filter((automation) => automation.notificationPolicy !== "none").map(toHomeAutomation).filter((automation) => Boolean(automation)).sort((a, b) => Date.parse(a.nextRunAt) - Date.parse(b.nextRunAt));
+		const upcomingAutomations = automations.filter((automation) => automation.delivery.notificationPolicy !== "none").map(toHomeAutomation).filter((automation) => Boolean(automation)).sort((a, b) => Date.parse(a.nextRunAt) - Date.parse(b.nextRunAt));
 		const latestAutomationRuns = [...automationRuns].sort((left, right) => right.createdAtMs - left.createdAtMs).filter((run, index, all) => all.findIndex((candidate) => candidate.automationId === run.automationId) === index);
 		const automationsById = new Map(automations.map((automation) => [automation.id, automation]));
 		const decisionCandidates = [...activeTasks.map((task) => decisionFromTask(this.#projector.project(task), task.projectId ? projectsById.get(task.projectId)?.name : void 0, locale)).filter((item) => Boolean(item)), ...connectorApprovals.filter((approval) => Date.parse(approval.expiresAt) > nowMs).map((approval) => ({
@@ -353,7 +354,7 @@ var HomeQueryService = class {
 			href: `/workflows?runId=${encodeURIComponent(run.id)}`,
 			updatedAt: run.completedAtMs ?? run.startedAtMs ?? run.createdAtMs,
 			conversationId: run.conversationId
-		})), ...latestAutomationRuns.filter((run) => run.status === "failed" || run.status === "timeout").filter((run) => automationsById.get(run.automationId)?.notificationPolicy !== "none").filter((run) => !isHomeAttentionAcknowledged("automation_run", run.id)).map((run) => ({
+		})), ...latestAutomationRuns.filter((run) => run.status === "failed" || run.status === "timeout").filter((run) => automationsById.get(run.automationId)?.delivery.notificationPolicy !== "none").filter((run) => !isHomeAttentionAcknowledged("automation_run", run.id)).map((run) => ({
 			id: `automation_run:${run.id}`,
 			kind: "automation_run",
 			runId: run.id,

package/dist/src/tasks/task-contract-planner.js

@@ -69,7 +69,7 @@ var ModelTaskContractPlanner = class {
 				apiKey: await resolveModelCallApiKey(model),
 				maxTokens: 2e3,
 				temperature: 0
-			})).content));
+			}, void 0, { operation: "task.plan_contract" })).content));
 			return planned ? {
 				...planned,
 				objective: fallback.objective

package/dist/src/tui/backends/embedded-backend.js

@@ -5,9 +5,8 @@ import { appendComposerInputHistory, listComposerInputHistory } from "../../stor
 import { getSessionMetadata } from "../../storage/sqlite/session-read-repository.js";
 import { init_sqlite } from "../../storage/sqlite/index.js";
 import { normalizeAgentId } from "../../agent-catalog/id.js";
-import { init_agent_scope, listAgentEntries } from "../../agent/agent-scope.js";
-import { AgentCatalogService } from "../../agent-catalog/service.js";
 import { getAgentDefaultModelRef, init_schema } from "../../config/schema.js";
+import { init_agent_scope, listAgentEntries } from "../../agent/agent-scope.js";
 import { createConversation, init_conversation_repository } from "../../storage/sqlite/conversation-repository.js";
 import { init_agent_session_key, resolveAgentIdFromConversationId } from "../../routing/agent-session-key.js";
 import { effectiveWorkspacePathForSession } from "../../session/session-workspace.js";
@@ -18,6 +17,7 @@ import { buildSessionTimeline } from "../../session/transcript-outline.js";
 import { transcriptRowsToClientHistory } from "../../session/client-history.js";
 import { createCreateShareTool, isShareToolAvailable } from "../../agent/tools/create-share-tool.js";
 import { evictEmbeddedSessionRunner } from "../../agent/embedded/session-runner.js";
+import { AgentCatalogService } from "../../agent-catalog/service.js";
 import { SessionIndex } from "../../session/manager.js";
 import "../../session/index.js";
 import { parseModelRef } from "../../agent/models/selection.js";

package/dist/src/tui/tui-commands.js

@@ -2,8 +2,8 @@ import { closeXopcDatabase, isXopcDatabaseOpen, openXopcDatabase } from "../stor
 import { TaskRepository } from "../tasks/task-repository.js";
 import { getSessionMetadata } from "../storage/sqlite/session-read-repository.js";
 import { init_sqlite } from "../storage/sqlite/index.js";
-import { resolveConfigPath } from "../config/paths.js";
 import { init_agent_session_key, isValidAgentId, normalizeAgentId } from "../routing/agent-session-key.js";
+import { resolveConfigPath } from "../config/paths.js";
 import { init_providers, providerSupportsOAuth } from "../providers/index.js";
 import { ProjectService } from "../projects/project-service.js";
 import { isValidProjectAgentId, normalizeProjectAgentId } from "../projects/project-agent.js";

package/dist/src/tui/tui-startup-resources.js

@@ -1,6 +1,6 @@
 import { resolveStateDir } from "../config/paths-state.js";
-import { init_paths, resolveAgentProfileDir, resolveBundledSkillsDir } from "../config/paths.js";
 import { resolveEffectiveAgentProfileForSession } from "../config/agent-profile.js";
+import { init_paths, resolveAgentProfileDir, resolveBundledSkillsDir } from "../config/paths.js";
 import { resolveBootstrapFilesSync } from "../agent/bootstrap/bootstrap-files.js";
 import { listConnectorInstances } from "../connectors/instances.js";
 import { createSkillConfigManager, isSkillEnabled } from "../agent/skills/config.js";

package/dist/src/tui/tui.js

@@ -1,8 +1,8 @@
 import { version } from "../../package.js";
 import { resolveStateDir, resolveXopcDatabasePath } from "../config/paths-state.js";
-import { init_paths } from "../config/paths.js";
 import { init_agent_session_key, normalizeAgentId } from "../routing/agent-session-key.js";
 import { resolveEffectiveAgentProfileForSession } from "../config/agent-profile.js";
+import { init_paths } from "../config/paths.js";
 import { getAllModels, getAllProviders, getApiKey, init_providers, isProviderConfiguredSync } from "../providers/index.js";
 import { isTerminalWorkflowRunStatus } from "../workflows/domain/run.js";
 import "../workflows/domain/index.js";

package/dist/src/usage/cost.d.ts

@@ -0,0 +1,4 @@
+import type { AiPricingSnapshot, AiUsageCostSource, AiUsageModel } from './types.js';
+export declare function pricingSnapshot(model: AiUsageModel): AiPricingSnapshot;
+export declare function classifyCostSource(model: AiUsageModel): AiUsageCostSource;
+export declare function dollarsToMicrousd(value: number | undefined): number | undefined;

package/dist/src/usage/cost.js

@@ -0,0 +1,28 @@
+//#region src/usage/cost.ts
+function isLocalBaseUrl(baseUrl) {
+	if (!baseUrl) return false;
+	try {
+		const host = new URL(baseUrl).hostname.toLowerCase();
+		return host === "localhost" || host === "::1" || host === "[::1]" || host.startsWith("127.") || host.startsWith("10.") || host.startsWith("192.168.") || /^172\.(1[6-9]|2\d|3[0-1])\./.test(host);
+	} catch {
+		return false;
+	}
+}
+function pricingSnapshot(model) {
+	return {
+		input: model.cost?.input ?? 0,
+		output: model.cost?.output ?? 0,
+		cacheRead: model.cost?.cacheRead ?? 0,
+		cacheWrite: model.cost?.cacheWrite ?? 0
+	};
+}
+function classifyCostSource(model) {
+	if (isLocalBaseUrl(model.baseUrl)) return "local";
+	return Object.values(pricingSnapshot(model)).some((price) => price > 0) ? "model_catalog" : "unknown";
+}
+function dollarsToMicrousd(value) {
+	if (value === void 0 || !Number.isFinite(value) || value < 0) return void 0;
+	return Math.round(value * 1e6);
+}
+//#endregion
+export { classifyCostSource, dollarsToMicrousd, pricingSnapshot };

package/dist/src/usage/recorder.d.ts

@@ -0,0 +1,13 @@
+import type { AssistantMessage, AssistantMessageEventStream } from '@earendil-works/pi-ai/compat';
+import type { AiUsageContext, AiUsageFinish, AiUsageModel } from './types.js';
+export type AiUsageCall = {
+    id: string;
+    finish: (result: AiUsageFinish) => void;
+};
+export declare function continuesAfterTool(messages: readonly {
+    role?: string;
+}[]): boolean;
+export declare function startAiUsageCall(model: AiUsageModel, context: AiUsageContext): AiUsageCall | undefined;
+export declare function observeAiUsageStream(stream: AssistantMessageEventStream, call: AiUsageCall | undefined): AssistantMessageEventStream;
+export declare function trackAiUsageStream(model: AiUsageModel, context: AiUsageContext, create: () => AssistantMessageEventStream | Promise<AssistantMessageEventStream>): AssistantMessageEventStream | Promise<AssistantMessageEventStream>;
+export declare function finishAiUsageFromMessage(call: AiUsageCall | undefined, message: AssistantMessage): void;

package/dist/src/usage/recorder.js

@@ -0,0 +1,117 @@
+import { createLogger } from "../utils/logger/index.js";
+import { init_logger } from "../utils/logger.js";
+import { init_connection, isXopcDatabaseOpen } from "../storage/sqlite/connection.js";
+import { finishAiUsageEvent, insertAiUsageEvent } from "../storage/sqlite/ai-usage-repository.js";
+import { classifyCostSource, dollarsToMicrousd, pricingSnapshot } from "./cost.js";
+import { resolveAiUsageScenario } from "./scenario-registry.js";
+import { randomUUID } from "node:crypto";
+//#region src/usage/recorder.ts
+init_connection();
+init_logger();
+const log = createLogger("AiUsage");
+function continuesAfterTool(messages) {
+	for (let index = messages.length - 1; index >= 0; index -= 1) {
+		const role = messages[index]?.role;
+		if (role === "system") continue;
+		return role === "toolResult";
+	}
+	return false;
+}
+function startAiUsageCall(model, context) {
+	if (!isXopcDatabaseOpen()) return void 0;
+	const operation = context.operation;
+	const scenario = resolveAiUsageScenario(operation);
+	const id = randomUUID();
+	const startedAt = Date.now();
+	const costSource = classifyCostSource(model);
+	try {
+		insertAiUsageEvent({
+			id,
+			traceId: context.traceId ?? context.runId ?? id,
+			parentEventId: context.parentEventId,
+			conversationId: context.conversationId,
+			runId: context.runId,
+			agentId: context.agentId,
+			category: scenario.category,
+			operation,
+			trigger: context.trigger ?? scenario.trigger,
+			reasonKey: scenario.reasonKey,
+			provider: model.provider,
+			model: model.id,
+			status: "running",
+			startedAt,
+			costSource,
+			pricingSnapshot: costSource === "unknown" ? void 0 : pricingSnapshot(model)
+		});
+	} catch (err) {
+		log.error({
+			err,
+			operation,
+			provider: model.provider,
+			modelId: model.id
+		}, "Failed to start AI usage record");
+		return;
+	}
+	let completed = false;
+	return {
+		id,
+		finish(result) {
+			if (completed) return;
+			completed = true;
+			try {
+				finishAiUsageEvent(id, {
+					...result,
+					estimatedCostMicrousd: costSource === "local" ? 0 : costSource === "unknown" ? void 0 : dollarsToMicrousd(result.usage?.cost.total)
+				});
+			} catch (err) {
+				log.error({
+					err,
+					usageEventId: id,
+					operation
+				}, "Failed to finish AI usage record");
+			}
+		}
+	};
+}
+function finishFromMessage(call, message) {
+	if (!call) return;
+	const status = message.stopReason === "aborted" ? "aborted" : message.stopReason === "error" ? "failed" : "succeeded";
+	call.finish({
+		status,
+		usage: message.usage,
+		errorSummary: message.errorMessage
+	});
+}
+function observeAiUsageStream(stream, call) {
+	if (!call) return stream;
+	stream.result().then((message) => finishFromMessage(call, message), (error) => call.finish({
+		status: "failed",
+		errorSummary: error instanceof Error ? error.message : String(error)
+	}));
+	return stream;
+}
+function trackAiUsageStream(model, context, create) {
+	const call = startAiUsageCall(model, context);
+	try {
+		const created = create();
+		if (created instanceof Promise) return created.then((stream) => observeAiUsageStream(stream, call), (error) => {
+			call?.finish({
+				status: "failed",
+				errorSummary: error instanceof Error ? error.message : String(error)
+			});
+			throw error;
+		});
+		return observeAiUsageStream(created, call);
+	} catch (error) {
+		call?.finish({
+			status: "failed",
+			errorSummary: error instanceof Error ? error.message : String(error)
+		});
+		throw error;
+	}
+}
+function finishAiUsageFromMessage(call, message) {
+	finishFromMessage(call, message);
+}
+//#endregion
+export { continuesAfterTool, finishAiUsageFromMessage, observeAiUsageStream, startAiUsageCall, trackAiUsageStream };

package/dist/src/usage/scenario-registry.d.ts

@@ -0,0 +1,8 @@
+import type { AiUsageCategory, AiUsageTrigger } from './types.js';
+type Scenario = {
+    category: AiUsageCategory;
+    reasonKey: string;
+    trigger: AiUsageTrigger;
+};
+export declare function resolveAiUsageScenario(operation: string): Scenario;
+export {};

package/dist/src/usage/scenario-registry.js

@@ -0,0 +1,117 @@
+//#region src/usage/scenario-registry.ts
+const SCENARIOS = {
+	"agent.answer": {
+		category: "chat",
+		reasonKey: "usage.reason.agentAnswer",
+		trigger: "user"
+	},
+	"agent.continue_after_tool": {
+		category: "tool_loop",
+		reasonKey: "usage.reason.continueAfterTool",
+		trigger: "agent"
+	},
+	"agent.background_review": {
+		category: "delegation",
+		reasonKey: "usage.reason.backgroundReview",
+		trigger: "agent"
+	},
+	"session.compact": {
+		category: "compaction",
+		reasonKey: "usage.reason.sessionCompaction",
+		trigger: "system"
+	},
+	"session.generate_title": {
+		category: "session_title",
+		reasonKey: "usage.reason.sessionTitle",
+		trigger: "system"
+	},
+	"media.understand_image": {
+		category: "image_understanding",
+		reasonKey: "usage.reason.imageUnderstanding",
+		trigger: "user"
+	},
+	"tool.web_extract": {
+		category: "web_extract",
+		reasonKey: "usage.reason.webExtract",
+		trigger: "agent"
+	},
+	"tool.session_search_summarize": {
+		category: "session_search",
+		reasonKey: "usage.reason.sessionSearch",
+		trigger: "agent"
+	},
+	"note.generate": {
+		category: "note_generation",
+		reasonKey: "usage.reason.noteGeneration",
+		trigger: "user"
+	},
+	"automation.generate_draft": {
+		category: "automation",
+		reasonKey: "usage.reason.automationDraft",
+		trigger: "user"
+	},
+	"scene.execute": {
+		category: "scene",
+		reasonKey: "usage.reason.sceneExecution",
+		trigger: "scheduled"
+	},
+	"task.plan_contract": {
+		category: "task_planning",
+		reasonKey: "usage.reason.taskPlanning",
+		trigger: "agent"
+	},
+	"task.judge_result": {
+		category: "task_judging",
+		reasonKey: "usage.reason.taskJudging",
+		trigger: "agent"
+	},
+	"voice.summarize_for_tts": {
+		category: "voice_summary",
+		reasonKey: "usage.reason.voiceSummary",
+		trigger: "user"
+	},
+	"voice.select": {
+		category: "voice_summary",
+		reasonKey: "usage.reason.voiceSelection",
+		trigger: "user"
+	},
+	"home.generate_advice": {
+		category: "home_intelligence",
+		reasonKey: "usage.reason.homeAdvice",
+		trigger: "system"
+	},
+	"work_discovery.analyze": {
+		category: "work_discovery",
+		reasonKey: "usage.reason.workDiscovery",
+		trigger: "user"
+	},
+	"work_discovery.investigate": {
+		category: "work_discovery",
+		reasonKey: "usage.reason.workInvestigation",
+		trigger: "user"
+	},
+	"discussion.analyze": {
+		category: "discussion_analysis",
+		reasonKey: "usage.reason.discussionAnalysis",
+		trigger: "user"
+	},
+	"text_assist.generate": {
+		category: "other",
+		reasonKey: "usage.reason.textAssist",
+		trigger: "user"
+	},
+	"user_model.interpret": {
+		category: "other",
+		reasonKey: "usage.reason.userModelInterpretation",
+		trigger: "system"
+	}
+};
+function resolveAiUsageScenario(operation) {
+	return SCENARIOS[operation] ?? {
+		category: "other",
+		reasonKey: "usage.reason.other",
+		trigger: "system"
+	};
+}
+//#endregion
+export { resolveAiUsageScenario };