@cloudflare/codemode 0.0.8 → 0.1.1

package/dist/types-CpkEgXwN.js

@@ -0,0 +1,392 @@
+import { createAuxiliaryTypeStore, createTypeAlias, printNode, zodToTs } from "zod-to-ts";
+
+//#region src/types.ts
+const JS_RESERVED = new Set([
+	"abstract",
+	"arguments",
+	"await",
+	"boolean",
+	"break",
+	"byte",
+	"case",
+	"catch",
+	"char",
+	"class",
+	"const",
+	"continue",
+	"debugger",
+	"default",
+	"delete",
+	"do",
+	"double",
+	"else",
+	"enum",
+	"eval",
+	"export",
+	"extends",
+	"false",
+	"final",
+	"finally",
+	"float",
+	"for",
+	"function",
+	"goto",
+	"if",
+	"implements",
+	"import",
+	"in",
+	"instanceof",
+	"int",
+	"interface",
+	"let",
+	"long",
+	"native",
+	"new",
+	"null",
+	"package",
+	"private",
+	"protected",
+	"public",
+	"return",
+	"short",
+	"static",
+	"super",
+	"switch",
+	"synchronized",
+	"this",
+	"throw",
+	"throws",
+	"transient",
+	"true",
+	"try",
+	"typeof",
+	"undefined",
+	"var",
+	"void",
+	"volatile",
+	"while",
+	"with",
+	"yield"
+]);
+/**
+* Sanitize a tool name into a valid JavaScript identifier.
+* Replaces hyphens, dots, and spaces with `_`, strips other invalid chars,
+* prefixes digit-leading names with `_`, and appends `_` to JS reserved words.
+*/
+function sanitizeToolName(name) {
+	if (!name) return "_";
+	let sanitized = name.replace(/[-.\s]/g, "_");
+	sanitized = sanitized.replace(/[^a-zA-Z0-9_$]/g, "");
+	if (!sanitized) return "_";
+	if (/^[0-9]/.test(sanitized)) sanitized = "_" + sanitized;
+	if (JS_RESERVED.has(sanitized)) sanitized = sanitized + "_";
+	return sanitized;
+}
+function toCamelCase(str) {
+	return str.replace(/_([a-z])/g, (_, letter) => letter.toUpperCase()).replace(/^[a-z]/, (letter) => letter.toUpperCase());
+}
+/**
+* Extract field descriptions from a schema and format as @param lines.
+* Returns an array of `@param input.fieldName - description` lines.
+*/
+function extractParamDescriptions(schema) {
+	const descriptions = extractDescriptions(schema);
+	return Object.entries(descriptions).map(([fieldName, desc]) => `@param input.${fieldName} - ${desc}`);
+}
+/**
+* Check if a value is a Zod schema (has _zod property).
+*/
+function isZodSchema(value) {
+	return value !== null && typeof value === "object" && "_zod" in value && value._zod !== void 0;
+}
+/**
+* Check if a value is an AI SDK jsonSchema wrapper.
+* The jsonSchema wrapper has a [Symbol] with jsonSchema property.
+*/
+function isJsonSchemaWrapper(value) {
+	if (value === null || typeof value !== "object") return false;
+	if ("jsonSchema" in value) return true;
+	const symbols = Object.getOwnPropertySymbols(value);
+	for (const sym of symbols) {
+		const symValue = value[sym];
+		if (symValue && typeof symValue === "object" && "jsonSchema" in symValue) return true;
+	}
+	return false;
+}
+/**
+* Extract JSON schema from an AI SDK jsonSchema wrapper.
+*/
+function extractJsonSchema(wrapper) {
+	if (wrapper === null || typeof wrapper !== "object") return null;
+	if ("jsonSchema" in wrapper) return wrapper.jsonSchema;
+	const symbols = Object.getOwnPropertySymbols(wrapper);
+	for (const sym of symbols) {
+		const symValue = wrapper[sym];
+		if (symValue && typeof symValue === "object" && "jsonSchema" in symValue) return symValue.jsonSchema;
+	}
+	return null;
+}
+/**
+* Check if a property name needs quoting in TypeScript.
+*/
+function needsQuotes(name) {
+	return !/^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(name);
+}
+/**
+* Escape a character as a unicode escape sequence if it is a control character.
+*/
+function escapeControlChar(ch) {
+	const code = ch.charCodeAt(0);
+	if (code <= 31 || code === 127) return "\\u" + code.toString(16).padStart(4, "0");
+	return ch;
+}
+/**
+* Quote a property name if needed.
+* Escapes backslashes, quotes, and control characters.
+*/
+function quoteProp(name) {
+	if (needsQuotes(name)) {
+		let escaped = "";
+		for (const ch of name) if (ch === "\\") escaped += "\\\\";
+		else if (ch === "\"") escaped += "\\\"";
+		else if (ch === "\n") escaped += "\\n";
+		else if (ch === "\r") escaped += "\\r";
+		else if (ch === "	") escaped += "\\t";
+		else if (ch === "\u2028") escaped += "\\u2028";
+		else if (ch === "\u2029") escaped += "\\u2029";
+		else escaped += escapeControlChar(ch);
+		return `"${escaped}"`;
+	}
+	return name;
+}
+/**
+* Escape a string for use inside a double-quoted TypeScript string literal.
+* Handles backslashes, quotes, newlines, control characters, and line/paragraph separators.
+*/
+function escapeStringLiteral(s) {
+	let out = "";
+	for (const ch of s) if (ch === "\\") out += "\\\\";
+	else if (ch === "\"") out += "\\\"";
+	else if (ch === "\n") out += "\\n";
+	else if (ch === "\r") out += "\\r";
+	else if (ch === "	") out += "\\t";
+	else if (ch === "\u2028") out += "\\u2028";
+	else if (ch === "\u2029") out += "\\u2029";
+	else out += escapeControlChar(ch);
+	return out;
+}
+/**
+* Escape a string for use inside a JSDoc comment.
+* Prevents premature comment closure from star-slash sequences.
+*/
+function escapeJsDoc(text) {
+	return text.replace(/\*\//g, "*\\/");
+}
+/**
+* Resolve an internal JSON Pointer $ref (e.g. #/definitions/Foo) against the root schema.
+* Returns null for external URLs or unresolvable paths.
+*/
+function resolveRef(ref, root) {
+	if (ref === "#") return root;
+	if (!ref.startsWith("#/")) return null;
+	const segments = ref.slice(2).split("/").map((s) => s.replace(/~1/g, "/").replace(/~0/g, "~"));
+	let current = root;
+	for (const seg of segments) {
+		if (current === null || typeof current !== "object") return null;
+		current = current[seg];
+		if (current === void 0) return null;
+	}
+	if (typeof current === "boolean") return current;
+	if (current === null || typeof current !== "object") return null;
+	return current;
+}
+/**
+* Convert a JSON Schema to a TypeScript type string.
+* This is a direct conversion without going through Zod.
+*/
+function jsonSchemaToTypeString(schema, indent, ctx) {
+	if (typeof schema === "boolean") return schema ? "unknown" : "never";
+	if (ctx.depth >= ctx.maxDepth) return "unknown";
+	if (ctx.seen.has(schema)) return "unknown";
+	const nextCtx = {
+		...ctx,
+		depth: ctx.depth + 1,
+		seen: new Set([...ctx.seen, schema])
+	};
+	if (schema.$ref) {
+		const resolved = resolveRef(schema.$ref, ctx.root);
+		if (!resolved) return "unknown";
+		return applyNullable(jsonSchemaToTypeString(resolved, indent, nextCtx), schema);
+	}
+	if (schema.anyOf) return applyNullable(schema.anyOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" | "), schema);
+	if (schema.oneOf) return applyNullable(schema.oneOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" | "), schema);
+	if (schema.allOf) return applyNullable(schema.allOf.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(" & "), schema);
+	if (schema.enum) {
+		if (schema.enum.length === 0) return "never";
+		return applyNullable(schema.enum.map((v) => {
+			if (v === null) return "null";
+			if (typeof v === "string") return "\"" + escapeStringLiteral(v) + "\"";
+			if (typeof v === "object") return JSON.stringify(v) ?? "unknown";
+			return String(v);
+		}).join(" | "), schema);
+	}
+	if (schema.const !== void 0) return applyNullable(schema.const === null ? "null" : typeof schema.const === "string" ? "\"" + escapeStringLiteral(schema.const) + "\"" : typeof schema.const === "object" ? JSON.stringify(schema.const) ?? "unknown" : String(schema.const), schema);
+	const type = schema.type;
+	if (type === "string") return applyNullable("string", schema);
+	if (type === "number" || type === "integer") return applyNullable("number", schema);
+	if (type === "boolean") return applyNullable("boolean", schema);
+	if (type === "null") return "null";
+	if (type === "array") {
+		const prefixItems = schema.prefixItems;
+		if (Array.isArray(prefixItems)) return applyNullable(`[${prefixItems.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(", ")}]`, schema);
+		if (Array.isArray(schema.items)) return applyNullable(`[${schema.items.map((s) => jsonSchemaToTypeString(s, indent, nextCtx)).join(", ")}]`, schema);
+		if (schema.items) return applyNullable(`${jsonSchemaToTypeString(schema.items, indent, nextCtx)}[]`, schema);
+		return applyNullable("unknown[]", schema);
+	}
+	if (type === "object" || schema.properties) {
+		const props = schema.properties || {};
+		const required = new Set(schema.required || []);
+		const lines = [];
+		for (const [propName, propSchema] of Object.entries(props)) {
+			if (typeof propSchema === "boolean") {
+				const boolType = propSchema ? "unknown" : "never";
+				const optionalMark = required.has(propName) ? "" : "?";
+				lines.push(`${indent}    ${quoteProp(propName)}${optionalMark}: ${boolType};`);
+				continue;
+			}
+			const isRequired = required.has(propName);
+			const propType = jsonSchemaToTypeString(propSchema, indent + "    ", nextCtx);
+			const desc = propSchema.description;
+			const format = propSchema.format;
+			if (desc || format) {
+				const descText = desc ? escapeJsDoc(desc.replace(/\r?\n/g, " ")) : void 0;
+				const formatTag = format ? `@format ${escapeJsDoc(format)}` : void 0;
+				if (descText && formatTag) {
+					lines.push(`${indent}    /**`);
+					lines.push(`${indent}     * ${descText}`);
+					lines.push(`${indent}     * ${formatTag}`);
+					lines.push(`${indent}     */`);
+				} else lines.push(`${indent}    /** ${descText ?? formatTag} */`);
+			}
+			const quotedName = quoteProp(propName);
+			const optionalMark = isRequired ? "" : "?";
+			lines.push(`${indent}    ${quotedName}${optionalMark}: ${propType};`);
+		}
+		if (schema.additionalProperties) {
+			const valueType = schema.additionalProperties === true ? "unknown" : jsonSchemaToTypeString(schema.additionalProperties, indent + "    ", nextCtx);
+			lines.push(`${indent}    [key: string]: ${valueType};`);
+		}
+		if (lines.length === 0) {
+			if (schema.additionalProperties === false) return applyNullable("{}", schema);
+			return applyNullable("Record<string, unknown>", schema);
+		}
+		return applyNullable(`{\n${lines.join("\n")}\n${indent}}`, schema);
+	}
+	if (Array.isArray(type)) return applyNullable(type.map((t) => {
+		if (t === "string") return "string";
+		if (t === "number" || t === "integer") return "number";
+		if (t === "boolean") return "boolean";
+		if (t === "null") return "null";
+		if (t === "array") return "unknown[]";
+		if (t === "object") return "Record<string, unknown>";
+		return "unknown";
+	}).join(" | "), schema);
+	return "unknown";
+}
+/**
+* Apply OpenAPI 3.0 `nullable: true` to a type result.
+*/
+function applyNullable(result, schema) {
+	if (result !== "unknown" && result !== "never" && schema?.nullable === true) return `${result} | null`;
+	return result;
+}
+/**
+* Extract field descriptions from a schema.
+* Works with Zod schemas (via .shape) and jsonSchema wrappers (via .properties).
+*/
+function extractDescriptions(schema) {
+	const descriptions = {};
+	const shape = schema.shape;
+	if (shape && typeof shape === "object") {
+		for (const [fieldName, fieldSchema] of Object.entries(shape)) {
+			const desc = fieldSchema.description;
+			if (desc) descriptions[fieldName] = desc;
+		}
+		return descriptions;
+	}
+	if (isJsonSchemaWrapper(schema)) {
+		const jsonSchema = extractJsonSchema(schema);
+		if (jsonSchema?.properties) {
+			for (const [fieldName, propSchema] of Object.entries(jsonSchema.properties)) if (propSchema && typeof propSchema === "object" && propSchema.description) descriptions[fieldName] = propSchema.description;
+		}
+	}
+	return descriptions;
+}
+/**
+* Safely convert a schema to TypeScript type string.
+* Handles Zod schemas and AI SDK jsonSchema wrappers.
+* Returns "unknown" if the schema cannot be represented in TypeScript.
+*/
+function safeSchemaToTs(schema, typeName, auxiliaryTypeStore) {
+	try {
+		if (isZodSchema(schema)) return printNode(createTypeAlias(zodToTs(schema, { auxiliaryTypeStore }).node, typeName));
+		if (isJsonSchemaWrapper(schema)) {
+			const jsonSchema = extractJsonSchema(schema);
+			if (jsonSchema) return `type ${typeName} = ${jsonSchemaToTypeString(jsonSchema, "", {
+				root: jsonSchema,
+				depth: 0,
+				seen: /* @__PURE__ */ new Set(),
+				maxDepth: 20
+			})}`;
+		}
+		return `type ${typeName} = unknown`;
+	} catch {
+		return `type ${typeName} = unknown`;
+	}
+}
+/**
+* Generate TypeScript type definitions from tool descriptors or an AI SDK ToolSet.
+* These types can be included in tool descriptions to help LLMs write correct code.
+*/
+function generateTypes(tools) {
+	let availableTools = "";
+	let availableTypes = "";
+	const auxiliaryTypeStore = createAuxiliaryTypeStore();
+	for (const [toolName, tool] of Object.entries(tools)) {
+		const safeName = sanitizeToolName(toolName);
+		const camelName = toCamelCase(safeName);
+		try {
+			const inputSchema = "inputSchema" in tool ? tool.inputSchema : tool.parameters;
+			const outputSchema = "outputSchema" in tool ? tool.outputSchema : void 0;
+			const description = tool.description;
+			const inputType = safeSchemaToTs(inputSchema, `${camelName}Input`, auxiliaryTypeStore);
+			const outputType = outputSchema ? safeSchemaToTs(outputSchema, `${camelName}Output`, auxiliaryTypeStore) : `type ${camelName}Output = unknown`;
+			availableTypes += `\n${inputType.trim()}`;
+			availableTypes += `\n${outputType.trim()}`;
+			const paramDescs = inputSchema ? extractParamDescriptions(inputSchema) : [];
+			const jsdocLines = [];
+			if (description?.trim()) jsdocLines.push(escapeJsDoc(description.trim().replace(/\r?\n/g, " ")));
+			else jsdocLines.push(escapeJsDoc(toolName));
+			for (const pd of paramDescs) jsdocLines.push(escapeJsDoc(pd.replace(/\r?\n/g, " ")));
+			const jsdocBody = jsdocLines.map((l) => `\t * ${l}`).join("\n");
+			availableTools += `\n\t/**\n${jsdocBody}\n\t */`;
+			availableTools += `\n\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;
+			availableTools += "\n";
+		} catch {
+			availableTypes += `\ntype ${camelName}Input = unknown`;
+			availableTypes += `\ntype ${camelName}Output = unknown`;
+			availableTools += `\n\t/**\n\t * ${escapeJsDoc(toolName)}\n\t */`;
+			availableTools += `\n\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;
+			availableTools += "\n";
+		}
+	}
+	availableTools = `\ndeclare const codemode: {${availableTools}}`;
+	return `
+${availableTypes}
+${availableTools}
+  `.trim();
+}
+
+//#endregion
+export { sanitizeToolName as n, generateTypes as t };
+//# sourceMappingURL=types-CpkEgXwN.js.map

package/dist/types-CpkEgXwN.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types-CpkEgXwN.js","names":["printNodeZodToTs"],"sources":["../src/types.ts"],"sourcesContent":["import {\n  zodToTs,\n  printNode as printNodeZodToTs,\n  createTypeAlias,\n  createAuxiliaryTypeStore\n} from \"zod-to-ts\";\nimport type { ZodType } from \"zod\";\nimport type { ToolSet } from \"ai\";\nimport type { JSONSchema7, JSONSchema7Definition } from \"json-schema\";\n\ninterface ConversionContext {\n  root: JSONSchema7;\n  depth: number;\n  seen: Set<unknown>;\n  maxDepth: number;\n}\n\nconst JS_RESERVED = new Set([\n  \"abstract\",\n  \"arguments\",\n  \"await\",\n  \"boolean\",\n  \"break\",\n  \"byte\",\n  \"case\",\n  \"catch\",\n  \"char\",\n  \"class\",\n  \"const\",\n  \"continue\",\n  \"debugger\",\n  \"default\",\n  \"delete\",\n  \"do\",\n  \"double\",\n  \"else\",\n  \"enum\",\n  \"eval\",\n  \"export\",\n  \"extends\",\n  \"false\",\n  \"final\",\n  \"finally\",\n  \"float\",\n  \"for\",\n  \"function\",\n  \"goto\",\n  \"if\",\n  \"implements\",\n  \"import\",\n  \"in\",\n  \"instanceof\",\n  \"int\",\n  \"interface\",\n  \"let\",\n  \"long\",\n  \"native\",\n  \"new\",\n  \"null\",\n  \"package\",\n  \"private\",\n  \"protected\",\n  \"public\",\n  \"return\",\n  \"short\",\n  \"static\",\n  \"super\",\n  \"switch\",\n  \"synchronized\",\n  \"this\",\n  \"throw\",\n  \"throws\",\n  \"transient\",\n  \"true\",\n  \"try\",\n  \"typeof\",\n  \"undefined\",\n  \"var\",\n  \"void\",\n  \"volatile\",\n  \"while\",\n  \"with\",\n  \"yield\"\n]);\n\n/**\n * Sanitize a tool name into a valid JavaScript identifier.\n * Replaces hyphens, dots, and spaces with `_`, strips other invalid chars,\n * prefixes digit-leading names with `_`, and appends `_` to JS reserved words.\n */\nexport function sanitizeToolName(name: string): string {\n  if (!name) return \"_\";\n\n  // Replace common separators with underscores\n  let sanitized = name.replace(/[-.\\s]/g, \"_\");\n\n  // Strip any remaining non-identifier characters\n  sanitized = sanitized.replace(/[^a-zA-Z0-9_$]/g, \"\");\n\n  if (!sanitized) return \"_\";\n\n  // Prefix with _ if starts with a digit\n  if (/^[0-9]/.test(sanitized)) {\n    sanitized = \"_\" + sanitized;\n  }\n\n  // Append _ to reserved words\n  if (JS_RESERVED.has(sanitized)) {\n    sanitized = sanitized + \"_\";\n  }\n\n  return sanitized;\n}\n\nfunction toCamelCase(str: string) {\n  return str\n    .replace(/_([a-z])/g, (_, letter) => letter.toUpperCase())\n    .replace(/^[a-z]/, (letter) => letter.toUpperCase());\n}\n\n/**\n * Extract field descriptions from a schema and format as @param lines.\n * Returns an array of `@param input.fieldName - description` lines.\n */\nfunction extractParamDescriptions(schema: unknown): string[] {\n  const descriptions = extractDescriptions(schema);\n  return Object.entries(descriptions).map(\n    ([fieldName, desc]) => `@param input.${fieldName} - ${desc}`\n  );\n}\n\nexport interface ToolDescriptor {\n  description?: string;\n  inputSchema: ZodType;\n  outputSchema?: ZodType;\n  execute?: (args: unknown) => Promise<unknown>;\n}\n\nexport type ToolDescriptors = Record<string, ToolDescriptor>;\n\n/**\n * Check if a value is a Zod schema (has _zod property).\n */\nfunction isZodSchema(value: unknown): value is ZodType {\n  return (\n    value !== null &&\n    typeof value === \"object\" &&\n    \"_zod\" in value &&\n    (value as { _zod?: unknown })._zod !== undefined\n  );\n}\n\n/**\n * Check if a value is an AI SDK jsonSchema wrapper.\n * The jsonSchema wrapper has a [Symbol] with jsonSchema property.\n */\nfunction isJsonSchemaWrapper(\n  value: unknown\n): value is { jsonSchema: JSONSchema7 } {\n  if (value === null || typeof value !== \"object\") return false;\n\n  // AI SDK jsonSchema wrapper stores data in a symbol property\n  // but also exposes jsonSchema directly in some versions\n  if (\"jsonSchema\" in value) {\n    return true;\n  }\n\n  // Check for symbol-based storage (AI SDK internal)\n  const symbols = Object.getOwnPropertySymbols(value);\n  for (const sym of symbols) {\n    const symValue = (value as Record<symbol, unknown>)[sym];\n    if (symValue && typeof symValue === \"object\" && \"jsonSchema\" in symValue) {\n      return true;\n    }\n  }\n\n  return false;\n}\n\n/**\n * Extract JSON schema from an AI SDK jsonSchema wrapper.\n */\nfunction extractJsonSchema(wrapper: unknown): JSONSchema7 | null {\n  if (wrapper === null || typeof wrapper !== \"object\") return null;\n\n  // Direct property access\n  if (\"jsonSchema\" in wrapper) {\n    return (wrapper as { jsonSchema: JSONSchema7 }).jsonSchema;\n  }\n\n  // Symbol-based storage\n  const symbols = Object.getOwnPropertySymbols(wrapper);\n  for (const sym of symbols) {\n    const symValue = (wrapper as Record<symbol, unknown>)[sym];\n    if (symValue && typeof symValue === \"object\" && \"jsonSchema\" in symValue) {\n      return (symValue as { jsonSchema: JSONSchema7 }).jsonSchema;\n    }\n  }\n\n  return null;\n}\n\n/**\n * Check if a property name needs quoting in TypeScript.\n */\nfunction needsQuotes(name: string): boolean {\n  // Valid JS identifier: starts with letter, $, or _, followed by letters, digits, $, _\n  return !/^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(name);\n}\n\n/**\n * Escape a character as a unicode escape sequence if it is a control character.\n */\nfunction escapeControlChar(ch: string): string {\n  const code = ch.charCodeAt(0);\n  if (code <= 0x1f || code === 0x7f) {\n    return \"\\\\u\" + code.toString(16).padStart(4, \"0\");\n  }\n  return ch;\n}\n\n/**\n * Quote a property name if needed.\n * Escapes backslashes, quotes, and control characters.\n */\nfunction quoteProp(name: string): string {\n  if (needsQuotes(name)) {\n    let escaped = \"\";\n    for (const ch of name) {\n      if (ch === \"\\\\\") escaped += \"\\\\\\\\\";\n      else if (ch === '\"') escaped += '\\\\\"';\n      else if (ch === \"\\n\") escaped += \"\\\\n\";\n      else if (ch === \"\\r\") escaped += \"\\\\r\";\n      else if (ch === \"\\t\") escaped += \"\\\\t\";\n      else if (ch === \"\\u2028\") escaped += \"\\\\u2028\";\n      else if (ch === \"\\u2029\") escaped += \"\\\\u2029\";\n      else escaped += escapeControlChar(ch);\n    }\n    return `\"${escaped}\"`;\n  }\n  return name;\n}\n\n/**\n * Escape a string for use inside a double-quoted TypeScript string literal.\n * Handles backslashes, quotes, newlines, control characters, and line/paragraph separators.\n */\nfunction escapeStringLiteral(s: string): string {\n  let out = \"\";\n  for (const ch of s) {\n    if (ch === \"\\\\\") out += \"\\\\\\\\\";\n    else if (ch === '\"') out += '\\\\\"';\n    else if (ch === \"\\n\") out += \"\\\\n\";\n    else if (ch === \"\\r\") out += \"\\\\r\";\n    else if (ch === \"\\t\") out += \"\\\\t\";\n    else if (ch === \"\\u2028\") out += \"\\\\u2028\";\n    else if (ch === \"\\u2029\") out += \"\\\\u2029\";\n    else out += escapeControlChar(ch);\n  }\n  return out;\n}\n\n/**\n * Escape a string for use inside a JSDoc comment.\n * Prevents premature comment closure from star-slash sequences.\n */\nfunction escapeJsDoc(text: string): string {\n  return text.replace(/\\*\\//g, \"*\\\\/\");\n}\n\n/**\n * Resolve an internal JSON Pointer $ref (e.g. #/definitions/Foo) against the root schema.\n * Returns null for external URLs or unresolvable paths.\n */\nfunction resolveRef(\n  ref: string,\n  root: JSONSchema7\n): JSONSchema7Definition | null {\n  // \"#\" is a valid self-reference to the root schema\n  if (ref === \"#\") return root;\n\n  if (!ref.startsWith(\"#/\")) return null;\n\n  const segments = ref\n    .slice(2)\n    .split(\"/\")\n    .map((s) => s.replace(/~1/g, \"/\").replace(/~0/g, \"~\"));\n\n  let current: unknown = root;\n  for (const seg of segments) {\n    if (current === null || typeof current !== \"object\") return null;\n    current = (current as Record<string, unknown>)[seg];\n    if (current === undefined) return null;\n  }\n\n  // Allow both object schemas and boolean schemas (true = any, false = never)\n  if (typeof current === \"boolean\") return current;\n  if (current === null || typeof current !== \"object\") return null;\n  return current as JSONSchema7;\n}\n\n/**\n * Convert a JSON Schema to a TypeScript type string.\n * This is a direct conversion without going through Zod.\n */\nfunction jsonSchemaToTypeString(\n  schema: JSONSchema7Definition,\n  indent: string,\n  ctx: ConversionContext\n): string {\n  // Handle boolean schemas\n  if (typeof schema === \"boolean\") {\n    return schema ? \"unknown\" : \"never\";\n  }\n\n  // Depth guard\n  if (ctx.depth >= ctx.maxDepth) return \"unknown\";\n\n  // Circular reference guard\n  if (ctx.seen.has(schema)) return \"unknown\";\n\n  const nextCtx: ConversionContext = {\n    ...ctx,\n    depth: ctx.depth + 1,\n    seen: new Set([...ctx.seen, schema])\n  };\n\n  // Handle $ref\n  if (schema.$ref) {\n    const resolved = resolveRef(schema.$ref, ctx.root);\n    if (!resolved) return \"unknown\";\n    return applyNullable(\n      jsonSchemaToTypeString(resolved, indent, nextCtx),\n      schema\n    );\n  }\n\n  // Handle anyOf/oneOf (union types)\n  if (schema.anyOf) {\n    const types = schema.anyOf.map((s) =>\n      jsonSchemaToTypeString(s, indent, nextCtx)\n    );\n    return applyNullable(types.join(\" | \"), schema);\n  }\n  if (schema.oneOf) {\n    const types = schema.oneOf.map((s) =>\n      jsonSchemaToTypeString(s, indent, nextCtx)\n    );\n    return applyNullable(types.join(\" | \"), schema);\n  }\n\n  // Handle allOf (intersection types)\n  if (schema.allOf) {\n    const types = schema.allOf.map((s) =>\n      jsonSchemaToTypeString(s, indent, nextCtx)\n    );\n    return applyNullable(types.join(\" & \"), schema);\n  }\n\n  // Handle enum\n  if (schema.enum) {\n    if (schema.enum.length === 0) return \"never\";\n    const result = schema.enum\n      .map((v) => {\n        if (v === null) return \"null\";\n        if (typeof v === \"string\") return '\"' + escapeStringLiteral(v) + '\"';\n        if (typeof v === \"object\") return JSON.stringify(v) ?? \"unknown\";\n        return String(v);\n      })\n      .join(\" | \");\n    return applyNullable(result, schema);\n  }\n\n  // Handle const\n  if (schema.const !== undefined) {\n    const result =\n      schema.const === null\n        ? \"null\"\n        : typeof schema.const === \"string\"\n          ? '\"' + escapeStringLiteral(schema.const) + '\"'\n          : typeof schema.const === \"object\"\n            ? (JSON.stringify(schema.const) ?? \"unknown\")\n            : String(schema.const);\n    return applyNullable(result, schema);\n  }\n\n  // Handle type\n  const type = schema.type;\n\n  if (type === \"string\") return applyNullable(\"string\", schema);\n  if (type === \"number\" || type === \"integer\")\n    return applyNullable(\"number\", schema);\n  if (type === \"boolean\") return applyNullable(\"boolean\", schema);\n  if (type === \"null\") return \"null\";\n\n  if (type === \"array\") {\n    // Tuple support: prefixItems (JSON Schema 2020-12)\n    const prefixItems = (schema as Record<string, unknown>)\n      .prefixItems as JSONSchema7Definition[];\n    if (Array.isArray(prefixItems)) {\n      const types = prefixItems.map((s) =>\n        jsonSchemaToTypeString(s, indent, nextCtx)\n      );\n      return applyNullable(`[${types.join(\", \")}]`, schema);\n    }\n\n    // Tuple support: items as array (draft-07)\n    if (Array.isArray(schema.items)) {\n      const types = schema.items.map((s) =>\n        jsonSchemaToTypeString(s, indent, nextCtx)\n      );\n      return applyNullable(`[${types.join(\", \")}]`, schema);\n    }\n\n    if (schema.items) {\n      const itemType = jsonSchemaToTypeString(schema.items, indent, nextCtx);\n      return applyNullable(`${itemType}[]`, schema);\n    }\n    return applyNullable(\"unknown[]\", schema);\n  }\n\n  if (type === \"object\" || schema.properties) {\n    const props = schema.properties || {};\n    const required = new Set(schema.required || []);\n    const lines: string[] = [];\n\n    for (const [propName, propSchema] of Object.entries(props)) {\n      if (typeof propSchema === \"boolean\") {\n        const boolType = propSchema ? \"unknown\" : \"never\";\n        const optionalMark = required.has(propName) ? \"\" : \"?\";\n        lines.push(\n          `${indent}    ${quoteProp(propName)}${optionalMark}: ${boolType};`\n        );\n        continue;\n      }\n\n      const isRequired = required.has(propName);\n      const propType = jsonSchemaToTypeString(\n        propSchema,\n        indent + \"    \",\n        nextCtx\n      );\n      const desc = propSchema.description;\n      const format = propSchema.format;\n\n      if (desc || format) {\n        const descText = desc\n          ? escapeJsDoc(desc.replace(/\\r?\\n/g, \" \"))\n          : undefined;\n        const formatTag = format ? `@format ${escapeJsDoc(format)}` : undefined;\n\n        if (descText && formatTag) {\n          // Multi-line JSDoc when both description and format are present\n          lines.push(`${indent}    /**`);\n          lines.push(`${indent}     * ${descText}`);\n          lines.push(`${indent}     * ${formatTag}`);\n          lines.push(`${indent}     */`);\n        } else {\n          lines.push(`${indent}    /** ${descText ?? formatTag} */`);\n        }\n      }\n\n      const quotedName = quoteProp(propName);\n      const optionalMark = isRequired ? \"\" : \"?\";\n      lines.push(`${indent}    ${quotedName}${optionalMark}: ${propType};`);\n    }\n\n    // Handle additionalProperties\n    // NOTE: In TypeScript, an index signature [key: string]: T requires all\n    // named properties to be assignable to T. If any named property has an\n    // incompatible type, the generated type is invalid. We emit it anyway\n    // since it's more informative for LLMs consuming these types.\n    if (schema.additionalProperties) {\n      const valueType =\n        schema.additionalProperties === true\n          ? \"unknown\"\n          : jsonSchemaToTypeString(\n              schema.additionalProperties,\n              indent + \"    \",\n              nextCtx\n            );\n      lines.push(`${indent}    [key: string]: ${valueType};`);\n    }\n\n    if (lines.length === 0) {\n      // additionalProperties: false means no keys allowed → empty object\n      if (schema.additionalProperties === false) {\n        return applyNullable(\"{}\", schema);\n      }\n      return applyNullable(\"Record<string, unknown>\", schema);\n    }\n\n    const result = `{\\n${lines.join(\"\\n\")}\\n${indent}}`;\n    return applyNullable(result, schema);\n  }\n\n  // Handle array of types (e.g., [\"string\", \"null\"])\n  if (Array.isArray(type)) {\n    const types = type.map((t) => {\n      if (t === \"string\") return \"string\";\n      if (t === \"number\" || t === \"integer\") return \"number\";\n      if (t === \"boolean\") return \"boolean\";\n      if (t === \"null\") return \"null\";\n      if (t === \"array\") return \"unknown[]\";\n      if (t === \"object\") return \"Record<string, unknown>\";\n      return \"unknown\";\n    });\n    return applyNullable(types.join(\" | \"), schema);\n  }\n\n  return \"unknown\";\n}\n\n/**\n * Apply OpenAPI 3.0 `nullable: true` to a type result.\n */\nfunction applyNullable(result: string, schema: unknown): string {\n  if (\n    result !== \"unknown\" &&\n    result !== \"never\" &&\n    (schema as Record<string, unknown>)?.nullable === true\n  ) {\n    return `${result} | null`;\n  }\n  return result;\n}\n\n/**\n * Extract field descriptions from a schema.\n * Works with Zod schemas (via .shape) and jsonSchema wrappers (via .properties).\n */\nfunction extractDescriptions(schema: unknown): Record<string, string> {\n  const descriptions: Record<string, string> = {};\n\n  // Try Zod schema shape\n  const shape = (schema as { shape?: Record<string, ZodType> }).shape;\n  if (shape && typeof shape === \"object\") {\n    for (const [fieldName, fieldSchema] of Object.entries(shape)) {\n      const desc = (fieldSchema as { description?: string }).description;\n      if (desc) {\n        descriptions[fieldName] = desc;\n      }\n    }\n    return descriptions;\n  }\n\n  // Try JSON Schema properties (for jsonSchema wrapper)\n  if (isJsonSchemaWrapper(schema)) {\n    const jsonSchema = extractJsonSchema(schema);\n    if (jsonSchema?.properties) {\n      for (const [fieldName, propSchema] of Object.entries(\n        jsonSchema.properties\n      )) {\n        if (\n          propSchema &&\n          typeof propSchema === \"object\" &&\n          propSchema.description\n        ) {\n          descriptions[fieldName] = propSchema.description;\n        }\n      }\n    }\n  }\n\n  return descriptions;\n}\n\n/**\n * Safely convert a schema to TypeScript type string.\n * Handles Zod schemas and AI SDK jsonSchema wrappers.\n * Returns \"unknown\" if the schema cannot be represented in TypeScript.\n */\nfunction safeSchemaToTs(\n  schema: unknown,\n  typeName: string,\n  auxiliaryTypeStore: ReturnType<typeof createAuxiliaryTypeStore>\n): string {\n  try {\n    // For Zod schemas, use zod-to-ts\n    if (isZodSchema(schema)) {\n      const result = zodToTs(schema, { auxiliaryTypeStore });\n      return printNodeZodToTs(createTypeAlias(result.node, typeName));\n    }\n\n    // For JSON Schema wrapper, convert directly to TypeScript\n    if (isJsonSchemaWrapper(schema)) {\n      const jsonSchema = extractJsonSchema(schema);\n      if (jsonSchema) {\n        const ctx: ConversionContext = {\n          root: jsonSchema,\n          depth: 0,\n          seen: new Set(),\n          maxDepth: 20\n        };\n        const typeBody = jsonSchemaToTypeString(jsonSchema, \"\", ctx);\n        return `type ${typeName} = ${typeBody}`;\n      }\n    }\n\n    return `type ${typeName} = unknown`;\n  } catch {\n    // If the schema cannot be represented, fall back to unknown\n    return `type ${typeName} = unknown`;\n  }\n}\n\n/**\n * Generate TypeScript type definitions from tool descriptors or an AI SDK ToolSet.\n * These types can be included in tool descriptions to help LLMs write correct code.\n */\nexport function generateTypes(tools: ToolDescriptors | ToolSet): string {\n  let availableTools = \"\";\n  let availableTypes = \"\";\n\n  const auxiliaryTypeStore = createAuxiliaryTypeStore();\n\n  for (const [toolName, tool] of Object.entries(tools)) {\n    const safeName = sanitizeToolName(toolName);\n    const camelName = toCamelCase(safeName);\n\n    try {\n      // Handle both our ToolDescriptor and AI SDK Tool types\n      const inputSchema =\n        \"inputSchema\" in tool ? tool.inputSchema : tool.parameters;\n      const outputSchema =\n        \"outputSchema\" in tool ? tool.outputSchema : undefined;\n      const description = tool.description;\n\n      const inputType = safeSchemaToTs(\n        inputSchema,\n        `${camelName}Input`,\n        auxiliaryTypeStore\n      );\n\n      const outputType = outputSchema\n        ? safeSchemaToTs(outputSchema, `${camelName}Output`, auxiliaryTypeStore)\n        : `type ${camelName}Output = unknown`;\n\n      availableTypes += `\\n${inputType.trim()}`;\n      availableTypes += `\\n${outputType.trim()}`;\n\n      // Build JSDoc comment with description and param descriptions\n      const paramDescs = inputSchema\n        ? extractParamDescriptions(inputSchema)\n        : [];\n      const jsdocLines: string[] = [];\n      if (description?.trim()) {\n        jsdocLines.push(escapeJsDoc(description.trim().replace(/\\r?\\n/g, \" \")));\n      } else {\n        jsdocLines.push(escapeJsDoc(toolName));\n      }\n      for (const pd of paramDescs) {\n        jsdocLines.push(escapeJsDoc(pd.replace(/\\r?\\n/g, \" \")));\n      }\n\n      const jsdocBody = jsdocLines.map((l) => `\\t * ${l}`).join(\"\\n\");\n      availableTools += `\\n\\t/**\\n${jsdocBody}\\n\\t */`;\n      availableTools += `\\n\\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;\n      availableTools += \"\\n\";\n    } catch {\n      // One bad tool should not break the others — emit unknown types\n      availableTypes += `\\ntype ${camelName}Input = unknown`;\n      availableTypes += `\\ntype ${camelName}Output = unknown`;\n\n      availableTools += `\\n\\t/**\\n\\t * ${escapeJsDoc(toolName)}\\n\\t */`;\n      availableTools += `\\n\\t${safeName}: (input: ${camelName}Input) => Promise<${camelName}Output>;`;\n      availableTools += \"\\n\";\n    }\n  }\n\n  availableTools = `\\ndeclare const codemode: {${availableTools}}`;\n\n  return `\n${availableTypes}\n${availableTools}\n  `.trim();\n}\n"],"mappings":";;;AAiBA,MAAM,cAAc,IAAI,IAAI;CAC1B;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;;;;;;AAOF,SAAgB,iBAAiB,MAAsB;AACrD,KAAI,CAAC,KAAM,QAAO;CAGlB,IAAI,YAAY,KAAK,QAAQ,WAAW,IAAI;AAG5C,aAAY,UAAU,QAAQ,mBAAmB,GAAG;AAEpD,KAAI,CAAC,UAAW,QAAO;AAGvB,KAAI,SAAS,KAAK,UAAU,CAC1B,aAAY,MAAM;AAIpB,KAAI,YAAY,IAAI,UAAU,CAC5B,aAAY,YAAY;AAG1B,QAAO;;AAGT,SAAS,YAAY,KAAa;AAChC,QAAO,IACJ,QAAQ,cAAc,GAAG,WAAW,OAAO,aAAa,CAAC,CACzD,QAAQ,WAAW,WAAW,OAAO,aAAa,CAAC;;;;;;AAOxD,SAAS,yBAAyB,QAA2B;CAC3D,MAAM,eAAe,oBAAoB,OAAO;AAChD,QAAO,OAAO,QAAQ,aAAa,CAAC,KACjC,CAAC,WAAW,UAAU,gBAAgB,UAAU,KAAK,OACvD;;;;;AAeH,SAAS,YAAY,OAAkC;AACrD,QACE,UAAU,QACV,OAAO,UAAU,YACjB,UAAU,SACT,MAA6B,SAAS;;;;;;AAQ3C,SAAS,oBACP,OACsC;AACtC,KAAI,UAAU,QAAQ,OAAO,UAAU,SAAU,QAAO;AAIxD,KAAI,gBAAgB,MAClB,QAAO;CAIT,MAAM,UAAU,OAAO,sBAAsB,MAAM;AACnD,MAAK,MAAM,OAAO,SAAS;EACzB,MAAM,WAAY,MAAkC;AACpD,MAAI,YAAY,OAAO,aAAa,YAAY,gBAAgB,SAC9D,QAAO;;AAIX,QAAO;;;;;AAMT,SAAS,kBAAkB,SAAsC;AAC/D,KAAI,YAAY,QAAQ,OAAO,YAAY,SAAU,QAAO;AAG5D,KAAI,gBAAgB,QAClB,QAAQ,QAAwC;CAIlD,MAAM,UAAU,OAAO,sBAAsB,QAAQ;AACrD,MAAK,MAAM,OAAO,SAAS;EACzB,MAAM,WAAY,QAAoC;AACtD,MAAI,YAAY,OAAO,aAAa,YAAY,gBAAgB,SAC9D,QAAQ,SAAyC;;AAIrD,QAAO;;;;;AAMT,SAAS,YAAY,MAAuB;AAE1C,QAAO,CAAC,6BAA6B,KAAK,KAAK;;;;;AAMjD,SAAS,kBAAkB,IAAoB;CAC7C,MAAM,OAAO,GAAG,WAAW,EAAE;AAC7B,KAAI,QAAQ,MAAQ,SAAS,IAC3B,QAAO,QAAQ,KAAK,SAAS,GAAG,CAAC,SAAS,GAAG,IAAI;AAEnD,QAAO;;;;;;AAOT,SAAS,UAAU,MAAsB;AACvC,KAAI,YAAY,KAAK,EAAE;EACrB,IAAI,UAAU;AACd,OAAK,MAAM,MAAM,KACf,KAAI,OAAO,KAAM,YAAW;WACnB,OAAO,KAAK,YAAW;WACvB,OAAO,KAAM,YAAW;WACxB,OAAO,KAAM,YAAW;WACxB,OAAO,IAAM,YAAW;WACxB,OAAO,SAAU,YAAW;WAC5B,OAAO,SAAU,YAAW;MAChC,YAAW,kBAAkB,GAAG;AAEvC,SAAO,IAAI,QAAQ;;AAErB,QAAO;;;;;;AAOT,SAAS,oBAAoB,GAAmB;CAC9C,IAAI,MAAM;AACV,MAAK,MAAM,MAAM,EACf,KAAI,OAAO,KAAM,QAAO;UACf,OAAO,KAAK,QAAO;UACnB,OAAO,KAAM,QAAO;UACpB,OAAO,KAAM,QAAO;UACpB,OAAO,IAAM,QAAO;UACpB,OAAO,SAAU,QAAO;UACxB,OAAO,SAAU,QAAO;KAC5B,QAAO,kBAAkB,GAAG;AAEnC,QAAO;;;;;;AAOT,SAAS,YAAY,MAAsB;AACzC,QAAO,KAAK,QAAQ,SAAS,OAAO;;;;;;AAOtC,SAAS,WACP,KACA,MAC8B;AAE9B,KAAI,QAAQ,IAAK,QAAO;AAExB,KAAI,CAAC,IAAI,WAAW,KAAK,CAAE,QAAO;CAElC,MAAM,WAAW,IACd,MAAM,EAAE,CACR,MAAM,IAAI,CACV,KAAK,MAAM,EAAE,QAAQ,OAAO,IAAI,CAAC,QAAQ,OAAO,IAAI,CAAC;CAExD,IAAI,UAAmB;AACvB,MAAK,MAAM,OAAO,UAAU;AAC1B,MAAI,YAAY,QAAQ,OAAO,YAAY,SAAU,QAAO;AAC5D,YAAW,QAAoC;AAC/C,MAAI,YAAY,OAAW,QAAO;;AAIpC,KAAI,OAAO,YAAY,UAAW,QAAO;AACzC,KAAI,YAAY,QAAQ,OAAO,YAAY,SAAU,QAAO;AAC5D,QAAO;;;;;;AAOT,SAAS,uBACP,QACA,QACA,KACQ;AAER,KAAI,OAAO,WAAW,UACpB,QAAO,SAAS,YAAY;AAI9B,KAAI,IAAI,SAAS,IAAI,SAAU,QAAO;AAGtC,KAAI,IAAI,KAAK,IAAI,OAAO,CAAE,QAAO;CAEjC,MAAM,UAA6B;EACjC,GAAG;EACH,OAAO,IAAI,QAAQ;EACnB,MAAM,IAAI,IAAI,CAAC,GAAG,IAAI,MAAM,OAAO,CAAC;EACrC;AAGD,KAAI,OAAO,MAAM;EACf,MAAM,WAAW,WAAW,OAAO,MAAM,IAAI,KAAK;AAClD,MAAI,CAAC,SAAU,QAAO;AACtB,SAAO,cACL,uBAAuB,UAAU,QAAQ,QAAQ,EACjD,OACD;;AAIH,KAAI,OAAO,MAIT,QAAO,cAHO,OAAO,MAAM,KAAK,MAC9B,uBAAuB,GAAG,QAAQ,QAAQ,CAC3C,CAC0B,KAAK,MAAM,EAAE,OAAO;AAEjD,KAAI,OAAO,MAIT,QAAO,cAHO,OAAO,MAAM,KAAK,MAC9B,uBAAuB,GAAG,QAAQ,QAAQ,CAC3C,CAC0B,KAAK,MAAM,EAAE,OAAO;AAIjD,KAAI,OAAO,MAIT,QAAO,cAHO,OAAO,MAAM,KAAK,MAC9B,uBAAuB,GAAG,QAAQ,QAAQ,CAC3C,CAC0B,KAAK,MAAM,EAAE,OAAO;AAIjD,KAAI,OAAO,MAAM;AACf,MAAI,OAAO,KAAK,WAAW,EAAG,QAAO;AASrC,SAAO,cARQ,OAAO,KACnB,KAAK,MAAM;AACV,OAAI,MAAM,KAAM,QAAO;AACvB,OAAI,OAAO,MAAM,SAAU,QAAO,OAAM,oBAAoB,EAAE,GAAG;AACjE,OAAI,OAAO,MAAM,SAAU,QAAO,KAAK,UAAU,EAAE,IAAI;AACvD,UAAO,OAAO,EAAE;IAChB,CACD,KAAK,MAAM,EACe,OAAO;;AAItC,KAAI,OAAO,UAAU,OASnB,QAAO,cAPL,OAAO,UAAU,OACb,SACA,OAAO,OAAO,UAAU,WACtB,OAAM,oBAAoB,OAAO,MAAM,GAAG,OAC1C,OAAO,OAAO,UAAU,WACrB,KAAK,UAAU,OAAO,MAAM,IAAI,YACjC,OAAO,OAAO,MAAM,EACD,OAAO;CAItC,MAAM,OAAO,OAAO;AAEpB,KAAI,SAAS,SAAU,QAAO,cAAc,UAAU,OAAO;AAC7D,KAAI,SAAS,YAAY,SAAS,UAChC,QAAO,cAAc,UAAU,OAAO;AACxC,KAAI,SAAS,UAAW,QAAO,cAAc,WAAW,OAAO;AAC/D,KAAI,SAAS,OAAQ,QAAO;AAE5B,KAAI,SAAS,SAAS;EAEpB,MAAM,cAAe,OAClB;AACH,MAAI,MAAM,QAAQ,YAAY,CAI5B,QAAO,cAAc,IAHP,YAAY,KAAK,MAC7B,uBAAuB,GAAG,QAAQ,QAAQ,CAC3C,CAC8B,KAAK,KAAK,CAAC,IAAI,OAAO;AAIvD,MAAI,MAAM,QAAQ,OAAO,MAAM,CAI7B,QAAO,cAAc,IAHP,OAAO,MAAM,KAAK,MAC9B,uBAAuB,GAAG,QAAQ,QAAQ,CAC3C,CAC8B,KAAK,KAAK,CAAC,IAAI,OAAO;AAGvD,MAAI,OAAO,MAET,QAAO,cAAc,GADJ,uBAAuB,OAAO,OAAO,QAAQ,QAAQ,CACrC,KAAK,OAAO;AAE/C,SAAO,cAAc,aAAa,OAAO;;AAG3C,KAAI,SAAS,YAAY,OAAO,YAAY;EAC1C,MAAM,QAAQ,OAAO,cAAc,EAAE;EACrC,MAAM,WAAW,IAAI,IAAI,OAAO,YAAY,EAAE,CAAC;EAC/C,MAAM,QAAkB,EAAE;AAE1B,OAAK,MAAM,CAAC,UAAU,eAAe,OAAO,QAAQ,MAAM,EAAE;AAC1D,OAAI,OAAO,eAAe,WAAW;IACnC,MAAM,WAAW,aAAa,YAAY;IAC1C,MAAM,eAAe,SAAS,IAAI,SAAS,GAAG,KAAK;AACnD,UAAM,KACJ,GAAG,OAAO,MAAM,UAAU,SAAS,GAAG,aAAa,IAAI,SAAS,GACjE;AACD;;GAGF,MAAM,aAAa,SAAS,IAAI,SAAS;GACzC,MAAM,WAAW,uBACf,YACA,SAAS,QACT,QACD;GACD,MAAM,OAAO,WAAW;GACxB,MAAM,SAAS,WAAW;AAE1B,OAAI,QAAQ,QAAQ;IAClB,MAAM,WAAW,OACb,YAAY,KAAK,QAAQ,UAAU,IAAI,CAAC,GACxC;IACJ,MAAM,YAAY,SAAS,WAAW,YAAY,OAAO,KAAK;AAE9D,QAAI,YAAY,WAAW;AAEzB,WAAM,KAAK,GAAG,OAAO,SAAS;AAC9B,WAAM,KAAK,GAAG,OAAO,SAAS,WAAW;AACzC,WAAM,KAAK,GAAG,OAAO,SAAS,YAAY;AAC1C,WAAM,KAAK,GAAG,OAAO,SAAS;UAE9B,OAAM,KAAK,GAAG,OAAO,UAAU,YAAY,UAAU,KAAK;;GAI9D,MAAM,aAAa,UAAU,SAAS;GACtC,MAAM,eAAe,aAAa,KAAK;AACvC,SAAM,KAAK,GAAG,OAAO,MAAM,aAAa,aAAa,IAAI,SAAS,GAAG;;AAQvE,MAAI,OAAO,sBAAsB;GAC/B,MAAM,YACJ,OAAO,yBAAyB,OAC5B,YACA,uBACE,OAAO,sBACP,SAAS,QACT,QACD;AACP,SAAM,KAAK,GAAG,OAAO,qBAAqB,UAAU,GAAG;;AAGzD,MAAI,MAAM,WAAW,GAAG;AAEtB,OAAI,OAAO,yBAAyB,MAClC,QAAO,cAAc,MAAM,OAAO;AAEpC,UAAO,cAAc,2BAA2B,OAAO;;AAIzD,SAAO,cADQ,MAAM,MAAM,KAAK,KAAK,CAAC,IAAI,OAAO,IACpB,OAAO;;AAItC,KAAI,MAAM,QAAQ,KAAK,CAUrB,QAAO,cATO,KAAK,KAAK,MAAM;AAC5B,MAAI,MAAM,SAAU,QAAO;AAC3B,MAAI,MAAM,YAAY,MAAM,UAAW,QAAO;AAC9C,MAAI,MAAM,UAAW,QAAO;AAC5B,MAAI,MAAM,OAAQ,QAAO;AACzB,MAAI,MAAM,QAAS,QAAO;AAC1B,MAAI,MAAM,SAAU,QAAO;AAC3B,SAAO;GACP,CACyB,KAAK,MAAM,EAAE,OAAO;AAGjD,QAAO;;;;;AAMT,SAAS,cAAc,QAAgB,QAAyB;AAC9D,KACE,WAAW,aACX,WAAW,WACV,QAAoC,aAAa,KAElD,QAAO,GAAG,OAAO;AAEnB,QAAO;;;;;;AAOT,SAAS,oBAAoB,QAAyC;CACpE,MAAM,eAAuC,EAAE;CAG/C,MAAM,QAAS,OAA+C;AAC9D,KAAI,SAAS,OAAO,UAAU,UAAU;AACtC,OAAK,MAAM,CAAC,WAAW,gBAAgB,OAAO,QAAQ,MAAM,EAAE;GAC5D,MAAM,OAAQ,YAAyC;AACvD,OAAI,KACF,cAAa,aAAa;;AAG9B,SAAO;;AAIT,KAAI,oBAAoB,OAAO,EAAE;EAC/B,MAAM,aAAa,kBAAkB,OAAO;AAC5C,MAAI,YAAY,YACd;QAAK,MAAM,CAAC,WAAW,eAAe,OAAO,QAC3C,WAAW,WACZ,CACC,KACE,cACA,OAAO,eAAe,YACtB,WAAW,YAEX,cAAa,aAAa,WAAW;;;AAM7C,QAAO;;;;;;;AAQT,SAAS,eACP,QACA,UACA,oBACQ;AACR,KAAI;AAEF,MAAI,YAAY,OAAO,CAErB,QAAOA,UAAiB,gBADT,QAAQ,QAAQ,EAAE,oBAAoB,CAAC,CACP,MAAM,SAAS,CAAC;AAIjE,MAAI,oBAAoB,OAAO,EAAE;GAC/B,MAAM,aAAa,kBAAkB,OAAO;AAC5C,OAAI,WAQF,QAAO,QAAQ,SAAS,KADP,uBAAuB,YAAY,IANrB;IAC7B,MAAM;IACN,OAAO;IACP,sBAAM,IAAI,KAAK;IACf,UAAU;IACX,CAC2D;;AAKhE,SAAO,QAAQ,SAAS;SAClB;AAEN,SAAO,QAAQ,SAAS;;;;;;;AAQ5B,SAAgB,cAAc,OAA0C;CACtE,IAAI,iBAAiB;CACrB,IAAI,iBAAiB;CAErB,MAAM,qBAAqB,0BAA0B;AAErD,MAAK,MAAM,CAAC,UAAU,SAAS,OAAO,QAAQ,MAAM,EAAE;EACpD,MAAM,WAAW,iBAAiB,SAAS;EAC3C,MAAM,YAAY,YAAY,SAAS;AAEvC,MAAI;GAEF,MAAM,cACJ,iBAAiB,OAAO,KAAK,cAAc,KAAK;GAClD,MAAM,eACJ,kBAAkB,OAAO,KAAK,eAAe;GAC/C,MAAM,cAAc,KAAK;GAEzB,MAAM,YAAY,eAChB,aACA,GAAG,UAAU,QACb,mBACD;GAED,MAAM,aAAa,eACf,eAAe,cAAc,GAAG,UAAU,SAAS,mBAAmB,GACtE,QAAQ,UAAU;AAEtB,qBAAkB,KAAK,UAAU,MAAM;AACvC,qBAAkB,KAAK,WAAW,MAAM;GAGxC,MAAM,aAAa,cACf,yBAAyB,YAAY,GACrC,EAAE;GACN,MAAM,aAAuB,EAAE;AAC/B,OAAI,aAAa,MAAM,CACrB,YAAW,KAAK,YAAY,YAAY,MAAM,CAAC,QAAQ,UAAU,IAAI,CAAC,CAAC;OAEvE,YAAW,KAAK,YAAY,SAAS,CAAC;AAExC,QAAK,MAAM,MAAM,WACf,YAAW,KAAK,YAAY,GAAG,QAAQ,UAAU,IAAI,CAAC,CAAC;GAGzD,MAAM,YAAY,WAAW,KAAK,MAAM,QAAQ,IAAI,CAAC,KAAK,KAAK;AAC/D,qBAAkB,YAAY,UAAU;AACxC,qBAAkB,OAAO,SAAS,YAAY,UAAU,oBAAoB,UAAU;AACtF,qBAAkB;UACZ;AAEN,qBAAkB,UAAU,UAAU;AACtC,qBAAkB,UAAU,UAAU;AAEtC,qBAAkB,iBAAiB,YAAY,SAAS,CAAC;AACzD,qBAAkB,OAAO,SAAS,YAAY,UAAU,oBAAoB,UAAU;AACtF,qBAAkB;;;AAItB,kBAAiB,8BAA8B,eAAe;AAE9D,QAAO;EACP,eAAe;EACf,eAAe;IACb,MAAM"}

package/e2e/codemode.spec.ts

@@ -0,0 +1,124 @@
+import { test, expect } from "@playwright/test";
+
+/**
+ * E2E tests for @cloudflare/codemode with a real AI binding.
+ *
+ * These verify the full pipeline:
+ *   user prompt → LLM generates code via createCodeTool → DynamicWorkerExecutor
+ *   runs the code in an isolated Worker → tool functions called via RPC → result returned.
+ *
+ * Uses Workers AI (@cf/zai-org/glm-4.7-flash) — no API key needed.
+ */
+
+async function runChat(
+  request: import("@playwright/test").APIRequestContext,
+  baseURL: string,
+  userMessage: string
+): Promise<string> {
+  const res = await request.post(`${baseURL}/run`, {
+    headers: { "Content-Type": "application/json" },
+    data: {
+      messages: [
+        {
+          id: `msg-${crypto.randomUUID()}`,
+          role: "user",
+          parts: [{ type: "text", text: userMessage }]
+        }
+      ]
+    },
+    timeout: 45_000
+  });
+  expect(res.ok()).toBe(true);
+  return res.text();
+}
+
+test.describe("codemode e2e (Workers AI)", () => {
+  test.setTimeout(45_000);
+
+  test("LLM generates and executes code that calls addNumbers tool", async ({
+    request,
+    baseURL
+  }) => {
+    const response = await runChat(
+      request,
+      baseURL!,
+      "What is 17 + 25? Use the codemode tool with the addNumbers function to calculate this."
+    );
+
+    // The response stream should contain the answer 42 somewhere
+    // (either in the tool result or the LLM's text response)
+    expect(response).toContain("42");
+  });
+
+  test("LLM generates and executes code that calls getWeather tool", async ({
+    request,
+    baseURL
+  }) => {
+    const response = await runChat(
+      request,
+      baseURL!,
+      "What is the weather in London? Use the codemode tool with the getWeather function."
+    );
+
+    // The getWeather tool returns { city: "London", temperature: 22, condition: "Sunny" }
+    // The LLM should mention London or the weather data in its response
+    const lower = response.toLowerCase();
+    expect(
+      lower.includes("london") ||
+        lower.includes("22") ||
+        lower.includes("sunny")
+    ).toBe(true);
+  });
+
+  test("LLM generates and executes code that calls listProjects tool", async ({
+    request,
+    baseURL
+  }) => {
+    const response = await runChat(
+      request,
+      baseURL!,
+      "List all projects using the codemode tool with the listProjects function."
+    );
+
+    // listProjects returns Alpha and Beta
+    const lower = response.toLowerCase();
+    expect(lower.includes("alpha") || lower.includes("beta")).toBe(true);
+  });
+
+  test("LLM generates code with multiple tool calls", async ({
+    request,
+    baseURL
+  }) => {
+    const response = await runChat(
+      request,
+      baseURL!,
+      "Using the codemode tool, first get the weather in Paris, then add the numbers 10 and 5. Return both results."
+    );
+
+    // Should contain evidence of both tool calls completing
+    const lower = response.toLowerCase();
+    expect(
+      lower.includes("paris") ||
+        lower.includes("22") ||
+        lower.includes("15") ||
+        lower.includes("sunny")
+    ).toBe(true);
+  });
+
+  test("generateTypes returns valid type definitions", async ({
+    request,
+    baseURL
+  }) => {
+    const res = await request.get(`${baseURL}/types`);
+    expect(res.ok()).toBe(true);
+
+    const data = await res.json();
+    const types = data.types as string;
+
+    expect(types).toContain("declare const codemode");
+    expect(types).toContain("addNumbers");
+    expect(types).toContain("getWeather");
+    expect(types).toContain("createProject");
+    expect(types).toContain("listProjects");
+  });
+});

package/e2e/playwright.config.ts

@@ -0,0 +1,24 @@
+import { defineConfig } from "@playwright/test";
+import { dirname, join } from "node:path";
+import { fileURLToPath } from "node:url";
+
+const PORT = 8798;
+const e2eDir = dirname(fileURLToPath(import.meta.url));
+const configPath = join(e2eDir, "wrangler.jsonc");
+
+export default defineConfig({
+  testDir: e2eDir,
+  testMatch: "*.spec.ts",
+  timeout: 60_000,
+  retries: 2,
+  workers: 1,
+  use: {
+    baseURL: `http://localhost:${PORT}`
+  },
+  webServer: {
+    command: `lsof -ti tcp:${PORT} | xargs kill -9 2>/dev/null; npx wrangler dev --config ${configPath} --port ${PORT}`,
+    port: PORT,
+    reuseExistingServer: !process.env.CI,
+    timeout: 30_000
+  }
+});