Method register

src/node/services/ptc/toolBridge.ts:79–113

* Register all bridgeable tools on the runtime under `mux` namespace. * * Tools receive the runtime's abort signal, which is aborted when: * - The sandbox timeout is exceeded * - runtime.abort() is called (e.g., from the parent's abort signal) * * This ensures nested tool calls are

(runtime: IJSRuntime)

   * not just when the parent stream is cancelled.
   */
  register(runtime: IJSRuntime): void {
    const muxObj: Record<string, (...args: unknown[]) => Promise<unknown>> = {};

    for (const [name, tool] of this.bridgeableTools) {
      // Capture tool for closure
      const boundTool = tool;
      const toolName = name;

      muxObj[name] = async (args: unknown) => {
        // Get the runtime's abort signal - this is aborted on timeout or manual abort
        const abortSignal = runtime.getAbortSignal();

        // Check if already aborted before executing
        if (abortSignal?.aborted) {
          throw new Error("Execution aborted");
        }

        // Validate args against tool's Zod schema
        const validatedArgs = this.validateArgs(toolName, boundTool, args);

        // Execute tool with full options (toolCallId and messages are required by type
        // but not used by most tools - generate synthetic values for sandbox context)
        const result: unknown = await boundTool.execute!(validatedArgs, {
          abortSignal,
          toolCallId: `ptc-${toolName}-${Date.now()}`,
          messages: [],
        });

        // Ensure result is JSON-serializable
        return this.serializeResult(result);
      };
    }

    runtime.registerObject("mux", muxObj);
  }

  private hasExecute(tool: Tool): tool is Tool & { execute: NonNullable<Tool["execute"]> } {
    return typeof tool.execute === "function";