@keystrokehq/keystroke 1.0.8 → 1.0.9

package/dist/agent.d.cts

@@ -1,5 +1,5 @@
 import { l as ThinkingLevel, u as getModel } from "./index-BByNfhJU.cjs";
-import { A as forceToolChoice, B as runAgentPrompt, C as SubagentToolMetadata, D as defineSubagentTool, E as defineAgent, F as prompt, G as AgentModelId, H as snapshot, I as resolveAgentAssets, J as ThinkingLevelSchema, K as AgentModelIdSchema, L as resolveAgentId, M as messages, N as normalizeAgentDefinition, O as defineTool, P as prepareAgentSession, R as resolveAgentTools, S as SubagentToolInherit, T as createAgent, U as AgentCreateInput, V as runLlm, W as AgentCreateInputSchema, X as resolveAgentModel, Y as parseAgentCreateInput, Z as resolveThinkingLevel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentTool, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as getSubagentToolMetadata, k as errorMessage, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as DEFAULT_THINKING_LEVEL, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SessionLifecycle, w as buildAgentRuntime, x as SubagentToolDetails, y as SessionLifecycleContext, z as resolveAgentWorkspaceRoot } from "./index-BY9SPR2h.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-nUAXCgEe.cjs";
+import { A as forceToolChoice, B as runAgentPrompt, C as SubagentToolMetadata, D as defineSubagentTool, E as defineAgent, F as prompt, G as AgentModelId, H as snapshot, I as resolveAgentAssets, J as ThinkingLevelSchema, K as AgentModelIdSchema, L as resolveAgentId, M as messages, N as normalizeAgentDefinition, O as defineTool, P as prepareAgentSession, R as resolveAgentTools, S as SubagentToolInherit, T as createAgent, U as AgentCreateInput, V as runLlm, W as AgentCreateInputSchema, X as resolveAgentModel, Y as parseAgentCreateInput, Z as resolveThinkingLevel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentTool, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as getSubagentToolMetadata, k as errorMessage, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as DEFAULT_THINKING_LEVEL, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SessionLifecycle, w as buildAgentRuntime, x as SubagentToolDetails, y as SessionLifecycleContext, z as resolveAgentWorkspaceRoot } from "./index-Ohwnfidc.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-BTz2OMAM.cjs";
 import { z as loadAssetManifest } from "./index-DstR-eGs.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 SessionLifecycle, type SessionLifecycleContext, 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,5 +1,5 @@
 import { l as ThinkingLevel, u as getModel } from "./index-ClGiGBy7.mjs";
-import { A as forceToolChoice, B as runAgentPrompt, C as SubagentToolMetadata, D as defineSubagentTool, E as defineAgent, F as prompt, G as AgentModelId, H as snapshot, I as resolveAgentAssets, J as ThinkingLevelSchema, K as AgentModelIdSchema, L as resolveAgentId, M as messages, N as normalizeAgentDefinition, O as defineTool, P as prepareAgentSession, R as resolveAgentTools, S as SubagentToolInherit, T as createAgent, U as AgentCreateInput, V as runLlm, W as AgentCreateInputSchema, X as resolveAgentModel, Y as parseAgentCreateInput, Z as resolveThinkingLevel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentTool, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as getSubagentToolMetadata, k as errorMessage, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as DEFAULT_THINKING_LEVEL, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SessionLifecycle, w as buildAgentRuntime, x as SubagentToolDetails, y as SessionLifecycleContext, z as resolveAgentWorkspaceRoot } from "./index-BeBgntxT.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-8_UEbXaF.mjs";
+import { A as forceToolChoice, B as runAgentPrompt, C as SubagentToolMetadata, D as defineSubagentTool, E as defineAgent, F as prompt, G as AgentModelId, H as snapshot, I as resolveAgentAssets, J as ThinkingLevelSchema, K as AgentModelIdSchema, L as resolveAgentId, M as messages, N as normalizeAgentDefinition, O as defineTool, P as prepareAgentSession, R as resolveAgentTools, S as SubagentToolInherit, T as createAgent, U as AgentCreateInput, V as runLlm, W as AgentCreateInputSchema, X as resolveAgentModel, Y as parseAgentCreateInput, Z as resolveThinkingLevel, _ as SessionAgentMismatchError, a as AgentDefinitionInput, b as SubagentTool, c as BuildAgentRuntimeContext, d as PromptInput, f as PromptResponse, g as SandboxedAgentRuntime, h as SUBAGENT_TOOL, i as AgentDefinition, j as getSubagentToolMetadata, k as errorMessage, l as DefineSubagentToolInput, m as RunAgentPromptOptions, n as AgentAssetInput, o as AgentRuntime, p as ResolvedAgentAssets, q as DEFAULT_THINKING_LEVEL, r as AgentConfig, s as AgentSnapshot, t as Agent, u as ForcedToolChoice, v as SessionLifecycle, w as buildAgentRuntime, x as SubagentToolDetails, y as SessionLifecycleContext, z as resolveAgentWorkspaceRoot } from "./index-DdAOmbQc.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-Cq2F2XdS.mjs";
 import { z as loadAssetManifest } from "./index-B3iWqXUd.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 SessionLifecycle, type SessionLifecycleContext, 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,6 +1,6 @@
 import { i as __require, r as __exportAll, t as __commonJSMin } from "./chunk-D0VCBZRD.mjs";
-import { E as toolParameters, c as getWorkflowRunHandle, d as isWithinActionExecution, l as isAction, m as resolveActionTool, s as getRunSignal } from "./dist-DqFdFpiB.mjs";
-import { A as resolveRunSourceFromTraceContext, C as clearLiveMessage, D as getSession, E as getProjectScopeId, F as getTraceContext, L as withSpan, M as setSessionTitle, N as touchSession, O as listMessageEvents, P as captureConsole, S as appendEvent, T as getAgentByRoute, a as resolveActionCredentials, b as MESSAGE_EVENT_TYPE, c as connectMcpServer, d as isMcp, i as isCredentialError, j as setSessionLiveMessage, k as recordLlmUsageFromAssistantMessage, l as connectMcpStdio, n as captureCredentialToolErrors, o as resolveMcpTools, r as createCredentialResolver, s as connectMcpDefinition, t as buildCredentialRunContext, u as defineMcp, w as createSession, x as addAgentSessionDuration } from "./dist-IUl7Bexl.mjs";
+import { O as toolParameters, c as getRunSignal, f as isWithinActionExecution, h as resolveActionTool, l as getWorkflowRunHandle, u as isAction } from "./dist-CppO7361.mjs";
+import { A as recordLlmUsageFromAssistantMessage, C as appendEvent, D as getProjectScopeId, E as getAgentByRoute, F as captureConsole, I as getTraceContext, M as setSessionLiveMessage, N as setSessionTitle, O as getSession, P as touchSession, R as withSpan, S as addAgentSessionDuration, T as createSession, a as resolveActionCredentials, c as connectMcpDefinition, d as defineMcp, f as isMcp, i as isCredentialError, j as resolveRunSourceFromTraceContext, k as listMessageEvents, l as connectMcpServer, n as captureCredentialToolErrors, o as resolveMcpTools, r as createCredentialResolver, s as withCredentialAssignments, t as buildCredentialRunContext, u as connectMcpStdio, w as clearLiveMessage, x as MESSAGE_EVENT_TYPE } from "./dist-UzTnzZem.mjs";
 import { $t as _Object_, An as IsFunction$1, At as IsNull$2, Bn as IsNull$1, Bt as IsBoolean$2, Cn as IsAsyncIterator$1, Ct as EnumToUnion, Dn as IsConstructor$1, Dt as IsSymbol$2, En as IsClassInstance, Et as String$1, Fn as IsLessEqualThan$1, Ft as IsLiteralString, Gn as IsSymbol$1, Gt as Unreachable, Hn as IsObject$1, Ht as IsCodec, In as IsLessThan$1, It as Literal, J as getBoundAppRoot, Jn as IsValueLike, Jt as IsCyclic, Kn as IsUndefined$1, Kt as IsIntersect, Ln as IsMaxLength$2, Lt as IsIterator$2, Mn as IsGreaterThan$1, Mt as IsLiteralBigInt, Nn as IsInteger$1, Nt as IsLiteralBoolean, On as IsDeepEqual$1, Ot as IsNumber$2, Pn as IsIterator$1, Pt as IsLiteralNumber, Qt as IsObject$2, Rn as IsMinLength$2, Rt as IsInteger$2, Sn as IsArray$1, St as IsTuple, Tn as IsBoolean$1, Tt as IsString$2, Un as IsObjectNotArray$1, Ut as Hash, Vn as IsNumber$1, Vt as IsBigInt$2, Wn as IsString$1, Wt as hash_exports, X as materializeSandbox, Xn as Symbols, Xt as IsUnion, Y as loadAssetManifest, Yn as Keys$1, Yt as Unknown, Zn as guard_exports, Zt as Union, _n as Entries$2, _t as IsVoid, a as defaultWorkspacesRoot, an as IsAsyncIterator$2, bn as EveryAll, bt as RecordPattern, cn as IsBase, ct as ensureWorkspaceDir, dn as IsPromise, dt as resolveSessionRoot, en as IsNever, fn as IsSchema$1, ft as Evaluate$1, gn as IsTypeArray, gt as IsTemplateLiteral, hn as IsSet, ht as Compare, i as createSandbox, in as IsConstructor$2, jn as IsGreaterEqualThan$1, jt as IsLiteral, kn as IsEqual$1, kt as Number$1, l as materializeSkills, ln as IsOptional, lt as resolveAgentRoot, mn as IsMap, mt as Options, n as createInvokeToolBridge, nn as Ref$1, o as defineSkill, on as IsArray$2, p as sandboxSystemPromptInjection, pn as Get$3, pt as Instantiate, qn as IsUnsafePropertyKey, qt as IsEnum$1, rn as IsFunction$2, sn as Base, tn as IsRef$1, tt as resolveSandboxDefinition, un as Optional, vn as EntriesRegExp, vt as IsUndefined$2, wn as IsBigInt$1, wt as TemplateLiteralDecode, xn as HasPropertyKey$1, xt as RecordValue, yn as Every$1, yt as IsRecord, zn as IsMultipleOf$1, zt as Boolean$1 } from "./dist-CRsUBf24.mjs";
 import "./env-api-keys-24ayFQZr.mjs";
 import { i as getModel, t as AssistantMessageEventStream } from "./event-stream-C7YBbzsc.mjs";
@@ -901,7 +901,7 @@ function loadGoogleVertexProviderModule() {
 	return googleVertexProviderModulePromise;
 }
 function loadMistralProviderModule() {
-	mistralProviderModulePromise ||= import("./mistral-CMDejXQt.mjs").then((module) => {
+	mistralProviderModulePromise ||= import("./mistral-StBMjPiK.mjs").then((module) => {
 		const provider = module;
 		return {
 			stream: provider.streamMistral,
@@ -14235,7 +14235,7 @@ async function resolveAgentWorkspaceRoot(agentId, sessionId, options) {
 		sessionRoot
 	};
 }
-async function resolveAgentTools(tools, resolver) {
+async function resolveAgentTools(tools, resolver, options) {
 	const prepared = [];
 	const mcpConnections = [];
 	for (const item of tools) {
@@ -14254,8 +14254,10 @@ async function resolveAgentTools(tools, resolver) {
 				}),
 				consumer: {
 					kind: "tool",
-					name: item.slug
-				}
+					name: item.slug,
+					id: item.slug
+				},
+				mcpCredentialContext: options?.mcpCredentialContext
 			}));
 			continue;
 		}
@@ -14273,7 +14275,7 @@ async function resolveMemoryFactory(memoryOption, defMemory, injectedFactory) {
 	return createDefaultMemory(typeof option === "object" ? option : {});
 }
 /** Sandbox tools + host tools → pi agent from a {@link AgentDefinition}. */
-async function buildAgentRuntime(def, ctx, runPrompt = {}) {
+async function buildAgentRuntime(def, ctx, runPrompt = {}, agentSlug) {
 	const { agentRoot, sessionRoot } = await resolveAgentWorkspaceRoot(ctx.agentId, ctx.sessionId, {
 		workspacesRoot: runPrompt.workspacesRoot,
 		agentRoot: runPrompt.agentRoot,
@@ -14291,12 +14293,21 @@ async function buildAgentRuntime(def, ctx, runPrompt = {}) {
 		agentId: ctx.agentId,
 		sessionId: ctx.sessionId
 	}) : void 0;
+	const credentialContext = buildCredentialRunContext({ request: runPrompt.credentials });
+	const assignmentTargetKey = agentSlug ?? runPrompt.tracing?.agentKey;
 	const resolver = createCredentialResolver({
-		context: buildCredentialRunContext({ request: runPrompt.credentials }),
+		context: assignmentTargetKey ? await withCredentialAssignments(credentialContext, {
+			targetType: "agent",
+			targetKey: assignmentTargetKey
+		}) : credentialContext,
 		oauthAdapter: runPrompt.oauthAdapter
 	});
 	const memoryTools = memory ? [memory.tool] : [];
-	const resolved = await resolveAgentTools(def.tools ?? [], resolver);
+	const mcpCredentialContext = assignmentTargetKey ? { assignmentTarget: {
+		type: "agent",
+		key: assignmentTargetKey
+	} } : void 0;
+	const resolved = await resolveAgentTools(def.tools ?? [], resolver, { mcpCredentialContext });
 	const captured = captureCredentialToolErrors([
 		...memoryTools,
 		...resolved.tools,
@@ -14409,7 +14420,7 @@ function defineAgent(input) {
 		slug,
 		...input.name !== void 0 ? { name: input.name } : {},
 		...input.description !== void 0 ? { description: input.description } : {},
-		buildRuntime: (ctx, runPrompt) => buildAgentRuntime(def, ctx, runPrompt),
+		buildRuntime: (ctx, runPrompt) => buildAgentRuntime(def, ctx, runPrompt, slug),
 		prompt: (promptInput, runPrompt) => {
 			const handle = getWorkflowRunHandle();
 			if (handle?.agentRunner && !isWithinActionExecution() && !isWithinAgentPromptExecution()) {