@polka-codes/cli 0.10.23 → 0.10.24

package/dist/file-attachments.test.js

@@ -1,214 +0,0 @@
-import { describe, expect, it } from 'bun:test';
-import { attachFilesToContent, countFilesByType, createUserMessageWithFiles, filterFilesByMediaType, getDocuments, getImages, } from './file-attachments';
-describe('file-attachments', () => {
-    describe('attachFilesToContent', () => {
-        it('should return content unchanged when no files provided', () => {
-            const content = 'Hello, world!';
-            const result = attachFilesToContent(content);
-            expect(result).toBe(content);
-        });
-        it('should return content unchanged when empty array provided', () => {
-            const content = 'Hello, world!';
-            const result = attachFilesToContent(content, []);
-            expect(result).toBe(content);
-        });
-        it('should attach files to string content', () => {
-            const content = 'Hello, world!';
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-            ];
-            const result = attachFilesToContent(content, files);
-            expect(Array.isArray(result)).toBe(true);
-            if (Array.isArray(result)) {
-                expect(result).toHaveLength(2);
-                expect(result[0]).toEqual({ type: 'text', text: content });
-                expect(result[1]).toEqual(files[0]);
-            }
-        });
-        it('should attach files to array content', () => {
-            const content = [{ type: 'text', text: 'Hello' }];
-            const files = [
-                {
-                    type: 'image',
-                    image: {
-                        type: 'base64',
-                        value: 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==',
-                    },
-                    mediaType: 'image/png',
-                },
-            ];
-            const result = attachFilesToContent(content, files);
-            expect(Array.isArray(result)).toBe(true);
-            if (Array.isArray(result)) {
-                expect(result).toHaveLength(2);
-                expect(result[0]).toEqual(content[0]);
-                expect(result[1]).toEqual(files[0]);
-            }
-        });
-        it('should attach multiple files', () => {
-            const content = 'Check these files';
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-                {
-                    type: 'image',
-                    image: {
-                        type: 'base64',
-                        value: 'iVBORw0KGgoAAAANSUhEUgAAAAEAAAABCAYAAAAfFcSJAAAADUlEQVR42mNk+M9QDwADhgGAWjR9awAAAABJRU5ErkJggg==',
-                    },
-                    mediaType: 'image/png',
-                },
-            ];
-            const result = attachFilesToContent(content, files);
-            expect(Array.isArray(result)).toBe(true);
-            if (Array.isArray(result)) {
-                expect(result).toHaveLength(3);
-                expect(result[0]).toEqual({ type: 'text', text: content });
-                expect(result[1]).toEqual(files[0]);
-                expect(result[2]).toEqual(files[1]);
-            }
-        });
-    });
-    describe('createUserMessageWithFiles', () => {
-        it('should create user message without files', () => {
-            const result = createUserMessageWithFiles('Hello');
-            expect(result).toEqual({
-                role: 'user',
-                content: [{ type: 'text', text: 'Hello' }],
-            });
-        });
-        it('should create user message with files', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-            ];
-            const result = createUserMessageWithFiles('Check this file', files);
-            expect(result.role).toBe('user');
-            expect(Array.isArray(result.content)).toBe(true);
-            if (Array.isArray(result.content)) {
-                expect(result.content).toHaveLength(2);
-                expect(result.content[0]).toEqual({ type: 'text', text: 'Check this file' });
-                expect(result.content[1]).toEqual(files[0]);
-            }
-        });
-    });
-    describe('filterFilesByMediaType', () => {
-        const files = [
-            { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-            { type: 'file', data: { type: 'base64', value: 'eyJ0ZXN0IjoidHJ1ZSJ9' }, filename: 'test.json', mediaType: 'application/json' },
-            { type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' },
-            { type: 'image', image: { type: 'base64', value: 'DEF' }, mediaType: 'image/jpeg' },
-            { type: 'file', data: { type: 'base64', value: 'V29ybGQ=' }, filename: 'world.txt', mediaType: 'text/plain' },
-        ];
-        it('should filter files by string media type', () => {
-            const result = filterFilesByMediaType(files, 'text/plain');
-            expect(result).toHaveLength(2);
-            expect(result.every((f) => f.mediaType === 'text/plain')).toBe(true);
-        });
-        it('should filter files by RegExp media type', () => {
-            const result = filterFilesByMediaType(files, /^image\//);
-            expect(result).toHaveLength(2);
-            expect(result.every((f) => f.mediaType?.startsWith('image/'))).toBe(true);
-        });
-        it('should filter files by application media type', () => {
-            const result = filterFilesByMediaType(files, 'application/json');
-            expect(result).toHaveLength(1);
-            expect(result[0].mediaType).toBe('application/json');
-        });
-        it('should return empty array when no files match', () => {
-            const result = filterFilesByMediaType(files, 'video/mp4');
-            expect(result).toHaveLength(0);
-        });
-    });
-    describe('getImages', () => {
-        it('should extract image files from mixed array', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-                { type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' },
-                { type: 'image', image: { type: 'base64', value: 'DEF' }, mediaType: 'image/jpeg' },
-            ];
-            const result = getImages(files);
-            expect(result).toHaveLength(2);
-            expect(result.every((f) => f.type === 'image')).toBe(true);
-        });
-        it('should return empty array when no images', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-            ];
-            const result = getImages(files);
-            expect(result).toHaveLength(0);
-        });
-        it('should return empty array for empty input', () => {
-            const result = getImages([]);
-            expect(result).toHaveLength(0);
-        });
-    });
-    describe('getDocuments', () => {
-        it('should extract document files from mixed array', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-                { type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' },
-                { type: 'file', data: { type: 'base64', value: 'REF=' }, filename: 'test.pdf', mediaType: 'application/pdf' },
-            ];
-            const result = getDocuments(files);
-            expect(result).toHaveLength(2);
-            expect(result.every((f) => f.type === 'file')).toBe(true);
-        });
-        it('should return empty array when no documents', () => {
-            const files = [{ type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' }];
-            const result = getDocuments(files);
-            expect(result).toHaveLength(0);
-        });
-        it('should return empty array for empty input', () => {
-            const result = getDocuments([]);
-            expect(result).toHaveLength(0);
-        });
-    });
-    describe('countFilesByType', () => {
-        it('should count files by type', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-                { type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' },
-                { type: 'image', image: { type: 'base64', value: 'DEF' }, mediaType: 'image/jpeg' },
-                { type: 'file', data: { type: 'base64', value: 'REF=' }, filename: 'test.pdf', mediaType: 'application/pdf' },
-            ];
-            const result = countFilesByType(files);
-            expect(result).toEqual({
-                images: 2,
-                documents: 2,
-                total: 4,
-            });
-        });
-        it('should return zero counts for empty array', () => {
-            const result = countFilesByType([]);
-            expect(result).toEqual({
-                images: 0,
-                documents: 0,
-                total: 0,
-            });
-        });
-        it('should count only images', () => {
-            const files = [
-                { type: 'image', image: { type: 'base64', value: 'ABC' }, mediaType: 'image/png' },
-                { type: 'image', image: { type: 'base64', value: 'DEF' }, mediaType: 'image/jpeg' },
-            ];
-            const result = countFilesByType(files);
-            expect(result).toEqual({
-                images: 2,
-                documents: 0,
-                total: 2,
-            });
-        });
-        it('should count only files', () => {
-            const files = [
-                { type: 'file', data: { type: 'base64', value: 'SGVsbG8=' }, filename: 'test.txt', mediaType: 'text/plain' },
-                { type: 'file', data: { type: 'base64', value: 'REF=' }, filename: 'test.pdf', mediaType: 'application/pdf' },
-            ];
-            const result = countFilesByType(files);
-            expect(result).toEqual({
-                images: 0,
-                documents: 2,
-                total: 2,
-            });
-        });
-    });
-});
-//# sourceMappingURL=file-attachments.test.js.map

package/dist/file-attachments.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"file-attachments.test.js","sourceRoot":"","sources":["../src/file-attachments.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,UAAU,CAAA;AAE/C,OAAO,EACL,oBAAoB,EACpB,gBAAgB,EAChB,0BAA0B,EAC1B,sBAAsB,EACtB,YAAY,EACZ,SAAS,GACV,MAAM,oBAAoB,CAAA;AAE3B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,wDAAwD,EAAE,GAAG,EAAE;YAChE,MAAM,OAAO,GAAG,eAAe,CAAA;YAC/B,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,CAAC,CAAA;YAC5C,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC9B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2DAA2D,EAAE,GAAG,EAAE;YACnE,MAAM,OAAO,GAAG,eAAe,CAAA;YAC/B,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAA;YAChD,MAAM,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;QAC9B,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,OAAO,GAAG,eAAe,CAAA;YAC/B,MAAM,KAAK,GAAmB;gBAC5B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;aAC7G,CAAA;YACD,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;YAEnD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACxC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;gBAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;gBAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YACrC,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAA0C,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;YACxF,MAAM,KAAK,GAAoB;gBAC7B;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,kGAAkG;qBAC1G;oBACD,SAAS,EAAE,WAAW;iBACvB;aACF,CAAA;YACD,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;YAEnD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACxC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;gBAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAA;gBACrC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YACrC,CAAC;QACH,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;YACtC,MAAM,OAAO,GAAG,mBAAmB,CAAA;YACnC,MAAM,KAAK,GAAqC;gBAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;gBAC5G;oBACE,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE;wBACL,IAAI,EAAE,QAAQ;wBACd,KAAK,EAAE,kGAAkG;qBAC1G;oBACD,SAAS,EAAE,WAAW;iBACvB;aACF,CAAA;YACD,MAAM,MAAM,GAAG,oBAAoB,CAAC,OAAO,EAAE,KAAK,CAAC,CAAA;YAEnD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YACxC,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC1B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;gBAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;gBAC1D,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;gBACnC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YACrC,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,4BAA4B,EAAE,GAAG,EAAE;QAC1C,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAG,0BAA0B,CAAC,OAAO,CAAC,CAAA;YAElD,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;aAC3C,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;YAC/C,MAAM,KAAK,GAAmB;gBAC5B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;aAC7G,CAAA;YACD,MAAM,MAAM,GAAG,0BAA0B,CAAC,iBAAiB,EAAE,KAAK,CAAC,CAAA;YAEnE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAA;YAChC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;YAChD,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;gBACtC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC,CAAA;gBAC5E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAA;YAC7C,CAAC;QACH,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAqC;YAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;YAC5G,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,sBAAsB,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,kBAAkB,EAAE;YAC/H,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;YAClF,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;YACnF,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,WAAW,EAAE,SAAS,EAAE,YAAY,EAAE;SAC9G,CAAA;QAED,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAG,sBAAsB,CAAC,KAAK,EAAE,YAAY,CAAC,CAAA;YAE1D,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,YAAY,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QACtE,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,MAAM,GAAG,sBAAsB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAA;YAExD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,EAAE,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3E,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAG,sBAAsB,CAAC,KAAK,EAAE,kBAAkB,CAAC,CAAA;YAEhE,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAA;QACtD,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;YACvD,MAAM,MAAM,GAAG,sBAAsB,CAAC,KAAK,EAAE,WAAW,CAAC,CAAA;YAEzD,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;QACzB,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,KAAK,GAAqC;gBAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;gBAC5G,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;gBAClF,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;aACpF,CAAA;YAED,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,CAAA;YAE/B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC5D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,KAAK,GAAmB;gBAC5B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;aAC7G,CAAA;YAED,MAAM,MAAM,GAAG,SAAS,CAAC,KAAK,CAAC,CAAA;YAE/B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,CAAC,CAAA;YAE5B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,gDAAgD,EAAE,GAAG,EAAE;YACxD,MAAM,KAAK,GAAqC;gBAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;gBAC5G,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;gBAClF,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,iBAAiB,EAAE;aAC9G,CAAA;YAED,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;YAElC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;YAC9B,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;QAC3D,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,KAAK,GAAoB,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE,CAAC,CAAA;YAEnH,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,CAAC,CAAA;YAElC,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,YAAY,CAAC,EAAE,CAAC,CAAA;YAE/B,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAA;QAChC,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;IAEF,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,KAAK,GAAqC;gBAC9C,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;gBAC5G,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;gBAClF,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;gBACnF,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,iBAAiB,EAAE;aAC9G,CAAA;YAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAA;YAEtC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,MAAM,GAAG,gBAAgB,CAAC,EAAE,CAAC,CAAA;YAEnC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,0BAA0B,EAAE,GAAG,EAAE;YAClC,MAAM,KAAK,GAAoB;gBAC7B,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,WAAW,EAAE;gBAClF,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,YAAY,EAAE;aACpF,CAAA;YAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAA;YAEtC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;QAEF,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;YACjC,MAAM,KAAK,GAAmB;gBAC5B,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE;gBAC5G,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,SAAS,EAAE,iBAAiB,EAAE;aAC9G,CAAA;YAED,MAAM,MAAM,GAAG,gBAAgB,CAAC,KAAK,CAAC,CAAA;YAEtC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,KAAK,EAAE,CAAC;aACT,CAAC,CAAA;QACJ,CAAC,CAAC,CAAA;IACJ,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/getModel.js

@@ -1,229 +0,0 @@
-import { appendFileSync } from 'node:fs';
-import { inspect } from 'node:util';
-import { createAnthropic } from '@ai-sdk/anthropic';
-import { createDeepSeek } from '@ai-sdk/deepseek';
-import { createGoogleGenerativeAI } from '@ai-sdk/google';
-import { createVertex } from '@ai-sdk/google-vertex';
-import { createOpenAI } from '@ai-sdk/openai';
-import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
-import { createOpenRouter } from '@openrouter/ai-sdk-provider';
-import { getEnv } from './env';
-function headersToObject(headers) {
-    if (!headers) {
-        return undefined;
-    }
-    if (headers instanceof Headers) {
-        return Object.fromEntries(headers.entries());
-    }
-    if (Array.isArray(headers)) {
-        return Object.fromEntries(headers);
-    }
-    return headers;
-}
-function redactHeaders(headers) {
-    if (!headers) {
-        return undefined;
-    }
-    const redactedHeaders = {};
-    const sensitiveKeywords = ['authorization', 'cookie', 'key', 'token'];
-    for (const [key, value] of Object.entries(headers)) {
-        if (sensitiveKeywords.some((keyword) => key.toLowerCase().includes(keyword))) {
-            redactedHeaders[key] = 'REDACTED';
-        }
-        else {
-            redactedHeaders[key] = value;
-        }
-    }
-    return redactedHeaders;
-}
-export var AiProvider;
-(function (AiProvider) {
-    AiProvider["Anthropic"] = "anthropic";
-    AiProvider["DeepSeek"] = "deepseek";
-    AiProvider["OpenRouter"] = "openrouter";
-    AiProvider["OpenAI"] = "openai";
-    AiProvider["OpenAICompatible"] = "openai-compatible";
-    AiProvider["GoogleVertex"] = "google-vertex";
-    AiProvider["Google"] = "google";
-})(AiProvider || (AiProvider = {}));
-export const getModel = (config, debugLogging = false) => {
-    const { TRACING_FILE } = getEnv();
-    const fetchOverride = debugLogging || TRACING_FILE
-        ? (async (url, options) => {
-            const requestBody = options?.body ? JSON.parse(options.body) : undefined;
-            if (debugLogging) {
-                console.error('-> Request URL:', url);
-                console.error('-> Request Headers:', options?.headers);
-                console.error('-> Request Body:');
-                console.error(inspect(requestBody, { depth: null, colors: process.stderr.isTTY }));
-            }
-            if (TRACING_FILE) {
-                appendFileSync(TRACING_FILE, `${JSON.stringify({
-                    type: 'request',
-                    timestamp: new Date().toISOString(),
-                    url,
-                    headers: redactHeaders(headersToObject(options?.headers)),
-                    body: requestBody,
-                }, null, 2)}\n`);
-            }
-            const res = await fetch(url, options);
-            if (debugLogging) {
-                console.error('<- Response Status:', res.status);
-            }
-            const contentType = res.headers.get('content-type') || '';
-            if (contentType.includes('text/event-stream') && res.body) {
-                const [branch, clientStream] = res.body.tee();
-                (async () => {
-                    const reader = branch.getReader();
-                    const decoder = new TextDecoder();
-                    try {
-                        let done = false;
-                        while (!done) {
-                            const { value, done: d } = await reader.read();
-                            done = d;
-                            if (value) {
-                                const text = decoder.decode(value);
-                                if (debugLogging) {
-                                    console.error('<- Stream chunk:', text.replace(/\n/g, '\\n'));
-                                }
-                                if (TRACING_FILE) {
-                                    for (const line of text.split('\n')) {
-                                        if (line.startsWith('data:')) {
-                                            const content = line.slice('data:'.length).trim();
-                                            if (content) {
-                                                try {
-                                                    const json = JSON.parse(content);
-                                                    appendFileSync(TRACING_FILE, `${JSON.stringify({
-                                                        type: 'response-chunk',
-                                                        timestamp: new Date().toISOString(),
-                                                        chunk: json,
-                                                    }, null, 2)}\n`);
-                                                }
-                                                catch (_e) {
-                                                    appendFileSync(TRACING_FILE, `${JSON.stringify({
-                                                        type: 'response-chunk',
-                                                        timestamp: new Date().toISOString(),
-                                                        chunk: content,
-                                                    }, null, 2)}\n`);
-                                                }
-                                            }
-                                        }
-                                    }
-                                }
-                            }
-                        }
-                    }
-                    finally {
-                        // Ensure the reader is properly closed to prevent memory leaks
-                        reader.releaseLock();
-                    }
-                })().catch((error) => {
-                    if (debugLogging) {
-                        console.error('Stream reading error:', error);
-                    }
-                });
-                return new Response(clientStream, {
-                    headers: res.headers,
-                    status: res.status,
-                });
-            }
-            const full = await res.text();
-            let responseBody;
-            try {
-                responseBody = JSON.parse(full);
-            }
-            catch (_error) {
-                // If parsing fails, treat the body as a string (e.g. HTML error page)
-                responseBody = full;
-            }
-            if (debugLogging) {
-                console.error('<- Response Body:');
-                console.error(inspect(responseBody, { depth: null, colors: process.stderr.isTTY }));
-            }
-            if (TRACING_FILE) {
-                appendFileSync(TRACING_FILE, `${JSON.stringify({
-                    type: 'response',
-                    timestamp: new Date().toISOString(),
-                    status: res.status,
-                    headers: redactHeaders(Object.fromEntries(res.headers.entries())),
-                    body: responseBody,
-                }, null, 2)}\n`);
-            }
-            return new Response(full, {
-                headers: res.headers,
-                status: res.status,
-            });
-        })
-        : undefined;
-    switch (config.provider) {
-        case AiProvider.Anthropic: {
-            const anthropic = createAnthropic({
-                apiKey: config.apiKey,
-                baseURL: config.baseUrl,
-                fetch: fetchOverride,
-            });
-            return anthropic(config.model);
-        }
-        case AiProvider.DeepSeek: {
-            const deepseek = createDeepSeek({
-                apiKey: config.apiKey,
-                baseURL: config.baseUrl,
-                fetch: fetchOverride,
-            });
-            return deepseek(config.model);
-        }
-        case AiProvider.OpenRouter: {
-            const openrouter = createOpenRouter({
-                apiKey: config.apiKey,
-                baseURL: config.baseUrl,
-                fetch: fetchOverride,
-                headers: {
-                    'HTTP-Referer': 'https://polka.codes',
-                    'X-Title': 'Polka Codes',
-                },
-            });
-            return openrouter.chat(config.model, {
-                usage: { include: true },
-            });
-        }
-        case AiProvider.OpenAI: {
-            const openai = createOpenAI({
-                apiKey: config.apiKey,
-                baseURL: config.baseUrl,
-                fetch: fetchOverride,
-            });
-            return openai(config.model);
-        }
-        case AiProvider.OpenAICompatible: {
-            if (!config.baseUrl) {
-                throw new Error('OpenAI-compatible providers require a baseUrl');
-            }
-            const openaiCompatible = createOpenAICompatible({
-                apiKey: config.apiKey,
-                baseURL: config.baseUrl,
-                name: config.name || 'OpenAI Compatible',
-                fetch: fetchOverride,
-            });
-            return openaiCompatible(config.model);
-        }
-        case AiProvider.GoogleVertex: {
-            const vertex = createVertex({
-                fetch: fetchOverride,
-                location: config.location,
-                project: config.project,
-                googleAuthOptions: {
-                    keyFile: config.keyFile,
-                },
-            });
-            return vertex(config.model);
-        }
-        case AiProvider.Google: {
-            const google = createGoogleGenerativeAI({
-                fetch: fetchOverride,
-                apiKey: config.apiKey,
-            });
-            return google(config.model);
-        }
-    }
-};
-//# sourceMappingURL=getModel.js.map

package/dist/getModel.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"getModel.js","sourceRoot":"","sources":["../src/getModel.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,SAAS,CAAA;AACxC,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAA;AACnC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAA;AACnD,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAA;AACjD,OAAO,EAAE,wBAAwB,EAAE,MAAM,gBAAgB,CAAA;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,uBAAuB,CAAA;AACpD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,2BAA2B,CAAA;AAElE,OAAO,EAAE,gBAAgB,EAAE,MAAM,6BAA6B,CAAA;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAA;AAE9B,SAAS,eAAe,CAAC,OAAgC;IACvD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAA;IAClB,CAAC;IACD,IAAI,OAAO,YAAY,OAAO,EAAE,CAAC;QAC/B,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAA;IAC9C,CAAC;IACD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,OAAO,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAA;IACpC,CAAC;IACD,OAAO,OAAO,CAAA;AAChB,CAAC;AAED,SAAS,aAAa,CAAC,OAA2C;IAChE,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO,SAAS,CAAA;IAClB,CAAC;IACD,MAAM,eAAe,GAA2B,EAAE,CAAA;IAClD,MAAM,iBAAiB,GAAG,CAAC,eAAe,EAAE,QAAQ,EAAE,KAAK,EAAE,OAAO,CAAC,CAAA;IACrE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACnD,IAAI,iBAAiB,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;YAC7E,eAAe,CAAC,GAAG,CAAC,GAAG,UAAU,CAAA;QACnC,CAAC;aAAM,CAAC;YACN,eAAe,CAAC,GAAG,CAAC,GAAG,KAAK,CAAA;QAC9B,CAAC;IACH,CAAC;IACD,OAAO,eAAe,CAAA;AACxB,CAAC;AACD,MAAM,CAAN,IAAY,UAQX;AARD,WAAY,UAAU;IACpB,qCAAuB,CAAA;IACvB,mCAAqB,CAAA;IACrB,uCAAyB,CAAA;IACzB,+BAAiB,CAAA;IACjB,oDAAsC,CAAA;IACtC,4CAA8B,CAAA;IAC9B,+BAAiB,CAAA;AACnB,CAAC,EARW,UAAU,KAAV,UAAU,QAQrB;AAaD,MAAM,CAAC,MAAM,QAAQ,GAAG,CAAC,MAAmB,EAAE,YAAY,GAAG,KAAK,EAAmB,EAAE;IACrF,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,EAAE,CAAA;IAEjC,MAAM,aAAa,GACjB,YAAY,IAAI,YAAY;QAC1B,CAAC,CAAE,CAAC,KAAK,EAAE,GAAG,EAAE,OAAO,EAAE,EAAE;YACvB,MAAM,WAAW,GAAG,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,IAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAA;YAElF,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAA;gBACrC,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,OAAO,EAAE,OAAO,CAAC,CAAA;gBACtD,OAAO,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAA;gBACjC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;YACpF,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,cAAc,CACZ,YAAY,EACZ,GAAG,IAAI,CAAC,SAAS,CACf;oBACE,IAAI,EAAE,SAAS;oBACf,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,GAAG;oBACH,OAAO,EAAE,aAAa,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;oBACzD,IAAI,EAAE,WAAW;iBAClB,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAA;YACH,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,OAAO,CAAC,CAAA;YAErC,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qBAAqB,EAAE,GAAG,CAAC,MAAM,CAAC,CAAA;YAClD,CAAC;YAED,MAAM,WAAW,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC,IAAI,EAAE,CAAA;YACzD,IAAI,WAAW,CAAC,QAAQ,CAAC,mBAAmB,CAAC,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC;gBAC1D,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,GAAG,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,CAC5C;gBAAA,CAAC,KAAK,IAAI,EAAE;oBACX,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAA;oBACjC,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAA;oBACjC,IAAI,CAAC;wBACH,IAAI,IAAI,GAAG,KAAK,CAAA;wBAChB,OAAO,CAAC,IAAI,EAAE,CAAC;4BACb,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAA;4BAC9C,IAAI,GAAG,CAAC,CAAA;4BACR,IAAI,KAAK,EAAE,CAAC;gCACV,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;gCAClC,IAAI,YAAY,EAAE,CAAC;oCACjB,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAA;gCAC/D,CAAC;gCACD,IAAI,YAAY,EAAE,CAAC;oCACjB,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;wCACpC,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;4CAC7B,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAA;4CACjD,IAAI,OAAO,EAAE,CAAC;gDACZ,IAAI,CAAC;oDACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAA;oDAChC,cAAc,CACZ,YAAY,EACZ,GAAG,IAAI,CAAC,SAAS,CACf;wDACE,IAAI,EAAE,gBAAgB;wDACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;wDACnC,KAAK,EAAE,IAAI;qDACZ,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAA;gDACH,CAAC;gDAAC,OAAO,EAAE,EAAE,CAAC;oDACZ,cAAc,CACZ,YAAY,EACZ,GAAG,IAAI,CAAC,SAAS,CACf;wDACE,IAAI,EAAE,gBAAgB;wDACtB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;wDACnC,KAAK,EAAE,OAAO;qDACf,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAA;gDACH,CAAC;4CACH,CAAC;wCACH,CAAC;oCACH,CAAC;gCACH,CAAC;4BACH,CAAC;wBACH,CAAC;oBACH,CAAC;4BAAS,CAAC;wBACT,+DAA+D;wBAC/D,MAAM,CAAC,WAAW,EAAE,CAAA;oBACtB,CAAC;gBACH,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;oBACnB,IAAI,YAAY,EAAE,CAAC;wBACjB,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAA;oBAC/C,CAAC;gBACH,CAAC,CAAC,CAAA;gBACF,OAAO,IAAI,QAAQ,CAAC,YAAY,EAAE;oBAChC,OAAO,EAAE,GAAG,CAAC,OAAO;oBACpB,MAAM,EAAE,GAAG,CAAC,MAAM;iBACnB,CAAC,CAAA;YACJ,CAAC;YAED,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,IAAI,EAAE,CAAA;YAC7B,IAAI,YAAqB,CAAA;YACzB,IAAI,CAAC;gBACH,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;YACjC,CAAC;YAAC,OAAO,MAAM,EAAE,CAAC;gBAChB,sEAAsE;gBACtE,YAAY,GAAG,IAAI,CAAA;YACrB,CAAC;YAED,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAA;gBAClC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAA;YACrF,CAAC;YACD,IAAI,YAAY,EAAE,CAAC;gBACjB,cAAc,CACZ,YAAY,EACZ,GAAG,IAAI,CAAC,SAAS,CACf;oBACE,IAAI,EAAE,UAAU;oBAChB,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;oBACnC,MAAM,EAAE,GAAG,CAAC,MAAM;oBAClB,OAAO,EAAE,aAAa,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,CAAC;oBACjE,IAAI,EAAE,YAAY;iBACnB,EACD,IAAI,EACJ,CAAC,CACF,IAAI,CACN,CAAA;YACH,CAAC;YAED,OAAO,IAAI,QAAQ,CAAC,IAAI,EAAE;gBACxB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,MAAM,EAAE,GAAG,CAAC,MAAM;aACnB,CAAC,CAAA;QACJ,CAAC,CAAkB;QACrB,CAAC,CAAC,SAAS,CAAA;IAEf,QAAQ,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxB,KAAK,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;YAC1B,MAAM,SAAS,GAAG,eAAe,CAAC;gBAChC,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAA;YACF,OAAO,SAAS,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAChC,CAAC;QAED,KAAK,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;YACzB,MAAM,QAAQ,GAAG,cAAc,CAAC;gBAC9B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAA;YACF,OAAO,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC/B,CAAC;QAED,KAAK,UAAU,CAAC,UAAU,CAAC,CAAC,CAAC;YAC3B,MAAM,UAAU,GAAG,gBAAgB,CAAC;gBAClC,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,aAAa;gBACpB,OAAO,EAAE;oBACP,cAAc,EAAE,qBAAqB;oBACrC,SAAS,EAAE,aAAa;iBACzB;aACF,CAAC,CAAA;YACF,OAAO,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE;gBACnC,KAAK,EAAE,EAAE,OAAO,EAAE,IAAI,EAAE;aACzB,CAAC,CAAA;QACJ,CAAC;QAED,KAAK,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACvB,MAAM,MAAM,GAAG,YAAY,CAAC;gBAC1B,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,KAAK,EAAE,aAAa;aACrB,CAAC,CAAA;YACF,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC7B,CAAC;QAED,KAAK,UAAU,CAAC,gBAAgB,CAAC,CAAC,CAAC;YACjC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAA;YAClE,CAAC;YACD,MAAM,gBAAgB,GAAG,sBAAsB,CAAC;gBAC9C,MAAM,EAAE,MAAM,CAAC,MAAM;gBACrB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,mBAAmB;gBACxC,KAAK,EAAE,aAAa;aACrB,CAAC,CAAA;YACF,OAAO,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QACvC,CAAC;QAED,KAAK,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,YAAY,CAAC;gBAC1B,KAAK,EAAE,aAAa;gBACpB,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,OAAO,EAAE,MAAM,CAAC,OAAO;gBACvB,iBAAiB,EAAE;oBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;iBACxB;aACF,CAAC,CAAA;YACF,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC7B,CAAC;QACD,KAAK,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACvB,MAAM,MAAM,GAAG,wBAAwB,CAAC;gBACtC,KAAK,EAAE,aAAa;gBACpB,MAAM,EAAE,MAAM,CAAC,MAAM;aACtB,CAAC,CAAA;YACF,OAAO,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;QAC7B,CAAC;IACH,CAAC;AACH,CAAC,CAAA"}

package/dist/getModel.test.js

@@ -1,67 +0,0 @@
-// generated by polka.codes
-import { describe, expect, it } from 'bun:test';
-import { AiProvider, getModel } from './getModel';
-describe('getModel', () => {
-    it('should create Anthropic model', () => {
-        const model = getModel({
-            provider: AiProvider.Anthropic,
-            model: 'claude-3-5-sonnet-20241022',
-            apiKey: 'test-key',
-        });
-        expect(model).toBeDefined();
-        expect(model.modelId).toBe('claude-3-5-sonnet-20241022');
-    });
-    it('should create OpenAI model', () => {
-        const model = getModel({
-            provider: AiProvider.OpenAI,
-            model: 'gpt-4',
-            apiKey: 'test-key',
-        });
-        expect(model).toBeDefined();
-        expect(model.modelId).toBe('gpt-4');
-    });
-    it('should create Google model', () => {
-        const model = getModel({
-            provider: AiProvider.Google,
-            model: 'gemini-2.0-flash-exp',
-            apiKey: 'test-key',
-        });
-        expect(model).toBeDefined();
-    });
-    it('should create Google Vertex model', () => {
-        const model = getModel({
-            provider: AiProvider.GoogleVertex,
-            model: 'gemini-2.0-flash-exp',
-            project: 'test-project',
-            location: 'us-central1',
-        });
-        expect(model).toBeDefined();
-    });
-    it('should create DeepSeek model', () => {
-        const model = getModel({
-            provider: AiProvider.DeepSeek,
-            model: 'deepseek-chat',
-            apiKey: 'test-key',
-        });
-        expect(model).toBeDefined();
-    });
-    it('should create OpenRouter model', () => {
-        const model = getModel({
-            provider: AiProvider.OpenRouter,
-            model: 'anthropic/claude-3.5-sonnet',
-            apiKey: 'test-key',
-        });
-        expect(model).toBeDefined();
-    });
-    it('should create OpenAI-compatible model', () => {
-        const model = getModel({
-            provider: AiProvider.OpenAICompatible,
-            model: 'gpt-4o',
-            apiKey: 'test-key',
-            baseUrl: 'https://api.together.xyz/v1',
-        });
-        expect(model).toBeDefined();
-        expect(model.modelId).toBe('gpt-4o');
-    });
-});
-//# sourceMappingURL=getModel.test.js.map

package/dist/getModel.test.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"getModel.test.js","sourceRoot":"","sources":["../src/getModel.test.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAC3B,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,UAAU,CAAA;AAC/C,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAEjD,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;IACxB,EAAE,CAAC,+BAA+B,EAAE,GAAG,EAAE;QACvC,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,SAAS;YAC9B,KAAK,EAAE,4BAA4B;YACnC,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAA;IAC1D,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,MAAM;YAC3B,KAAK,EAAE,OAAO;YACd,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAA;IACrC,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,MAAM;YAC3B,KAAK,EAAE,sBAAsB;YAC7B,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,YAAY;YACjC,KAAK,EAAE,sBAAsB;YAC7B,OAAO,EAAE,cAAc;YACvB,QAAQ,EAAE,aAAa;SACxB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,8BAA8B,EAAE,GAAG,EAAE;QACtC,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,QAAQ;YAC7B,KAAK,EAAE,eAAe;YACtB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,gCAAgC,EAAE,GAAG,EAAE;QACxC,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,UAAU;YAC/B,KAAK,EAAE,6BAA6B;YACpC,MAAM,EAAE,UAAU;SACnB,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;IAC7B,CAAC,CAAC,CAAA;IAEF,EAAE,CAAC,uCAAuC,EAAE,GAAG,EAAE;QAC/C,MAAM,KAAK,GAAG,QAAQ,CAAC;YACrB,QAAQ,EAAE,UAAU,CAAC,gBAAgB;YACrC,KAAK,EAAE,QAAQ;YACf,MAAM,EAAE,UAAU;YAClB,OAAO,EAAE,6BAA6B;SACvC,CAAC,CAAA;QAEF,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAA;QAC3B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAA;IACtC,CAAC,CAAC,CAAA;AACJ,CAAC,CAAC,CAAA"}

package/dist/getProviderOptions.js

@@ -1,58 +0,0 @@
-// generated by polka.codes
-import { AiProvider } from './getModel';
-import models from './prices';
-/**
- * Generates provider-specific options for enabling thinking capabilities.
- *
- * @param options The options for generating provider options.
- * @returns A provider options object for the `streamText` call.
- */
-export function getProviderOptions(options) {
-    const { provider, modelId, parameters, supportThinking } = options;
-    const thinkingBudgetTokens = parameters?.thinkingBudgetTokens ?? 0;
-    if (thinkingBudgetTokens <= 0 || !provider || !modelId) {
-        return {};
-    }
-    const modelInfo = models[provider]?.[modelId];
-    const providerSupportsThinking = provider === AiProvider.OpenRouter || !!modelInfo?.supportsThinking;
-    const shouldSupportThinking = supportThinking ?? providerSupportsThinking;
-    if (!shouldSupportThinking) {
-        return {};
-    }
-    const providerOptions = {};
-    switch (provider) {
-        case AiProvider.Anthropic:
-            providerOptions.anthropic = {
-                thinking: { type: 'enabled', budgetTokens: thinkingBudgetTokens },
-            };
-            break;
-        case AiProvider.OpenRouter:
-            if (modelId.startsWith('anthropic/')) {
-                // TODO: support thinking for anthropic models
-                // openrouter ai sdk doesn't appear to expose reasining details
-                // so we can't construct multi step messages correctly when thinking is enabled
-                break;
-            }
-            providerOptions.openrouter = {
-                reasoning: {
-                    max_tokens: thinkingBudgetTokens,
-                },
-            };
-            break;
-        case AiProvider.GoogleVertex:
-        case AiProvider.Google:
-            providerOptions.google = {
-                thinkingConfig: {
-                    includeThoughts: true,
-                    thinkingBudget: thinkingBudgetTokens,
-                },
-            };
-            break;
-        case AiProvider.DeepSeek:
-            break;
-        case AiProvider.OpenAI:
-            break;
-    }
-    return providerOptions;
-}
-//# sourceMappingURL=getProviderOptions.js.map

package/dist/getProviderOptions.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"getProviderOptions.js","sourceRoot":"","sources":["../src/getProviderOptions.ts"],"names":[],"mappings":"AAAA,2BAA2B;AAE3B,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAA;AACvC,OAAO,MAAM,MAAM,UAAU,CAAA;AAY7B;;;;;GAKG;AACH,MAAM,UAAU,kBAAkB,CAAC,OAA+B;IAChE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,GAAG,OAAO,CAAA;IAClE,MAAM,oBAAoB,GAAG,UAAU,EAAE,oBAAoB,IAAI,CAAC,CAAA;IAElE,IAAI,oBAAoB,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,OAAO,EAAE,CAAC;QACvD,OAAO,EAAE,CAAA;IACX,CAAC;IAED,MAAM,SAAS,GAAI,MAAM,CAAC,QAA+B,CAAyB,EAAE,CAAC,OAAO,CAAC,CAAA;IAC7F,MAAM,wBAAwB,GAAG,QAAQ,KAAK,UAAU,CAAC,UAAU,IAAI,CAAC,CAAC,SAAS,EAAE,gBAAgB,CAAA;IACpG,MAAM,qBAAqB,GAAG,eAAe,IAAI,wBAAwB,CAAA;IAEzE,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC3B,OAAO,EAAE,CAAA;IACX,CAAC;IAED,MAAM,eAAe,GAAQ,EAAE,CAAA;IAE/B,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,UAAU,CAAC,SAAS;YACvB,eAAe,CAAC,SAAS,GAAG;gBAC1B,QAAQ,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,YAAY,EAAE,oBAAoB,EAAE;aAClE,CAAA;YACD,MAAK;QACP,KAAK,UAAU,CAAC,UAAU;YACxB,IAAI,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;gBACrC,8CAA8C;gBAC9C,+DAA+D;gBAC/D,+EAA+E;gBAC/E,MAAK;YACP,CAAC;YACD,eAAe,CAAC,UAAU,GAAG;gBAC3B,SAAS,EAAE;oBACT,UAAU,EAAE,oBAAoB;iBACjC;aACF,CAAA;YACD,MAAK;QACP,KAAK,UAAU,CAAC,YAAY,CAAC;QAC7B,KAAK,UAAU,CAAC,MAAM;YACpB,eAAe,CAAC,MAAM,GAAG;gBACvB,cAAc,EAAE;oBACd,eAAe,EAAE,IAAI;oBACrB,cAAc,EAAE,oBAAoB;iBACrC;aACF,CAAA;YACD,MAAK;QACP,KAAK,UAAU,CAAC,QAAQ;YACtB,MAAK;QACP,KAAK,UAAU,CAAC,MAAM;YACpB,MAAK;IACT,CAAC;IAED,OAAO,eAAe,CAAA;AACxB,CAAC"}