calculator.ts

import { z } from 'zod'
import AIFunctionBuilder from '../src'
import NodeExec from '../src/backends/node'
import { anthropic } from '@ai-sdk/anthropic'

// Provide a LLM model
const llm = anthropic.chat('claude-3-5-sonnet-20240620')

// Create a new AI Function Executor
const backend = new NodeExec({ debug: true })
const ai = new AIFunctionBuilder(llm, backend)

// Define the input parameters and output parameters of the function
const parameters = z.object({ a: z.number(), b: z.number() })
const output = z.number()

// Generate the function
const f = await ai.function('add values provided', parameters, output)

// Call the function and log the result
// const result = await f({ a: 1, b: 2 })
// console.log(result)

// Call the function 10 times
for (let i = 0; i < 10; i++) {
  const result = await f({ a: i, b: i + 1 })
  console.log(result)
}