@tanstack/ai-code-mode 0.2.12 → 0.3.0

package/dist/esm/type-generator/json-schema-to-ts.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"json-schema-to-ts.js","sources":["../../../src/type-generator/json-schema-to-ts.ts"],"sourcesContent":["import type { ToolBinding } from '../types'\n\n/**\n * Options for type stub generation\n */\nexport interface TypeGeneratorOptions {\n  /**\n   * Include JSDoc comments with descriptions\n   * @default true\n   */\n  includeDescriptions?: boolean\n}\n\n/**\n * Generate TypeScript type stubs for all tool bindings\n *\n * These stubs are included in the LLM system prompt so it knows\n * the exact type signatures of available tools.\n *\n * Tool names match the actual function names injected into the sandbox.\n */\nexport function generateTypeStubs(\n  bindings: Record<string, ToolBinding>,\n  options: TypeGeneratorOptions = {},\n): string {\n  const { includeDescriptions = true } = options\n\n  const declarations: Array<string> = []\n\n  for (const [name, binding] of Object.entries(bindings)) {\n    const inputTypeName = `${capitalize(name)}Input`\n    const outputTypeName = `${capitalize(name)}Output`\n\n    // Generate input type\n    const inputType = jsonSchemaToTypeScript(binding.inputSchema, inputTypeName)\n    if (inputType.declaration) {\n      declarations.push(inputType.declaration)\n    }\n\n    // Generate output type if present\n    let outputTypeRef = 'unknown'\n    if (binding.outputSchema) {\n      const outputType = jsonSchemaToTypeScript(\n        binding.outputSchema,\n        outputTypeName,\n      )\n      if (outputType.declaration) {\n        declarations.push(outputType.declaration)\n      }\n      outputTypeRef = outputType.name\n    }\n\n    // Generate function declaration matching the actual sandbox function name\n    const description =\n      includeDescriptions && binding.description\n        ? `/** ${binding.description} */\\n`\n        : ''\n\n    declarations.push(\n      `${description}declare function ${name}(input: ${inputType.name}): Promise<${outputTypeRef}>;`,\n    )\n  }\n\n  return declarations.join('\\n\\n')\n}\n\ninterface TypeResult {\n  name: string\n  declaration: string\n}\n\n/**\n * Convert a JSON Schema to a TypeScript type\n *\n * Supports basic types: string, number, boolean, object, array\n */\nexport function jsonSchemaToTypeScript(\n  schema: Record<string, unknown>,\n  typeName: string,\n): TypeResult {\n  const type = schemaToType(schema)\n\n  // For object schemas with properties, create a named interface\n  if (\n    schema.type === 'object' &&\n    schema.properties &&\n    Object.keys(schema.properties).length > 0\n  ) {\n    return {\n      name: typeName,\n      declaration: `interface ${typeName} ${type}`,\n    }\n  }\n\n  // For simple types or empty objects, create a type alias\n  return {\n    name: type,\n    declaration: '',\n  }\n}\n\n/**\n * Convert a JSON Schema to a TypeScript type string\n */\nfunction schemaToType(schema: Record<string, unknown>): string {\n  if (typeof schema !== 'object') {\n    return 'unknown'\n  }\n\n  const schemaType = schema.type\n\n  // Handle basic types\n  if (schemaType === 'string') return 'string'\n  if (schemaType === 'number' || schemaType === 'integer') return 'number'\n  if (schemaType === 'boolean') return 'boolean'\n  if (schemaType === 'null') return 'null'\n\n  // Handle arrays\n  if (schemaType === 'array') {\n    const items = schema.items as Record<string, unknown> | undefined\n    const itemType = items ? schemaToType(items) : 'unknown'\n    return `Array<${itemType}>`\n  }\n\n  // Handle objects with properties\n  if (schemaType === 'object' && schema.properties) {\n    const properties = schema.properties as Record<\n      string,\n      Record<string, unknown>\n    >\n    const required = new Set(\n      (schema.required as Array<string> | undefined) ?? [],\n    )\n\n    const props = Object.entries(properties)\n      .map(([key, propSchema]) => {\n        const optional = required.has(key) ? '' : '?'\n        const propType = schemaToType(propSchema)\n        // Handle property names that need quoting\n        const safeName = /^[a-zA-Z_$][a-zA-Z0-9_$]*$/.test(key)\n          ? key\n          : `\"${key}\"`\n        return `  ${safeName}${optional}: ${propType};`\n      })\n      .join('\\n')\n\n    return `{\\n${props}\\n}`\n  }\n\n  // Handle enums\n  if (schema.enum) {\n    const enumValues = schema.enum as Array<unknown>\n    return enumValues.map((v) => JSON.stringify(v)).join(' | ')\n  }\n\n  // Handle union types (anyOf, oneOf)\n  if (schema.anyOf || schema.oneOf) {\n    const variants = (schema.anyOf || schema.oneOf) as Array<\n      Record<string, unknown>\n    >\n    return variants.map((v) => schemaToType(v)).join(' | ')\n  }\n\n  // Handle type arrays (e.g., [\"string\", \"null\"])\n  if (Array.isArray(schemaType)) {\n    return schemaType\n      .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 'Array<unknown>'\n        if (t === 'object') return 'object'\n        return 'unknown'\n      })\n      .join(' | ')\n  }\n\n  // Fallback for unknown schemas\n  return 'unknown'\n}\n\n/**\n * Capitalize the first letter of a string\n */\nfunction capitalize(str: string): string {\n  return str.charAt(0).toUpperCase() + str.slice(1)\n}\n"],"names":[],"mappings":"AAqBO,SAAS,kBACd,UACA,UAAgC,IACxB;AACR,QAAM,EAAE,sBAAsB,KAAA,IAAS;AAEvC,QAAM,eAA8B,CAAA;AAEpC,aAAW,CAAC,MAAM,OAAO,KAAK,OAAO,QAAQ,QAAQ,GAAG;AACtD,UAAM,gBAAgB,GAAG,WAAW,IAAI,CAAC;AACzC,UAAM,iBAAiB,GAAG,WAAW,IAAI,CAAC;AAG1C,UAAM,YAAY,uBAAuB,QAAQ,aAAa,aAAa;AAC3E,QAAI,UAAU,aAAa;AACzB,mBAAa,KAAK,UAAU,WAAW;AAAA,IACzC;AAGA,QAAI,gBAAgB;AACpB,QAAI,QAAQ,cAAc;AACxB,YAAM,aAAa;AAAA,QACjB,QAAQ;AAAA,QACR;AAAA,MAAA;AAEF,UAAI,WAAW,aAAa;AAC1B,qBAAa,KAAK,WAAW,WAAW;AAAA,MAC1C;AACA,sBAAgB,WAAW;AAAA,IAC7B;AAGA,UAAM,cACJ,uBAAuB,QAAQ,cAC3B,OAAO,QAAQ,WAAW;AAAA,IAC1B;AAEN,iBAAa;AAAA,MACX,GAAG,WAAW,oBAAoB,IAAI,WAAW,UAAU,IAAI,cAAc,aAAa;AAAA,IAAA;AAAA,EAE9F;AAEA,SAAO,aAAa,KAAK,MAAM;AACjC;AAYO,SAAS,uBACd,QACA,UACY;AACZ,QAAM,OAAO,aAAa,MAAM;AAGhC,MACE,OAAO,SAAS,YAChB,OAAO,cACP,OAAO,KAAK,OAAO,UAAU,EAAE,SAAS,GACxC;AACA,WAAO;AAAA,MACL,MAAM;AAAA,MACN,aAAa,aAAa,QAAQ,IAAI,IAAI;AAAA,IAAA;AAAA,EAE9C;AAGA,SAAO;AAAA,IACL,MAAM;AAAA,IACN,aAAa;AAAA,EAAA;AAEjB;AAKA,SAAS,aAAa,QAAyC;AAC7D,MAAI,OAAO,WAAW,UAAU;AAC9B,WAAO;AAAA,EACT;AAEA,QAAM,aAAa,OAAO;AAG1B,MAAI,eAAe,SAAU,QAAO;AACpC,MAAI,eAAe,YAAY,eAAe,UAAW,QAAO;AAChE,MAAI,eAAe,UAAW,QAAO;AACrC,MAAI,eAAe,OAAQ,QAAO;AAGlC,MAAI,eAAe,SAAS;AAC1B,UAAM,QAAQ,OAAO;AACrB,UAAM,WAAW,QAAQ,aAAa,KAAK,IAAI;AAC/C,WAAO,SAAS,QAAQ;AAAA,EAC1B;AAGA,MAAI,eAAe,YAAY,OAAO,YAAY;AAChD,UAAM,aAAa,OAAO;AAI1B,UAAM,WAAW,IAAI;AAAA,MAClB,OAAO,YAA0C,CAAA;AAAA,IAAC;AAGrD,UAAM,QAAQ,OAAO,QAAQ,UAAU,EACpC,IAAI,CAAC,CAAC,KAAK,UAAU,MAAM;AAC1B,YAAM,WAAW,SAAS,IAAI,GAAG,IAAI,KAAK;AAC1C,YAAM,WAAW,aAAa,UAAU;AAExC,YAAM,WAAW,6BAA6B,KAAK,GAAG,IAClD,MACA,IAAI,GAAG;AACX,aAAO,KAAK,QAAQ,GAAG,QAAQ,KAAK,QAAQ;AAAA,IAC9C,CAAC,EACA,KAAK,IAAI;AAEZ,WAAO;AAAA,EAAM,KAAK;AAAA;AAAA,EACpB;AAGA,MAAI,OAAO,MAAM;AACf,UAAM,aAAa,OAAO;AAC1B,WAAO,WAAW,IAAI,CAAC,MAAM,KAAK,UAAU,CAAC,CAAC,EAAE,KAAK,KAAK;AAAA,EAC5D;AAGA,MAAI,OAAO,SAAS,OAAO,OAAO;AAChC,UAAM,WAAY,OAAO,SAAS,OAAO;AAGzC,WAAO,SAAS,IAAI,CAAC,MAAM,aAAa,CAAC,CAAC,EAAE,KAAK,KAAK;AAAA,EACxD;AAGA,MAAI,MAAM,QAAQ,UAAU,GAAG;AAC7B,WAAO,WACJ,IAAI,CAAC,MAAM;AACV,UAAI,MAAM,SAAU,QAAO;AAC3B,UAAI,MAAM,YAAY,MAAM,UAAW,QAAO;AAC9C,UAAI,MAAM,UAAW,QAAO;AAC5B,UAAI,MAAM,OAAQ,QAAO;AACzB,UAAI,MAAM,QAAS,QAAO;AAC1B,UAAI,MAAM,SAAU,QAAO;AAC3B,aAAO;AAAA,IACT,CAAC,EACA,KAAK,KAAK;AAAA,EACf;AAGA,SAAO;AACT;AAKA,SAAS,WAAW,KAAqB;AACvC,SAAO,IAAI,OAAO,CAAC,EAAE,gBAAgB,IAAI,MAAM,CAAC;AAClD;"}

package/dist/esm/types.d.ts

@@ -1,216 +0,0 @@
-import { SchemaInput, ServerTool, ToolExecutionContext } from '@tanstack/ai';
-/**
- * Interface for isolate/sandbox drivers
- * Each runtime environment implements this to provide sandboxed code execution
- */
-export interface IsolateDriver {
-    /**
-     * Create a new isolated execution context with tool bindings
-     */
-    createContext: (config: IsolateConfig) => Promise<IsolateContext>;
-}
-/**
- * Configuration for creating an isolate context
- */
-export interface IsolateConfig {
-    /**
-     * Tools transformed into callable bindings for the sandbox
-     */
-    bindings: Record<string, ToolBinding>;
-    /**
-     * Execution timeout in milliseconds (default: 30000)
-     */
-    timeout?: number;
-    /**
-     * Memory limit in MB (default: 128)
-     */
-    memoryLimit?: number;
-}
-/**
- * Isolated execution context with tool bindings injected
- */
-export interface IsolateContext {
-    /**
-     * Execute generated code and return results
-     */
-    execute: <T = unknown>(code: string) => Promise<ExecutionResult<T>>;
-    /**
-     * Clean up sandbox resources
-     */
-    dispose: () => Promise<void>;
-}
-/**
- * Result of code execution in the sandbox
- */
-export interface ExecutionResult<T = unknown> {
-    /**
-     * Whether execution completed without errors
-     */
-    success: boolean;
-    /**
-     * Return value from the executed code (if successful)
-     */
-    value?: T;
-    /**
-     * Normalized error information (if failed)
-     */
-    error?: NormalizedError;
-    /**
-     * Console output captured during execution
-     */
-    logs?: Array<string>;
-}
-/**
- * Normalized error format for cross-runtime compatibility
- */
-export interface NormalizedError {
-    /**
-     * Error name/type
-     */
-    name: string;
-    /**
-     * Error message
-     */
-    message: string;
-    /**
-     * Stack trace (if available)
-     */
-    stack?: string;
-    /**
-     * Error code (if available)
-     */
-    code?: string;
-}
-/**
- * A tool transformed into a format suitable for sandbox injection
- */
-export interface ToolBinding {
-    /**
-     * Unique tool identifier
-     */
-    name: string;
-    /**
-     * Human-readable description for the LLM
-     */
-    description: string;
-    /**
-     * JSON Schema for tool input parameters
-     */
-    inputSchema: Record<string, unknown>;
-    /**
-     * JSON Schema for tool output (optional)
-     */
-    outputSchema?: Record<string, unknown> | undefined;
-    /**
-     * The execute function that will be injected into the sandbox.
-     * Accepts optional context for emitting custom events.
-     */
-    execute: (args: unknown, context?: ToolExecutionContext) => Promise<unknown>;
-}
-export type { ToolExecutionContext };
-/**
- * Server-side tool types that can be passed to Code Mode.
- *
- * Code Mode executes tools inside a server-managed sandbox. Client tools and
- * bare tool definitions are intentionally excluded because they do not provide
- * a server execution implementation for the sandbox binding.
- */
-export type CodeModeTool = ServerTool<SchemaInput, SchemaInput, string, unknown>;
-/**
- * Configuration for createCodeModeTool
- */
-export interface CodeModeToolConfig {
-    /**
-     * Isolate driver for sandboxed code execution
-     */
-    driver: IsolateDriver;
-    /**
-     * Tools to expose as external_* functions in the sandbox
-     */
-    tools: Array<CodeModeTool>;
-    /**
-     * Execution timeout in milliseconds (default: 30000)
-     */
-    timeout?: number;
-    /**
-     * Memory limit for isolate in MB (default: 128)
-     */
-    memoryLimit?: number;
-    /**
-     * Optional function to get additional bindings dynamically.
-     * Called at execution time (each execute_typescript call) to get current skill bindings.
-     * These are merged with the static external_* bindings.
-     *
-     * @returns Record of skill bindings with skill_ prefix
-     *
-     * @example
-     * ```typescript
-     * getSkillBindings: async () => {
-     *   const skills = await storage.loadAll()
-     *   return skillsToBindings(skills, 'skill_')
-     * }
-     * ```
-     */
-    getSkillBindings?: () => Promise<Record<string, ToolBinding>>;
-    /**
-     * Optional escape hatch to swap out the TypeScript-stripping step.
-     *
-     * Receives the raw model-generated code and must return runnable JavaScript
-     * with all TypeScript syntax removed. Defaults to the built-in
-     * {@link stripTypeScript}, which uses sucrase and is safe to bundle for
-     * browsers and edge runtimes (Cloudflare Workers/Pages etc.).
-     *
-     * Provide your own only to trade the edge-safe default for a faster
-     * Node-only transpiler. A custom transpiler MUST tolerate top-level `return`
-     * and `await` in its input (the default wraps the code in an async function
-     * internally to allow this).
-     *
-     * NOTE: This only affects `createCodeModeTool`. The skills helpers
-     * (`skillsToTools`, `codeModeWithSkills` in `@tanstack/ai-code-mode-skills`)
-     * call the exported `stripTypeScript` directly, so they ignore this hook — but
-     * they still get the edge-safe sucrase default, so #487 is fixed for them too;
-     * they just can't be pointed at a different transpiler.
-     *
-     * @example
-     * ```typescript
-     * // Node-only fast path using esbuild (NOT edge-safe — Node only). esbuild
-     * // rejects top-level `return`, so reuse the same async-function wrapper the
-     * // default uses, then slice the body back out. `keepNames: false` stops
-     * // esbuild injecting `__name()` helpers the sandbox can't resolve.
-     * import { transformSync } from 'esbuild'
-     * transpile: (code) => {
-     *   const out = transformSync(`async function _w(){\n${code}\n}`, {
-     *     loader: 'ts',
-     *     keepNames: false,
-     *   }).code
-     *   return out.slice(out.indexOf('{') + 1, out.lastIndexOf('}'))
-     * }
-     * ```
-     */
-    transpile?: (code: string) => string | Promise<string>;
-}
-/**
- * Result returned by the execute_typescript tool
- */
-export interface CodeModeToolResult {
-    /**
-     * Whether execution completed without errors
-     */
-    success: boolean;
-    /**
-     * Return value from the executed code (if successful)
-     */
-    result?: unknown;
-    /**
-     * Console output captured during execution
-     */
-    logs?: Array<string>;
-    /**
-     * Error details if execution failed
-     */
-    error?: {
-        message: string;
-        name?: string | undefined;
-        line?: number | undefined;
-    } | undefined;
-}