@keystrokehq/keystroke 0.0.159 → 0.0.161

package/dist/agent.d.cts

@@ -1,4 +1,4 @@
-import { l as ThinkingLevel, u as getModel } from "./index-BcMCpDKh.cjs";
-import { A as normalizeAgentDefinition, B as AgentCreateInputSchema, C as defineAgent, D as getSubagentToolMetadata, E as errorMessage, F as resolveAgentTools, G as parseAgentCreateInput, H as AgentModelIdSchema, I as resolveAgentWorkspaceRoot, K as resolveAgentModel, L as runAgentPrompt, M as prompt, N as resolveAgentAssets, O as loadAssetManifest, P as resolveAgentId, R as snapshot, S as createAgent, T as defineTool, U as DEFAULT_THINKING_LEVEL, V as AgentModelId, W as ThinkingLevelSchema, _ as SubagentTool, a as AgentDefinitionInput, b as SubagentToolMetadata, c as BuildAgentRuntimeContext, d as PromptResponse, f as ResolvedAgentAssets, g as SessionAgentMismatchError, h as SandboxedAgentRuntime, i as AgentDefinition, j as prepareAgentSession, k as messages, l as DefineSubagentToolInput, m as SUBAGENT_TOOL, n as AgentAssetInput, o as AgentRuntime, p as RunAgentPromptOptions, q as resolveThinkingLevel, r as AgentConfig, s as AgentSnapshot, t as Agent, u as PromptInput, v as SubagentToolDetails, w as defineSubagentTool, x as buildAgentRuntime, y as SubagentToolInherit, z as AgentCreateInput } from "./index-DuYf0Fdc.cjs";
-import { a as McpDefinition, c as McpServerOptions, d as McpTransport, f as connectMcpDefinition, g as isMcp, h as defineMcp, i as McpConnection, l as McpStdioOptions, m as connectMcpStdio, n as McpAuthOptions, o as McpDefinitionInput, p as connectMcpServer, r as McpCallToolOptions, s as McpHttpTransport, t as ToolCredentialResolver, u as McpStdioTransport } from "./index-CUrU-w2Y.cjs";
-export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentModelId, AgentModelIdSchema, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type PromptInput, type PromptResponse, type ResolvedAgentAssets, type RunAgentPromptOptions, SUBAGENT_TOOL, type SandboxedAgentRuntime, SessionAgentMismatchError, type SubagentTool, type SubagentToolDetails, type SubagentToolInherit, type SubagentToolMetadata, type ThinkingLevel, ThinkingLevelSchema, type ToolCredentialResolver, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+import { l as ThinkingLevel, u as getModel } from "./index-Dr-3nl0A.cjs";
+import { A as loadAssetManifest, B as runLlm, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentAssets, G as AgentModelIdSchema, H as AgentCreateInput, I as resolveAgentId, J as parseAgentCreateInput, K as DEFAULT_THINKING_LEVEL, L as resolveAgentTools, M as normalizeAgentDefinition, N as prepareAgentSession, O as forceToolChoice, P as prompt, R as resolveAgentWorkspaceRoot, S as buildAgentRuntime, T as defineSubagentTool, U as AgentCreateInputSchema, V as snapshot, W as AgentModelId, X as resolveThinkingLevel, Y as resolveAgentModel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentToolInherit, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as messages, k as getSubagentToolMetadata, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as ThinkingLevelSchema, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as runAgentPrompt } from "./index-DIm3_7or.cjs";
+import { a as McpDefinition, c as McpServerOptions, d as McpTransport, f as connectMcpDefinition, g as isMcp, h as defineMcp, i as McpConnection, l as McpStdioOptions, m as connectMcpStdio, n as McpAuthOptions, o as McpDefinitionInput, p as connectMcpServer, r as McpCallToolOptions, s as McpHttpTransport, t as ToolCredentialResolver, u as McpStdioTransport } from "./index-CsGfxjDx.cjs";
+export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentModelId, AgentModelIdSchema, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineSubagentToolInput, type ForcedToolChoice, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type PromptInput, type PromptResponse, type ResolvedAgentAssets, type RunAgentPromptOptions, SUBAGENT_TOOL, type SandboxedAgentRuntime, SessionAgentMismatchError, type SubagentTool, type SubagentToolDetails, type SubagentToolInherit, type SubagentToolMetadata, type ThinkingLevel, ThinkingLevelSchema, type ToolCredentialResolver, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, forceToolChoice, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, runLlm, snapshot };

package/dist/agent.d.mts

@@ -1,4 +1,4 @@
-import { l as ThinkingLevel, u as getModel } from "./index-Bd_asjgz.mjs";
-import { A as normalizeAgentDefinition, B as AgentCreateInputSchema, C as defineAgent, D as getSubagentToolMetadata, E as errorMessage, F as resolveAgentTools, G as parseAgentCreateInput, H as AgentModelIdSchema, I as resolveAgentWorkspaceRoot, K as resolveAgentModel, L as runAgentPrompt, M as prompt, N as resolveAgentAssets, O as loadAssetManifest, P as resolveAgentId, R as snapshot, S as createAgent, T as defineTool, U as DEFAULT_THINKING_LEVEL, V as AgentModelId, W as ThinkingLevelSchema, _ as SubagentTool, a as AgentDefinitionInput, b as SubagentToolMetadata, c as BuildAgentRuntimeContext, d as PromptResponse, f as ResolvedAgentAssets, g as SessionAgentMismatchError, h as SandboxedAgentRuntime, i as AgentDefinition, j as prepareAgentSession, k as messages, l as DefineSubagentToolInput, m as SUBAGENT_TOOL, n as AgentAssetInput, o as AgentRuntime, p as RunAgentPromptOptions, q as resolveThinkingLevel, r as AgentConfig, s as AgentSnapshot, t as Agent, u as PromptInput, v as SubagentToolDetails, w as defineSubagentTool, x as buildAgentRuntime, y as SubagentToolInherit, z as AgentCreateInput } from "./index-CyCx29m7.mjs";
-import { a as McpDefinition, c as McpServerOptions, d as McpTransport, f as connectMcpDefinition, g as isMcp, h as defineMcp, i as McpConnection, l as McpStdioOptions, m as connectMcpStdio, n as McpAuthOptions, o as McpDefinitionInput, p as connectMcpServer, r as McpCallToolOptions, s as McpHttpTransport, t as ToolCredentialResolver, u as McpStdioTransport } from "./index-Qnwdoq7U.mjs";
-export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentModelId, AgentModelIdSchema, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type PromptInput, type PromptResponse, type ResolvedAgentAssets, type RunAgentPromptOptions, SUBAGENT_TOOL, type SandboxedAgentRuntime, SessionAgentMismatchError, type SubagentTool, type SubagentToolDetails, type SubagentToolInherit, type SubagentToolMetadata, type ThinkingLevel, ThinkingLevelSchema, type ToolCredentialResolver, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+import { l as ThinkingLevel, u as getModel } from "./index-0KyTNuR1.mjs";
+import { A as loadAssetManifest, B as runLlm, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentAssets, G as AgentModelIdSchema, H as AgentCreateInput, I as resolveAgentId, J as parseAgentCreateInput, K as DEFAULT_THINKING_LEVEL, L as resolveAgentTools, M as normalizeAgentDefinition, N as prepareAgentSession, O as forceToolChoice, P as prompt, R as resolveAgentWorkspaceRoot, S as buildAgentRuntime, T as defineSubagentTool, U as AgentCreateInputSchema, V as snapshot, W as AgentModelId, X as resolveThinkingLevel, Y as resolveAgentModel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentToolInherit, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as messages, k as getSubagentToolMetadata, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as ThinkingLevelSchema, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as runAgentPrompt } from "./index-IVnunyAF.mjs";
+import { a as McpDefinition, c as McpServerOptions, d as McpTransport, f as connectMcpDefinition, g as isMcp, h as defineMcp, i as McpConnection, l as McpStdioOptions, m as connectMcpStdio, n as McpAuthOptions, o as McpDefinitionInput, p as connectMcpServer, r as McpCallToolOptions, s as McpHttpTransport, t as ToolCredentialResolver, u as McpStdioTransport } from "./index-BAIg1_V_.mjs";
+export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentModelId, AgentModelIdSchema, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineSubagentToolInput, type ForcedToolChoice, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type PromptInput, type PromptResponse, type ResolvedAgentAssets, type RunAgentPromptOptions, SUBAGENT_TOOL, type SandboxedAgentRuntime, SessionAgentMismatchError, type SubagentTool, type SubagentToolDetails, type SubagentToolInherit, type SubagentToolMetadata, type ThinkingLevel, ThinkingLevelSchema, type ToolCredentialResolver, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, forceToolChoice, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, runLlm, snapshot };

package/dist/agent.mjs

@@ -1,9 +1,9 @@
 import { $ as Options, $t as IsMap, An as IsSymbol$1, At as Unreachable, Bn as __require, Bt as IsRef$1, Cn as IsMinLength$2, Ct as IsInteger$2, Dn as IsObject$1, Dt as IsCodec, En as IsNumber$1, Et as IsBigInt$2, Fn as Symbols, Ft as IsUnion, G as ensureWorkspaceDir, Gt as IsArray$2, Ht as IsFunction$2, In as guard_exports, It as Union, Jt as IsOptional, K as materializeSandbox, Kt as Base, Ln as __commonJSMin, Lt as IsObject$2, Mn as IsUnsafePropertyKey, Mt as IsEnum$1, Nn as IsValueLike, Nt as IsCyclic, On as IsObjectNotArray$1, Ot as Hash, Pn as Keys$1, Pt as Unknown, Q as Instantiate, Qt as Get$3, Rt as _Object_, Sn as IsMaxLength$2, St as IsIterator$2, Tn as IsNull$1, Tt as IsBoolean$2, Ut as IsConstructor$2, Vt as Ref$1, W as defineSandbox, Wt as IsAsyncIterator$2, X as resolveSandboxRoot, Xt as IsPromise, Yt as Optional, Z as Evaluate$1, Zt as IsSchema$1, _n as IsGreaterThan$1, _t as IsLiteralBigInt, a as createSandbox, an as EveryAll, at as RecordPattern, bn as IsLessEqualThan$1, bt as IsLiteralString, c as defineTool$1, cn as IsAsyncIterator$1, ct as EnumToUnion, d as resolveRunProvider, dn as IsClassInstance, dt as String$1, en as IsSet, et as Compare, fn as IsConstructor$1, ft as IsSymbol$2, gn as IsGreaterEqualThan$1, gt as IsLiteral, hn as IsFunction$1, ht as IsNull$2, i as createJustBashProvider, in as Every$1, it as IsRecord, jn as IsUndefined$1, jt as IsIntersect, kn as IsString$1, kt as hash_exports, ln as IsBigInt$1, lt as TemplateLiteralDecode, mn as IsEqual$1, mt as Number$1, nn as Entries$2, nt as IsVoid, o as defaultWorkspacesRoot, on as HasPropertyKey$1, ot as RecordValue, pn as IsDeepEqual$1, pt as IsNumber$2, q as packAssetDirs, qt as IsBase, r as createInvokeToolBridge, rn as EntriesRegExp, rt as IsUndefined$2, s as defineSkill, sn as IsArray$1, st as IsTuple, tn as IsTypeArray, tt as IsTemplateLiteral, u as materializeSkills, un as IsBoolean$1, ut as IsString$2, vn as IsInteger$1, vt as IsLiteralBoolean, wn as IsMultipleOf$1, wt as Boolean$1, xn as IsLessThan$1, xt as Literal, yn as IsIterator$1, yt as IsLiteralNumber, zn as __exportAll, zt as IsNever } from "./dist-Dv44b9V1.mjs";
-import { c as isAction, f as resolveActionTool, o as getRunSignal, s as getWorkflowRunHandle, u as isWithinActionExecution } from "./dist-XKCtZShd.mjs";
+import { T as toolParameters, c as isAction, f as resolveActionTool, o as getRunSignal, s as getWorkflowRunHandle, u as isWithinActionExecution } from "./dist-BIbZjRd7.mjs";
 import "./env-api-keys-mjSaOeuP.mjs";
 import { i as getModel, t as AssistantMessageEventStream } from "./event-stream-C7YBbzsc.mjs";
 import "./json-parse-Drlv-g_X.mjs";
-import { C as isMcp, S as defineMcp, a as resolveActionCredentials, b as connectMcpServer, c as appendEvent, d as getSession, f as listMessageEvents, g as getTraceContext, h as captureConsole, i as isCredentialError, l as createSession, m as touchSession, n as captureCredentialToolErrors, o as resolveMcpTools, p as resolveRunSourceFromTraceContext, r as createCredentialResolver, s as MESSAGE_EVENT_TYPE$1, t as buildCredentialRunContext, u as getAgentByRoute, v as withSpan, x as connectMcpStdio, y as connectMcpDefinition } from "./dist-lYNbBwe5.mjs";
+import { C as connectMcpDefinition, D as isMcp, E as defineMcp, S as withSpan, T as connectMcpStdio, _ as resolveRunSourceFromTraceContext, a as resolveActionCredentials, b as getTraceContext, c as appendEvent, d as finalizeAgentRunFromPrompt, f as getAgentByRoute, g as resolveAgentRunSourceFromSession, h as recordLlmUsageFromAssistantMessage, i as isCredentialError, l as createAgentRun, m as listMessageEvents, n as captureCredentialToolErrors, o as resolveMcpTools, p as getSession, r as createCredentialResolver, s as MESSAGE_EVENT_TYPE$1, t as buildCredentialRunContext, u as createSession, v as touchSession, w as connectMcpServer, y as captureConsole } from "./dist-DXzvN8aq.mjs";
 import { createRequire } from "node:module";
 import { z } from "zod";
 import { AsyncLocalStorage } from "node:async_hooks";
@@ -900,7 +900,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= import("./mistral-CWlaNuqc.mjs").then((module) => {
+	mistralProviderModulePromise ||= import("./mistral-CDuUHkpO.mjs").then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -1025,6 +1025,9 @@ function resolveApiProvider(api) {
 function streamSimple(model, context, options) {
 	return resolveApiProvider(model.api).streamSimple(model, context, options);
 }
+async function completeSimple(model, context, options) {
+	return streamSimple(model, context, options).result();
+}
 //#endregion
 //#region ../../node_modules/.pnpm/typebox@1.1.38/node_modules/typebox/build/schema/types/_guard.mjs
 function IsGuardInterface(value) {
@@ -13384,7 +13387,7 @@ function readMd(path, limit) {
 		chars: content.length
 	};
 }
-function extractText(content) {
+function extractText$1(content) {
 	if (typeof content === "string") return content;
 	if (Array.isArray(content)) {
 		const parts = [];
@@ -13438,7 +13441,7 @@ function extractSessionRows(file) {
 			continue;
 		}
 		if (obj.type !== "message" || !obj.message) continue;
-		const text = extractText(obj.message.content);
+		const text = extractText$1(obj.message.content);
 		if (!text) continue;
 		rows.push({
 			line_no: i + 1,
@@ -13730,13 +13733,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 appendEvent(sessionId, eventType, payload);
+	const { sessionId, runId, memoryConfig, eventType, payload } = options;
+	const seq = await appendEvent(sessionId, eventType, payload, { runId });
 	if (memoryConfig) try {
 		await appendSessionLine(memoryConfig, sessionId, {
 			seq,
@@ -13749,15 +13771,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: MESSAGE_EVENT_TYPE$1,
 		payload: event.message
@@ -13775,49 +13801,93 @@ async function prepareAgentSession(agentId, sessionId, options) {
 async function runAgentPrompt(agent, agentId, input, options = {}) {
 	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
 	const priorMessages = await listMessageEvents(sessionId);
+	const runSource = await resolveAgentRunSourceFromSession(sessionId);
+	const runId = await 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: 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 = getRunSignal();
-		const onAbort = () => runtime.pi.abort();
-		if (signal.aborted) onAbort();
-		else signal.addEventListener("abort", onAbort, { once: true });
 		try {
-			const snapshot = await captureConsole(async () => prompt(runtime, input.message));
-			const credentialError = runtime.takeCredentialError?.();
+			runtime = await agent.buildRuntime({
+				agentId,
+				sessionId,
+				messages: priorMessages
+			}, {
+				...options,
+				credentials: 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 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 captureConsole(async () => prompt(activeRuntime, input.message));
+			const credentialError = activeRuntime.takeCredentialError?.();
 			if (credentialError && isCredentialError(credentialError)) throw credentialError;
 			await 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 finalizeAgentRunFromPrompt({
+				runId,
+				status: promptError ? "failed" : signal.aborted ? "canceled" : "completed",
+				error: promptError,
+				finishedAt
+			});
 		}
 	};
 	if (options.skipAgentSessionSpan) return runWithinAgentPromptExecution(runPrompt);
@@ -14161,7 +14231,165 @@ function defineSubagentTool(input) {
 	subagentTool[SUBAGENT_TOOL] = metadata;
 	return subagentTool;
 }
+/**
+* Best-effort provider `tool_choice` for forcing a named tool. Returns `undefined`
+* when the API shape is unknown so callers can rely on prompt + validate + retry.
+*/
+function forceToolChoice(api, toolName) {
+	switch (api) {
+		case "anthropic-messages": return {
+			type: "tool",
+			name: toolName
+		};
+		case "openai-completions":
+		case "openai-responses":
+		case "azure-openai-responses": return {
+			type: "function",
+			function: { name: toolName }
+		};
+		case "bedrock-converse-stream": return { tool: { name: toolName } };
+		case "google-generative-ai":
+		case "google-vertex": return "any";
+		default: return;
+	}
+}
+const RESPOND_TOOL_NAME = "respond";
+const STRUCTURED_OUTPUT_INSTRUCTION = "Respond only by calling `respond` with arguments matching the schema; do not answer in plain text.";
+const MAX_OBJECT_ATTEMPTS = 3;
+/** Same validation as `defineAgent` — format check before catalog lookup. */
+function resolvePromptModel(model) {
+	return resolveAgentModel(AgentModelIdSchema.parse(model));
+}
+/**
+* pi-ai types tool `parameters` as TypeBox `TSchema`, but providers consume JSON Schema
+* at runtime. Same bridge as `resolveActionTool` in `@keystrokehq/action`.
+*/
+function buildToolParameters(schema) {
+	return toolParameters(schema);
+}
+function extractText(message) {
+	return message.content.filter((part) => part.type === "text").map((part) => part.text).join("");
+}
+function extractToolCall(message) {
+	return message.content.find((part) => part.type === "toolCall");
+}
+function needsEnvelope(schema) {
+	const json = z.toJSONSchema(schema, { target: "draft-2020-12" });
+	if (typeof json !== "object" || json === null || !("type" in json)) return true;
+	return json.type !== "object";
+}
+function wrapSchemaForTool(schema) {
+	if (!needsEnvelope(schema)) return {
+		toolSchema: schema,
+		unwrap: (args) => args
+	};
+	const envelopeSchema = z.object({ result: schema });
+	return {
+		toolSchema: envelopeSchema,
+		unwrap: (args) => envelopeSchema.parse(args).result
+	};
+}
+function buildRespondTool(schema) {
+	return {
+		name: RESPOND_TOOL_NAME,
+		description: "Return the final result.",
+		parameters: buildToolParameters(schema)
+	};
+}
+function resolvePromptReasoning(thinkingLevel) {
+	if (!thinkingLevel || thinkingLevel === "off") return;
+	return thinkingLevel;
+}
+function formatValidationError(error) {
+	return error instanceof Error ? error.message : String(error);
+}
+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, {
+		systemPrompt: opts.system,
+		messages: [{
+			role: "user",
+			content: opts.prompt,
+			timestamp: Date.now()
+		}]
+	}, {
+		...reasoning !== void 0 ? { reasoning } : {},
+		...opts.temperature !== void 0 ? { temperature: opts.temperature } : {},
+		...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, hooks) {
+	const schema = opts.outputSchema;
+	if (!schema) throw new Error("outputSchema is required for structured LLM output");
+	const model = resolvePromptModel(opts.model);
+	const reasoning = resolvePromptReasoning(opts.thinkingLevel);
+	const { toolSchema, unwrap } = wrapSchemaForTool(schema);
+	const tool = buildRespondTool(toolSchema);
+	const toolChoice = forceToolChoice(model.api, RESPOND_TOOL_NAME);
+	const systemPrompt = [opts.system, STRUCTURED_OUTPUT_INSTRUCTION].filter(Boolean).join("\n\n");
+	const messages = [{
+		role: "user",
+		content: opts.prompt,
+		timestamp: Date.now()
+	}];
+	let lastError;
+	for (let attempt = 0; attempt < MAX_OBJECT_ATTEMPTS; attempt++) {
+		let message;
+		let toolCall;
+		try {
+			message = await completeSimple(model, {
+				systemPrompt,
+				messages,
+				tools: [tool]
+			}, {
+				...reasoning !== void 0 ? { reasoning } : {},
+				...opts.temperature !== void 0 ? { temperature: opts.temperature } : {},
+				...opts.maxTokens !== void 0 ? { maxTokens: opts.maxTokens } : {},
+				...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));
+		} catch (error) {
+			lastError = error;
+			if (attempt === MAX_OBJECT_ATTEMPTS - 1) break;
+			const correction = `Validation failed: ${formatValidationError(error)}. Call respond again with valid arguments.`;
+			if (message && toolCall) messages.push(message, {
+				role: "toolResult",
+				toolCallId: toolCall.id,
+				toolName: toolCall.name,
+				content: [{
+					type: "text",
+					text: correction
+				}],
+				isError: true,
+				timestamp: Date.now()
+			});
+			else if (message) messages.push(message, {
+				role: "user",
+				content: correction,
+				timestamp: Date.now()
+			});
+		}
+	}
+	throw lastError instanceof Error ? lastError : new Error(String(lastError));
+}
+/** One-shot LLM generation for workflow `promptLlm` steps (pi-ai transport). */
+async function runLlm(opts, hooks) {
+	if (opts.outputSchema) return runPromptObject(opts, hooks);
+	return runPromptText(opts, hooks);
+}
 //#endregion
-export { AgentCreateInputSchema, AgentModelIdSchema, DEFAULT_THINKING_LEVEL, SUBAGENT_TOOL, SessionAgentMismatchError, ThinkingLevelSchema, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+export { AgentCreateInputSchema, AgentModelIdSchema, DEFAULT_THINKING_LEVEL, SUBAGENT_TOOL, SessionAgentMismatchError, ThinkingLevelSchema, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, forceToolChoice, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentId, resolveAgentModel, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, runLlm, snapshot };
 
 //# sourceMappingURL=agent.mjs.map