Function createContext

packages/ai-isolate-node/src/isolate-driver.ts:138–232

(isolateConfig: IsolateConfig)

  return {
    async createContext(isolateConfig: IsolateConfig): Promise<IsolateContext> {
      const memoryLimit = isolateConfig.memoryLimit ?? defaultMemoryLimit
      const timeout = isolateConfig.timeout ?? defaultTimeout

      // Create isolate with memory limit
      const isolate = new ivm.Isolate({ memoryLimit })

      // Create context
      const context = await isolate.createContext()

      // Get reference to global object
      const jail = context.global

      // Set up global reference
      await jail.set('global', jail.derefInto())

      // Set up console.log capture
      const logs: Array<string> = []
      await jail.set(
        '__captureLog',
        new ivm.Reference((msg: string) => {
          logs.push(msg)
        }),
      )

      // Inject console object
      await context.eval(`
        const console = {
          log: (...args) => {
            const msg = args.map(a =>
              typeof a === 'object' ? JSON.stringify(a) : String(a)
            ).join(' ');
            __captureLog.applySync(undefined, [msg]);
          },
          error: (...args) => {
            const msg = 'ERROR: ' + args.map(a =>
              typeof a === 'object' ? JSON.stringify(a) : String(a)
            ).join(' ');
            __captureLog.applySync(undefined, [msg]);
          },
          warn: (...args) => {
            const msg = 'WARN: ' + args.map(a =>
              typeof a === 'object' ? JSON.stringify(a) : String(a)
            ).join(' ');
            __captureLog.applySync(undefined, [msg]);
          },
          info: (...args) => {
            const msg = 'INFO: ' + args.map(a =>
              typeof a === 'object' ? JSON.stringify(a) : String(a)
            ).join(' ');
            __captureLog.applySync(undefined, [msg]);
          },
        };
      `)

      // Inject each tool binding
      for (const [name, binding] of Object.entries(isolateConfig.bindings)) {
        // Create an async Reference that executes the tool