ai-functions 0.2.18 → 0.3.0

package/dist/templates/result.test.js

@@ -0,0 +1,75 @@
+import { describe, expect, it } from 'vitest';
+import { createTemplateResult, parseTemplateInput } from './result';
+describe('Template Result', () => {
+    describe('createTemplateResult', () => {
+        const mockTemplateFn = async (prompt, options) => `Result: ${prompt} with ${JSON.stringify(options)}`;
+        it('should create template result with basic functionality', async () => {
+            const result = createTemplateResult('test prompt', {}, mockTemplateFn);
+            expect(result).toBeDefined();
+            expect(typeof result.then).toBe('function');
+            expect(typeof result[Symbol.asyncIterator]).toBe('function');
+        });
+        it('should support async iteration', async () => {
+            const result = createTemplateResult('test prompt', {}, mockTemplateFn);
+            const chunks = [];
+            for await (const chunk of result) {
+                chunks.push(chunk);
+            }
+            expect(chunks.length).toBe(1);
+            expect(chunks[0]).toMatch(/^Result: test prompt/);
+        });
+        it('should support promise-like behavior', async () => {
+            const result = createTemplateResult('test prompt', {}, mockTemplateFn);
+            const text = await result;
+            expect(text).toMatch(/^Result: test prompt/);
+        });
+        it('should support option overrides', async () => {
+            const result = createTemplateResult('test prompt', { temperature: 0.5 }, mockTemplateFn);
+            const text = await result({ temperature: 0.7 });
+            expect(text).toMatch(/"temperature":0.7/);
+        });
+    });
+    describe('parseTemplateInput', () => {
+        const defaultOptions = { temperature: 0.5 };
+        it('should handle undefined input', () => {
+            const result = parseTemplateInput(undefined, [], defaultOptions);
+            expect(result).toEqual({
+                prompt: '',
+                options: defaultOptions
+            });
+        });
+        it('should parse template strings array', () => {
+            const strings = Object.assign(['Hello ', ' world'], { raw: ['Hello ', ' world'] });
+            const values = ['beautiful'];
+            const result = parseTemplateInput(strings, values, defaultOptions);
+            expect(result).toEqual({
+                prompt: 'Hello beautiful world',
+                options: defaultOptions
+            });
+        });
+        // it('should handle options in last value', () => {
+        //   const strings = Object.assign(['Hello ', ''], { raw: ['Hello ', ''] }) as TemplateStringsArray
+        //   const values = ['world', { temperature: 0.7 }]
+        //   const result = parseTemplateInput(strings, values, defaultOptions)
+        //   expect(result).toEqual({
+        //     prompt: 'Hello world',
+        //     options: { temperature: 0.7 }
+        //   })
+        // })
+        it('should throw on mismatched template slots', () => {
+            const strings = Object.assign(['a', 'b', 'c'], { raw: ['a', 'b', 'c'] });
+            const values = ['x'];
+            expect(() => parseTemplateInput(strings, values, defaultOptions))
+                .toThrow('Template literal slots must match provided values');
+        });
+        it('should handle direct options object', () => {
+            const options = { temperature: 0.7 };
+            const result = parseTemplateInput(options, [], defaultOptions);
+            expect(result).toEqual({
+                prompt: '',
+                options: { temperature: 0.7 }
+            });
+        });
+    });
+});
+//# sourceMappingURL=result.test.js.map

package/dist/templates/result.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"result.test.js","sourceRoot":"","sources":["../../src/templates/result.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAA;AAC7C,OAAO,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAA;AAGnE,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,MAAM,cAAc,GAAG,KAAK,EAAE,MAAc,EAAE,OAA0B,EAAE,EAAE,CAC1E,WAAW,MAAM,SAAS,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,EAAE,CAAA;QAErD,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,MAAM,GAAG,oBAAoB,CAAC,aAAa,EAAE,EAAE,EAAE,cAAc,CAAC,CAAA;YACtE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAA;YAC5B,MAAM,CAAC,OAAO,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;YAC3C,MAAM,CAAC,OAAO,MAAM,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAA;QAC9D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAC9C,MAAM,MAAM,GAAG,oBAAoB,CAAC,aAAa,EAAE,EAAE,EAAE,cAAc,CAAC,CAAA;YACtE,MAAM,MAAM,GAAa,EAAE,CAAA;YAC3B,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACpB,CAAC;YACD,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAA;YAC7B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;QACnD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,MAAM,GAAG,oBAAoB,CAAC,aAAa,EAAE,EAAE,EAAE,cAAc,CAAC,CAAA;YACtE,MAAM,IAAI,GAAG,MAAM,MAAM,CAAA;YACzB,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;QAC9C,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,iCAAiC,EAAE,KAAK,IAAI,EAAE;YAC/C,MAAM,MAAM,GAAG,oBAAoB,CAAC,aAAa,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,EAAE,cAAc,CAAC,CAAA;YACxF,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAA;YAC/C,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,mBAAmB,CAAC,CAAA;QAC3C,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,MAAM,cAAc,GAAsB,EAAE,WAAW,EAAE,GAAG,EAAE,CAAA;QAE9D,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;YACvC,MAAM,MAAM,GAAG,kBAAkB,CAAC,SAAS,EAAE,EAAE,EAAE,cAAc,CAAC,CAAA;YAChE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,EAAE;gBACV,OAAO,EAAE,cAAc;aACxB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,QAAQ,EAAE,QAAQ,CAAC,EAAE,CAAyB,CAAA;YAC1G,MAAM,MAAM,GAAG,CAAC,WAAW,CAAC,CAAA;YAC5B,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,cAAc,CAAC,CAAA;YAClE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,uBAAuB;gBAC/B,OAAO,EAAE,cAAc;aACxB,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,oDAAoD;QACpD,mGAAmG;QACnG,mDAAmD;QACnD,uEAAuE;QACvE,6BAA6B;QAC7B,6BAA6B;QAC7B,oCAAoC;QACpC,OAAO;QACP,KAAK;QAEL,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,EAAE,CAAyB,CAAA;YAChG,MAAM,MAAM,GAAG,CAAC,GAAG,CAAC,CAAA;YACpB,MAAM,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,OAAO,EAAE,MAAM,EAAE,cAAc,CAAC,CAAC;iBAC9D,OAAO,CAAC,mDAAmD,CAAC,CAAA;QACjE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,MAAM,OAAO,GAAG,EAAE,WAAW,EAAE,GAAG,EAAE,CAAA;YACpC,MAAM,MAAM,GAAG,kBAAkB,CAAC,OAAO,EAAE,EAAE,EAAE,cAAc,CAAC,CAAA;YAC9D,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,EAAE;gBACV,OAAO,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE;aAC9B,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/test/setup.d.ts

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

package/dist/test/setup.d.ts.map

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

package/dist/test/setup.js

@@ -0,0 +1,21 @@
+import { config } from 'dotenv';
+import { resolve } from 'path';
+import { openai } from '@ai-sdk/openai';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
+// Load environment variables from .env file
+config({ path: resolve(process.cwd(), '.env') });
+// Configure OpenAI provider with custom endpoint if AI_GATEWAY is set
+if (process.env.AI_GATEWAY) {
+    const customProvider = createOpenAICompatible({
+        baseURL: process.env.AI_GATEWAY,
+        headers: {
+            'Authorization': `Bearer ${process.env.OPENAI_API_KEY || ''}`
+        },
+        name: 'openai-compatible'
+    });
+    // Replace the default provider with our custom configured one
+    Object.assign(openai, customProvider);
+}
+// Set default model for tests
+process.env.OPENAI_DEFAULT_MODEL = 'gpt-4o';
+//# sourceMappingURL=setup.js.map

package/dist/test/setup.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"setup.js","sourceRoot":"","sources":["../../src/test/setup.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAA;AAC/B,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAA;AAC9B,OAAO,EAAE,MAAM,EAAE,MAAM,gBAAgB,CAAA;AACvC,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAA;AAElE,4CAA4C;AAC5C,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,EAAE,CAAC,CAAA;AAEhD,sEAAsE;AACtE,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,EAAE,CAAC;IAC3B,MAAM,cAAc,GAAG,sBAAsB,CAAC;QAC5C,OAAO,EAAE,OAAO,CAAC,GAAG,CAAC,UAAU;QAC/B,OAAO,EAAE;YACP,eAAe,EAAE,UAAU,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE,EAAE;SAC9D;QACD,IAAI,EAAE,mBAAmB;KAC1B,CAAC,CAAA;IACF,8DAA8D;IAC9D,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,cAAc,CAAC,CAAA;AACvC,CAAC;AAED,8BAA8B;AAC9B,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,QAAQ,CAAA"}

package/dist/test-types.d.ts

@@ -0,0 +1,13 @@
+import { type GenerateTextResult, type JSONValue, type CoreTool } from 'ai';
+import { Response } from 'undici';
+export type MockGenerateTextResult = GenerateTextResult<Record<string, CoreTool<any, any>>, JSONValue>;
+export type MockGenerateObjectResult = GenerateTextResult<Record<string, CoreTool<any, any>>, JSONValue> & {
+    object: JSONValue;
+    toJsonResponse: () => Response;
+};
+export declare const createMockTextResponse: (text: string) => MockGenerateTextResult;
+export declare const createMockObjectResponse: (object: JSONValue) => MockGenerateObjectResult;
+export declare const createMockStreamResponse: (chunks: string[]) => MockGenerateTextResult & {
+    experimental_stream: AsyncIterable<string>;
+};
+//# sourceMappingURL=test-types.d.ts.map

package/dist/test-types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-types.d.ts","sourceRoot":"","sources":["../src/test-types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,kBAAkB,EAA6B,KAAK,SAAS,EAAE,KAAK,QAAQ,EAAE,MAAM,IAAI,CAAA;AACtG,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAA;AAEjC,MAAM,MAAM,sBAAsB,GAAG,kBAAkB,CACrD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAClC,SAAS,CACV,CAAA;AAED,MAAM,MAAM,wBAAwB,GAAG,kBAAkB,CACvD,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,EAClC,SAAS,CACV,GAAG;IACF,MAAM,EAAE,SAAS,CAAA;IACjB,cAAc,EAAE,MAAM,QAAQ,CAAA;CAC/B,CAAA;AAED,eAAO,MAAM,sBAAsB,SAAU,MAAM,KAAG,sBAkBpD,CAAA;AAEF,eAAO,MAAM,wBAAwB,WAAY,SAAS,KAAG,wBAwB3D,CAAA;AAEF,eAAO,MAAM,wBAAwB,WAAY,MAAM,EAAE,KAAG,sBAAsB,GAAG;IACnF,mBAAmB,EAAE,aAAa,CAAC,MAAM,CAAC,CAAA;CAU1C,CAAA"}

package/dist/test-types.js

@@ -0,0 +1,55 @@
+import { Response } from 'undici';
+export const createMockTextResponse = (text) => ({
+    text,
+    usage: { promptTokens: 10, completionTokens: 20, totalTokens: 30 },
+    response: {
+        id: '1',
+        timestamp: new Date(),
+        modelId: 'test-model',
+        messages: [{ role: 'assistant', content: text }],
+    },
+    finishReason: 'stop',
+    warnings: [],
+    request: {},
+    logprobs: undefined,
+    toolCalls: [],
+    toolResults: [],
+    steps: [],
+    experimental_output: null,
+    experimental_providerMetadata: {},
+});
+export const createMockObjectResponse = (object) => ({
+    text: JSON.stringify(object),
+    object,
+    usage: { promptTokens: 10, completionTokens: 20, totalTokens: 30 },
+    response: {
+        id: '1',
+        timestamp: new Date(),
+        modelId: 'test-model',
+        messages: [{ role: 'assistant', content: JSON.stringify(object) }],
+    },
+    finishReason: 'stop',
+    warnings: [],
+    request: {},
+    logprobs: undefined,
+    toolCalls: [],
+    toolResults: [],
+    steps: [],
+    experimental_output: object,
+    experimental_providerMetadata: {},
+    toJsonResponse: () => new Response(JSON.stringify(object), {
+        status: 200,
+        headers: { 'Content-Type': 'application/json' },
+    }),
+});
+export const createMockStreamResponse = (chunks) => ({
+    ...createMockTextResponse(chunks.join('')),
+    experimental_stream: {
+        [Symbol.asyncIterator]: async function* () {
+            for (const chunk of chunks) {
+                yield chunk;
+            }
+        },
+    },
+});
+//# sourceMappingURL=test-types.js.map

package/dist/test-types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-types.js","sourceRoot":"","sources":["../src/test-types.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,QAAQ,CAAA;AAejC,MAAM,CAAC,MAAM,sBAAsB,GAAG,CAAC,IAAY,EAA0B,EAAE,CAAC,CAAC;IAC/E,IAAI;IACJ,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;IAClE,QAAQ,EAAE;QACR,EAAE,EAAE,GAAG;QACP,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,OAAO,EAAE,YAAY;QACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;KAC1D;IACD,YAAY,EAAE,MAAM;IACpB,QAAQ,EAAE,EAAE;IACZ,OAAO,EAAE,EAAE;IACX,QAAQ,EAAE,SAAS;IACnB,SAAS,EAAE,EAAE;IACb,WAAW,EAAE,EAAE;IACf,KAAK,EAAE,EAAE;IACT,mBAAmB,EAAE,IAAI;IACzB,6BAA6B,EAAE,EAAE;CAClC,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,MAAiB,EAA4B,EAAE,CAAC,CAAC;IACxF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;IAC5B,MAAM;IACN,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE;IAClE,QAAQ,EAAE;QACR,EAAE,EAAE,GAAG;QACP,SAAS,EAAE,IAAI,IAAI,EAAE;QACrB,OAAO,EAAE,YAAY;QACrB,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,WAAoB,EAAE,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;KAC5E;IACD,YAAY,EAAE,MAAM;IACpB,QAAQ,EAAE,EAAE;IACZ,OAAO,EAAE,EAAE;IACX,QAAQ,EAAE,SAAS;IACnB,SAAS,EAAE,EAAE;IACb,WAAW,EAAE,EAAE;IACf,KAAK,EAAE,EAAE;IACT,mBAAmB,EAAE,MAAM;IAC3B,6BAA6B,EAAE,EAAE;IACjC,cAAc,EAAE,GAAG,EAAE,CACnB,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE;QACnC,MAAM,EAAE,GAAG;QACX,OAAO,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE;KAChD,CAAC;CACL,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,MAAgB,EAEvD,EAAE,CAAC,CAAC;IACJ,GAAG,sBAAsB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC1C,mBAAmB,EAAE;QACnB,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,KAAK,SAAS,CAAC;YACrC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,MAAM,KAAK,CAAA;YACb,CAAC;QACH,CAAC;KACF;CACF,CAAC,CAAA"}

package/dist/types/index.d.ts

@@ -0,0 +1,4 @@
+export * from './schema';
+export * from './options';
+export * from './model';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,cAAc,WAAW,CAAA;AACzB,cAAc,SAAS,CAAA"}

package/dist/types/index.js

@@ -0,0 +1,4 @@
+export * from './schema';
+export * from './options';
+export * from './model';
+//# sourceMappingURL=index.js.map

package/dist/types/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/types/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAA;AACxB,cAAc,WAAW,CAAA;AACzB,cAAc,SAAS,CAAA"}

package/dist/types/list.d.ts

@@ -0,0 +1,10 @@
+export type ListFunction = {
+    (strings: string[] | TemplateStringsArray, ...values: any[]): AsyncIterableIterator<any>;
+    await: (strings: string[] | TemplateStringsArray, ...values: any[]) => Promise<string[]>;
+};
+export type ListOptions = {
+    model?: string;
+    system?: string;
+    seed?: number;
+};
+//# sourceMappingURL=list.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"list.d.ts","sourceRoot":"","sources":["../../src/types/list.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,YAAY,GAAG;IACzB,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,oBAAoB,EAAE,GAAG,MAAM,EAAE,GAAG,EAAE,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAA;IACxF,KAAK,EAAE,CAAC,OAAO,EAAE,MAAM,EAAE,GAAG,oBAAoB,EAAE,GAAG,MAAM,EAAE,GAAG,EAAE,KAAK,OAAO,CAAC,MAAM,EAAE,CAAC,CAAA;CACzF,CAAA;AAED,MAAM,MAAM,WAAW,GAAG;IACxB,KAAK,CAAC,EAAE,MAAM,CAAA;IACd,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,IAAI,CAAC,EAAE,MAAM,CAAA;CACd,CAAA"}

package/dist/types/list.js

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

package/dist/types/list.js.map

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

package/dist/types/model.d.ts

@@ -0,0 +1,7 @@
+import { OpenAIModel } from '@ai-sdk/openai';
+export type Model = string | OpenAIModel;
+export interface ExtendedOpenAIModel extends OpenAIModel {
+    type: 'openai';
+    model: string;
+}
+//# sourceMappingURL=model.d.ts.map

package/dist/types/model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.ts","sourceRoot":"","sources":["../../src/types/model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAA;AAE5C,MAAM,MAAM,KAAK,GAAG,MAAM,GAAG,WAAW,CAAA;AAExC,MAAM,WAAW,mBAAoB,SAAQ,WAAW;IACtD,IAAI,EAAE,QAAQ,CAAA;IACd,KAAK,EAAE,MAAM,CAAA;CACd"}

package/dist/types/model.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=model.js.map

package/dist/types/model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.js","sourceRoot":"","sources":["../../src/types/model.ts"],"names":[],"mappings":""}

package/dist/types/options.d.ts

@@ -0,0 +1,25 @@
+import { SimpleSchema } from './schema';
+import { Model } from './model';
+export type BaseOptions<T extends SimpleSchema> = {
+    model?: Model;
+    schema: T;
+    prompt?: string;
+    system?: string;
+    stream?: boolean;
+    seed?: number;
+    temperature?: number;
+    maxTokens?: number;
+};
+export type GenerateObjectOptions<T extends SimpleSchema> = BaseOptions<T> & {
+    output?: 'single' | 'array';
+};
+export type StreamObjectOptions<T extends SimpleSchema> = BaseOptions<T>;
+export type AIFunctionOptions = {
+    stream?: boolean;
+    system?: string;
+    model?: Model;
+    seed?: number;
+    temperature?: number;
+    maxTokens?: number;
+};
+//# sourceMappingURL=options.d.ts.map

package/dist/types/options.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.d.ts","sourceRoot":"","sources":["../../src/types/options.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,UAAU,CAAA;AACvC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAA;AAE/B,MAAM,MAAM,WAAW,CAAC,CAAC,SAAS,YAAY,IAAI;IAChD,KAAK,CAAC,EAAE,KAAK,CAAA;IACb,MAAM,EAAE,CAAC,CAAA;IACT,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,MAAM,CAAC,EAAE,OAAO,CAAA;IAChB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB,CAAA;AAED,MAAM,MAAM,qBAAqB,CAAC,CAAC,SAAS,YAAY,IAAI,WAAW,CAAC,CAAC,CAAC,GAAG;IAC3E,MAAM,CAAC,EAAE,QAAQ,GAAG,OAAO,CAAA;CAC5B,CAAA;AAED,MAAM,MAAM,mBAAmB,CAAC,CAAC,SAAS,YAAY,IAAI,WAAW,CAAC,CAAC,CAAC,CAAA;AAExE,MAAM,MAAM,iBAAiB,GAAG;IAC9B,MAAM,CAAC,EAAE,OAAO,CAAA;IAChB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,KAAK,CAAC,EAAE,KAAK,CAAA;IACb,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;CACnB,CAAA"}

package/dist/types/options.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=options.js.map

package/dist/types/options.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"options.js","sourceRoot":"","sources":["../../src/types/options.ts"],"names":[],"mappings":""}

package/dist/types/schema.d.ts

@@ -0,0 +1,5 @@
+export type SimpleSchema = {
+    [key: string]: SimpleSchemaValue;
+};
+export type SimpleSchemaValue = string | string[] | SimpleSchema;
+//# sourceMappingURL=schema.d.ts.map

package/dist/types/schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.ts","sourceRoot":"","sources":["../../src/types/schema.ts"],"names":[],"mappings":"AAEA,MAAM,MAAM,YAAY,GAAG;IACzB,CAAC,GAAG,EAAE,MAAM,GAAG,iBAAiB,CAAA;CACjC,CAAA;AAED,MAAM,MAAM,iBAAiB,GAAG,MAAM,GAAG,MAAM,EAAE,GAAG,YAAY,CAAA"}

package/dist/types/schema.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=schema.js.map

package/dist/types/schema.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.js","sourceRoot":"","sources":["../../src/types/schema.ts"],"names":[],"mappings":""}

package/dist/types.d.ts

@@ -0,0 +1,113 @@
+import { z } from 'zod';
+import type PQueue from 'p-queue';
+import type { LanguageModelV1 } from 'ai';
+export type Queue = Omit<PQueue, 'add'> & {
+    add<T, R = T>(fn: () => Promise<T> | AsyncGenerator<T, void, unknown>): Promise<R>;
+};
+export interface RetryOptions {
+    maxRetries: number;
+    initialDelay: number;
+    maxDelay: number;
+    backoffFactor: number;
+}
+export interface RateLimitOptions {
+    requestsPerMinute: number;
+    burstLimit?: number;
+    timeoutMs?: number;
+}
+export interface RequestHandlingOptions {
+    retry?: RetryOptions;
+    rateLimit?: RateLimitOptions;
+    timeout?: number;
+    maxRetries?: number;
+    retryDelay?: number;
+    streamingTimeout?: number;
+    concurrency?: number;
+    requestHandling?: {
+        retry?: RetryOptions;
+        rateLimit?: RateLimitOptions;
+        timeout?: number;
+        concurrency?: number;
+    };
+}
+export interface StreamProgress {
+    type: 'token' | 'chunk' | 'complete';
+    tokensGenerated?: number;
+    totalTokens?: number;
+    chunk?: string;
+    estimatedTimeRemaining?: number;
+}
+export type ProgressCallback = (progress: StreamProgress) => void;
+export interface StreamingOptions {
+    onProgress?: ProgressCallback;
+    enableTokenCounting?: boolean;
+    estimateTimeRemaining?: boolean;
+}
+export interface AIFunctionOptions {
+    model?: LanguageModelV1;
+    prompt?: string;
+    outputFormat?: 'json';
+    schema?: z.ZodType | Record<string, unknown>;
+    structuredOutputs?: boolean;
+    system?: string;
+    temperature?: number;
+    maxTokens?: number;
+    topP?: number;
+    frequencyPenalty?: number;
+    presencePenalty?: number;
+    stop?: string | string[];
+    seed?: number;
+    concurrency?: number;
+    requestHandling?: RequestHandlingOptions;
+    streaming?: StreamingOptions;
+}
+export type AIFunction<T extends z.ZodTypeAny = z.ZodTypeAny> = {
+    (): Promise<{
+        schema: T;
+    }>;
+    (args: z.infer<T>): Promise<z.infer<T>>;
+    (args: z.infer<T>, options: AIFunctionOptions): Promise<z.infer<T>>;
+    schema?: T;
+    queue?: Queue;
+};
+export type AsyncIterablePromise<T> = Promise<T> & {
+    (options?: AIFunctionOptions): Promise<T>;
+    then: Promise<T>['then'];
+    catch: Promise<T>['catch'];
+    finally: Promise<T>['finally'];
+    [Symbol.asyncIterator]: () => AsyncIterator<string>;
+};
+export type TemplateResult = {
+    (options?: AIFunctionOptions): Promise<string>;
+    then: Promise<string>['then'];
+    catch: Promise<string>['catch'];
+    finally: Promise<string>['finally'];
+    [Symbol.asyncIterator]: () => AsyncIterator<string>;
+    call: (options?: AIFunctionOptions) => Promise<string>;
+};
+export interface BaseTemplateFunction {
+    (strings: TemplateStringsArray, ...values: unknown[]): TemplateResult;
+    (options?: AIFunctionOptions): TemplateResult;
+    [Symbol.asyncIterator]: () => AsyncIterator<string>;
+    withOptions: (options?: AIFunctionOptions) => Promise<string>;
+    queue?: Queue;
+}
+export type AITemplateFunction = BaseTemplateFunction & {
+    (strings: TemplateStringsArray, ...values: unknown[]): TemplateResult;
+    withOptions: (options?: AIFunctionOptions) => AsyncIterablePromise<string>;
+};
+export interface AI extends AITemplateFunction {
+    categorizeProduct: AIFunction<z.ZodObject<{
+        productType: z.ZodEnum<['App', 'API', 'Marketplace', 'Platform', 'Packaged Service', 'Professional Service', 'Website']>;
+        customer: z.ZodString;
+        solution: z.ZodString;
+        description: z.ZodString;
+    }>>;
+}
+export type ListFunction = BaseTemplateFunction;
+export declare class AIRequestError extends Error {
+    readonly cause?: unknown | undefined;
+    readonly retryable: boolean;
+    constructor(message: string, cause?: unknown | undefined, retryable?: boolean);
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,KAAK,MAAM,MAAM,SAAS,CAAA;AACjC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,IAAI,CAAA;AAEzC,MAAM,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,GAAG;IACxC,GAAG,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC,CAAC,CAAC,GAAG,cAAc,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAA;CACnF,CAAA;AAED,MAAM,WAAW,YAAY;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,gBAAgB;IAC/B,iBAAiB,EAAE,MAAM,CAAC;IAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,sBAAsB;IACrC,KAAK,CAAC,EAAE,YAAY,CAAC;IACrB,SAAS,CAAC,EAAE,gBAAgB,CAAC;IAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,eAAe,CAAC,EAAE;QAChB,KAAK,CAAC,EAAE,YAAY,CAAC;QACrB,SAAS,CAAC,EAAE,gBAAgB,CAAC;QAC7B,OAAO,CAAC,EAAE,MAAM,CAAC;QACjB,WAAW,CAAC,EAAE,MAAM,CAAC;KACtB,CAAC;CACH;AAED,MAAM,WAAW,cAAc;IAC7B,IAAI,EAAE,OAAO,GAAG,OAAO,GAAG,UAAU,CAAC;IACrC,eAAe,CAAC,EAAE,MAAM,CAAC;IACzB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,sBAAsB,CAAC,EAAE,MAAM,CAAC;CACjC;AAED,MAAM,MAAM,gBAAgB,GAAG,CAAC,QAAQ,EAAE,cAAc,KAAK,IAAI,CAAC;AAElE,MAAM,WAAW,gBAAgB;IAC/B,UAAU,CAAC,EAAE,gBAAgB,CAAC;IAC9B,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,qBAAqB,CAAC,EAAE,OAAO,CAAC;CACjC;AAED,MAAM,WAAW,iBAAiB;IAChC,KAAK,CAAC,EAAE,eAAe,CAAA;IACvB,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,YAAY,CAAC,EAAE,MAAM,CAAA;IACrB,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;IAC5C,iBAAiB,CAAC,EAAE,OAAO,CAAA;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAA;IACf,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,SAAS,CAAC,EAAE,MAAM,CAAA;IAClB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,gBAAgB,CAAC,EAAE,MAAM,CAAA;IACzB,eAAe,CAAC,EAAE,MAAM,CAAA;IACxB,IAAI,CAAC,EAAE,MAAM,GAAG,MAAM,EAAE,CAAA;IACxB,IAAI,CAAC,EAAE,MAAM,CAAA;IACb,WAAW,CAAC,EAAE,MAAM,CAAA;IACpB,eAAe,CAAC,EAAE,sBAAsB,CAAC;IACzC,SAAS,CAAC,EAAE,gBAAgB,CAAC;CAC9B;AAED,MAAM,MAAM,UAAU,CAAC,CAAC,SAAS,CAAC,CAAC,UAAU,GAAG,CAAC,CAAC,UAAU,IAAI;IAC9D,IAAI,OAAO,CAAC;QAAE,MAAM,EAAE,CAAC,CAAA;KAAE,CAAC,CAAA;IAC1B,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;IACvC,CAAC,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;IACnE,MAAM,CAAC,EAAE,CAAC,CAAA;IACV,KAAK,CAAC,EAAE,KAAK,CAAA;CACd,CAAA;AAED,MAAM,MAAM,oBAAoB,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,CAAC,GAAG;IACjD,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;IAC1C,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IACzB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;IAC3B,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;CACrD,CAAA;AAED,MAAM,MAAM,cAAc,GAAG;IAC3B,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAC/C,IAAI,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,CAAC,CAAC;IAC9B,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC;IAChC,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC;IACpC,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;IACpD,IAAI,EAAE,CAAC,OAAO,CAAC,EAAE,iBAAiB,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACxD,CAAA;AAED,MAAM,WAAW,oBAAoB;IACnC,CAAC,OAAO,EAAE,oBAAoB,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,GAAG,cAAc,CAAC;IACtE,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,cAAc,CAAC;IAC9C,CAAC,MAAM,CAAC,aAAa,CAAC,EAAE,MAAM,aAAa,CAAC,MAAM,CAAC,CAAC;IACpD,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE,iBAAiB,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IAC9D,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,MAAM,MAAM,kBAAkB,GAAG,oBAAoB,GAAG;IACtD,CAAC,OAAO,EAAE,oBAAoB,EAAE,GAAG,MAAM,EAAE,OAAO,EAAE,GAAG,cAAc,CAAA;IACrE,WAAW,EAAE,CAAC,OAAO,CAAC,EAAE,iBAAiB,KAAK,oBAAoB,CAAC,MAAM,CAAC,CAAA;CAC3E,CAAA;AAED,MAAM,WAAW,EAAG,SAAQ,kBAAkB;IAC5C,iBAAiB,EAAE,UAAU,CAC3B,CAAC,CAAC,SAAS,CAAC;QACV,WAAW,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,kBAAkB,EAAE,sBAAsB,EAAE,SAAS,CAAC,CAAC,CAAA;QACxH,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAA;QACrB,QAAQ,EAAE,CAAC,CAAC,SAAS,CAAA;QACrB,WAAW,EAAE,CAAC,CAAC,SAAS,CAAA;KACzB,CAAC,CACH,CAAA;CACF;AAED,MAAM,MAAM,YAAY,GAAG,oBAAoB,CAAA;AAE/C,qBAAa,cAAe,SAAQ,KAAK;aAGrB,KAAK,CAAC,EAAE,OAAO;aACf,SAAS,EAAE,OAAO;gBAFlC,OAAO,EAAE,MAAM,EACC,KAAK,CAAC,EAAE,OAAO,YAAA,EACf,SAAS,GAAE,OAAc;CAK5C"}

package/dist/types.js

@@ -0,0 +1,9 @@
+export class AIRequestError extends Error {
+    constructor(message, cause, retryable = true) {
+        super(message);
+        this.cause = cause;
+        this.retryable = retryable;
+        this.name = 'AIRequestError';
+    }
+}
+//# sourceMappingURL=types.js.map

package/dist/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AA2HA,MAAM,OAAO,cAAe,SAAQ,KAAK;IACvC,YACE,OAAe,EACC,KAAe,EACf,YAAqB,IAAI;QAEzC,KAAK,CAAC,OAAO,CAAC,CAAC;QAHC,UAAK,GAAL,KAAK,CAAU;QACf,cAAS,GAAT,SAAS,CAAgB;QAGzC,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;IAC/B,CAAC;CACF"}

package/dist/utils/__tests__/request-handler.test.d.ts

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

package/dist/utils/__tests__/request-handler.test.d.ts.map

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

package/dist/utils/__tests__/request-handler.test.js

@@ -0,0 +1,134 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { createRequestHandler } from '../request-handler';
+import { AIRequestError } from '../../types';
+describe('RequestHandler', () => {
+    beforeEach(() => {
+        vi.useFakeTimers();
+    });
+    afterEach(() => {
+        vi.useRealTimers();
+    });
+    it('should successfully execute an operation', async () => {
+        const handler = createRequestHandler();
+        const operation = vi.fn().mockResolvedValue('success');
+        const promise = handler.execute(operation);
+        await vi.runAllTimersAsync();
+        const result = await promise;
+        expect(result).toBe('success');
+        expect(operation).toHaveBeenCalledTimes(1);
+    });
+    it('should retry on failure', async () => {
+        const handler = createRequestHandler({
+            requestHandling: {
+                retry: {
+                    maxRetries: 2,
+                    initialDelay: 100,
+                    maxDelay: 1000,
+                    backoffFactor: 2,
+                }
+            }
+        });
+        const operation = vi.fn()
+            .mockRejectedValueOnce(new Error('Temporary error'))
+            .mockRejectedValueOnce(new Error('Another temporary error'))
+            .mockResolvedValueOnce('success');
+        const promise = handler.execute(operation);
+        // Advance time for first retry
+        await vi.advanceTimersByTimeAsync(100);
+        // Advance time for second retry
+        await vi.advanceTimersByTimeAsync(200);
+        // Advance remaining timers
+        await vi.runAllTimersAsync();
+        const result = await promise;
+        expect(result).toBe('success');
+        expect(operation).toHaveBeenCalledTimes(3);
+    });
+    it('should respect rate limits', async () => {
+        const handler = createRequestHandler({
+            requestHandling: {
+                rateLimit: {
+                    requestsPerMinute: 60,
+                    burstLimit: 2,
+                }
+            }
+        });
+        const operation = vi.fn().mockResolvedValue('success');
+        // First two requests should be immediate
+        const promise1 = handler.execute(operation);
+        const promise2 = handler.execute(operation);
+        await vi.runAllTimersAsync();
+        await Promise.all([promise1, promise2]);
+        // Third request should be delayed
+        const startTime = Date.now();
+        const promise3 = handler.execute(operation);
+        await vi.advanceTimersByTimeAsync(1000);
+        await vi.runAllTimersAsync();
+        await promise3;
+        const duration = Date.now() - startTime;
+        expect(duration).toBeGreaterThanOrEqual(1000);
+        expect(operation).toHaveBeenCalledTimes(3);
+    });
+    it('should handle timeouts', async () => {
+        const timeout = 1000;
+        const handler = createRequestHandler({
+            requestHandling: {
+                timeout,
+                retry: {
+                    maxRetries: 0,
+                    initialDelay: 100,
+                    maxDelay: 1000,
+                    backoffFactor: 2
+                }
+            }
+        });
+        const operation = vi.fn(() => new Promise(resolve => {
+            setTimeout(resolve, 2000, 'delayed result');
+        }));
+        const promise = handler.execute(operation);
+        await vi.advanceTimersByTimeAsync(timeout);
+        await vi.runAllTimersAsync();
+        await expect(promise).rejects.toThrow(`Request timed out after ${timeout}ms`);
+    });
+    it('should not retry on non-retryable errors', async () => {
+        const handler = createRequestHandler();
+        const operation = vi.fn().mockRejectedValue(new AIRequestError('Non-retryable error', undefined, false));
+        const promise = handler.execute(operation);
+        await vi.runAllTimersAsync();
+        await expect(promise).rejects.toThrow('Non-retryable error');
+        expect(operation).toHaveBeenCalledTimes(1);
+    });
+    it('should respect max retries', async () => {
+        const handler = createRequestHandler({
+            requestHandling: {
+                retry: {
+                    maxRetries: 2,
+                    initialDelay: 100,
+                    maxDelay: 1000,
+                    backoffFactor: 2,
+                }
+            }
+        });
+        const operation = vi.fn().mockRejectedValue(new Error('Persistent error'));
+        const promise = handler.execute(operation);
+        // Advance time for all retries
+        await vi.advanceTimersByTimeAsync(100);
+        await vi.advanceTimersByTimeAsync(200);
+        await vi.advanceTimersByTimeAsync(400);
+        await vi.runAllTimersAsync();
+        await expect(promise).rejects.toThrow('Failed after 3 attempts');
+        expect(operation).toHaveBeenCalledTimes(3);
+    });
+    it('should handle concurrent requests with queue', async () => {
+        const handler = createRequestHandler({
+            requestHandling: { concurrency: 2 }
+        });
+        const operation = vi.fn().mockResolvedValue('success');
+        const tasks = Array(5).fill(null).map(() => handler.execute(operation));
+        await vi.runAllTimersAsync();
+        const results = await Promise.all(tasks);
+        expect(results).toHaveLength(5);
+        results.forEach(result => expect(result).toBe('success'));
+        expect(operation).toHaveBeenCalledTimes(5);
+    });
+});
+//# sourceMappingURL=request-handler.test.js.map

package/dist/utils/__tests__/request-handler.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"request-handler.test.js","sourceRoot":"","sources":["../../../src/utils/__tests__/request-handler.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAA;AACxE,OAAO,EAAkB,oBAAoB,EAAE,MAAM,oBAAoB,CAAA;AACzE,OAAO,EAAE,cAAc,EAAE,MAAM,aAAa,CAAA;AAE5C,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,EAAE,CAAC,aAAa,EAAE,CAAA;IACpB,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAA;QACtC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QAEtD,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAA;QAE5B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC9B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE;aACtB,qBAAqB,CAAC,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;aACnD,qBAAqB,CAAC,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;aAC3D,qBAAqB,CAAC,SAAS,CAAC,CAAA;QAEnC,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAE1C,+BAA+B;QAC/B,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,gCAAgC;QAChC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,2BAA2B;QAC3B,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAA;QAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAA;QAC9B,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,SAAS,EAAE;oBACT,iBAAiB,EAAE,EAAE;oBACrB,UAAU,EAAE,CAAC;iBACd;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QAEtD,yCAAyC;QACzC,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAA;QAEvC,kCAAkC;QAClC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAA;QAC5B,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC3C,MAAM,EAAE,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAA;QACvC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,QAAQ,CAAA;QAEd,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAA;QACvC,MAAM,CAAC,QAAQ,CAAC,CAAC,sBAAsB,CAAC,IAAI,CAAC,CAAA;QAC7C,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;QACtC,MAAM,OAAO,GAAG,IAAI,CAAC;QACrB,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,OAAO;gBACP,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YAClD,UAAU,CAAC,OAAO,EAAE,IAAI,EAAE,gBAAgB,CAAC,CAAA;QAC7C,CAAC,CAAC,CAAC,CAAA;QAEH,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,2BAA2B,OAAO,IAAI,CAAC,CAAA;IAC/E,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAA;QACtC,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CACzC,IAAI,cAAc,CAAC,qBAAqB,EAAE,SAAS,EAAE,KAAK,CAAC,CAC5D,CAAA;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAC1C,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,qBAAqB,CAAC,CAAA;QAC5D,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE;gBACf,KAAK,EAAE;oBACL,UAAU,EAAE,CAAC;oBACb,YAAY,EAAE,GAAG;oBACjB,QAAQ,EAAE,IAAI;oBACd,aAAa,EAAE,CAAC;iBACjB;aACF;SACF,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,IAAI,KAAK,CAAC,kBAAkB,CAAC,CAAC,CAAA;QAC1E,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAA;QAE1C,+BAA+B;QAC/B,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,wBAAwB,CAAC,GAAG,CAAC,CAAA;QACtC,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAE5B,MAAM,MAAM,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CAAA;QAChE,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;QAC5D,MAAM,OAAO,GAAG,oBAAoB,CAAC;YACnC,eAAe,EAAE,EAAE,WAAW,EAAE,CAAC,EAAE;SACpC,CAAC,CAAA;QAEF,MAAM,SAAS,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAA;QACtD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAA;QAEvE,MAAM,EAAE,CAAC,iBAAiB,EAAE,CAAA;QAC5B,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAA;QAExC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAC/B,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAA;QACzD,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAA;IAC5C,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/utils/__tests__/schema.test.d.ts

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

package/dist/utils/__tests__/schema.test.d.ts.map

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