@xopcai/xopc 0.0.332 → 0.0.334

package/dist/src/tui/tui.js

@@ -1,9 +1,9 @@
 import { version } from "../../package.js";
 import { resolveStateDir, resolveXopcDatabasePath } from "../config/paths-state.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 { init_agent_session_key, normalizeAgentId } from "../routing/agent-session-key.js";
 import { getAllModels, getAllProviders, getApiKey, init_providers, isProviderConfiguredSync } from "../providers/index.js";
+import { resolveEffectiveAgentProfile } from "../config/agent-profile.js";
 import { isTerminalWorkflowRunStatus } from "../workflows/domain/run.js";
 import "../workflows/domain/index.js";
 import { ProjectTrustStore, hasTrustRequiringProjectResources } from "../project-trust/trust-store.js";
@@ -26,7 +26,6 @@ import { transcriptTreeEntryIdToRowNumber } from "./tui-transcript-tree.js";
 import { formatTuiCompactionResult, formatTuiShareResult } from "./tui-command-formatters.js";
 import { formatTuiStartupText } from "./tui-startup-text.js";
 import { createTuiCommandHandler, getSlashCommands } from "./tui-commands.js";
-import { GENERATED_TUI_SESSION_SHELL_PATCH, cleanupAbandonedTuiSessions, deleteGeneratedTuiSessionIfEmpty } from "./tui-empty-session-cleanup.js";
 import { extensionCustomMessageContentToText, extensionCustomMessageToTurnText, extensionUserMessageContentToText } from "./tui-extension-user-message.js";
 import { resolveFdPath } from "./tui-fd-path.js";
 import { createBackspaceDeduper, drainAndStopTuiSafely, isIgnorableTuiStopError, resolveCtrlCAction, stopTuiSafely } from "./tui-lifecycle.js";
@@ -45,6 +44,7 @@ import { installTuiStdioFilter } from "./tui-stdio-filter.js";
 import { createEditorSubmitHandler, createSubmitBurstCoalescer, shouldEnableWindowsGitBashPasteFallback } from "./tui-submit.js";
 import { withTuiSuspended } from "./tui-suspend.js";
 import { createInitialState } from "./tui-types.js";
+import { TuiDraftFileStore } from "./backends/embedded-drafts.js";
 import { EmbeddedBackend } from "./backends/embedded-backend.js";
 import { GatewayRealtimeBackend } from "./backends/gateway-realtime-backend.js";
 import { ChatLog } from "./components/chat-log.js";
@@ -196,7 +196,6 @@ async function runTui(opts) {
 	const generatedStartupConversationId = !opts.session?.trim() ? startup.conversationId : null;
 	const generatedStartupConversationIds = new Set(generatedStartupConversationId ? [generatedStartupConversationId] : []);
 	const isGeneratedTuiConversationId = (id) => generatedStartupConversationIds.has(id);
-	let startupSessionHadUserTurn = false;
 	const startupWorkingDirectory = resolveStartupWorkingDirectory(opts, isLocalMode);
 	const implicitTrustedWorkspace = startupWorkingDirectory ?? (isLocalMode ? process.cwd() : void 0);
 	const projectTrustStore = new ProjectTrustStore();
@@ -392,6 +391,7 @@ async function runTui(opts) {
 		tui.requestRender();
 	};
 	editor.onChange = (text) => {
+		scheduleComposerDraftSave();
 		const trimmed = text.trimStart();
 		const nextExclude = trimmed.startsWith("!!");
 		const nextBash = editorMode === "shell" || trimmed.startsWith("!");
@@ -401,6 +401,37 @@ async function runTui(opts) {
 			updateEditorBorderColor();
 		}
 	};
+	let composerDrafts;
+	let composerDraftTimer;
+	const saveComposerDraft = () => {
+		if (!composerDrafts) return;
+		try {
+			const text = editor.getText();
+			if (text || pendingImageAttachments.length) composerDrafts.save(state.currentConversationId, {
+				text,
+				attachments: pendingImageAttachments
+			});
+			else composerDrafts.remove(state.currentConversationId);
+		} catch (error) {
+			chatLog.addSystem(`Draft could not be saved: ${String(error)}`);
+		}
+	};
+	const scheduleComposerDraftSave = () => {
+		clearTimeout(composerDraftTimer);
+		composerDraftTimer = setTimeout(saveComposerDraft, 300);
+	};
+	const restoreComposerDraft = () => {
+		if (!composerDrafts || editor.getText() || pendingImageAttachments.length) return;
+		try {
+			const saved = composerDrafts.read(state.currentConversationId);
+			if (saved) {
+				pendingImageAttachments = saved.attachments;
+				editor.setText(saved.text);
+			}
+		} catch (error) {
+			chatLog.addSystem(`Draft could not be restored: ${String(error)}`);
+		}
+	};
 	const copyDefaultEditorAppHandlers = (target) => {
 		const customEditor = target;
 		if (!(customEditor.actionHandlers instanceof Map)) return;
@@ -558,7 +589,7 @@ async function runTui(opts) {
 		}
 	};
 	const getAvailableProviderCount = () => getAllProviders().filter((provider) => isProviderConfiguredSync(provider)).length;
-	const getExtensionSystemPrompt = () => resolveEffectiveAgentProfileForSession(state.currentConversationId).customInstructions ?? "";
+	const getExtensionSystemPrompt = () => resolveEffectiveAgentProfile(currentAgentId).customInstructions ?? "";
 	const waitForTuiIdle = async () => {
 		while (state.activeRunId != null || state.isCompacting || state.activityStatus === "sending" || state.activityStatus === "waiting" || state.activityStatus === "streaming" || state.activityStatus === "running" || state.activityStatus === "compacting") await new Promise((resolveDelay) => setTimeout(resolveDelay, 50));
 	};
@@ -952,6 +983,8 @@ async function runTui(opts) {
 	let exitResult = { exitReason: "exit" };
 	const requestExit = (options) => {
 		if (state.exitRequested) return;
+		clearTimeout(composerDraftTimer);
+		saveComposerDraft();
 		state.exitRequested = true;
 		if (elapsedTimerId) {
 			clearInterval(elapsedTimerId);
@@ -963,21 +996,15 @@ async function runTui(opts) {
 		}
 		tui.terminal.setProgress(false);
 		(async () => {
-			const removedEmptyStartupSessions = /* @__PURE__ */ new Set();
-			if (!state.activeRunId) for (const conversationId of generatedStartupConversationIds) {
-				if (conversationId === state.currentConversationId && startupSessionHadUserTurn) continue;
-				if (await deleteGeneratedTuiSessionIfEmpty(client, conversationId).catch(() => false)) removedEmptyStartupSessions.add(conversationId);
-			}
 			client.stop();
 			await drainAndStopTuiSafely(tui);
 			restoreStdio();
-			const currentSessionWasRemoved = removedEmptyStartupSessions.has(state.currentConversationId);
-			if (options?.showResumeHint !== false && !currentSessionWasRemoved) process.stdout.write(`\nTo resume this session: ${formatTuiResumeCommand(opts, state.currentConversationId)}\n`);
+			if (options?.showResumeHint !== false) process.stdout.write(`\nTo resume this session: ${formatTuiResumeCommand(opts, state.currentConversationId)}\n`);
 			finishTui?.();
 			process.exit(0);
 		})();
 	};
-	const { refreshSessionInfo, loadHistory: loadSessionHistory, loadHistoryWindow, setSession, abortActive, resetCurrentSession, clearChatForSessionSwitch } = createSessionActions({
+	const { refreshSessionInfo, loadHistory: loadSessionHistory, loadHistoryWindow, setSession: changeSession, abortActive, resetCurrentSession, clearChatForSessionSwitch } = createSessionActions({
 		client,
 		chatLog,
 		tui,
@@ -992,6 +1019,15 @@ async function runTui(opts) {
 		},
 		onSessionInfoChange: () => persistCurrentNewSessionContext()
 	});
+	const setSession = async (key) => {
+		clearTimeout(composerDraftTimer);
+		saveComposerDraft();
+		pendingImageAttachments = [];
+		editor.setText("");
+		clearTimeout(composerDraftTimer);
+		await changeSession(key);
+		restoreComposerDraft();
+	};
 	const persistCurrentNewSessionContext = () => {
 		const current = getTuiNewSessionPreferences(tuiSettings, newSessionPreferenceKey);
 		const normalizedProjectId = state.sessionInfo.projectId?.trim() || null;
@@ -1012,8 +1048,7 @@ async function runTui(opts) {
 		if (initialAgentConfig || projectId) {
 			await client.patchSession(state.currentConversationId, {
 				...initialAgentConfig,
-				...projectId ? { projectId } : {},
-				...isGeneratedTuiConversationId(state.currentConversationId) ? GENERATED_TUI_SESSION_SHELL_PATCH : {}
+				...projectId ? { projectId } : {}
 			});
 			await refreshSessionInfo();
 		}
@@ -1050,10 +1085,7 @@ async function runTui(opts) {
 	const applyStartupWorkingDirectory = async () => {
 		if (startupWorkingDirectoryApplied || !startupWorkingDirectory) return;
 		startupWorkingDirectoryApplied = true;
-		await client.patchSession(state.currentConversationId, {
-			workingDirectory: startupWorkingDirectory,
-			...isGeneratedTuiConversationId(state.currentConversationId) ? GENERATED_TUI_SESSION_SHELL_PATCH : {}
-		});
+		await client.patchSession(state.currentConversationId, { workingDirectory: startupWorkingDirectory });
 		state.sessionInfo.effectiveWorkspacePath = startupWorkingDirectory;
 		state.sessionInfo.workingDirectoryLocked = true;
 	};
@@ -1073,18 +1105,13 @@ async function runTui(opts) {
 			generatedStartupConversationIds.add(targetConversationId);
 			await setSession(targetConversationId);
 			currentAgentId = projectAgentId;
-			if (isGeneratedTuiConversationId(state.currentConversationId)) await client.patchSession(state.currentConversationId, GENERATED_TUI_SESSION_SHELL_PATCH);
 			await client.patchSession(state.currentConversationId, {
 				projectId: result.project.id,
-				workingDirectory: startupWorkingDirectory,
-				...isGeneratedTuiConversationId(state.currentConversationId) ? GENERATED_TUI_SESSION_SHELL_PATCH : {}
+				workingDirectory: startupWorkingDirectory
 			});
 			state.sessionInfo.effectiveWorkspacePath = startupWorkingDirectory;
 			state.sessionInfo.workingDirectoryLocked = true;
-		} else await client.patchSession(state.currentConversationId, {
-			projectId: result.project.id,
-			...isGeneratedTuiConversationId(state.currentConversationId) ? GENERATED_TUI_SESSION_SHELL_PATCH : {}
-		});
+		} else await client.patchSession(state.currentConversationId, { projectId: result.project.id });
 		const verb = result.created ? "Created project" : "Using project";
 		chatLog.addStatus(theme.dim(`${verb}: ${result.project.name}`));
 	};
@@ -1531,7 +1558,6 @@ async function runTui(opts) {
 			return;
 		}
 		chatLog.addUser(userDisplayContentForAttachments(text, attachments));
-		startupSessionHadUserTurn = true;
 		sessionSnapshot.appendMessage("user", messageText);
 		lastRetryMessageText = messageText;
 		markRunSending(state);
@@ -1686,11 +1712,19 @@ async function runTui(opts) {
 		tui.requestRender();
 	};
 	const retryLastMessage = async () => {
-		const message = lastRetryMessageText?.trim();
+		const interrupted = state.chatInputState.inputs.find((input) => input.status === "interrupted");
+		const message = interrupted?.content.trim() || lastRetryMessageText?.trim();
 		if (!message) {
 			chatLog.addSystem("No previous user message to retry.");
 			return;
 		}
+		if (interrupted) {
+			if (pendingImageAttachments.length || pendingNextTurnCustomMessages.length) {
+				chatLog.addSystem("Send or clear the current attachments and context before retrying an interrupted input.");
+				return;
+			}
+			pendingImageAttachments = interrupted.attachments ?? [];
+		}
 		if (state.activeRunId) await abortActive({ clearUi: false });
 		sendMessage(message);
 	};
@@ -2112,8 +2146,10 @@ async function runTui(opts) {
 	});
 	defaultEditor.onAction("app.editor.external", openExternalEditor);
 	defaultEditor.onPasteImage = () => {
+		const conversationId = state.currentConversationId;
 		(async () => {
 			const image = await readClipboardImage();
+			if (conversationId !== state.currentConversationId) return;
 			if (!image) {
 				chatLog.addStatus(theme.dim("No image found on clipboard."));
 				tui.requestRender();
@@ -2138,6 +2174,7 @@ async function runTui(opts) {
 				name,
 				size: image.bytes.byteLength
 			});
+			scheduleComposerDraftSave();
 			chatLog.addStatus(theme.dim(`Attached image: ${name} (${pendingImageAttachments.length} pending). Send a message to include it.`));
 			tui.requestRender();
 		})();
@@ -2269,6 +2306,10 @@ async function runTui(opts) {
 		setConnectionStatus(isLocalMode ? "local ready" : "gateway connected");
 		touchStreamingActivity();
 		(async () => {
+			if (!composerDrafts) {
+				composerDrafts = new TuiDraftFileStore(await client.getComposerDraftDirectory());
+				restoreComposerDraft();
+			}
 			if (!startupConversationCreated) {
 				await client.createConversation(currentAgentId, state.currentConversationId);
 				startupConversationCreated = true;
@@ -2314,7 +2355,6 @@ async function runTui(opts) {
 			} catch {}
 			await refreshModelChoices();
 			await loadSessionHistory({ merge: true });
-			cleanupAbandonedTuiSessions(client, state.currentConversationId).catch(() => {});
 			showStartupCardOnce();
 			refreshStartupResources().then(() => {
 				updateFooter();

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 };

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

@@ -0,0 +1,55 @@
+import type { Model, Api, Usage } from '@earendil-works/pi-ai/compat';
+export type AiUsageStatus = 'running' | 'succeeded' | 'failed' | 'aborted' | 'unknown';
+export type AiUsageTrigger = 'user' | 'scheduled' | 'system' | 'agent' | 'retry';
+export type AiUsageCostSource = 'model_catalog' | 'local' | 'unknown';
+export type AiUsageCategory = 'chat' | 'tool_loop' | 'delegation' | 'compaction' | 'session_title' | 'image_understanding' | 'web_extract' | 'session_search' | 'note_generation' | 'automation' | 'scene' | 'task_planning' | 'task_judging' | 'voice_summary' | 'home_intelligence' | 'work_discovery' | 'discussion_analysis' | 'other';
+export type AiUsageContext = {
+    operation: string;
+    trigger?: AiUsageTrigger;
+    traceId?: string;
+    parentEventId?: string;
+    conversationId?: string;
+    runId?: string;
+    agentId?: string;
+};
+export type AiPricingSnapshot = {
+    input: number;
+    output: number;
+    cacheRead: number;
+    cacheWrite: number;
+};
+export type AiUsageEvent = {
+    id: string;
+    traceId: string;
+    parentEventId?: string;
+    conversationId?: string;
+    runId?: string;
+    agentId?: string;
+    category: AiUsageCategory;
+    operation: string;
+    trigger: AiUsageTrigger;
+    reasonKey: string;
+    provider: string;
+    model: string;
+    status: AiUsageStatus;
+    startedAt: number;
+    finishedAt?: number;
+    durationMs?: number;
+    inputTokens?: number;
+    outputTokens?: number;
+    cacheReadTokens?: number;
+    cacheWriteTokens?: number;
+    reasoningTokens?: number;
+    totalTokens?: number;
+    estimatedCostMicrousd?: number;
+    costSource: AiUsageCostSource;
+    pricingSnapshot?: AiPricingSnapshot;
+    errorSummary?: string;
+};
+export type AiUsageFinish = {
+    status: Exclude<AiUsageStatus, 'running'>;
+    usage?: Usage;
+    errorSummary?: string;
+    finishedAt?: number;
+};
+export type AiUsageModel = Pick<Model<Api>, 'provider' | 'id' | 'baseUrl' | 'cost'>;

package/dist/src/usage/types.js

@@ -0,0 +1 @@
+export {};

package/dist/src/voice/realtime/persona-context.js

@@ -1,6 +1,6 @@
 import { init_agent_scope, resolveAgentProfileDir } from "../../agent/agent-scope.js";
-import { resolveEffectiveAgentProfileForSession } from "../../config/agent-profile.js";
 import { DEFAULT_IDENTITY_FILENAME, DEFAULT_SOUL_FILENAME } from "../../agent/context/workspace.js";
+import { resolveEffectiveAgentProfileForSession } from "../../config/agent-profile.js";
 import { loadProfileBootstrapFiles } from "../../agent/bootstrap/load-bootstrap-files.js";
 import { join } from "node:path";
 import { statSync } from "node:fs";

package/dist/src/voice/tts/summarize.js

@@ -71,7 +71,7 @@ async function summarizeForTts(options) {
 			maxTokens: Math.min(Math.ceil(targetLength / 2), 4096),
 			temperature: .3,
 			signal: mergedSignal
-		});
+		}, void 0, { operation: "voice.summarize_for_tts" });
 		let out = "";
 		if (Array.isArray(result.content)) {
 			for (const c of result.content) if (c && typeof c === "object" && c.type === "text") out += String(c.text || "");

package/dist/src/work-discovery/analyzer.js

@@ -229,7 +229,7 @@ async function analyzeWorkContext(input) {
 		maxTokens: input.projectOverviewOnly ? 2e3 : WORK_ANALYSIS_MAX_TOKENS,
 		temperature: .1,
 		signal: input.signal
-	});
+	}, void 0, { operation: "work_discovery.analyze" });
 	const raw = extractText(response.content);
 	const parsed = parseJson(raw);
 	if (!parsed) throw invalidJsonError("Analysis", response, raw);
@@ -324,7 +324,7 @@ async function analyzeUnderstandingBatch(input) {
 		maxTokens: 3e3,
 		temperature: .1,
 		signal: input.signal
-	});
+	}, void 0, { operation: "work_discovery.analyze" });
 	const raw = extractText(response.content);
 	const parsed = parseJson(raw);
 	if (!parsed) throw invalidJsonError("Understanding source analysis", response, raw);

package/dist/src/work-discovery/investigator.js

@@ -132,7 +132,7 @@ async function chooseAction(input) {
 		maxTokens: 1200,
 		temperature: .1,
 		signal: input.signal
-	})).content)), new Set(input.snapshot.structure.sampledPaths));
+	}, void 0, { operation: "work_discovery.investigate" })).content)), new Set(input.snapshot.structure.sampledPaths));
 }
 function contentHash(content) {
 	return createHash("sha256").update(content).digest("hex");

package/dist/src/work-discovery/service.js

@@ -8,15 +8,15 @@ import { ActivityService } from "../activity/service.js";
 import { getAgentDefaultModelRef, init_schema } from "../config/schema.js";
 import { init_session_key, resolveConversationId } from "../routing/session-key.js";
 import { getDefaultAgentId, init_resolve_route } from "../routing/resolve-route.js";
-import { resolveEffectiveAgentProfile } from "../config/agent-profile.js";
 import { init_providers, resolveModel } from "../providers/index.js";
+import { isLocalModelBaseUrl } from "../providers/model-call.js";
+import { resolveEffectiveAgentProfile } from "../config/agent-profile.js";
 import { writeKnowledgeItem } from "../knowledge-memory/repository.js";
 import "../knowledge-memory/index.js";
 import { createWorkDiscoveryRun, getProjectUnderstandingRun, getWorkDiscoveryOnboardingState, getWorkDiscoveryRun, getWorkDiscoveryRunByIdempotencyKey, listPendingProjectUnderstandingRuns, setWorkDiscoveryFeedback, setWorkDiscoveryOnboardingState, updateWorkDiscoveryRun } from "./repository.js";
 import { linkAssertionEvidence, reconcileAssertion } from "../user-model/repository.js";
 import { createPriorityWindow, createUserGoal } from "../user-model/goals.js";
 import "../user-model/index.js";
-import { isLocalModelBaseUrl } from "../providers/model-call.js";
 import { createUnderstandingSourceRun, getUnderstandingSourceGrant, getUnderstandingSourceRun, updateUnderstandingSourceGrantCheckpoint, updateUnderstandingSourceGrantPolicies, updateUnderstandingSourceRun, upsertUnderstandingSourceGrant } from "../user-context/sources/repository.js";
 import { allowsRemoteSourceProcessing } from "../user-context/sources/processing-policy.js";
 import { analyzeUnderstandingSources, analyzeWorkContext, workDiscoveryResultMarkdown } from "./analyzer.js";
@@ -237,10 +237,7 @@ function sessionMetadata(agentId, peerId) {
 			peerKind: "direct",
 			peerId
 		},
-		customData: {
-			workDiscovery: true,
-			genericNewChatShell: false
-		}
+		customData: { workDiscovery: true }
 	};
 }
 function errorCode(error) {