ai-functions/dist/factory/__tests__/list.test.js

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

import { describe, it, expect, beforeEach } from 'vitest';
import { createListFunction } from '../list';
import { openai } from '@ai-sdk/openai';
beforeEach(() => {
    process.env.OPENAI_API_KEY = process.env.OPENAI_API_KEY || 'test-key';
    // Ensure AI gateway is configured for cached responses
    process.env.AI_GATEWAY = process.env.AI_GATEWAY || 'https://api.openai.com/v1';
});
const model = openai(process.env.OPENAI_DEFAULT_MODEL || 'gpt-4o');
describe('createListFunction', () => {
    it('should generate a list of items', async () => {
        const list = createListFunction();
        const result = await list `fun things to do in Miami`({ model });
        expect(result).toBeDefined();
        expect(typeof result).toBe('string');
        expect(result.split('\n').length).toBeGreaterThan(0);
    });
    it('should support async iteration', async () => {
        const list = createListFunction();
        const items = [];
        for await (const item of list `fun things to do in Miami`) {
            items.push(item);
        }
        expect(items.length).toBeGreaterThan(0);
    });
    it('should support configuration options', async () => {
        const list = createListFunction();
        const result = await list `fun things to do in Miami`({ model });
        expect(result).toBeDefined();
        expect(typeof result).toBe('string');
        expect(result.split('\n').length).toBeGreaterThan(0);
    });
    it('should handle empty input', async () => {
        const list = createListFunction();
        const result = await list ``({ model });
        expect(result).toBeDefined();
        expect(typeof result).toBe('string');
    });
    it('should handle concurrent list operations', async () => {
        const list = createListFunction();
        const topics = ['cities', 'foods', 'sports'];
        const tasks = topics.map(topic => list `5 popular ${topic}`({ model, concurrency: 2 }));
        const results = await Promise.all(tasks);
        expect(results).toHaveLength(3);
        results.forEach((result) => {
            expect(result.split('\n').length).toBeGreaterThan(0);
        });
    });
    it('should handle concurrent streaming list operations', async () => {
        const list = createListFunction();
        const topics = ['movies', 'books', 'games'];
        // Configure the list function with options once
        const configuredList = createListFunction({
            model,
            concurrency: 2
        });
        // Create an array of promises that will resolve to arrays of items
        const promises = topics.map(async (topic) => {
            const items = [];
            // Use async iteration directly on the template result
            for await (const item of configuredList `3 popular ${topic}`) {
                items.push(item);
            }
            return items;
        });
        // Wait for all promises to complete
        const results = await Promise.all(promises);
        expect(results).toHaveLength(3);
        results.forEach((items) => {
            expect(items.length).toBeGreaterThan(0);
            expect(items.every((item) => typeof item === 'string')).toBe(true);
        });
    });
    it('should handle errors in concurrent list operations', async () => {
        const list = createListFunction();
        const tasks = [
            list `valid request`({ model }),
            list `trigger error`({ model }),
            list `another valid request`({ model })
        ].map(promise => promise.catch(err => {
            // If any request fails, we'll handle it gracefully
            return 'error occurred';
        }));
        const results = await Promise.all(tasks);
        expect(results).toHaveLength(3);
        results.forEach(result => {
            expect(typeof result).toBe('string');
            expect(result.length).toBeGreaterThan(0);
        });
    });
});
//# sourceMappingURL=list.test.js.map