ai-functions 0.2.18 → 0.3.0

package/dist/factory/index.test.js

@@ -0,0 +1,287 @@
+import { describe, expect, it, beforeEach } from 'vitest';
+import { createTemplateFunction } from './index';
+import { openai } from '@ai-sdk/openai';
+// import { anthropic } from '@ai-sdk/anthropic' // Commented out as it's just for test demonstration
+describe('Template Function', () => {
+    beforeEach(() => {
+        process.env.OPENAI_API_KEY = process.env.OPENAI_API_KEY || 'test-key';
+    });
+    const model = openai('gpt-4o-mini');
+    describe('Template Literals', () => {
+        it('should support markdown generation', async () => {
+            const ai = createTemplateFunction();
+            const title = 'AI Functions';
+            const result = await ai `write a blog post in markdown starting with "# ${title}"`({ model });
+            expect(result).toMatch(/^# AI Functions/);
+            expect(result).toContain('# ');
+            expect(result).toContain('\n');
+        }, 30000);
+        // it('should support complex object summarization', async () => {
+        //   const ai = createTemplateFunction()
+        //   const itinerary = {
+        //     dates: ['2024-01-01', '2024-01-02'],
+        //     location: 'Miami Beach',
+        //     activities: ['Swimming', 'Surfing']
+        //   }
+        //   const result = await ai`Summarize the itinerary: ${JSON.stringify(itinerary)}`({ model })
+        //   expect(result).toContain('Miami Beach')
+        //   expect(result).toMatch(/[Ss]wimming/)
+        // })
+        it('should support multiple interpolated values', async () => {
+            const ai = createTemplateFunction();
+            const city = 'Paris';
+            const days = 3;
+            const interests = ['art', 'food'];
+            const result = await ai `Create an itinerary for ${days} days in ${city} focusing on ${interests.join(' and ')}`({ model });
+            expect(result).toContain('Paris');
+            expect(result).toContain('art');
+            expect(result).toContain('food');
+        }, 30000);
+    });
+    // describe('Structured Outputs', () => {
+    //   it('should support getting and using schema functions', async () => {
+    //     const ai = new Proxy(createTemplateFunction(), {
+    //       get: (target, prop) => {
+    //         if (prop === 'categorizeProduct') {
+    //           return createAIFunction(z.object({
+    //             productType: z.enum(['App', 'API', 'Marketplace', 'Platform', 'Service', 'Website']),
+    //             customer: z.string().describe('ideal customer profile'),
+    //             solution: z.string().describe('describe the offer'),
+    //             description: z.string().describe('website meta description')
+    //           }))
+    //         }
+    //         return target[prop]
+    //       }
+    //     })
+    //     // Get the function and use it later
+    //     const categorizeProduct = ai.categorizeProduct
+    //     const result1 = await categorizeProduct({ domain: 'stripe.com' }, { model })
+    //     expect(result1).toHaveProperty('productType')
+    //     expect(result1).toHaveProperty('customer')
+    //     expect(result1).toHaveProperty('solution')
+    //     expect(result1).toHaveProperty('description')
+    //     expect(['App', 'API', 'Marketplace', 'Platform', 'Service', 'Website']).toContain(result1.productType)
+    //     // Immediate invocation pattern
+    //     const result2 = await ai.categorizeProduct({
+    //       productType: 'App | API | Marketplace | Platform | Service | Website',
+    //       customer: 'ideal customer profile',
+    //       solution: 'describe the offer',
+    //       description: 'website meta description',
+    //     })({ name: 'drively AI' }, { model })
+    //     expect(result2).toHaveProperty('productType')
+    //     expect(result2).toHaveProperty('customer')
+    //     expect(result2).toHaveProperty('solution')
+    //     expect(result2).toHaveProperty('description')
+    //     expect(['App', 'API', 'Marketplace', 'Platform', 'Service', 'Website']).toContain(result2.productType)
+    //   }, 30000)
+    //   it('should support enum output format', async () => {
+    //     const ai = new Proxy(createTemplateFunction(), {
+    //       get: (target, prop) => {
+    //         if (prop === 'classifyMovie') {
+    //           return createAIFunction(z.enum(['action', 'comedy', 'drama', 'horror', 'sci-fi']))
+    //         }
+    //         return target[prop]
+    //       }
+    //     })
+    //     // Get the function and use it later
+    //     const classifyMovie = ai.classifyMovie
+    //     const result1 = await classifyMovie({
+    //       plot: 'A group of astronauts travel through a wormhole in search of a new habitable planet for humanity.'
+    //     }, { model })
+    //     expect(['action', 'comedy', 'drama', 'horror', 'sci-fi']).toContain(result1)
+    //     // Immediate invocation pattern
+    //     const result2 = await ai.classifyMovie({
+    //       genre: ['action', 'comedy', 'drama', 'horror', 'sci-fi']
+    //     })({ 
+    //       plot: 'A detective investigates a series of mysterious disappearances in a small town.'
+    //     }, { model })
+    //     expect(['action', 'comedy', 'drama', 'horror', 'sci-fi']).toContain(result2)
+    //   }, 30000)
+    // })
+    describe('Configuration', () => {
+        it('should support model specification', async () => {
+            const ai = createTemplateFunction();
+            const result = await ai `Hello`({ model: openai('gpt-4o-mini') });
+            expect(result).toBeDefined();
+        });
+        it('should support system prompts', async () => {
+            const ai = createTemplateFunction();
+            const result = await ai `List fun activities`({
+                model,
+                system: 'You are an expert tour guide',
+                temperature: 0.2
+            });
+            expect(result).toBeDefined();
+        });
+        it('should support concurrency limits', async () => {
+            const ai = createTemplateFunction();
+            const startTime = Date.now();
+            const results = await Promise.all([
+                ai `task 1`({ model, concurrency: 2 }),
+                ai `task 2`({ model, concurrency: 2 }),
+                ai `task 3`({ model, concurrency: 2 })
+            ]);
+            const endTime = Date.now();
+            expect(results).toHaveLength(3);
+            // With concurrency of 2, it should take at least 2 batches
+            expect(endTime - startTime).toBeGreaterThan(100);
+        });
+    });
+    //   describe('Composable Functions & Workflows', () => {
+    //     it('should support function composition', async () => {
+    //       const ai = createTemplateFunction()
+    //       const list = createTemplateFunction()
+    //       const listBlogPosts = (count: number, topic: string) => 
+    //         list`${count} blog post titles about ${topic}`({ model })
+    //       const writeBlogPost = (title: string) => 
+    //         ai`write a blog post in markdown starting with "# ${title}"`({ model })
+    //       async function* writeBlog(count: number, topic: string) {
+    //         for await (const title of await listBlogPosts(count, topic)) {
+    //           const content = await writeBlogPost(title)
+    //           yield { title, content }
+    //         }
+    //       }
+    //       const posts = []
+    //       for await (const post of writeBlog(2, 'future of car sales')) {
+    //         posts.push(post)
+    //       }
+    //       expect(posts).toHaveLength(2)
+    //       posts.forEach(post => {
+    //         expect(post).toHaveProperty('title')
+    //         expect(post).toHaveProperty('content')
+    //         expect(post.content).toMatch(new RegExp(`^# ${post.title}`))
+    //       })
+    //     })
+    //     it('should support nested template functions', async () => {
+    //       const ai = createTemplateFunction()
+    //       const generateName = (type: string) => ai`generate a name for a ${type}`({ model })
+    //       const generateFunction = (name: string) => 
+    //         ai`write a function in TypeScript called ${name}`({ model })
+    //       const name = await generateName('utility function')
+    //       const result = await generateFunction(name)
+    //       expect(result).toContain('function')
+    //       expect(result).toContain(name)
+    //       expect(result).toContain('export')
+    //     })
+    //   })
+    //   describe('Alternative Providers', () => {
+    //     it('should support OpenAI provider', async () => {
+    //       const ai = createTemplateFunction()
+    //       const result = await ai`Hello`({ model: openai('gpt-4o-mini') })
+    //       expect(result).toBeDefined()
+    //     })
+    //     // This test is commented out as we don't want to actually import anthropic
+    //     // but it demonstrates how the test would look
+    //     /*
+    //     it('should support Anthropic provider', async () => {
+    //       const ai = createTemplateFunction()
+    //       const result = await ai`write a function in TypeScript called ${name}`({ 
+    //         model: anthropic('claude-3-5-sonnet-20241022')
+    //       })
+    //       expect(result).toContain('function')
+    //       expect(result).toContain('export')
+    //     })
+    //     */
+    //     it('should support custom provider configuration', async () => {
+    //       const ai = createTemplateFunction()
+    //       const result = await ai`Hello`({ 
+    //         model: openai('gpt-4o-mini', { structuredOutputs: true })
+    //       })
+    //       expect(result).toBeDefined()
+    //     })
+    //   })
+    //   describe('Advanced Features', () => {
+    //     it('should support streaming with onChunk callback', async () => {
+    //       const ai = createTemplateFunction()
+    //       const chunks: string[] = []
+    //       const result = await ai`Write a short story`({
+    //         model,
+    //         streaming: {
+    //           onProgress: (chunk) => chunks.push(chunk)
+    //         }
+    //       })
+    //       expect(result).toBeDefined()
+    //       expect(chunks.length).toBeGreaterThan(0)
+    //       expect(chunks.join('')).toBe(result)
+    //     })
+    //     it('should support structured outputs with OpenAI', async () => {
+    //       const schema = z.object({
+    //         name: z.string(),
+    //         ingredients: z.array(z.object({
+    //           name: z.string(),
+    //           amount: z.string().nullable() // Note: using nullable instead of optional
+    //         })),
+    //         steps: z.array(z.string())
+    //       })
+    //       const ai = createTemplateFunction()
+    //       const result = await ai`Generate a recipe`({
+    //         model: openai('gpt-4o-mini', { structuredOutputs: true }),
+    //         outputFormat: 'object',
+    //         schema,
+    //         schemaName: 'recipe',
+    //         schemaDescription: 'A cooking recipe with ingredients and steps.'
+    //       })
+    //       const parsed = JSON.parse(result)
+    //       expect(() => schema.parse(parsed)).not.toThrow()
+    //     })
+    //     it('should support long text generation', async () => {
+    //       const ai = createTemplateFunction()
+    //       const result = await ai`Write a detailed essay about the history of Rome`({
+    //         model,
+    //         maxSteps: 3,
+    //         experimental_continueSteps: true,
+    //         system: 'Stop when sufficient information was provided.'
+    //       })
+    //       expect(result.length).toBeGreaterThan(1000) // Assuming it generates substantial content
+    //       expect(result).toContain('Rome')
+    //     })
+    //     it('should support predicted outputs', async () => {
+    //       const ai = createTemplateFunction()
+    //       const existingCode = `interface User {
+    //         Username: string;
+    //         Age: number;
+    //       }`
+    //       const result = await ai`Replace the Username property with an Email property`({
+    //         model,
+    //         experimental_providerMetadata: {
+    //           openai: {
+    //             prediction: {
+    //               type: 'content',
+    //               content: existingCode
+    //             }
+    //           }
+    //         }
+    //       })
+    //       expect(result).toContain('interface User')
+    //       expect(result).toContain('Email: string')
+    //       expect(result).not.toContain('Username')
+    //     })
+    //     it('should support streaming with tool calls', async () => {
+    //       const ai = createTemplateFunction()
+    //       const toolCalls: any[] = []
+    //       const toolResults: any[] = []
+    //       await ai`What are some San Francisco tourist attractions?`({
+    //         model,
+    //         tools: {
+    //           cityAttractions: {
+    //             parameters: z.object({ city: z.string() }),
+    //             execute: async ({ city }) => ({
+    //               attractions: ['Alcatraz', 'Golden Gate Bridge', 'Fisherman\'s Wharf']
+    //             })
+    //           }
+    //         },
+    //         streaming: {
+    //           onProgress: (chunk) => {
+    //             if (chunk.type === 'tool-call') toolCalls.push(chunk)
+    //             if (chunk.type === 'tool-result') toolResults.push(chunk)
+    //           }
+    //         }
+    //       })
+    //       expect(toolCalls.length).toBeGreaterThan(0)
+    //       expect(toolResults.length).toBeGreaterThan(0)
+    //       expect(toolCalls[0].toolName).toBe('cityAttractions')
+    //     })
+    //   })
+    // })
+});
+//# sourceMappingURL=index.test.js.map

package/dist/factory/index.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/factory/index.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AAEzD,OAAO,EAAE,sBAAsB,EAAoB,MAAM,SAAS,CAAA;AAElE,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACvC,qGAAqG;AAErG,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,UAAU,CAAA;IACvE,CAAC,CAAC,CAAA;IAEF,MAAM,KAAK,GAAG,MAAM,CAAC,aAAa,CAAC,CAAA;IAEnC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAA;YACnC,MAAM,KAAK,GAAG,cAAc,CAAA;YAC5B,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,kDAAkD,KAAK,GAAG,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAC5F,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAA;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,IAAI,CAAC,CAAA;QAChC,CAAC,EAAE,KAAK,CAAC,CAAA;QAET,kEAAkE;QAClE,wCAAwC;QACxC,wBAAwB;QACxB,2CAA2C;QAC3C,+BAA+B;QAC/B,0CAA0C;QAC1C,MAAM;QACN,8FAA8F;QAC9F,4CAA4C;QAC5C,0CAA0C;QAC1C,KAAK;QAEL,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAA;YACnC,MAAM,IAAI,GAAG,OAAO,CAAA;YACpB,MAAM,IAAI,GAAG,CAAC,CAAA;YACd,MAAM,SAAS,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAA;YACjC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,2BAA2B,IAAI,YAAY,IAAI,gBAAgB,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAC1H,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAA;YACjC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,KAAK,CAAC,CAAA;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAA;QAClC,CAAC,EAAE,KAAK,CAAC,CAAA;IACX,CAAC,CAAC,CAAA;IAEF,yCAAyC;IACzC,0EAA0E;IAC1E,uDAAuD;IACvD,iCAAiC;IACjC,8CAA8C;IAC9C,+CAA+C;IAC/C,oGAAoG;IACpG,uEAAuE;IACvE,mEAAmE;IACnE,2EAA2E;IAC3E,gBAAgB;IAChB,YAAY;IACZ,8BAA8B;IAC9B,UAAU;IACV,SAAS;IAET,2CAA2C;IAC3C,qDAAqD;IACrD,mFAAmF;IACnF,oDAAoD;IACpD,iDAAiD;IACjD,iDAAiD;IACjD,oDAAoD;IACpD,6GAA6G;IAE7G,sCAAsC;IACtC,mDAAmD;IACnD,+EAA+E;IAC/E,4CAA4C;IAC5C,wCAAwC;IACxC,iDAAiD;IACjD,4CAA4C;IAE5C,oDAAoD;IACpD,iDAAiD;IACjD,iDAAiD;IACjD,oDAAoD;IACpD,6GAA6G;IAC7G,cAAc;IAEd,0DAA0D;IAC1D,uDAAuD;IACvD,iCAAiC;IACjC,0CAA0C;IAC1C,+FAA+F;IAC/F,YAAY;IACZ,8BAA8B;IAC9B,UAAU;IACV,SAAS;IAET,2CAA2C;IAC3C,6CAA6C;IAC7C,4CAA4C;IAC5C,kHAAkH;IAClH,oBAAoB;IACpB,mFAAmF;IAEnF,sCAAsC;IACtC,+CAA+C;IAC/C,iEAAiE;IACjE,YAAY;IACZ,gGAAgG;IAChG,oBAAoB;IACpB,mFAAmF;IACnF,cAAc;IACd,KAAK;IAEL,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAA;YACnC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,OAAO,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC,CAAA;YAChE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;QAC9B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAA;YACnC,MAAM,MAAM,GAAG,MAAM,EAAE,CAAA,qBAAqB,CAAC;gBAC3C,KAAK;gBACL,MAAM,EAAE,8BAA8B;gBACtC,WAAW,EAAE,GAAG;aACjB,CAAC,CAAA;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;QAC9B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACjD,MAAM,EAAE,GAAG,sBAAsB,EAAE,CAAA;YACnC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAE5B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAChC,EAAE,CAAA,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;gBACrC,EAAE,CAAA,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;gBACrC,EAAE,CAAA,QAAQ,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC;aACtC,CAAC,CAAA;YAEF,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;YAC1B,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC/B,2DAA2D;YAC3D,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,CAAA;QAClD,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEJ,yDAAyD;IACzD,8DAA8D;IAC9D,4CAA4C;IAC5C,8CAA8C;IAE9C,iEAAiE;IACjE,oEAAoE;IACpE,kDAAkD;IAClD,kFAAkF;IAElF,kEAAkE;IAClE,yEAAyE;IACzE,uDAAuD;IACvD,qCAAqC;IACrC,YAAY;IACZ,UAAU;IAEV,yBAAyB;IACzB,wEAAwE;IACxE,2BAA2B;IAC3B,UAAU;IAEV,sCAAsC;IACtC,gCAAgC;IAChC,+CAA+C;IAC/C,iDAAiD;IACjD,uEAAuE;IACvE,WAAW;IACX,SAAS;IAET,mEAAmE;IACnE,4CAA4C;IAC5C,4FAA4F;IAC5F,oDAAoD;IACpD,uEAAuE;IAEvE,4DAA4D;IAC5D,oDAAoD;IAEpD,6CAA6C;IAC7C,uCAAuC;IACvC,2CAA2C;IAC3C,SAAS;IACT,OAAO;IAEP,8CAA8C;IAC9C,yDAAyD;IACzD,4CAA4C;IAC5C,yEAAyE;IACzE,qCAAqC;IACrC,SAAS;IAET,kFAAkF;IAClF,qDAAqD;IACrD,SAAS;IACT,4DAA4D;IAC5D,4CAA4C;IAC5C,kFAAkF;IAClF,yDAAyD;IACzD,WAAW;IACX,6CAA6C;IAC7C,2CAA2C;IAC3C,SAAS;IACT,SAAS;IAET,uEAAuE;IACvE,4CAA4C;IAC5C,0CAA0C;IAC1C,oEAAoE;IACpE,WAAW;IACX,qCAAqC;IACrC,SAAS;IACT,OAAO;IAEP,0CAA0C;IAC1C,yEAAyE;IACzE,4CAA4C;IAC5C,oCAAoC;IAEpC,uDAAuD;IACvD,iBAAiB;IACjB,uBAAuB;IACvB,sDAAsD;IACtD,YAAY;IACZ,WAAW;IAEX,qCAAqC;IACrC,iDAAiD;IACjD,6CAA6C;IAC7C,SAAS;IAET,wEAAwE;IACxE,kCAAkC;IAClC,4BAA4B;IAC5B,0CAA0C;IAC1C,8BAA8B;IAC9B,sFAAsF;IACtF,eAAe;IACf,qCAAqC;IACrC,WAAW;IAEX,4CAA4C;IAC5C,qDAAqD;IACrD,qEAAqE;IACrE,kCAAkC;IAClC,kBAAkB;IAClB,gCAAgC;IAChC,4EAA4E;IAC5E,WAAW;IAEX,0CAA0C;IAC1C,yDAAyD;IACzD,SAAS;IAET,8DAA8D;IAC9D,4CAA4C;IAC5C,oFAAoF;IACpF,iBAAiB;IACjB,uBAAuB;IACvB,4CAA4C;IAC5C,mEAAmE;IACnE,WAAW;IAEX,iGAAiG;IACjG,yCAAyC;IACzC,SAAS;IAET,2DAA2D;IAC3D,4CAA4C;IAC5C,+CAA+C;IAC/C,4BAA4B;IAC5B,uBAAuB;IACvB,WAAW;IAEX,wFAAwF;IACxF,iBAAiB;IACjB,2CAA2C;IAC3C,sBAAsB;IACtB,4BAA4B;IAC5B,iCAAiC;IACjC,sCAAsC;IACtC,gBAAgB;IAChB,cAAc;IACd,YAAY;IACZ,WAAW;IAEX,mDAAmD;IACnD,kDAAkD;IAClD,iDAAiD;IACjD,SAAS;IAET,mEAAmE;IACnE,4CAA4C;IAC5C,oCAAoC;IACpC,sCAAsC;IAEtC,qEAAqE;IACrE,iBAAiB;IACjB,mBAAmB;IACnB,+BAA+B;IAC/B,0DAA0D;IAC1D,8CAA8C;IAC9C,sFAAsF;IACtF,iBAAiB;IACjB,cAAc;IACd,aAAa;IACb,uBAAuB;IACvB,qCAAqC;IACrC,oEAAoE;IACpE,wEAAwE;IACxE,cAAc;IACd,YAAY;IACZ,WAAW;IAEX,oDAAoD;IACpD,sDAAsD;IACtD,8DAA8D;IAC9D,SAAS;IACT,OAAO;IACP,KAAK;AACL,CAAC,CAAC,CAAA"}

package/dist/factory/list.d.ts

@@ -0,0 +1,3 @@
+import type { AIFunctionOptions, BaseTemplateFunction } from '../types';
+export declare function createListFunction(defaultOptions?: AIFunctionOptions): BaseTemplateFunction;
+//# sourceMappingURL=list.d.ts.map

package/dist/factory/list.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"list.d.ts","sourceRoot":"","sources":["../../src/factory/list.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,iBAAiB,EAAE,oBAAoB,EAAyB,MAAM,UAAU,CAAA;AAc9F,wBAAgB,kBAAkB,CAAC,cAAc,GAAE,iBAAsB,GAAG,oBAAoB,CAwO/F"}

package/dist/factory/list.js

@@ -0,0 +1,221 @@
+import { streamObject } from 'ai';
+import { openai } from '@ai-sdk/openai';
+import { z } from 'zod';
+import PQueue from 'p-queue';
+function createQueue(options) {
+    if (!options.concurrency) {
+        return undefined;
+    }
+    return new PQueue({
+        concurrency: options.concurrency,
+        autoStart: true,
+        carryoverConcurrencyCount: true,
+    });
+}
+export function createListFunction(defaultOptions = {}) {
+    let currentPrompt;
+    let currentQueue;
+    let queueOptions;
+    const getQueue = (options) => {
+        if (!options.concurrency) {
+            return undefined;
+        }
+        // Create a new queue if options have changed
+        if (!currentQueue ||
+            !queueOptions?.concurrency ||
+            queueOptions.concurrency !== options.concurrency) {
+            currentQueue = createQueue(options);
+            queueOptions = options;
+        }
+        return currentQueue;
+    };
+    const executeRequest = async (prompt, options) => {
+        const modelParams = {
+            temperature: options.temperature,
+            maxTokens: options.maxTokens,
+            topP: options.topP,
+            frequencyPenalty: options.frequencyPenalty,
+            presencePenalty: options.presencePenalty,
+            stopSequences: options.stop ? Array.isArray(options.stop) ? options.stop : [options.stop] : undefined,
+            seed: options.seed
+        };
+        const performRequest = async () => {
+            try {
+                const model = options.model || openai(process.env.OPENAI_DEFAULT_MODEL || 'gpt-4o');
+                const streamOptions = {
+                    model,
+                    output: 'array',
+                    schema: z.string(),
+                    prompt: `Generate a list of items based on this prompt: ${prompt}`,
+                    system: options.system,
+                    ...modelParams
+                };
+                const { elementStream } = streamObject(streamOptions);
+                const elements = [];
+                for await (const item of elementStream) {
+                    elements.push(item);
+                }
+                return elements.join('\n');
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    throw error;
+                }
+                throw new Error('Failed to generate list');
+            }
+        };
+        const queue = getQueue(options);
+        const result = queue
+            ? await queue.add(performRequest)
+            : await performRequest();
+        return result;
+    };
+    const executeStreamingRequest = async function* (prompt, options) {
+        const modelParams = {
+            temperature: options.temperature,
+            maxTokens: options.maxTokens,
+            topP: options.topP,
+            frequencyPenalty: options.frequencyPenalty,
+            presencePenalty: options.presencePenalty,
+            stopSequences: options.stop ? Array.isArray(options.stop) ? options.stop : [options.stop] : undefined,
+            seed: options.seed
+        };
+        const performRequest = async function* () {
+            try {
+                const model = options.model || openai(process.env.OPENAI_DEFAULT_MODEL || 'gpt-4o');
+                const streamOptions = {
+                    model,
+                    output: 'array',
+                    schema: z.string(),
+                    prompt: `Generate a list of items based on this prompt: ${prompt}`,
+                    system: options.system,
+                    ...modelParams
+                };
+                const { elementStream } = streamObject(streamOptions);
+                for await (const item of elementStream) {
+                    yield item;
+                }
+            }
+            catch (error) {
+                if (error instanceof Error) {
+                    if (error.name === 'AbortError') {
+                        throw new Error('Stream was aborted');
+                    }
+                    else if (error.name === 'TimeoutError') {
+                        throw new Error('Stream timed out');
+                    }
+                    else {
+                        throw error;
+                    }
+                }
+                throw new Error('Failed to generate list: Unknown error occurred');
+            }
+        };
+        const queue = getQueue(options);
+        if (queue) {
+            const operation = performRequest();
+            // Queue the operation based on its type
+            if (operation[Symbol.asyncIterator]) {
+                // For AsyncGenerator, queue each value
+                for await (const value of operation) {
+                    yield await queue.add(async () => value);
+                }
+            }
+            else {
+                // For Promise, queue the entire operation
+                yield await queue.add(async () => {
+                    const result = await operation;
+                    return result;
+                });
+            }
+        }
+        else {
+            yield* performRequest();
+        }
+    };
+    const templateFn = async (prompt, options = defaultOptions) => {
+        currentPrompt = prompt;
+        const mergedOptions = { ...defaultOptions, ...options };
+        try {
+            return await executeRequest(prompt, mergedOptions);
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw error;
+            }
+            throw new Error('Failed to generate list');
+        }
+    };
+    const asyncIterator = async function* (prompt, options = defaultOptions) {
+        currentPrompt = prompt;
+        const mergedOptions = { ...defaultOptions, ...options };
+        try {
+            for await (const item of executeStreamingRequest(prompt, mergedOptions)) {
+                yield item;
+            }
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                throw error;
+            }
+            throw new Error('Failed to generate list');
+        }
+    };
+    function createTemplateResult(prompt, options) {
+        const promise = templateFn(prompt, options);
+        const result = Object.assign((opts) => templateFn(prompt, { ...options, ...opts }), {
+            [Symbol.asyncIterator]: () => asyncIterator(prompt, options),
+            call: (opts) => templateFn(prompt, { ...options, ...opts }),
+            then: promise.then.bind(promise),
+            catch: promise.catch.bind(promise),
+            finally: promise.finally.bind(promise),
+        });
+        return result;
+    }
+    function createBaseFunction() {
+        function fn(stringsOrOptions, ...values) {
+            if (!stringsOrOptions) {
+                return createTemplateResult('', defaultOptions);
+            }
+            if (Array.isArray(stringsOrOptions)) {
+                const strings = stringsOrOptions;
+                if (strings.length - 1 !== values.length) {
+                    throw new Error('Template literal slots must match provided values');
+                }
+                const lastValue = values[values.length - 1];
+                const options = typeof lastValue === 'object' && !Array.isArray(lastValue) && lastValue !== null
+                    ? { ...defaultOptions, ...lastValue }
+                    : defaultOptions;
+                const actualValues = typeof lastValue === 'object' && !Array.isArray(lastValue) && lastValue !== null
+                    ? values.slice(0, -1)
+                    : values;
+                const prompt = strings.reduce((acc, str, i) => acc + str + (actualValues[i] || ''), '');
+                currentPrompt = prompt;
+                return createTemplateResult(prompt, options);
+            }
+            const options = { ...defaultOptions, ...stringsOrOptions };
+            return createTemplateResult('', options);
+        }
+        // Add the asyncIterator to the base function
+        Object.defineProperty(fn, Symbol.asyncIterator, {
+            value: () => asyncIterator(currentPrompt || '', defaultOptions),
+            enumerable: false,
+            configurable: true
+        });
+        // Add the withOptions method
+        Object.defineProperty(fn, 'withOptions', {
+            value: (opts) => templateFn(currentPrompt || '', { ...defaultOptions, ...opts }),
+            enumerable: false,
+            configurable: true
+        });
+        // Add the queue property
+        Object.defineProperty(fn, 'queue', {
+            get: () => currentQueue,
+            enumerable: false,
+            configurable: true
+        });
+        return fn;
+    }
+    return createBaseFunction();
+}
+//# sourceMappingURL=list.js.map

package/dist/factory/list.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"list.js","sourceRoot":"","sources":["../../src/factory/list.ts"],"names":[],"mappings":"AAAA,OAAO,EAAgB,YAAY,EAAE,MAAM,IAAI,CAAA;AAC/C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACvC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,MAAM,MAAM,SAAS,CAAA;AAG5B,SAAS,WAAW,CAAC,OAA0B;IAC7C,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QACzB,OAAO,SAAS,CAAA;IAClB,CAAC;IAED,OAAO,IAAI,MAAM,CAAC;QAChB,WAAW,EAAE,OAAO,CAAC,WAAW;QAChC,SAAS,EAAE,IAAI;QACf,yBAAyB,EAAE,IAAI;KAChC,CAAU,CAAA;AACb,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,iBAAoC,EAAE;IACvE,IAAI,aAAiC,CAAA;IACrC,IAAI,YAA+B,CAAA;IACnC,IAAI,YAA2C,CAAA;IAE/C,MAAM,QAAQ,GAAG,CAAC,OAA0B,EAAqB,EAAE;QACjE,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;YACzB,OAAO,SAAS,CAAA;QAClB,CAAC;QAED,6CAA6C;QAC7C,IAAI,CAAC,YAAY;YACb,CAAC,YAAY,EAAE,WAAW;YAC1B,YAAY,CAAC,WAAW,KAAK,OAAO,CAAC,WAAW,EAAE,CAAC;YACrD,YAAY,GAAG,WAAW,CAAC,OAAO,CAAC,CAAA;YACnC,YAAY,GAAG,OAAO,CAAA;QACxB,CAAC;QAED,OAAO,YAAY,CAAA;IACrB,CAAC,CAAA;IAED,MAAM,cAAc,GAAG,KAAK,EAAE,MAAc,EAAE,OAA0B,EAAmB,EAAE;QAC3F,MAAM,WAAW,GAAG;YAClB,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YACrG,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAA;QAED,MAAM,cAAc,GAAG,KAAK,IAAqB,EAAE;YACjD,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,QAAQ,CAAC,CAAA;gBACnF,MAAM,aAAa,GAAG;oBACpB,KAAK;oBACL,MAAM,EAAE,OAAgB;oBACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;oBAClB,MAAM,EAAE,kDAAkD,MAAM,EAAE;oBAClE,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,GAAG,WAAW;iBACf,CAAA;gBAED,MAAM,EAAE,aAAa,EAAE,GAAG,YAAY,CAAC,aAAa,CAAC,CAAA;gBACrD,MAAM,QAAQ,GAAa,EAAE,CAAA;gBAE7B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;oBACvC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;gBACrB,CAAC;gBAED,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAC5B,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,MAAM,KAAK,CAAA;gBACb,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;YAC5C,CAAC;QACH,CAAC,CAAA;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAA;QAC/B,MAAM,MAAM,GAAG,KAAK;YAClB,CAAC,CAAC,MAAM,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC;YACjC,CAAC,CAAC,MAAM,cAAc,EAAE,CAAA;QAE1B,OAAO,MAAM,CAAA;IACf,CAAC,CAAA;IAED,MAAM,uBAAuB,GAAG,KAAK,SAAS,CAAC,EAAE,MAAc,EAAE,OAA0B;QACzF,MAAM,WAAW,GAAG;YAClB,WAAW,EAAE,OAAO,CAAC,WAAW;YAChC,SAAS,EAAE,OAAO,CAAC,SAAS;YAC5B,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,gBAAgB,EAAE,OAAO,CAAC,gBAAgB;YAC1C,eAAe,EAAE,OAAO,CAAC,eAAe;YACxC,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YACrG,IAAI,EAAE,OAAO,CAAC,IAAI;SACnB,CAAA;QAED,MAAM,cAAc,GAAG,KAAK,SAAS,CAAC;YACpC,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,QAAQ,CAAC,CAAA;gBACnF,MAAM,aAAa,GAAG;oBACpB,KAAK;oBACL,MAAM,EAAE,OAAgB;oBACxB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;oBAClB,MAAM,EAAE,kDAAkD,MAAM,EAAE;oBAClE,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,GAAG,WAAW;iBACf,CAAA;gBAED,MAAM,EAAE,aAAa,EAAE,GAAG,YAAY,CAAC,aAAa,CAAC,CAAA;gBAErD,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;oBACvC,MAAM,IAAI,CAAA;gBACZ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;oBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBAChC,MAAM,IAAI,KAAK,CAAC,oBAAoB,CAAC,CAAA;oBACvC,CAAC;yBAAM,IAAI,KAAK,CAAC,IAAI,KAAK,cAAc,EAAE,CAAC;wBACzC,MAAM,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAA;oBACrC,CAAC;yBAAM,CAAC;wBACN,MAAM,KAAK,CAAA;oBACb,CAAC;gBACH,CAAC;gBACD,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAA;YACpE,CAAC;QACH,CAAC,CAAA;QAED,MAAM,KAAK,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAA;QAC/B,IAAI,KAAK,EAAE,CAAC;YACV,MAAM,SAAS,GAAG,cAAc,EAAE,CAAA;YAElC,wCAAwC;YACxC,IAAI,SAAS,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,CAAC;gBACpC,uCAAuC;gBACvC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;oBACpC,MAAM,MAAM,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE,CAAC,KAAK,CAAC,CAAA;gBAC1C,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,0CAA0C;gBAC1C,MAAM,MAAM,KAAK,CAAC,GAAG,CAAC,KAAK,IAAI,EAAE;oBAC/B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAA;oBAC9B,OAAO,MAAM,CAAA;gBACf,CAAC,CAAC,CAAA;YACJ,CAAC;QACH,CAAC;aAAM,CAAC;YACN,KAAK,CAAC,CAAC,cAAc,EAAE,CAAA;QACzB,CAAC;IACH,CAAC,CAAA;IAED,MAAM,UAAU,GAAG,KAAK,EAAE,MAAc,EAAE,UAA6B,cAAc,EAAmB,EAAE;QACxG,aAAa,GAAG,MAAM,CAAA;QACtB,MAAM,aAAa,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAA;QAEvD,IAAI,CAAC;YACH,OAAO,MAAM,cAAc,CAAC,MAAM,EAAE,aAAa,CAAC,CAAA;QACpD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;QAC5C,CAAC;IACH,CAAC,CAAA;IAED,MAAM,aAAa,GAAG,KAAK,SAAS,CAAC,EAAE,MAAc,EAAE,UAA6B,cAAc;QAChG,aAAa,GAAG,MAAM,CAAA;QACtB,MAAM,aAAa,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,OAAO,EAAE,CAAA;QAEvD,IAAI,CAAC;YACH,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,uBAAuB,CAAC,MAAM,EAAE,aAAa,CAAC,EAAE,CAAC;gBACxE,MAAM,IAAI,CAAA;YACZ,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAA;YACb,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAA;QAC5C,CAAC;IACH,CAAC,CAAA;IAED,SAAS,oBAAoB,CAAC,MAAc,EAAE,OAA0B;QACtE,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;QAC3C,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAC1B,CAAC,IAAwB,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC,EACzE;YACE,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,EAAE,OAAO,CAAC;YAC5D,IAAI,EAAE,CAAC,IAAwB,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,EAAE,GAAG,OAAO,EAAE,GAAG,IAAI,EAAE,CAAC;YAC/E,IAAI,EAAE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC;YAChC,KAAK,EAAE,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC;YAClC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC;SACvC,CACgB,CAAA;QAEnB,OAAO,MAAM,CAAA;IACf,CAAC;IAED,SAAS,kBAAkB;QACzB,SAAS,EAAE,CAAC,gBAA2D,EAAE,GAAG,MAAiB;YAC3F,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,oBAAoB,CAAC,EAAE,EAAE,cAAc,CAAC,CAAA;YACjD,CAAC;YAED,IAAI,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,EAAE,CAAC;gBACpC,MAAM,OAAO,GAAG,gBAAwC,CAAA;gBACxD,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,KAAK,MAAM,CAAC,MAAM,EAAE,CAAC;oBACzC,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAA;gBACtE,CAAC;gBAED,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC,CAAA;gBAC3C,MAAM,OAAO,GAAG,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,SAAS,KAAK,IAAI;oBAC9F,CAAC,CAAC,EAAE,GAAG,cAAc,EAAE,GAAG,SAA8B,EAAE;oBAC1D,CAAC,CAAC,cAAc,CAAA;gBAClB,MAAM,YAAY,GAAG,OAAO,SAAS,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,SAAS,KAAK,IAAI;oBACnG,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;oBACrB,CAAC,CAAC,MAAM,CAAA;gBAEV,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAA;gBACvF,aAAa,GAAG,MAAM,CAAA;gBACtB,OAAO,oBAAoB,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;YAC9C,CAAC;YAED,MAAM,OAAO,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,gBAAgB,EAAE,CAAA;YAC1D,OAAO,oBAAoB,CAAC,EAAE,EAAE,OAAO,CAAC,CAAA;QAC1C,CAAC;QAED,6CAA6C;QAC7C,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,MAAM,CAAC,aAAa,EAAE;YAC9C,KAAK,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,aAAa,IAAI,EAAE,EAAE,cAAc,CAAC;YAC/D,UAAU,EAAE,KAAK;YACjB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAA;QAEF,6BAA6B;QAC7B,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,aAAa,EAAE;YACvC,KAAK,EAAE,CAAC,IAAwB,EAAE,EAAE,CAAC,UAAU,CAAC,aAAa,IAAI,EAAE,EAAE,EAAE,GAAG,cAAc,EAAE,GAAG,IAAI,EAAE,CAAC;YACpG,UAAU,EAAE,KAAK;YACjB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAA;QAEF,yBAAyB;QACzB,MAAM,CAAC,cAAc,CAAC,EAAE,EAAE,OAAO,EAAE;YACjC,GAAG,EAAE,GAAG,EAAE,CAAC,YAAY;YACvB,UAAU,EAAE,KAAK;YACjB,YAAY,EAAE,IAAI;SACnB,CAAC,CAAA;QAEF,OAAO,EAA0B,CAAA;IACnC,CAAC;IAED,OAAO,kBAAkB,EAAE,CAAA;AAC7B,CAAC"}

package/dist/factory/list.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=list.test.d.ts.map

package/dist/factory/list.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"list.test.d.ts","sourceRoot":"","sources":["../../src/factory/list.test.ts"],"names":[],"mappings":""}

package/dist/factory/list.test.js

@@ -0,0 +1,84 @@
+import { describe, expect, it, beforeEach } from 'vitest';
+import { createListFunction } from './list';
+import { openai } from '@ai-sdk/openai';
+describe('List Function', () => {
+    beforeEach(() => {
+        process.env.OPENAI_API_KEY = process.env.OPENAI_API_KEY || 'test-key';
+    });
+    const model = openai('gpt-4o-mini');
+    describe('Template Literals', () => {
+        it('should generate list using template literal', async () => {
+            const list = createListFunction();
+            const result = await list `fun things to do in Miami`({ model });
+            expect(result).toBeDefined();
+            const items = result.split('\n');
+            expect(items.length).toBeGreaterThan(0);
+            items.forEach(item => expect(typeof item).toBe('string'));
+        });
+        it('should support async iteration', async () => {
+            const list = createListFunction();
+            const items = [];
+            for await (const item of list `fun things to do in Miami`({ model })) {
+                items.push(item);
+            }
+            expect(items.length).toBeGreaterThan(0);
+            items.forEach(item => expect(typeof item).toBe('string'));
+        });
+        it('should support interpolation', async () => {
+            const list = createListFunction();
+            const city = 'Paris';
+            const category = 'museums';
+            const result = await list `top ${category} to visit in ${city}`({ model });
+            expect(result).toContain('Louvre');
+        });
+    });
+    describe('Configuration', () => {
+        it('should support system prompts', async () => {
+            const list = createListFunction();
+            const result = await list `fun things to do in Miami`({
+                model,
+                system: 'You are an expert tour guide',
+                temperature: 0.2
+            });
+            expect(result).toBeDefined();
+            expect(result.split('\n').length).toBeGreaterThan(0);
+        });
+        it('should support concurrency for multiple lists', async () => {
+            const list = createListFunction();
+            const cities = ['Paris', 'London', 'Tokyo'];
+            const results = await Promise.all(cities.map(city => list `top attractions in ${city}`({ model, concurrency: 2 })));
+            expect(results).toHaveLength(3);
+            results.forEach(result => {
+                expect(result.split('\n').length).toBeGreaterThan(0);
+            });
+        });
+    });
+    describe('Streaming', () => {
+        it('should stream list items', async () => {
+            const list = createListFunction();
+            const items = [];
+            const progressItems = [];
+            for await (const item of list `fun things to do in Miami`({
+                model,
+                streaming: {
+                    onProgress: chunk => progressItems.push(chunk)
+                }
+            })) {
+                items.push(item);
+            }
+            expect(items.length).toBeGreaterThan(0);
+            expect(progressItems.length).toBeGreaterThan(0);
+        });
+        it('should handle streaming errors gracefully', async () => {
+            const list = createListFunction();
+            await expect(async () => {
+                for await (const item of list `fun things to do in Miami`({
+                    model: undefined // Force an error
+                })) {
+                    console.log(item);
+                }
+            }).rejects.toThrow();
+        });
+    });
+});
+//# sourceMappingURL=list.test.js.map

package/dist/factory/list.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"list.test.js","sourceRoot":"","sources":["../../src/factory/list.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAA;AACzD,OAAO,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAA;AAC3C,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AAEvC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,UAAU,CAAC,GAAG,EAAE;QACd,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,UAAU,CAAA;IACvE,CAAC,CAAC,CAAA;IAEF,MAAM,KAAK,GAAG,MAAM,CAAC,aAAa,CAAC,CAAA;IAEnC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAA,2BAA2B,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YAC/D,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;YAC5B,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YAChC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YACvC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAA,2BAA2B,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBACpE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAClB,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YACvC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,KAAK,IAAI,EAAE;YAC5C,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,IAAI,GAAG,OAAO,CAAA;YACpB,MAAM,QAAQ,GAAG,SAAS,CAAA;YAC1B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAA,OAAO,QAAQ,gBAAgB,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAA;YACzE,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAA;QACpC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;YAC7C,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAA,2BAA2B,CAAC;gBACnD,KAAK;gBACL,MAAM,EAAE,8BAA8B;gBACtC,WAAW,EAAE,GAAG;aACjB,CAAC,CAAA;YACF,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;YAC5B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,KAAK,IAAI,EAAE;YAC7D,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,MAAM,GAAG,CAAC,OAAO,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAA;YAC3C,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAA,sBAAsB,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC,CAChF,CAAA;YACD,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC/B,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE;gBACvB,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YACtD,CAAC,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;YACxC,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,KAAK,GAAa,EAAE,CAAA;YAC1B,MAAM,aAAa,GAAa,EAAE,CAAA;YAElC,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAA,2BAA2B,CAAC;gBACvD,KAAK;gBACL,SAAS,EAAE;oBACT,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,KAAK,CAAC;iBAC/C;aACF,CAAC,EAAE,CAAC;gBACH,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAClB,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;YACvC,MAAM,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAA;QACjD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,IAAI,GAAG,kBAAkB,EAAE,CAAA;YACjC,MAAM,MAAM,CAAC,KAAK,IAAI,EAAE;gBACtB,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAA,2BAA2B,CAAC;oBACvD,KAAK,EAAE,SAAgB,CAAC,iBAAiB;iBAC1C,CAAC,EAAE,CAAC;oBACH,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,CAAA;gBACnB,CAAC;YACH,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,CAAA;QACtB,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/generate/index.d.ts

@@ -0,0 +1,5 @@
+import type { GenerateObjectOptions, SimpleSchema } from '@/types';
+export declare function generateObject<T extends SimpleSchema>(options: GenerateObjectOptions<T>): Promise<import("ai").GenerateObjectResult<{
+    [x: string]: any;
+}>>;
+//# sourceMappingURL=index.d.ts.map

package/dist/generate/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/generate/index.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,qBAAqB,EAAE,YAAY,EAAE,MAAM,SAAS,CAAA;AAElE,wBAAsB,cAAc,CAAC,CAAC,SAAS,YAAY,EAAE,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC;;IAa7F"}