ai-functions
Version:
A powerful TypeScript library for building AI-powered applications with template literals and structured outputs
24 lines • 861 B
JavaScript
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