@keystrokehq/keystroke 0.0.101 → 0.0.102

package/dist/action.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_dist = require("./dist-DG2RS68d.cjs");
+const require_dist = require("./dist-DnqDVVpK.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;
@@ -14,6 +14,7 @@ exports.getWorkflowRunHandle = require_dist.getWorkflowRunHandle;
 exports.isAction = require_dist.isAction;
 exports.isCredentialInput = require_dist.isCredentialInput;
 exports.isStepInvocation = require_dist.isStepInvocation;
+exports.isWithinActionExecution = require_dist.isWithinActionExecution;
 exports.normalizeCredentialList = require_dist.normalizeCredentialList;
 exports.registerWorkflowRunGetter = require_dist.registerWorkflowRunGetter;
 exports.resolveActionTool = require_dist.resolveActionTool;

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-CiuCYKLw.cjs";
-import { _ as resolveActionTool, a as ResolveActionToolOptions, c as actionCoreSchema, d as executeAction, f as getActionCredentialRequirements, g as registerWorkflowRunGetter, h as isStepInvocation, i as ResolveActionCredentialsFn, l as createStepInvocation, m as isAction, n as ActionDefinition, o as StepInvocation, p as getWorkflowRunHandle, r as ActionDefinitionInput, s as WorkflowActionDefinition, t as ActionCredentialConsumer, u as defineAction, v as runOutsideActionExecution, y as runWithinActionExecution } from "./index-CfK7yBQI.cjs";
-export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, 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, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithinActionExecution, toCredentialRequirement };
+import { _ as isWithinActionExecution, a as ResolveActionCredentialsFn, b as runOutsideActionExecution, c as WorkflowActionDefinition, d as defineAction, f as executeAction, g as isStepInvocation, h as isAction, i as AgentRunOptions, l as actionCoreSchema, m as getWorkflowRunHandle, n as ActionDefinition, o as ResolveActionToolOptions, p as getActionCredentialRequirements, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as createStepInvocation, v as registerWorkflowRunGetter, x as runWithinActionExecution, y as resolveActionTool } from "./index-B6BYxmgO.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, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, 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-CiuCYKLw.mjs";
-import { _ as resolveActionTool, a as ResolveActionToolOptions, c as actionCoreSchema, d as executeAction, f as getActionCredentialRequirements, g as registerWorkflowRunGetter, h as isStepInvocation, i as ResolveActionCredentialsFn, l as createStepInvocation, m as isAction, n as ActionDefinition, o as StepInvocation, p as getWorkflowRunHandle, r as ActionDefinitionInput, s as WorkflowActionDefinition, t as ActionCredentialConsumer, u as defineAction, v as runOutsideActionExecution, y as runWithinActionExecution } from "./index-by6PXA-8.mjs";
-export { type ActionCredentialConsumer, type ActionDefinition, type ActionDefinitionInput, 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, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithinActionExecution, toCredentialRequirement };
+import { _ as isWithinActionExecution, a as ResolveActionCredentialsFn, b as runOutsideActionExecution, c as WorkflowActionDefinition, d as defineAction, f as executeAction, g as isStepInvocation, h as isAction, i as AgentRunOptions, l as actionCoreSchema, m as getWorkflowRunHandle, n as ActionDefinition, o as ResolveActionToolOptions, p as getActionCredentialRequirements, r as ActionDefinitionInput, s as StepInvocation, t as ActionCredentialConsumer, u as createStepInvocation, v as registerWorkflowRunGetter, x as runWithinActionExecution, y as resolveActionTool } from "./index-DtcAWajc.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, getWorkflowRunHandle, isAction, isCredentialInput, isStepInvocation, isWithinActionExecution, normalizeCredentialList, registerWorkflowRunGetter, resolveActionTool, runOutsideActionExecution, runWithinActionExecution, toCredentialRequirement };

package/dist/action.mjs

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

package/dist/agent.cjs

@@ -1,11 +1,12 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
 const require_dist$1 = require("./dist-CLqJza2Y.cjs");
-const require_dist$2 = require("./dist-DG2RS68d.cjs");
+const require_dist$2 = require("./dist-DnqDVVpK.cjs");
 require("./env-api-keys-BFptfIIs.cjs");
 const require_event_stream = require("./event-stream-CS-ls7wC.cjs");
 require("./json-parse-CpkTvzp8.cjs");
-const require_dist$3 = require("./dist-DJhg5KTL.cjs");
+const require_dist$3 = require("./dist-BkC9pJIu.cjs");
 let zod = require("zod");
+let node_async_hooks = require("node:async_hooks");
 let node_module = require("node:module");
 let node_fs = require("node:fs");
 let node_path = require("node:path");
@@ -928,7 +929,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-9mfXttcp.cjs")).then((module) => {
+	mistralProviderModulePromise ||= Promise.resolve().then(() => require("./mistral-DXb4X3t4.cjs")).then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -13680,6 +13681,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 node_async_hooks.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 require_dist$3.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: require_dist$3.MESSAGE_EVENT_TYPE,
+		payload: event.message
+	});
+}
+async function prepareAgentSession(agentId, sessionId) {
+	if (sessionId) {
+		const session = await require_dist$3.getSession(sessionId);
+		if (!session) await require_dist$3.createSession(agentId, sessionId, await require_dist$3.resolveRunSourceFromTraceContext());
+		else if (session.agentId !== agentId) throw new SessionAgentMismatchError(sessionId);
+		return { sessionId };
+	}
+	return { sessionId: (await require_dist$3.createSession(agentId, void 0, await require_dist$3.resolveRunSourceFromTraceContext())).id };
+}
+async function runAgentPrompt(agent, agentId, input, options = {}) {
+	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
+	const priorMessages = await require_dist$3.listMessageEvents(sessionId);
+	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);
+		});
+		try {
+			const snapshot = await require_dist$3.captureConsole(async () => prompt(runtime, input.message));
+			const credentialError = runtime.takeCredentialError?.();
+			if (credentialError && require_dist$3.isCredentialError(credentialError)) throw credentialError;
+			await require_dist$3.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 = require_dist$3.getTraceContext();
+	return require_dist$3.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 require_dist$3.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.
 *
@@ -13906,110 +14032,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 require_dist$3.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: require_dist$3.MESSAGE_EVENT_TYPE,
-		payload: event.message
-	});
-}
-async function prepareAgentSession(agentId, sessionId) {
-	if (sessionId) {
-		const session = await require_dist$3.getSession(sessionId);
-		if (!session) await require_dist$3.createSession(agentId, sessionId, await require_dist$3.resolveRunSourceFromTraceContext());
-		else if (session.agentId !== agentId) throw new SessionAgentMismatchError(sessionId);
-		return { sessionId };
-	}
-	return { sessionId: (await require_dist$3.createSession(agentId, void 0, await require_dist$3.resolveRunSourceFromTraceContext())).id };
-}
-async function runAgentPrompt(agent, agentId, input, options = {}) {
-	const { sessionId } = await prepareAgentSession(agentId, input.sessionId);
-	const priorMessages = await require_dist$3.listMessageEvents(sessionId);
-	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);
-		});
-		try {
-			const snapshot = await require_dist$3.captureConsole(async () => prompt(runtime, input.message));
-			const credentialError = runtime.takeCredentialError?.();
-			if (credentialError && require_dist$3.isCredentialError(credentialError)) throw credentialError;
-			await require_dist$3.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 = require_dist$3.getTraceContext();
-	return require_dist$3.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.
 */
@@ -14024,14 +14046,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 require_dist$3.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 = require_dist$2.getWorkflowRunHandle();
+			if (handle?.agentRunner && !require_dist$2.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;
@@ -14074,7 +14101,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: [{
@@ -14118,6 +14145,7 @@ exports.parseAgentCreateInput = parseAgentCreateInput;
 exports.prepareAgentSession = prepareAgentSession;
 exports.prompt = prompt;
 exports.resolveAgentAssets = resolveAgentAssets;
+exports.resolveAgentId = resolveAgentId;
 exports.resolveAgentTools = resolveAgentTools;
 exports.resolveAgentWorkspaceRoot = resolveAgentWorkspaceRoot;
 exports.resolveThinkingLevel = resolveThinkingLevel;