@keystrokehq/keystroke 0.0.113 → 0.1.0

package/dist/agent.cjs

@@ -5321,16 +5321,10 @@ function validateToolArguments(tool, toolCall) {
 	throw new Error(errorMessage);
 }
 //#endregion
-//#region ../agent/dist/schemas-D3fs0obg.mjs
+//#region ../agent/dist/schemas-i6pkwhd7.mjs
 const getModelLoose = require_event_stream.getModel;
-const LEGACY_MODEL_REF_SCHEMA = zod.z.object({
-	provider: zod.z.string().min(1),
-	id: zod.z.string().min(1)
-});
-const AgentModelIdSchema = zod.z.union([zod.z.string().regex(/^[a-z0-9-]+\/.+/, "model must be vendor/model-id"), LEGACY_MODEL_REF_SCHEMA.transform(({ provider, id }) => `${provider}/${id}`)]);
-function parseAgentModelId(input) {
-	return AgentModelIdSchema.parse(input);
-}
+const AGENT_MODEL_ID_PATTERN = /^[a-z0-9-]+\/.+/;
+const AgentModelIdSchema = zod.z.custom((value) => typeof value === "string" && AGENT_MODEL_ID_PATTERN.test(value), "model must be vendor/model-id");
 function splitAgentModelId(modelId) {
 	const slash = modelId.indexOf("/");
 	return {
@@ -5338,12 +5332,8 @@ function splitAgentModelId(modelId) {
 		directId: modelId.slice(slash + 1)
 	};
 }
-function isWorkerRuntime() {
-	return process.env.KEYSTROKE_MODE === "worker";
-}
-/** Gateway transport when hosted or when a local gateway key is configured. */
 function shouldUseGatewayTransport() {
-	return isWorkerRuntime() || !!process.env.AI_GATEWAY_API_KEY?.trim();
+	return !!process.env.AI_GATEWAY_API_KEY?.trim();
 }
 function directModelIdCandidates(directId) {
 	const candidates = [directId];
@@ -5353,20 +5343,28 @@ function directModelIdCandidates(directId) {
 	if (dashToDot !== directId) candidates.push(dashToDot);
 	return candidates;
 }
+function agentModelIdCandidates(modelId) {
+	const { vendor, directId } = splitAgentModelId(modelId);
+	return directModelIdCandidates(directId).map((candidate) => `${vendor}/${candidate}`);
+}
 function resolveGatewayBaseUrl() {
 	const configured = process.env.AI_GATEWAY_BASE_URL?.trim();
 	if (!configured) return;
 	return configured.replace(/\/$/, "");
 }
 function resolveGatewayModel(modelId) {
-	const registered = getModelLoose("vercel-ai-gateway", modelId);
-	if (!registered) throw new Error(`Unknown gateway model: ${modelId}`);
-	const baseUrl = resolveGatewayBaseUrl();
-	if (!baseUrl) return registered;
-	return {
-		...registered,
-		baseUrl
-	};
+	for (const candidate of agentModelIdCandidates(modelId)) {
+		const registered = getModelLoose("vercel-ai-gateway", candidate);
+		if (registered) {
+			const baseUrl = resolveGatewayBaseUrl();
+			if (!baseUrl) return registered;
+			return {
+				...registered,
+				baseUrl
+			};
+		}
+	}
+	throw new Error(`Unknown gateway model: ${modelId}`);
 }
 function resolveDirectModel(modelId) {
 	const { vendor, directId } = splitAgentModelId(modelId);
@@ -5381,15 +5379,6 @@ function resolveAgentModel(modelId) {
 	if (shouldUseGatewayTransport()) return resolveGatewayModel(modelId);
 	return resolveDirectModel(modelId);
 }
-/** Fail fast at discovery when a model id cannot be resolved in this runtime. */
-function validateAgentModel(modelId) {
-	resolveAgentModel(modelId);
-}
-/** @deprecated Use {@link AgentModelId} string (`vendor/model-id`) instead. */
-const ModelRefSchema = zod.z.object({
-	provider: zod.z.string(),
-	id: zod.z.string()
-});
 const ThinkingLevelSchema = zod.z.enum([
 	"off",
 	"minimal",
@@ -5406,11 +5395,7 @@ const AgentCreateInputSchema = zod.z.object({
 	thinkingLevel: ThinkingLevelSchema.optional()
 });
 function parseAgentCreateInput(input) {
-	const parsed = AgentCreateInputSchema.parse(input);
-	return {
-		...parsed,
-		model: parsed.model
-	};
+	return AgentCreateInputSchema.parse(input);
 }
 function resolveThinkingLevel(level) {
 	return level ?? "medium";
@@ -13995,8 +13980,7 @@ async function buildAgentRuntime(def, ctx, runPrompt = {}) {
 			model: resolveAgentModel(def.model),
 			thinkingLevel,
 			messages: ctx.messages,
-			tools,
-			...isWorkerRuntime() ? { getApiKey: async () => process.env.WORKER_INTERNAL_TOKEN?.trim() ?? "" } : {}
+			tools
 		}),
 		sandbox: handle.runtime,
 		tools,
@@ -14193,7 +14177,6 @@ function defineSubagentTool(input) {
 exports.AgentCreateInputSchema = AgentCreateInputSchema;
 exports.AgentModelIdSchema = AgentModelIdSchema;
 exports.DEFAULT_THINKING_LEVEL = DEFAULT_THINKING_LEVEL;
-exports.ModelRefSchema = ModelRefSchema;
 exports.SUBAGENT_TOOL = SUBAGENT_TOOL;
 exports.SessionAgentMismatchError = SessionAgentMismatchError;
 exports.ThinkingLevelSchema = ThinkingLevelSchema;
@@ -14210,12 +14193,10 @@ exports.errorMessage = errorMessage;
 exports.getModel = require_event_stream.getModel;
 exports.getSubagentToolMetadata = getSubagentToolMetadata;
 exports.isMcp = require_dist$3.isMcp;
-exports.isWorkerRuntime = isWorkerRuntime;
 exports.loadAssetManifest = loadAssetManifest;
 exports.messages = messages;
 exports.normalizeAgentDefinition = normalizeAgentDefinition;
 exports.parseAgentCreateInput = parseAgentCreateInput;
-exports.parseAgentModelId = parseAgentModelId;
 exports.prepareAgentSession = prepareAgentSession;
 exports.prompt = prompt;
 exports.resolveAgentAssets = resolveAgentAssets;
@@ -14225,8 +14206,6 @@ exports.resolveAgentTools = resolveAgentTools;
 exports.resolveAgentWorkspaceRoot = resolveAgentWorkspaceRoot;
 exports.resolveThinkingLevel = resolveThinkingLevel;
 exports.runAgentPrompt = runAgentPrompt;
-exports.shouldUseGatewayTransport = shouldUseGatewayTransport;
 exports.snapshot = snapshot;
-exports.validateAgentModel = validateAgentModel;
 
 //# sourceMappingURL=agent.cjs.map