@keystrokehq/keystroke 1.0.12 → 1.0.14

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-Ohwnfidc.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-BQEzTb7P.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-DdAOmbQc.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-ub0OqZXh.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

@@ -5314,10 +5314,29 @@ function validateToolArguments(tool, toolCall) {
 	throw new Error(errorMessage);
 }
 //#endregion
-//#region ../agent/dist/schemas-DCBOj0XY.mjs
+//#region ../agent/dist/schemas-T2usEGaN.mjs
 const getModelLoose = getModel;
+const DEFAULT_GATEWAY_ORIGIN = "https://ai-gateway.vercel.sh";
 const AGENT_MODEL_ID_PATTERN = /^[a-z0-9-]+\/.+/;
 const AgentModelIdSchema = z.custom((value) => typeof value === "string" && AGENT_MODEL_ID_PATTERN.test(value), "model must be vendor/model-id");
+const GatewayModelPricingSchema = z.object({
+	input: z.string().optional(),
+	output: z.string().optional(),
+	input_cache_read: z.string().optional()
+});
+const GatewayModelResponseSchema = z.object({
+	id: z.string().min(1),
+	name: z.string().optional(),
+	type: z.string().optional(),
+	context_window: z.number().int().positive().optional(),
+	max_tokens: z.number().int().positive().optional(),
+	tags: z.array(z.string()).optional(),
+	pricing: GatewayModelPricingSchema.optional()
+});
+function parseGatewayModelResponse(body) {
+	const parsed = GatewayModelResponseSchema.safeParse(body);
+	return parsed.success ? parsed.data : void 0;
+}
 function splitAgentModelId(modelId) {
 	const slash = modelId.indexOf("/");
 	return {
@@ -5348,11 +5367,95 @@ function resolveGatewayBaseUrl() {
 	const projectId = getProjectScopeId();
 	return `${platformUrl.replace(/\/$/, "")}/internal/projects/${encodeURIComponent(projectId)}/llm`;
 }
-function resolveGatewayModel(modelId) {
+function resolveGatewayOrigin(baseUrl) {
+	return baseUrl ?? DEFAULT_GATEWAY_ORIGIN;
+}
+function pricePerMillion(value) {
+	if (!value) return 0;
+	const perToken = Number(value);
+	if (!Number.isFinite(perToken) || perToken <= 0) return 0;
+	return perToken * 1e6;
+}
+function buildGatewayModel(raw, baseUrl) {
+	if (raw.type && raw.type !== "language") return;
+	const tags = raw.tags ?? [];
+	return {
+		id: raw.id,
+		name: raw.name ?? raw.id,
+		api: "anthropic-messages",
+		provider: "vercel-ai-gateway",
+		baseUrl: resolveGatewayOrigin(baseUrl),
+		reasoning: tags.includes("reasoning"),
+		input: tags.includes("vision") ? ["text", "image"] : ["text"],
+		cost: {
+			input: pricePerMillion(raw.pricing?.input),
+			output: pricePerMillion(raw.pricing?.output),
+			cacheRead: pricePerMillion(raw.pricing?.input_cache_read),
+			cacheWrite: 0
+		},
+		contextWindow: raw.context_window ?? 128e3,
+		maxTokens: raw.max_tokens ?? 8192
+	};
+}
+const gatewayModelCache = /* @__PURE__ */ new Map();
+const gatewayModelPending = /* @__PURE__ */ new Map();
+function gatewayModelCacheKey(modelId, baseUrl) {
+	return `${resolveGatewayOrigin(baseUrl)}::${modelId}`;
+}
+function readCachedGatewayModel(cacheKey) {
+	const entry = gatewayModelCache.get(cacheKey);
+	if (!entry) return { cached: false };
+	return {
+		cached: true,
+		model: entry.status === "found" ? entry.model : void 0
+	};
+}
+function writeCachedGatewayModel(cacheKey, model) {
+	gatewayModelCache.set(cacheKey, model ? {
+		status: "found",
+		model
+	} : { status: "missing" });
+}
+async function fetchGatewayModel(modelId, baseUrl) {
+	const cacheKey = gatewayModelCacheKey(modelId, baseUrl);
+	const cached = readCachedGatewayModel(cacheKey);
+	if (cached.cached) return cached.model;
+	const pending = gatewayModelPending.get(cacheKey);
+	if (pending) return pending;
+	const request = (async () => {
+		const gatewayApiKey = process.env.AI_GATEWAY_API_KEY?.trim();
+		const headers = {};
+		if (gatewayApiKey) headers.authorization = `Bearer ${gatewayApiKey}`;
+		const url = `${resolveGatewayOrigin(baseUrl)}/v1/models/${modelId}`;
+		try {
+			const response = await fetch(url, { headers });
+			if (!response.ok) {
+				writeCachedGatewayModel(cacheKey, void 0);
+				return;
+			}
+			const raw = parseGatewayModelResponse(await response.json());
+			if (!raw) {
+				writeCachedGatewayModel(cacheKey, void 0);
+				return;
+			}
+			const model = buildGatewayModel(raw, baseUrl);
+			writeCachedGatewayModel(cacheKey, model);
+			return model;
+		} catch {
+			writeCachedGatewayModel(cacheKey, void 0);
+			return;
+		} finally {
+			gatewayModelPending.delete(cacheKey);
+		}
+	})();
+	gatewayModelPending.set(cacheKey, request);
+	return request;
+}
+async function resolveGatewayModel(modelId) {
+	const baseUrl = resolveGatewayBaseUrl();
 	for (const candidate of agentModelIdCandidates(modelId)) {
 		const registered = getModelLoose("vercel-ai-gateway", candidate);
 		if (registered) {
-			const baseUrl = resolveGatewayBaseUrl();
 			if (!baseUrl) return registered;
 			return {
 				...registered,
@@ -5360,6 +5463,10 @@ function resolveGatewayModel(modelId) {
 			};
 		}
 	}
+	for (const candidate of agentModelIdCandidates(modelId)) {
+		const fetched = await fetchGatewayModel(candidate, baseUrl);
+		if (fetched) return fetched;
+	}
 	throw new Error(`Unknown gateway model: ${modelId}`);
 }
 function resolveDirectModel(modelId) {
@@ -5371,7 +5478,7 @@ function resolveDirectModel(modelId) {
 	throw new Error(`Unknown model: ${modelId}`);
 }
 /** Resolve a user-authored model id to a pi-ai {@link Model} for the current runtime env. */
-function resolveAgentModel(modelId) {
+async function resolveAgentModel(modelId) {
 	if (shouldUseGatewayTransport()) return resolveGatewayModel(modelId);
 	return resolveDirectModel(modelId);
 }
@@ -13929,7 +14036,7 @@ function normalizeTitle(raw) {
 async function generateSessionTitle(message) {
 	const input = message.trim();
 	if (!input) return null;
-	return normalizeTitle((await completeSimple(resolveAgentModel(TITLE_MODEL_ID), {
+	return normalizeTitle((await completeSimple(await resolveAgentModel(TITLE_MODEL_ID), {
 		systemPrompt: TITLE_SYSTEM_PROMPT,
 		messages: [{
 			role: "user",
@@ -14340,7 +14447,7 @@ async function buildAgentRuntime(def, ctx, runPrompt = {}, agentSlug) {
 	return {
 		...createAgent({
 			systemPrompt,
-			model: resolveAgentModel(def.model),
+			model: await resolveAgentModel(def.model),
 			thinkingLevel,
 			messages: ctx.messages,
 			tools
@@ -14518,7 +14625,7 @@ 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) {
+async function resolvePromptModel(model) {
 	return resolveAgentModel(AgentModelIdSchema.parse(model));
 }
 /**
@@ -14569,7 +14676,7 @@ async function recordUsageFromMessage(message, hooks) {
 	if (usage) await hooks?.onUsage?.(usage);
 }
 async function runPromptText(opts, hooks) {
-	const model = resolvePromptModel(opts.model);
+	const model = await resolvePromptModel(opts.model);
 	const reasoning = resolvePromptReasoning(opts.thinkingLevel);
 	const message = await completeSimple(model, {
 		systemPrompt: opts.system,
@@ -14590,7 +14697,7 @@ async function runPromptText(opts, hooks) {
 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 model = await resolvePromptModel(opts.model);
 	const reasoning = resolvePromptReasoning(opts.thinkingLevel);
 	const { toolSchema, unwrap } = wrapSchemaForTool(schema);
 	const tool = buildRespondTool(toolSchema);