@keystrokehq/keystroke 0.0.160 → 0.0.161

package/dist/action.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_dist = require("./dist-DFXV_Io5.cjs");
+const require_dist = require("./dist-lWDTuftA.cjs");
 exports.CREDENTIAL_SCOPE_LEVELS = require_dist.CREDENTIAL_SCOPE_LEVELS;
 exports.DEFAULT_CREDENTIAL_RESOLUTION_CHAIN = require_dist.DEFAULT_CREDENTIAL_RESOLUTION_CHAIN;
 exports.actionCoreSchema = require_dist.actionCoreSchema;

package/dist/action.mjs

@@ -1,2 +1,2 @@
-import { C as normalizeCredentialList, S as isCredentialInput, a as getActionCredentialRequirements, b as credentialInputSchema, c as isAction, d as registerWorkflowRunGetter, f as resolveActionTool, g as CREDENTIAL_SCOPE_LEVELS, h as runWithinActionExecution, i as executeAction, l as isStepInvocation, m as runWithRunSignal, n as createStepInvocation, o as getRunSignal, p as runOutsideActionExecution, r as defineAction, s as getWorkflowRunHandle, t as actionCoreSchema, u as isWithinActionExecution, v as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as toCredentialRequirement, x as defineCredential, y as credential } from "./dist-DM1Na5CA.mjs";
+import { C as normalizeCredentialList, S as isCredentialInput, a as getActionCredentialRequirements, b as credentialInputSchema, c as isAction, d as registerWorkflowRunGetter, f as resolveActionTool, g as CREDENTIAL_SCOPE_LEVELS, h as runWithinActionExecution, i as executeAction, l as isStepInvocation, m as runWithRunSignal, n as createStepInvocation, o as getRunSignal, p as runOutsideActionExecution, r as defineAction, s as getWorkflowRunHandle, t as actionCoreSchema, u as isWithinActionExecution, v as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as toCredentialRequirement, x as defineCredential, y as credential } from "./dist-BIbZjRd7.mjs";
 export { CREDENTIAL_SCOPE_LEVELS, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };

package/dist/agent.cjs

@@ -1,10 +1,10 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_dist$1 = require("./dist-DnGkWV6x.cjs");
-const require_dist$2 = require("./dist-DFXV_Io5.cjs");
+const require_dist$2 = require("./dist-lWDTuftA.cjs");
 require("./env-api-keys-BFptfIIs.cjs");
 const require_event_stream = require("./event-stream-CS-ls7wC.cjs");
 require("./json-parse-CwSaKHrs.cjs");
-const require_dist$3 = require("./dist-BkkOal_e.cjs");
+const require_dist$3 = require("./dist-DiqTL3ZF.cjs");
 let zod = require("zod");
 let node_async_hooks = require("node:async_hooks");
 let node_fs = require("node:fs");
@@ -901,7 +901,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-DUKMgFL1.cjs")).then((module) => {
+	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-DuG294d1.cjs")).then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -13744,13 +13744,32 @@ function runWithinAgentPromptExecution(fn) {
 function isWithinAgentPromptExecution() {
 	return executingAgentPrompt.getStore() === true;
 }
+function llmUsageFromAssistantMessage(message) {
+	const usage = message.usage;
+	if (!usage) return null;
+	return {
+		label: message.responseModel ?? message.model ?? "unknown-model",
+		quantity: usage.totalTokens,
+		amount: usage.cost.total,
+		metadata: {
+			provider: message.provider,
+			model: message.model,
+			responseModel: message.responseModel ?? null,
+			input: usage.input,
+			output: usage.output,
+			cacheRead: usage.cacheRead,
+			cacheWrite: usage.cacheWrite,
+			cost: usage.cost
+		}
+	};
+}
 const SKIPPED_EVENT_TYPES = new Set(["message_update", "tool_execution_update"]);
 function shouldPersistAgentEvent(event) {
 	return !SKIPPED_EVENT_TYPES.has(event.type);
 }
 async function persistSessionEvent(options) {
-	const { sessionId, memoryConfig, eventType, payload } = options;
-	const seq = await require_dist$3.appendEvent(sessionId, eventType, payload);
+	const { sessionId, runId, memoryConfig, eventType, payload } = options;
+	const seq = await require_dist$3.appendEvent(sessionId, eventType, payload, { runId });
 	if (memoryConfig) try {
 		await appendSessionLine(memoryConfig, sessionId, {
 			seq,
@@ -13763,15 +13782,19 @@ async function persistSessionEvent(options) {
 	}
 	return seq;
 }
-async function persistAgentEvent(sessionId, event, memoryConfig) {
+async function persistAgentEvent(sessionId, event, options) {
+	const runId = options?.runId ?? null;
+	const memoryConfig = options?.memoryConfig;
 	await persistSessionEvent({
 		sessionId,
+		runId,
 		memoryConfig,
 		eventType: event.type,
 		payload: event
 	});
 	if (event.type === "message_end") await persistSessionEvent({
 		sessionId,
+		runId,
 		memoryConfig,
 		eventType: require_dist$3.MESSAGE_EVENT_TYPE,
 		payload: event.message
@@ -13789,49 +13812,93 @@ async function prepareAgentSession(agentId, sessionId, options) {
 async function runAgentPrompt(agent, agentId, input, options = {}) {
 	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
 	const priorMessages = await require_dist$3.listMessageEvents(sessionId);
+	const runSource = await require_dist$3.resolveAgentRunSourceFromSession(sessionId);
+	const runId = await require_dist$3.createAgentRun({
+		sessionId,
+		agentId,
+		sourceKind: runSource.sourceKind,
+		sourceId: runSource.sourceId,
+		ranByUserId: runSource.ranByUserId
+	});
 	const runPrompt = async () => {
-		const runtime = await agent.buildRuntime({
-			agentId,
-			sessionId,
-			messages: priorMessages
-		}, {
-			...options,
-			credentials: require_dist$3.buildCredentialRunContext({ request: {
-				...options.credentials,
-				...input.context
-			} })
-		});
-		const memoryConfig = runtime.memory?.config;
-		const unsubscribe = runtime.pi.subscribe(async (event) => {
-			if (!shouldPersistAgentEvent(event)) return;
-			await persistAgentEvent(sessionId, event, memoryConfig);
-		});
+		let promptError = null;
+		let runtime;
+		let unsubscribe;
+		let onAbort;
+		const pendingHandlers = /* @__PURE__ */ new Set();
 		const signal = require_dist$2.getRunSignal();
-		const onAbort = () => runtime.pi.abort();
-		if (signal.aborted) onAbort();
-		else signal.addEventListener("abort", onAbort, { once: true });
 		try {
-			const snapshot = await require_dist$3.captureConsole(async () => prompt(runtime, input.message));
-			const credentialError = runtime.takeCredentialError?.();
+			runtime = await agent.buildRuntime({
+				agentId,
+				sessionId,
+				messages: priorMessages
+			}, {
+				...options,
+				credentials: require_dist$3.buildCredentialRunContext({ request: {
+					...options.credentials,
+					...input.context
+				} })
+			});
+			const activeRuntime = runtime;
+			const memoryConfig = activeRuntime.memory?.config;
+			unsubscribe = activeRuntime.pi.subscribe((event) => {
+				if (!shouldPersistAgentEvent(event)) return;
+				const tracked = (async () => {
+					await persistAgentEvent(sessionId, event, {
+						runId,
+						memoryConfig
+					});
+					if (event.type === "message_end" && event.message.role === "assistant") {
+						const usage = llmUsageFromAssistantMessage(event.message);
+						if (usage) await require_dist$3.recordLlmUsageFromAssistantMessage({
+							runId,
+							usage
+						});
+					}
+				})().finally(() => {
+					pendingHandlers.delete(tracked);
+				});
+				pendingHandlers.add(tracked);
+				return tracked;
+			});
+			onAbort = () => activeRuntime.pi.abort();
+			if (signal.aborted) onAbort();
+			else signal.addEventListener("abort", onAbort, { once: true });
+			const snapshot = await require_dist$3.captureConsole(async () => prompt(activeRuntime, input.message));
+			const credentialError = activeRuntime.takeCredentialError?.();
 			if (credentialError && require_dist$3.isCredentialError(credentialError)) throw credentialError;
 			await require_dist$3.touchSession(sessionId);
 			return {
 				sessionId,
+				runId,
 				messages: messages(snapshot),
 				error: errorMessage(snapshot) ?? null,
 				canceled: signal.aborted
 			};
+		} catch (error) {
+			promptError = error;
+			throw error;
 		} finally {
-			signal.removeEventListener("abort", onAbort);
-			unsubscribe();
-			try {
-				runtime.memory?.reindexSessions();
-			} catch (err) {
-				console.error("[memory] session reindex failed:", err);
-			}
-			runtime.memory?.close();
-			await Promise.all(runtime.mcpConnections.map((connection) => connection.close()));
-			await runtime.sandbox.dispose();
+			if (onAbort) signal.removeEventListener("abort", onAbort);
+			unsubscribe?.();
+			await Promise.allSettled([...pendingHandlers]);
+			if (runtime) {
+				try {
+					runtime.memory?.reindexSessions();
+				} catch (err) {
+					console.error("[memory] session reindex failed:", err);
+				}
+				runtime.memory?.close();
+				await Promise.all(runtime.mcpConnections.map((connection) => connection.close()));
+				await runtime.sandbox.dispose();
+			}
+			const finishedAt = /* @__PURE__ */ new Date();
+			await require_dist$3.finalizeAgentRunFromPrompt({
+				runId,
+				status: promptError ? "failed" : signal.aborted ? "canceled" : "completed",
+				error: promptError,
+				finishedAt
+			});
 		}
 	};
 	if (options.skipAgentSessionSpan) return runWithinAgentPromptExecution(runPrompt);
@@ -14247,7 +14314,11 @@ function resolvePromptReasoning(thinkingLevel) {
 function formatValidationError(error) {
 	return error instanceof Error ? error.message : String(error);
 }
-async function runPromptText(opts) {
+async function recordUsageFromMessage(message, hooks) {
+	const usage = llmUsageFromAssistantMessage(message);
+	if (usage) await hooks?.onUsage?.(usage);
+}
+async function runPromptText(opts, hooks) {
 	const model = resolvePromptModel(opts.model);
 	const reasoning = resolvePromptReasoning(opts.thinkingLevel);
 	const message = await completeSimple(model, {
@@ -14263,9 +14334,10 @@ async function runPromptText(opts) {
 		...opts.maxTokens !== void 0 ? { maxTokens: opts.maxTokens } : {}
 	});
 	if (message.stopReason === "error") throw new Error(message.errorMessage ?? "LLM request failed");
+	await recordUsageFromMessage(message, hooks);
 	return extractText(message);
 }
-async function runPromptObject(opts) {
+async function runPromptObject(opts, hooks) {
 	const schema = opts.outputSchema;
 	if (!schema) throw new Error("outputSchema is required for structured LLM output");
 	const model = resolvePromptModel(opts.model);
@@ -14295,6 +14367,7 @@ async function runPromptObject(opts) {
 				...toolChoice !== void 0 ? { toolChoice } : {}
 			});
 			if (message.stopReason === "error") throw new Error(message.errorMessage ?? "LLM request failed");
+			await recordUsageFromMessage(message, hooks);
 			toolCall = extractToolCall(message);
 			if (!toolCall) throw new Error("Model did not return a tool call");
 			return schema.parse(unwrap(toolCall.arguments));
@@ -14323,9 +14396,9 @@ async function runPromptObject(opts) {
 	throw lastError instanceof Error ? lastError : new Error(String(lastError));
 }
 /** One-shot LLM generation for workflow `promptLlm` steps (pi-ai transport). */
-async function runLlm(opts) {
-	if (opts.outputSchema) return runPromptObject(opts);
-	return runPromptText(opts);
+async function runLlm(opts, hooks) {
+	if (opts.outputSchema) return runPromptObject(opts, hooks);
+	return runPromptText(opts, hooks);
 }
 //#endregion
 exports.AgentCreateInputSchema = AgentCreateInputSchema;