@keystrokehq/keystroke 0.0.159 → 0.0.161

package/dist/action.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_dist = require("./dist-D5jH3nQI.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.d.cts

@@ -1,3 +1,3 @@
-import { C as normalizeCredentialList, S as isCredentialInput, _ as ResolvedCredentials, a as CredentialInput, b as credentialInputSchema, d as DefineCredentialInput, f as DefineOAuthCredentialInput, l as CredentialScopeLevel, m as NormalizeCredential, n as Credential, o as CredentialList, p as DefineStaticCredentialInput, r as CredentialAuthKind, s as CredentialRequirement, t as CREDENTIAL_SCOPE_LEVELS, u as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as toCredentialRequirement, x as defineCredential, y as credential } from "./index-BNcrFoZi.cjs";
-import { C as runWithinActionExecution, S as runWithRunSignal, _ as isStepInvocation, a as ResolveActionCredentialsFn, b as resolveActionTool, c as WorkflowActionDefinition, d as defineAction, f as executeAction, g as isAction, h as getWorkflowRunHandle, i as AgentRunOptions, l as actionCoreSchema, m as getRunSignal, n as ActionDefinition, o as ResolveActionToolOptions, p as getActionCredentialRequirements, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as createStepInvocation, v as isWithinActionExecution, x as runOutsideActionExecution, y as registerWorkflowRunGetter } from "./index-BmXIGJlQ.cjs";
-export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, type AgentRunOptions, CREDENTIAL_SCOPE_LEVELS, type Credential, type CredentialAuthKind, type CredentialInput, type CredentialList, type CredentialRequirement, type CredentialScopeLevel, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, type DefineCredentialInput, type DefineOAuthCredentialInput, type DefineStaticCredentialInput, type NormalizeCredential, type ResolveActionCredentialsFn, type ResolveActionToolOptions, type ResolvedCredentials, type StepInvocation, type WorkflowActionDefinition, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };
+import { C as defineCredential, E as toCredentialRequirement, S as credentialInputSchema, T as normalizeCredentialList, a as CredentialInput, d as DefineCredentialInput, f as DefineOAuthCredentialInput, l as CredentialScopeLevel, m as NormalizeCredential, n as Credential, o as CredentialList, p as DefineStaticCredentialInput, r as CredentialAuthKind, s as CredentialRequirement, t as CREDENTIAL_SCOPE_LEVELS, u as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as isCredentialInput, x as credential, y as ResolvedCredentials } from "./index-CaW3JRFU.cjs";
+import { C as runWithRunSignal, S as runOutsideActionExecution, _ as isAction, a as ResolveActionCredentialsFn, b as registerWorkflowRunGetter, c as WorkflowActionDefinition, d as createStepInvocation, f as defineAction, g as getWorkflowRunHandle, h as getRunSignal, i as AgentRunOptions, l as WorkflowRunHandle, m as getActionCredentialRequirements, n as ActionDefinition, o as ResolveActionToolOptions, p as executeAction, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as actionCoreSchema, v as isStepInvocation, w as runWithinActionExecution, x as resolveActionTool, y as isWithinActionExecution } from "./index-D7upoVsK.cjs";
+export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, type AgentRunOptions, CREDENTIAL_SCOPE_LEVELS, type Credential, type CredentialAuthKind, type CredentialInput, type CredentialList, type CredentialRequirement, type CredentialScopeLevel, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, type DefineCredentialInput, type DefineOAuthCredentialInput, type DefineStaticCredentialInput, type NormalizeCredential, type ResolveActionCredentialsFn, type ResolveActionToolOptions, type ResolvedCredentials, type StepInvocation, type WorkflowActionDefinition, type WorkflowRunHandle, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };

package/dist/action.d.mts

@@ -1,3 +1,3 @@
-import { C as normalizeCredentialList, S as isCredentialInput, _ as ResolvedCredentials, a as CredentialInput, b as credentialInputSchema, d as DefineCredentialInput, f as DefineOAuthCredentialInput, l as CredentialScopeLevel, m as NormalizeCredential, n as Credential, o as CredentialList, p as DefineStaticCredentialInput, r as CredentialAuthKind, s as CredentialRequirement, t as CREDENTIAL_SCOPE_LEVELS, u as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as toCredentialRequirement, x as defineCredential, y as credential } from "./index-BNcrFoZi.mjs";
-import { C as runWithinActionExecution, S as runWithRunSignal, _ as isStepInvocation, a as ResolveActionCredentialsFn, b as resolveActionTool, c as WorkflowActionDefinition, d as defineAction, f as executeAction, g as isAction, h as getWorkflowRunHandle, i as AgentRunOptions, l as actionCoreSchema, m as getRunSignal, n as ActionDefinition, o as ResolveActionToolOptions, p as getActionCredentialRequirements, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as createStepInvocation, v as isWithinActionExecution, x as runOutsideActionExecution, y as registerWorkflowRunGetter } from "./index-Bbe9CzYo.mjs";
-export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, type AgentRunOptions, CREDENTIAL_SCOPE_LEVELS, type Credential, type CredentialAuthKind, type CredentialInput, type CredentialList, type CredentialRequirement, type CredentialScopeLevel, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, type DefineCredentialInput, type DefineOAuthCredentialInput, type DefineStaticCredentialInput, type NormalizeCredential, type ResolveActionCredentialsFn, type ResolveActionToolOptions, type ResolvedCredentials, type StepInvocation, type WorkflowActionDefinition, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };
+import { C as defineCredential, E as toCredentialRequirement, S as credentialInputSchema, T as normalizeCredentialList, a as CredentialInput, d as DefineCredentialInput, f as DefineOAuthCredentialInput, l as CredentialScopeLevel, m as NormalizeCredential, n as Credential, o as CredentialList, p as DefineStaticCredentialInput, r as CredentialAuthKind, s as CredentialRequirement, t as CREDENTIAL_SCOPE_LEVELS, u as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, w as isCredentialInput, x as credential, y as ResolvedCredentials } from "./index-CaW3JRFU.mjs";
+import { C as runWithRunSignal, S as runOutsideActionExecution, _ as isAction, a as ResolveActionCredentialsFn, b as registerWorkflowRunGetter, c as WorkflowActionDefinition, d as createStepInvocation, f as defineAction, g as getWorkflowRunHandle, h as getRunSignal, i as AgentRunOptions, l as WorkflowRunHandle, m as getActionCredentialRequirements, n as ActionDefinition, o as ResolveActionToolOptions, p as executeAction, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as actionCoreSchema, v as isStepInvocation, w as runWithinActionExecution, x as resolveActionTool, y as isWithinActionExecution } from "./index-CgMisiAc.mjs";
+export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, type AgentRunOptions, CREDENTIAL_SCOPE_LEVELS, type Credential, type CredentialAuthKind, type CredentialInput, type CredentialList, type CredentialRequirement, type CredentialScopeLevel, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, type DefineCredentialInput, type DefineOAuthCredentialInput, type DefineStaticCredentialInput, type NormalizeCredential, type ResolveActionCredentialsFn, type ResolveActionToolOptions, type ResolvedCredentials, type StepInvocation, type WorkflowActionDefinition, type WorkflowRunHandle, actionCoreSchema, createStepInvocation, credential, credentialInputSchema, defineAction, defineCredential, executeAction, getActionCredentialRequirements, getRunSignal, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithRunSignal, runWithinActionExecution, toCredentialRequirement };

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-XKCtZShd.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-types.d.cts

@@ -1,2 +1,2 @@
-import { a as AgentState, c as AgentToolUpdateCallback, d as AssistantMessage, f as Message, h as UserMessage, i as AgentMessage, l as ThinkingLevel, m as TextContent, n as AgentOptions, o as AgentTool, p as Model, r as AgentEvent, s as AgentToolResult } from "./index-BcMCpDKh.cjs";
+import { a as AgentState, c as AgentToolUpdateCallback, f as AssistantMessage, g as UserMessage, h as TextContent, i as AgentMessage, l as ThinkingLevel, m as Model, n as AgentOptions, o as AgentTool, p as Message, r as AgentEvent, s as AgentToolResult } from "./index-Dr-3nl0A.cjs";
 export type { AgentEvent, AgentMessage, AgentOptions, AgentState, AgentTool, AgentToolResult, AgentToolUpdateCallback, AssistantMessage, Message, Model, TextContent, ThinkingLevel, UserMessage };

package/dist/agent-types.d.mts

@@ -1,2 +1,2 @@
-import { a as AgentState, c as AgentToolUpdateCallback, d as AssistantMessage, f as Message, h as UserMessage, i as AgentMessage, l as ThinkingLevel, m as TextContent, n as AgentOptions, o as AgentTool, p as Model, r as AgentEvent, s as AgentToolResult } from "./index-Bd_asjgz.mjs";
+import { a as AgentState, c as AgentToolUpdateCallback, f as AssistantMessage, g as UserMessage, h as TextContent, i as AgentMessage, l as ThinkingLevel, m as Model, n as AgentOptions, o as AgentTool, p as Message, r as AgentEvent, s as AgentToolResult } from "./index-0KyTNuR1.mjs";
 export type { AgentEvent, AgentMessage, AgentOptions, AgentState, AgentTool, AgentToolResult, AgentToolUpdateCallback, AssistantMessage, Message, Model, TextContent, ThinkingLevel, UserMessage };

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-D5jH3nQI.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-u0RYk4-X.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-DIpD1PUg.cjs")).then((module) => {
+	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-DuG294d1.cjs")).then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -1026,6 +1026,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) {
@@ -13395,7 +13398,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 = [];
@@ -13449,7 +13452,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,
@@ -13741,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,
@@ -13760,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
@@ -13786,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);
@@ -14172,6 +14242,164 @@ 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 require_dist$2.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 = zod.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 = zod.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
 exports.AgentCreateInputSchema = AgentCreateInputSchema;
 exports.AgentModelIdSchema = AgentModelIdSchema;
@@ -14189,6 +14417,7 @@ exports.defineMcp = require_dist$3.defineMcp;
 exports.defineSubagentTool = defineSubagentTool;
 exports.defineTool = defineTool;
 exports.errorMessage = errorMessage;
+exports.forceToolChoice = forceToolChoice;
 exports.getModel = require_event_stream.getModel;
 exports.getSubagentToolMetadata = getSubagentToolMetadata;
 exports.isMcp = require_dist$3.isMcp;
@@ -14205,6 +14434,7 @@ exports.resolveAgentTools = resolveAgentTools;
 exports.resolveAgentWorkspaceRoot = resolveAgentWorkspaceRoot;
 exports.resolveThinkingLevel = resolveThinkingLevel;
 exports.runAgentPrompt = runAgentPrompt;
+exports.runLlm = runLlm;
 exports.snapshot = snapshot;
 
 //# sourceMappingURL=agent.cjs.map