@tanstack/ai-code-mode 0.1.24 → 0.2.1

package/dist/esm/bindings/tool-to-binding.d.ts

@@ -1,3 +1,4 @@
+import { ToolExecutionContext } from '@tanstack/ai';
 import { CodeModeTool, ToolBinding } from '../types.js';
 /**
  * Convert an array of TanStack AI tools to a Record of ToolBindings
@@ -21,4 +22,4 @@ export declare function toolToBinding(tool: CodeModeTool, prefix?: string): Tool
  * @param bindings - Original tool bindings
  * @param emitCustomEvent - Callback to emit custom events to the stream
  */
-export declare function createEventAwareBindings(bindings: Record<string, ToolBinding>, emitCustomEvent: (eventName: string, data: Record<string, any>) => void): Record<string, ToolBinding>;
+export declare function createEventAwareBindings(bindings: Record<string, ToolBinding>, emitCustomEvent: ToolExecutionContext['emitCustomEvent']): Record<string, ToolBinding>;

package/dist/esm/bindings/tool-to-binding.js

@@ -13,21 +13,15 @@ function toolToBinding(tool, prefix = "") {
     properties: {}
   };
   const outputSchema = tool.outputSchema ? convertSchemaToJsonSchema(tool.outputSchema) : void 0;
-  let execute;
-  if ("execute" in tool && typeof tool.execute === "function") {
-    const toolExecute = tool.execute;
-    execute = (args, context) => {
-      return Promise.resolve(toolExecute(args, context));
-    };
-  } else if ("__toolSide" in tool && tool.__toolSide === "definition") {
-    throw new Error(
-      `Tool "${tool.name}" is a ToolDefinition without an execute function. Call .server(fn) to provide an implementation before using with Code Mode.`
-    );
-  } else {
+  if (typeof tool.execute !== "function") {
     throw new Error(
-      `Tool "${tool.name}" does not have an execute function. Code Mode requires tools with implementations.`
+      `Tool "${tool.name}" does not have an execute function. Code Mode requires server tools with implementations.`
     );
   }
+  const toolExecute = tool.execute;
+  const execute = (args, context) => {
+    return Promise.resolve(toolExecute(args, context));
+  };
   return {
     name: `${prefix}${tool.name}`,
     description: tool.description,

package/dist/esm/bindings/tool-to-binding.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-to-binding.js","sources":["../../../src/bindings/tool-to-binding.ts"],"sourcesContent":["import { convertSchemaToJsonSchema } from '@tanstack/ai'\nimport type { ToolExecutionContext } from '@tanstack/ai'\nimport type { CodeModeTool, ToolBinding } from '../types'\n\n/**\n * Convert an array of TanStack AI tools to a Record of ToolBindings\n *\n * @param tools - Array of tools to convert\n * @param prefix - Optional prefix to add to binding names (e.g., 'external_')\n */\nexport function toolsToBindings(\n  tools: Array<CodeModeTool>,\n  prefix: string = '',\n): Record<string, ToolBinding> {\n  const bindings: Record<string, ToolBinding> = {}\n\n  for (const tool of tools) {\n    const bindingName = `${prefix}${tool.name}`\n    bindings[bindingName] = toolToBinding(tool, prefix)\n  }\n\n  return bindings\n}\n\n/**\n * Convert a single TanStack AI tool to a ToolBinding\n *\n * @param tool - Tool to convert\n * @param prefix - Optional prefix to add to binding name (e.g., 'external_')\n * @throws Error if the tool doesn't have an execute function\n */\nexport function toolToBinding(\n  tool: CodeModeTool,\n  prefix: string = '',\n): ToolBinding {\n  // Convert schemas (Zod or Standard Schema) to JSON Schema\n  const inputSchema = convertSchemaToJsonSchema(tool.inputSchema) || {\n    type: 'object',\n    properties: {},\n  }\n\n  const outputSchema = tool.outputSchema\n    ? convertSchemaToJsonSchema(tool.outputSchema)\n    : undefined\n\n  // Get execute function\n  // ServerTool has execute, ToolDefinition (without .server()) does not\n  let execute: (\n    args: unknown,\n    context?: ToolExecutionContext,\n  ) => Promise<unknown>\n\n  if ('execute' in tool && typeof tool.execute === 'function') {\n    const toolExecute = tool.execute\n    execute = (args: unknown, context?: ToolExecutionContext) => {\n      // Pass context to the underlying tool so it can emit custom events\n      return Promise.resolve(toolExecute(args, context))\n    }\n  } else if ('__toolSide' in tool && tool.__toolSide === 'definition') {\n    throw new Error(\n      `Tool \"${tool.name}\" is a ToolDefinition without an execute function. ` +\n        `Call .server(fn) to provide an implementation before using with Code Mode.`,\n    )\n  } else {\n    throw new Error(\n      `Tool \"${tool.name}\" does not have an execute function. ` +\n        `Code Mode requires tools with implementations.`,\n    )\n  }\n\n  return {\n    name: `${prefix}${tool.name}`,\n    description: tool.description,\n    inputSchema,\n    outputSchema,\n    execute,\n  }\n}\n\n/**\n * Create event-aware bindings that emit custom events for each external function call.\n * Wraps each binding's execute function to emit events before and after execution.\n *\n * @param bindings - Original tool bindings\n * @param emitCustomEvent - Callback to emit custom events to the stream\n */\nexport function createEventAwareBindings(\n  bindings: Record<string, ToolBinding>,\n  emitCustomEvent: (eventName: string, data: Record<string, any>) => void,\n): Record<string, ToolBinding> {\n  const wrapped: Record<string, ToolBinding> = {}\n\n  for (const [name, binding] of Object.entries(bindings)) {\n    wrapped[name] = {\n      ...binding,\n      execute: async (args: unknown) => {\n        // Emit call event\n        emitCustomEvent('code_mode:external_call', {\n          function: name,\n          args,\n          timestamp: Date.now(),\n        })\n\n        const startTime = Date.now()\n        try {\n          // Create context for the underlying tool so it can also emit events\n          const toolContext: ToolExecutionContext = { emitCustomEvent }\n          const result = await binding.execute(args, toolContext)\n\n          // Emit result event\n          emitCustomEvent('code_mode:external_result', {\n            function: name,\n            result,\n            duration: Date.now() - startTime,\n          })\n\n          return result\n        } catch (error) {\n          // Emit error event\n          emitCustomEvent('code_mode:external_error', {\n            function: name,\n            error: error instanceof Error ? error.message : String(error),\n            duration: Date.now() - startTime,\n          })\n          throw error\n        }\n      },\n    }\n  }\n\n  return wrapped\n}\n"],"names":[],"mappings":";AAUO,SAAS,gBACd,OACA,SAAiB,IACY;AAC7B,QAAM,WAAwC,CAAA;AAE9C,aAAW,QAAQ,OAAO;AACxB,UAAM,cAAc,GAAG,MAAM,GAAG,KAAK,IAAI;AACzC,aAAS,WAAW,IAAI,cAAc,MAAM,MAAM;AAAA,EACpD;AAEA,SAAO;AACT;AASO,SAAS,cACd,MACA,SAAiB,IACJ;AAEb,QAAM,cAAc,0BAA0B,KAAK,WAAW,KAAK;AAAA,IACjE,MAAM;AAAA,IACN,YAAY,CAAA;AAAA,EAAC;AAGf,QAAM,eAAe,KAAK,eACtB,0BAA0B,KAAK,YAAY,IAC3C;AAIJ,MAAI;AAKJ,MAAI,aAAa,QAAQ,OAAO,KAAK,YAAY,YAAY;AAC3D,UAAM,cAAc,KAAK;AACzB,cAAU,CAAC,MAAe,YAAmC;AAE3D,aAAO,QAAQ,QAAQ,YAAY,MAAM,OAAO,CAAC;AAAA,IACnD;AAAA,EACF,WAAW,gBAAgB,QAAQ,KAAK,eAAe,cAAc;AACnE,UAAM,IAAI;AAAA,MACR,SAAS,KAAK,IAAI;AAAA,IAAA;AAAA,EAGtB,OAAO;AACL,UAAM,IAAI;AAAA,MACR,SAAS,KAAK,IAAI;AAAA,IAAA;AAAA,EAGtB;AAEA,SAAO;AAAA,IACL,MAAM,GAAG,MAAM,GAAG,KAAK,IAAI;AAAA,IAC3B,aAAa,KAAK;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAEJ;AASO,SAAS,yBACd,UACA,iBAC6B;AAC7B,QAAM,UAAuC,CAAA;AAE7C,aAAW,CAAC,MAAM,OAAO,KAAK,OAAO,QAAQ,QAAQ,GAAG;AACtD,YAAQ,IAAI,IAAI;AAAA,MACd,GAAG;AAAA,MACH,SAAS,OAAO,SAAkB;AAEhC,wBAAgB,2BAA2B;AAAA,UACzC,UAAU;AAAA,UACV;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,QAAI,CACrB;AAED,cAAM,YAAY,KAAK,IAAA;AACvB,YAAI;AAEF,gBAAM,cAAoC,EAAE,gBAAA;AAC5C,gBAAM,SAAS,MAAM,QAAQ,QAAQ,MAAM,WAAW;AAGtD,0BAAgB,6BAA6B;AAAA,YAC3C,UAAU;AAAA,YACV;AAAA,YACA,UAAU,KAAK,QAAQ;AAAA,UAAA,CACxB;AAED,iBAAO;AAAA,QACT,SAAS,OAAO;AAEd,0BAAgB,4BAA4B;AAAA,YAC1C,UAAU;AAAA,YACV,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,YAC5D,UAAU,KAAK,QAAQ;AAAA,UAAA,CACxB;AACD,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IAAA;AAAA,EAEJ;AAEA,SAAO;AACT;"}
+{"version":3,"file":"tool-to-binding.js","sources":["../../../src/bindings/tool-to-binding.ts"],"sourcesContent":["import { convertSchemaToJsonSchema } from '@tanstack/ai'\nimport type { ToolExecutionContext } from '@tanstack/ai'\nimport type { CodeModeTool, ToolBinding } from '../types'\n\n/**\n * Convert an array of TanStack AI tools to a Record of ToolBindings\n *\n * @param tools - Array of tools to convert\n * @param prefix - Optional prefix to add to binding names (e.g., 'external_')\n */\nexport function toolsToBindings(\n  tools: Array<CodeModeTool>,\n  prefix: string = '',\n): Record<string, ToolBinding> {\n  const bindings: Record<string, ToolBinding> = {}\n\n  for (const tool of tools) {\n    const bindingName = `${prefix}${tool.name}`\n    bindings[bindingName] = toolToBinding(tool, prefix)\n  }\n\n  return bindings\n}\n\n/**\n * Convert a single TanStack AI tool to a ToolBinding\n *\n * @param tool - Tool to convert\n * @param prefix - Optional prefix to add to binding name (e.g., 'external_')\n * @throws Error if the tool doesn't have an execute function\n */\nexport function toolToBinding(\n  tool: CodeModeTool,\n  prefix: string = '',\n): ToolBinding {\n  // Convert schemas (Zod or Standard Schema) to JSON Schema\n  const inputSchema = convertSchemaToJsonSchema(tool.inputSchema) || {\n    type: 'object',\n    properties: {},\n  }\n\n  const outputSchema = tool.outputSchema\n    ? convertSchemaToJsonSchema(tool.outputSchema)\n    : undefined\n\n  if (typeof tool.execute !== 'function') {\n    throw new Error(\n      `Tool \"${tool.name}\" does not have an execute function. ` +\n        `Code Mode requires server tools with implementations.`,\n    )\n  }\n\n  const toolExecute = tool.execute\n  const execute = (args: unknown, context?: ToolExecutionContext) => {\n    // Pass context to the underlying tool so it can emit custom events\n    return Promise.resolve(toolExecute(args, context))\n  }\n\n  return {\n    name: `${prefix}${tool.name}`,\n    description: tool.description,\n    inputSchema,\n    outputSchema,\n    execute,\n  }\n}\n\n/**\n * Create event-aware bindings that emit custom events for each external function call.\n * Wraps each binding's execute function to emit events before and after execution.\n *\n * @param bindings - Original tool bindings\n * @param emitCustomEvent - Callback to emit custom events to the stream\n */\nexport function createEventAwareBindings(\n  bindings: Record<string, ToolBinding>,\n  emitCustomEvent: ToolExecutionContext['emitCustomEvent'],\n): Record<string, ToolBinding> {\n  const wrapped: Record<string, ToolBinding> = {}\n\n  for (const [name, binding] of Object.entries(bindings)) {\n    wrapped[name] = {\n      ...binding,\n      execute: async (args: unknown) => {\n        // Emit call event\n        emitCustomEvent('code_mode:external_call', {\n          function: name,\n          args,\n          timestamp: Date.now(),\n        })\n\n        const startTime = Date.now()\n        try {\n          // Create context for the underlying tool so it can also emit events\n          const toolContext: ToolExecutionContext = { emitCustomEvent }\n          const result = await binding.execute(args, toolContext)\n\n          // Emit result event\n          emitCustomEvent('code_mode:external_result', {\n            function: name,\n            result,\n            duration: Date.now() - startTime,\n          })\n\n          return result\n        } catch (error) {\n          // Emit error event\n          emitCustomEvent('code_mode:external_error', {\n            function: name,\n            error: error instanceof Error ? error.message : String(error),\n            duration: Date.now() - startTime,\n          })\n          throw error\n        }\n      },\n    }\n  }\n\n  return wrapped\n}\n"],"names":[],"mappings":";AAUO,SAAS,gBACd,OACA,SAAiB,IACY;AAC7B,QAAM,WAAwC,CAAA;AAE9C,aAAW,QAAQ,OAAO;AACxB,UAAM,cAAc,GAAG,MAAM,GAAG,KAAK,IAAI;AACzC,aAAS,WAAW,IAAI,cAAc,MAAM,MAAM;AAAA,EACpD;AAEA,SAAO;AACT;AASO,SAAS,cACd,MACA,SAAiB,IACJ;AAEb,QAAM,cAAc,0BAA0B,KAAK,WAAW,KAAK;AAAA,IACjE,MAAM;AAAA,IACN,YAAY,CAAA;AAAA,EAAC;AAGf,QAAM,eAAe,KAAK,eACtB,0BAA0B,KAAK,YAAY,IAC3C;AAEJ,MAAI,OAAO,KAAK,YAAY,YAAY;AACtC,UAAM,IAAI;AAAA,MACR,SAAS,KAAK,IAAI;AAAA,IAAA;AAAA,EAGtB;AAEA,QAAM,cAAc,KAAK;AACzB,QAAM,UAAU,CAAC,MAAe,YAAmC;AAEjE,WAAO,QAAQ,QAAQ,YAAY,MAAM,OAAO,CAAC;AAAA,EACnD;AAEA,SAAO;AAAA,IACL,MAAM,GAAG,MAAM,GAAG,KAAK,IAAI;AAAA,IAC3B,aAAa,KAAK;AAAA,IAClB;AAAA,IACA;AAAA,IACA;AAAA,EAAA;AAEJ;AASO,SAAS,yBACd,UACA,iBAC6B;AAC7B,QAAM,UAAuC,CAAA;AAE7C,aAAW,CAAC,MAAM,OAAO,KAAK,OAAO,QAAQ,QAAQ,GAAG;AACtD,YAAQ,IAAI,IAAI;AAAA,MACd,GAAG;AAAA,MACH,SAAS,OAAO,SAAkB;AAEhC,wBAAgB,2BAA2B;AAAA,UACzC,UAAU;AAAA,UACV;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,QAAI,CACrB;AAED,cAAM,YAAY,KAAK,IAAA;AACvB,YAAI;AAEF,gBAAM,cAAoC,EAAE,gBAAA;AAC5C,gBAAM,SAAS,MAAM,QAAQ,QAAQ,MAAM,WAAW;AAGtD,0BAAgB,6BAA6B;AAAA,YAC3C,UAAU;AAAA,YACV;AAAA,YACA,UAAU,KAAK,QAAQ;AAAA,UAAA,CACxB;AAED,iBAAO;AAAA,QACT,SAAS,OAAO;AAEd,0BAAgB,4BAA4B;AAAA,YAC1C,UAAU;AAAA,YACV,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,YAC5D,UAAU,KAAK,QAAQ;AAAA,UAAA,CACxB;AACD,gBAAM;AAAA,QACR;AAAA,MACF;AAAA,IAAA;AAAA,EAEJ;AAEA,SAAO;AACT;"}

package/dist/esm/create-code-mode.d.ts

@@ -36,6 +36,6 @@ export declare function createCodeMode(config: CodeModeToolConfig): {
             name: import('zod').ZodOptional<import('zod').ZodString>;
             line: import('zod').ZodOptional<import('zod').ZodNumber>;
         }, import('zod/v4/core').$strip>>;
-    }, import('zod/v4/core').$strip>, "execute_typescript">;
+    }, import('zod/v4/core').$strip>, "execute_typescript", unknown>;
     systemPrompt: string;
 };

package/dist/esm/types.d.ts

@@ -1,4 +1,4 @@
-import { ServerTool, ToolDefinition, ToolExecutionContext } from '@tanstack/ai';
+import { SchemaInput, ServerTool, ToolExecutionContext } from '@tanstack/ai';
 /**
  * Interface for isolate/sandbox drivers
  * Each runtime environment implements this to provide sandboxed code execution
@@ -109,9 +109,13 @@ export interface ToolBinding {
 }
 export type { ToolExecutionContext };
 /**
- * Tool types that can be passed to Code Mode
+ * 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<any, any, any> | ToolDefinition<any, any, any>;
+export type CodeModeTool = ServerTool<SchemaInput, SchemaInput, string, unknown>;
 /**
  * Configuration for createCodeModeTool
  */

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-code-mode",
-  "version": "0.1.24",
+  "version": "0.2.1",
   "description": "Secure TypeScript Code Mode for TanStack AI agents to execute sandboxed tool orchestration programs.",
   "author": "",
   "license": "MIT",
@@ -46,12 +46,12 @@
   },
   "peerDependencies": {
     "zod": "^3.0.0 || ^4.0.0",
-    "@tanstack/ai": "0.23.1"
+    "@tanstack/ai": "0.25.0"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "4.0.14",
     "zod": "^4.2.0",
-    "@tanstack/ai": "0.23.1"
+    "@tanstack/ai": "0.25.0"
   },
   "scripts": {
     "build": "vite build",

package/src/bindings/tool-to-binding.ts

@@ -43,31 +43,19 @@ export function toolToBinding(
     ? convertSchemaToJsonSchema(tool.outputSchema)
     : undefined
 
-  // Get execute function
-  // ServerTool has execute, ToolDefinition (without .server()) does not
-  let execute: (
-    args: unknown,
-    context?: ToolExecutionContext,
-  ) => Promise<unknown>
-
-  if ('execute' in tool && typeof tool.execute === 'function') {
-    const toolExecute = tool.execute
-    execute = (args: unknown, context?: ToolExecutionContext) => {
-      // Pass context to the underlying tool so it can emit custom events
-      return Promise.resolve(toolExecute(args, context))
-    }
-  } else if ('__toolSide' in tool && tool.__toolSide === 'definition') {
-    throw new Error(
-      `Tool "${tool.name}" is a ToolDefinition without an execute function. ` +
-        `Call .server(fn) to provide an implementation before using with Code Mode.`,
-    )
-  } else {
+  if (typeof tool.execute !== 'function') {
     throw new Error(
       `Tool "${tool.name}" does not have an execute function. ` +
-        `Code Mode requires tools with implementations.`,
+        `Code Mode requires server tools with implementations.`,
     )
   }
 
+  const toolExecute = tool.execute
+  const execute = (args: unknown, context?: ToolExecutionContext) => {
+    // Pass context to the underlying tool so it can emit custom events
+    return Promise.resolve(toolExecute(args, context))
+  }
+
   return {
     name: `${prefix}${tool.name}`,
     description: tool.description,
@@ -86,7 +74,7 @@ export function toolToBinding(
  */
 export function createEventAwareBindings(
   bindings: Record<string, ToolBinding>,
-  emitCustomEvent: (eventName: string, data: Record<string, any>) => void,
+  emitCustomEvent: ToolExecutionContext['emitCustomEvent'],
 ): Record<string, ToolBinding> {
   const wrapped: Record<string, ToolBinding> = {}
 

package/src/types.ts

@@ -1,6 +1,6 @@
 import type {
+  SchemaInput,
   ServerTool,
-  ToolDefinition,
   ToolExecutionContext,
 } from '@tanstack/ai'
 
@@ -147,11 +147,13 @@ export type { ToolExecutionContext }
 // ============================================================================
 
 /**
- * Tool types that can be passed to Code Mode
+ * 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<any, any, any>
-  | ToolDefinition<any, any, any>
+export type CodeModeTool = ServerTool<SchemaInput, SchemaInput, string, unknown>
 
 /**
  * Configuration for createCodeModeTool