UNPKG

ai-functions

Version:

A powerful TypeScript library for building AI-powered applications with template literals and structured outputs

24 lines 861 B
import { generateObject, streamObject } from '@/index'; import { DEFAULT_PROXY_OPTIONS } from './constants'; import { generatePrompt } from './prompt'; export function createAIFunction(functionName, schema, options = {}) { return async function (input) { const { seed, ...restInput } = input; const prompt = generatePrompt(functionName, restInput); const baseOptions = { schema, prompt, model: options.model || DEFAULT_PROXY_OPTIONS.model, system: options.system, seed: parseInt(seed || '1', 10) // Convert seed to integer }; if (options.stream) { return streamObject(baseOptions); } return generateObject({ ...baseOptions, output: 'single' }); }; } //# sourceMappingURL=create-function.js.map