Function generateToolWrappers

packages/ai-isolate-cloudflare/src/worker/wrap-code.ts:86–121

(
  tools: Array<ToolSchema>,
  toolResults?: Record<string, ToolResultPayload>,
)

 * inputs contain non-deterministic values (e.g. random UUIDs).
 */
export function generateToolWrappers(
  tools: Array<ToolSchema>,
  toolResults?: Record<string, ToolResultPayload>,
): string {
  const wrappers: Array<string> = []

  for (const tool of tools) {
    assertSafeToolName(tool.name)
    if (toolResults) {
      wrappers.push(`
        async function ${tool.name}(input) {
          const callId = 'tc_' + (__toolCallIdx++);
          const result = __toolResults[callId];
          if (!result) {
            __pendingToolCalls.push({ id: callId, name: '${tool.name}', args: input });
            throw new __ToolCallNeeded(callId);
          }
          if (!result.success) {
            throw new Error(result.error || 'Tool call failed');
          }
          return result.value;
        }
      `)
    } else {
      wrappers.push(`
        async function ${tool.name}(input) {
          const callId = 'tc_' + (__toolCallIdx++);
          __pendingToolCalls.push({ id: callId, name: '${tool.name}', args: input });
          throw new __ToolCallNeeded(callId);
        }
      `)
    }
  }

  return wrappers.join('\n')
}

/**
 * Wrap user code in an async IIFE with tool wrappers