@keystrokehq/keystroke 0.0.101 → 0.0.102

package/dist/agent.d.cts

@@ -1,4 +1,4 @@
 import { l as ThinkingLevel, u as getModel } from "./index-BcMCpDKh.cjs";
-import { A as messages, B as AgentCreateInputSchema, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentTools, G as parseAgentCreateInput, H as ModelRef, I as resolveAgentWorkspaceRoot, K as resolveThinkingLevel, L as runAgentPrompt, M as prepareAgentSession, N as prompt, O as getSubagentToolMetadata, P as resolveAgentAssets, R as snapshot, S as buildAgentRuntime, T as defineSubagentTool, U as ModelRefSchema, V as DEFAULT_THINKING_LEVEL, W as ThinkingLevelSchema, _ 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 normalizeAgentDefinition, k as loadAssetManifest, l as DefineAgentOptions, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, r as AgentConfig, s as AgentSnapshot, t as Agent, u as DefineSubagentToolInput, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as AgentCreateInput } from "./index-L5a2mf72.cjs";
+import { A as messages, B as AgentCreateInput, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentId, G as ThinkingLevelSchema, H as DEFAULT_THINKING_LEVEL, I as resolveAgentTools, K as parseAgentCreateInput, L as resolveAgentWorkspaceRoot, M as prepareAgentSession, N as prompt, O as getSubagentToolMetadata, P as resolveAgentAssets, R as runAgentPrompt, S as buildAgentRuntime, T as defineSubagentTool, U as ModelRef, V as AgentCreateInputSchema, W as ModelRefSchema, _ 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 normalizeAgentDefinition, k as loadAssetManifest, l as DefineAgentOptions, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as resolveThinkingLevel, r as AgentConfig, s as AgentSnapshot, t as Agent, u as DefineSubagentToolInput, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as snapshot } from "./index-FlkwP4zn.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-MfPRY7cK.cjs";
-export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineAgentOptions, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type ModelRef, ModelRefSchema, 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, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineAgentOptions, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type ModelRef, ModelRefSchema, 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, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, 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 messages, B as AgentCreateInputSchema, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentTools, G as parseAgentCreateInput, H as ModelRef, I as resolveAgentWorkspaceRoot, K as resolveThinkingLevel, L as runAgentPrompt, M as prepareAgentSession, N as prompt, O as getSubagentToolMetadata, P as resolveAgentAssets, R as snapshot, S as buildAgentRuntime, T as defineSubagentTool, U as ModelRefSchema, V as DEFAULT_THINKING_LEVEL, W as ThinkingLevelSchema, _ 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 normalizeAgentDefinition, k as loadAssetManifest, l as DefineAgentOptions, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, r as AgentConfig, s as AgentSnapshot, t as Agent, u as DefineSubagentToolInput, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as AgentCreateInput } from "./index-CBgXyNH2.mjs";
+import { A as messages, B as AgentCreateInput, C as createAgent, D as errorMessage, E as defineTool, F as resolveAgentId, G as ThinkingLevelSchema, H as DEFAULT_THINKING_LEVEL, I as resolveAgentTools, K as parseAgentCreateInput, L as resolveAgentWorkspaceRoot, M as prepareAgentSession, N as prompt, O as getSubagentToolMetadata, P as resolveAgentAssets, R as runAgentPrompt, S as buildAgentRuntime, T as defineSubagentTool, U as ModelRef, V as AgentCreateInputSchema, W as ModelRefSchema, _ 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 normalizeAgentDefinition, k as loadAssetManifest, l as DefineAgentOptions, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as resolveThinkingLevel, r as AgentConfig, s as AgentSnapshot, t as Agent, u as DefineSubagentToolInput, v as SubagentTool, w as defineAgent, x as SubagentToolMetadata, y as SubagentToolDetails, z as snapshot } from "./index-TLlgJ1q1.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-ptRM3z9H.mjs";
-export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineAgentOptions, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type ModelRef, ModelRefSchema, 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, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+export { type Agent, type AgentAssetInput, type AgentConfig, type AgentCreateInput, AgentCreateInputSchema, type AgentDefinition, type AgentDefinitionInput, type AgentRuntime, type AgentSnapshot, type BuildAgentRuntimeContext, DEFAULT_THINKING_LEVEL, type DefineAgentOptions, type DefineSubagentToolInput, type McpAuthOptions, type McpCallToolOptions, type McpConnection, type McpDefinition, type McpDefinitionInput, type McpHttpTransport, type McpServerOptions, type McpStdioOptions, type McpStdioTransport, type McpTransport, type ModelRef, ModelRefSchema, 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, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };

package/dist/agent.mjs

@@ -1,11 +1,12 @@
 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, Y as resolveAppRootFromModule, 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-f_JGyGWx.mjs";
-import { s as isAction, u as resolveActionTool } from "./dist-CAGgGvND.mjs";
+import { d as resolveActionTool, l as isWithinActionExecution, o as getWorkflowRunHandle, s as isAction } from "./dist-SGAuX401.mjs";
 import "./env-api-keys-BSMeSmLq.mjs";
 import { i as getModel, t as AssistantMessageEventStream } from "./event-stream-C7YBbzsc.mjs";
 import "./json-parse-BhvKZ346.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-DzfOPx7X.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-BUfE2se6.mjs";
 import { createRequire } from "node:module";
 import { z } from "zod";
+import { AsyncLocalStorage } from "node:async_hooks";
 import { existsSync, mkdirSync, readFileSync, readdirSync, statSync, writeFileSync } from "node:fs";
 import { basename, dirname, join } from "node:path";
 import { DatabaseSync } from "node:sqlite";
@@ -927,7 +928,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= import("./mistral-CGHbHFQG.mjs").then((module) => {
+	mistralProviderModulePromise ||= import("./mistral-DUtxOD_C.mjs").then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -13669,6 +13670,131 @@ function errorMessage(snapshot) {
 function defineTool(tool) {
 	return tool;
 }
+/** Tracks an in-flight agent prompt (subagents/tools must not become workflow steps). */
+const executingAgentPrompt = new AsyncLocalStorage();
+function runWithinAgentPromptExecution(fn) {
+	return executingAgentPrompt.run(true, fn);
+}
+function isWithinAgentPromptExecution() {
+	return executingAgentPrompt.getStore() === true;
+}
+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);
+	if (memoryConfig) try {
+		await appendSessionLine(memoryConfig, sessionId, {
+			seq,
+			eventType,
+			payload,
+			createdAt: /* @__PURE__ */ new Date()
+		});
+	} catch (err) {
+		console.error("[memory] transcript append failed:", err);
+	}
+	return seq;
+}
+async function persistAgentEvent(sessionId, event, memoryConfig) {
+	await persistSessionEvent({
+		sessionId,
+		memoryConfig,
+		eventType: event.type,
+		payload: event
+	});
+	if (event.type === "message_end") await persistSessionEvent({
+		sessionId,
+		memoryConfig,
+		eventType: MESSAGE_EVENT_TYPE$1,
+		payload: event.message
+	});
+}
+async function prepareAgentSession(agentId, sessionId) {
+	if (sessionId) {
+		const session = await getSession(sessionId);
+		if (!session) await createSession(agentId, sessionId, await resolveRunSourceFromTraceContext());
+		else if (session.agentId !== agentId) throw new SessionAgentMismatchError(sessionId);
+		return { sessionId };
+	}
+	return { sessionId: (await createSession(agentId, void 0, await resolveRunSourceFromTraceContext())).id };
+}
+async function runAgentPrompt(agent, agentId, input, options = {}) {
+	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
+	const priorMessages = await listMessageEvents(sessionId);
+	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);
+		});
+		try {
+			const snapshot = await captureConsole(async () => prompt(runtime, input.message));
+			const credentialError = runtime.takeCredentialError?.();
+			if (credentialError && isCredentialError(credentialError)) throw credentialError;
+			await touchSession(sessionId);
+			return {
+				sessionId,
+				messages: messages(snapshot),
+				error: errorMessage(snapshot) ?? null
+			};
+		} finally {
+			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 (options.skipAgentSessionSpan) return runWithinAgentPromptExecution(runPrompt);
+	const parent = getTraceContext();
+	return withSpan({
+		kind: "agent_session",
+		name: agentId,
+		refId: sessionId,
+		traceId: parent?.traceId ?? sessionId,
+		metadata: {
+			agentId,
+			trigger: options.tracing?.trigger ?? (parent ? "subagent" : "api"),
+			...options.tracing
+		}
+	}, () => runWithinAgentPromptExecution(runPrompt));
+}
+var SessionAgentMismatchError = class extends Error {
+	constructor(sessionId) {
+		super(`Session ${sessionId} does not belong to this agent`);
+		this.name = "SessionAgentMismatchError";
+	}
+};
+/** Resolves an agent id from the DB registry by key when not passed explicitly. */
+async function resolveAgentId(agentKey, explicitId) {
+	if (explicitId) return explicitId;
+	const registered = await getAgentByRoute(`/agents/${agentKey}`);
+	if (!registered?.id) throw new Error(`Agent "${agentKey}" is not registered; pass agentId or sync the project before prompting.`);
+	return registered.id;
+}
+async function runDirectAgentPrompt(agent, agentKey, defaultRunPrompt, promptInput, runPrompt) {
+	return runAgentPrompt(agent, await resolveAgentId(agentKey, promptInput.agentId), promptInput, {
+		...defaultRunPrompt,
+		...runPrompt
+	});
+}
 /**
 * Resolve the host directory backing `/workspace` for an agent prompt.
 *
@@ -13895,110 +14021,6 @@ function normalizeAgentDefinition(input, options = {}) {
 		...input.memory !== void 0 ? { memory: input.memory } : {}
 	};
 }
-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);
-	if (memoryConfig) try {
-		await appendSessionLine(memoryConfig, sessionId, {
-			seq,
-			eventType,
-			payload,
-			createdAt: /* @__PURE__ */ new Date()
-		});
-	} catch (err) {
-		console.error("[memory] transcript append failed:", err);
-	}
-	return seq;
-}
-async function persistAgentEvent(sessionId, event, memoryConfig) {
-	await persistSessionEvent({
-		sessionId,
-		memoryConfig,
-		eventType: event.type,
-		payload: event
-	});
-	if (event.type === "message_end") await persistSessionEvent({
-		sessionId,
-		memoryConfig,
-		eventType: MESSAGE_EVENT_TYPE$1,
-		payload: event.message
-	});
-}
-async function prepareAgentSession(agentId, sessionId) {
-	if (sessionId) {
-		const session = await getSession(sessionId);
-		if (!session) await createSession(agentId, sessionId, await resolveRunSourceFromTraceContext());
-		else if (session.agentId !== agentId) throw new SessionAgentMismatchError(sessionId);
-		return { sessionId };
-	}
-	return { sessionId: (await createSession(agentId, void 0, await resolveRunSourceFromTraceContext())).id };
-}
-async function runAgentPrompt(agent, agentId, input, options = {}) {
-	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
-	const priorMessages = await listMessageEvents(sessionId);
-	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);
-		});
-		try {
-			const snapshot = await captureConsole(async () => prompt(runtime, input.message));
-			const credentialError = runtime.takeCredentialError?.();
-			if (credentialError && isCredentialError(credentialError)) throw credentialError;
-			await touchSession(sessionId);
-			return {
-				sessionId,
-				messages: messages(snapshot),
-				error: errorMessage(snapshot) ?? null
-			};
-		} finally {
-			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 (options.skipAgentSessionSpan) return runPrompt();
-	const parent = getTraceContext();
-	return withSpan({
-		kind: "agent_session",
-		name: agentId,
-		refId: sessionId,
-		traceId: parent?.traceId ?? sessionId,
-		metadata: {
-			agentId,
-			trigger: options.tracing?.trigger ?? (parent ? "subagent" : "api"),
-			...options.tracing
-		}
-	}, runPrompt);
-}
-var SessionAgentMismatchError = class extends Error {
-	constructor(sessionId) {
-		super(`Session ${sessionId} does not belong to this agent`);
-		this.name = "SessionAgentMismatchError";
-	}
-};
 /**
 * Define an agent: normalized config plus a pipeline that builds sandbox + pi runtime on demand.
 */
@@ -14013,14 +14035,19 @@ function defineAgent(input, options = {}) {
 		...def,
 		key,
 		buildRuntime: (ctx, runPrompt) => buildAgentRuntime(def, ctx, runPrompt ?? options.runPrompt),
-		prompt: async (promptInput, runPrompt) => {
-			let agentId = promptInput.agentId;
-			if (!agentId) agentId = (await getAgentByRoute(`/agents/${key}`))?.id;
-			if (!agentId) throw new Error("agentId is required when calling Agent.prompt(); pass agentId or define the agent with { key }");
-			return runAgentPrompt(agent, agentId, promptInput, {
-				...options.runPrompt,
-				...runPrompt
-			});
+		prompt: (promptInput, runPrompt) => {
+			const handle = getWorkflowRunHandle();
+			if (handle?.agentRunner && !isWithinActionExecution() && !isWithinAgentPromptExecution()) {
+				const { stepId, ...mergedRunPrompt } = {
+					...options.runPrompt,
+					...runPrompt
+				};
+				return handle.agentRunner(agent, promptInput, {
+					id: stepId,
+					runPrompt: mergedRunPrompt
+				});
+			}
+			return runDirectAgentPrompt(agent, key, options.runPrompt, promptInput, runPrompt);
 		}
 	};
 	return agent;
@@ -14063,7 +14090,7 @@ function defineSubagentTool(input) {
 					tool: toolDef.name,
 					type: "subagent"
 				}
-			}, () => agent.prompt({ message: toMessage(params) }, { tracing: { trigger: "subagent" } }));
+			}, async () => runDirectAgentPrompt(agent, agent.key, void 0, { message: toMessage(params) }, { tracing: { trigger: "subagent" } }));
 			if (result.error) throw new Error(result.error);
 			return {
 				content: [{
@@ -14081,6 +14108,6 @@ function defineSubagentTool(input) {
 	return subagentTool;
 }
 //#endregion
-export { AgentCreateInputSchema, DEFAULT_THINKING_LEVEL, ModelRefSchema, SUBAGENT_TOOL, SessionAgentMismatchError, ThinkingLevelSchema, buildAgentRuntime, connectMcpDefinition, connectMcpServer, connectMcpStdio, createAgent, defineAgent, defineMcp, defineSubagentTool, defineTool, errorMessage, getModel, getSubagentToolMetadata, isMcp, loadAssetManifest, messages, normalizeAgentDefinition, parseAgentCreateInput, prepareAgentSession, prompt, resolveAgentAssets, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
+export { AgentCreateInputSchema, DEFAULT_THINKING_LEVEL, ModelRefSchema, 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, resolveAgentTools, resolveAgentWorkspaceRoot, resolveThinkingLevel, runAgentPrompt, snapshot };
 
 //# sourceMappingURL=agent.mjs.map