@auto-engineer/ai-gateway 0.6.3 → 0.8.1

package/dist/config.specs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"config.specs.js","sourceRoot":"","sources":["../src/config.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,mBAAmB,EAAE,MAAM,UAAU,CAAC;AAE/C,6BAA6B;AAC7B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC;AAEhC,QAAQ,CAAC,wCAAwC,EAAE,GAAG,EAAE;IACtD,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,kEAAkE;QAClE,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC;IACnB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;QACnC,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,SAAS,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,wBAAwB,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,gBAAgB,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,iBAAiB,CAAC;YAE9D,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAErC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YAC9D,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6EAA6E,EAAE,GAAG,EAAE;YACrF,sCAAsC;YACtC,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,YAAY,CAAC;YAChD,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,yBAAyB,CAAC;YACjE,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,SAAS,CAAC,CAAC,6BAA6B;YACxE,oEAAoE;YAEpE,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,eAAe,CAAC;YAC7C,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,kBAAkB,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,QAAQ,CAAC;YAC5C,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,wBAAwB,CAAC;YAChE,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,YAAY,CAAC;YACnD,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,cAAc,CAAC;YAE3D,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAErC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,WAAW,EAAE,CAAC;YACvC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QACtC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,aAAa,CAAC;YACjD,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,6BAA6B,CAAC;YACrE,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,iBAAiB,CAAC;YACxD,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,mBAAmB,CAAC;YAEhE,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAErC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;YACpC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,aAAa,EAAE,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;YACtC,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,CAAC;QACrC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,kCAAkC;YAClC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YAClC,OAAO,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;YACrC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;YAClC,OAAO,OAAO,CAAC,GAAG,CAAC,WAAW,CAAC;YAC/B,OAAO,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;YACxC,OAAO,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC;YAC5C,OAAO,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC;YAC3C,OAAO,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;YAEjD,MAAM,CAAC,GAAG,EAAE,CAAC,mBAAmB,EAAE,CAAC,CAAC,OAAO,CAAC,6CAA6C,CAAC,CAAC;QAC7F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,MAAM,SAAS,GAAG;gBAChB;oBACE,IAAI,EAAE,eAAe;oBACrB,OAAO,EAAE,sCAAsC;oBAC/C,KAAK,EAAE,aAAa;iBACrB;gBACD;oBACE,IAAI,EAAE,cAAc;oBACpB,OAAO,EAAE,2BAA2B;oBACpC,KAAK,EAAE,aAAa;iBACrB;gBACD;oBACE,IAAI,EAAE,cAAc;oBACpB,OAAO,EAAE,qDAAqD;oBAC9D,KAAK,EAAE,OAAO;iBACf;aACF,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC7B,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,QAAQ,CAAC,IAAI,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,QAAQ,CAAC,OAAO,CAAC;gBACxD,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,UAAU,CAAC;gBACjD,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,QAAQ,CAAC,KAAK,CAAC;gBAE3D,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;gBAErC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;gBAChD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;gBACtD,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;YACvE,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,EAAE,CAAC;YACtC,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,EAAE,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,EAAE,CAAC;YACzC,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,EAAE,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,iBAAiB,GAAG,SAAS,CAAC,CAAC,6BAA6B;YAExE,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,aAAa,EAAE,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,GAAG,EAAE;YAC3D,OAAO,CAAC,GAAG,CAAC,oBAAoB,GAAG,WAAW,CAAC;YAC/C,OAAO,CAAC,GAAG,CAAC,wBAAwB,GAAG,0BAA0B,CAAC;YAClE,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,UAAU,CAAC;YACjD,OAAO,CAAC,GAAG,CAAC,6BAA6B,GAAG,mBAAmB,CAAC;YAEhE,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;YAErC,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;YAChE,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/constants.d.ts

@@ -0,0 +1,20 @@
+export declare enum AIProvider {
+    OpenAI = "openai",
+    Anthropic = "anthropic",
+    Google = "google",
+    XAI = "xai",
+    Custom = "custom"
+}
+export interface CustomProviderConfig {
+    name: string;
+    baseUrl: string;
+    apiKey: string;
+    defaultModel: string;
+}
+export declare const DEFAULT_MODELS: {
+    readonly openai: "gpt-4o-mini";
+    readonly anthropic: "claude-sonnet-4-20250514";
+    readonly google: "gemini-2.5-pro";
+    readonly xai: "grok-4";
+};
+//# sourceMappingURL=constants.d.ts.map

package/dist/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":"AAAA,oBAAY,UAAU;IACpB,MAAM,WAAW;IACjB,SAAS,cAAc;IACvB,MAAM,WAAW;IACjB,GAAG,QAAQ;IACX,MAAM,WAAW;CAClB;AAED,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,eAAO,MAAM,cAAc;;;;;CAKjB,CAAC"}

package/dist/constants.js

@@ -0,0 +1,15 @@
+export var AIProvider;
+(function (AIProvider) {
+    AIProvider["OpenAI"] = "openai";
+    AIProvider["Anthropic"] = "anthropic";
+    AIProvider["Google"] = "google";
+    AIProvider["XAI"] = "xai";
+    AIProvider["Custom"] = "custom";
+})(AIProvider || (AIProvider = {}));
+export const DEFAULT_MODELS = {
+    [AIProvider.OpenAI]: 'gpt-4o-mini',
+    [AIProvider.Anthropic]: 'claude-sonnet-4-20250514',
+    [AIProvider.Google]: 'gemini-2.5-pro',
+    [AIProvider.XAI]: 'grok-4',
+};
+//# sourceMappingURL=constants.js.map

package/dist/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../src/constants.ts"],"names":[],"mappings":"AAAA,MAAM,CAAN,IAAY,UAMX;AAND,WAAY,UAAU;IACpB,+BAAiB,CAAA;IACjB,qCAAuB,CAAA;IACvB,+BAAiB,CAAA;IACjB,yBAAW,CAAA;IACX,+BAAiB,CAAA;AACnB,CAAC,EANW,UAAU,KAAV,UAAU,QAMrB;AASD,MAAM,CAAC,MAAM,cAAc,GAAG;IAC5B,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,aAAa;IAClC,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,0BAA0B;IAClD,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,gBAAgB;IACrC,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,QAAQ;CAClB,CAAC"}

package/dist/index-custom.specs.d.ts

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

package/dist/index-custom.specs.d.ts.map

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

package/dist/index-custom.specs.js

@@ -0,0 +1,161 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { AIProvider } from './constants.js';
+import { getAvailableProviders, getDefaultModel, getDefaultAIProvider } from './index.js';
+// Mock the config module
+vi.mock('./config', () => ({
+    configureAIProvider: vi.fn(),
+}));
+// Mock the custom provider
+vi.mock('./providers/custom', () => ({
+    createCustomProvider: vi.fn(() => ({
+        languageModel: vi.fn(),
+    })),
+}));
+describe('Index Integration with Custom Providers', () => {
+    const originalEnv = process.env;
+    beforeEach(() => {
+        vi.resetModules();
+        process.env = { ...originalEnv };
+        vi.clearAllMocks();
+    });
+    afterEach(() => {
+        process.env = originalEnv;
+    });
+    describe('getAvailableProviders', () => {
+        it('should include custom provider when configured', async () => {
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                custom: {
+                    name: 'litellm',
+                    baseUrl: 'https://api.litellm.ai',
+                    apiKey: 'sk-key',
+                    defaultModel: 'claude-3-sonnet',
+                },
+                anthropic: { apiKey: 'sk-anthropic' },
+            });
+            const providers = getAvailableProviders();
+            expect(providers).toContain(AIProvider.Custom);
+            expect(providers).toContain(AIProvider.Anthropic);
+        });
+        it('should not include custom provider when not configured', async () => {
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                anthropic: { apiKey: 'sk-anthropic' },
+            });
+            const providers = getAvailableProviders();
+            expect(providers).not.toContain(AIProvider.Custom);
+            expect(providers).toContain(AIProvider.Anthropic);
+        });
+        it('should handle multiple providers including custom', async () => {
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                openai: { apiKey: 'sk-openai' },
+                anthropic: { apiKey: 'sk-anthropic' },
+                google: { apiKey: 'sk-google' },
+                xai: { apiKey: 'sk-xai' },
+                custom: {
+                    name: 'custom',
+                    baseUrl: 'https://api.custom.com',
+                    apiKey: 'sk-custom',
+                    defaultModel: 'custom-model',
+                },
+            });
+            const providers = getAvailableProviders();
+            expect(providers).toHaveLength(5);
+            expect(providers).toContain(AIProvider.OpenAI);
+            expect(providers).toContain(AIProvider.Anthropic);
+            expect(providers).toContain(AIProvider.Google);
+            expect(providers).toContain(AIProvider.XAI);
+            expect(providers).toContain(AIProvider.Custom);
+        });
+    });
+    describe('getDefaultModel', () => {
+        it('should return custom provider default model when provider is Custom', async () => {
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                custom: {
+                    name: 'litellm',
+                    baseUrl: 'https://api.litellm.ai',
+                    apiKey: 'sk-key',
+                    defaultModel: 'claude-3-sonnet',
+                },
+            });
+            const model = getDefaultModel(AIProvider.Custom);
+            expect(model).toBe('claude-3-sonnet');
+        });
+        it('should throw error when custom provider not configured but requested', async () => {
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                anthropic: { apiKey: 'sk-anthropic' },
+            });
+            expect(() => getDefaultModel(AIProvider.Custom)).toThrow('Custom provider not configured');
+        });
+        it('should use environment variable DEFAULT_AI_MODEL for custom provider', async () => {
+            process.env.DEFAULT_AI_MODEL = 'env-override-model';
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                custom: {
+                    name: 'custom',
+                    baseUrl: 'https://api.custom.com',
+                    apiKey: 'sk-key',
+                    defaultModel: 'config-model',
+                },
+            });
+            const model = getDefaultModel(AIProvider.Custom);
+            expect(model).toBe('env-override-model');
+        });
+        it('should handle different custom provider models', async () => {
+            const testCases = ['gpt-4o', 'claude-3-opus', 'llama3.1:70b', 'mistral-large', 'gemini-1.5-pro'];
+            const { configureAIProvider } = await import('./config.js');
+            testCases.forEach((testModel) => {
+                vi.mocked(configureAIProvider).mockReturnValue({
+                    custom: {
+                        name: 'test',
+                        baseUrl: 'https://api.test.com',
+                        apiKey: 'sk-key',
+                        defaultModel: testModel,
+                    },
+                });
+                const model = getDefaultModel(AIProvider.Custom);
+                expect(model).toBe(testModel);
+            });
+        });
+    });
+    describe('getDefaultAIProvider', () => {
+        it('should use custom provider as fallback when available', async () => {
+            process.env.DEFAULT_AI_PROVIDER = 'nonexistent';
+            const { configureAIProvider } = await import('./config.js');
+            vi.mocked(configureAIProvider).mockReturnValue({
+                custom: {
+                    name: 'custom',
+                    baseUrl: 'https://api.custom.com',
+                    apiKey: 'sk-key',
+                    defaultModel: 'custom-model',
+                },
+            });
+            // Mock getAvailableProviders to return custom
+            const mockGetAvailableProviders = vi.fn(() => [AIProvider.Custom]);
+            vi.doMock('./index', async () => ({
+                ...(await vi.importActual('./index')),
+                getAvailableProviders: mockGetAvailableProviders,
+            }));
+            const { getDefaultAIProvider: getDefaultAIProviderMocked } = await import('./index.js');
+            const provider = getDefaultAIProviderMocked();
+            expect(provider).toBe(AIProvider.Custom);
+        });
+        it('should respect explicit DEFAULT_AI_PROVIDER environment variable', () => {
+            const testCases = [
+                { env: 'openai', expected: AIProvider.OpenAI },
+                { env: 'anthropic', expected: AIProvider.Anthropic },
+                { env: 'google', expected: AIProvider.Google },
+                { env: 'xai', expected: AIProvider.XAI },
+            ];
+            testCases.forEach(({ env, expected }) => {
+                process.env.DEFAULT_AI_PROVIDER = env;
+                const provider = getDefaultAIProvider();
+                expect(provider).toBe(expected);
+            });
+        });
+    });
+});
+//# sourceMappingURL=index-custom.specs.js.map

package/dist/index-custom.specs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index-custom.specs.js","sourceRoot":"","sources":["../src/index-custom.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,qBAAqB,EAAE,eAAe,EAAE,oBAAoB,EAAE,MAAM,SAAS,CAAC;AAEvF,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IACzB,mBAAmB,EAAE,EAAE,CAAC,EAAE,EAAE;CAC7B,CAAC,CAAC,CAAC;AAEJ,2BAA2B;AAC3B,EAAE,CAAC,IAAI,CAAC,oBAAoB,EAAE,GAAG,EAAE,CAAC,CAAC;IACnC,oBAAoB,EAAE,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;QACjC,aAAa,EAAE,EAAE,CAAC,EAAE,EAAE;KACvB,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,yCAAyC,EAAE,GAAG,EAAE;IACvD,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC;IAEhC,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,YAAY,EAAE,CAAC;QAClB,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;QACjC,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,MAAM,EAAE;oBACN,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,QAAQ;oBAChB,YAAY,EAAE,iBAAiB;iBAChC;gBACD,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;aACtC,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAE1C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wDAAwD,EAAE,KAAK,IAAI,EAAE;YACtE,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;aACtC,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAE1C,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACnD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;YACjE,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,MAAM,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;gBAC/B,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;gBACrC,MAAM,EAAE,EAAE,MAAM,EAAE,WAAW,EAAE;gBAC/B,GAAG,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE;gBACzB,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,WAAW;oBACnB,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAE1C,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAClD,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,MAAM,EAAE;oBACN,IAAI,EAAE,SAAS;oBACf,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,QAAQ;oBAChB,YAAY,EAAE,iBAAiB;iBAChC;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,SAAS,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;aACtC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,gCAAgC,CAAC,CAAC;QAC7F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,KAAK,IAAI,EAAE;YACpF,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,oBAAoB,CAAC;YAEpD,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,QAAQ;oBAChB,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,SAAS,GAAG,CAAC,QAAQ,EAAE,eAAe,EAAE,cAAc,EAAE,eAAe,EAAE,gBAAgB,CAAC,CAAC;YAEjG,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YAEzD,SAAS,CAAC,OAAO,CAAC,CAAC,SAAS,EAAE,EAAE;gBAC9B,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;oBAC7C,MAAM,EAAE;wBACN,IAAI,EAAE,MAAM;wBACZ,OAAO,EAAE,sBAAsB;wBAC/B,MAAM,EAAE,QAAQ;wBAChB,YAAY,EAAE,SAAS;qBACxB;iBACF,CAAC,CAAC;gBAEH,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;gBACjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YAChC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,aAAa,CAAC;YAEhD,MAAM,EAAE,mBAAmB,EAAE,GAAG,MAAM,MAAM,CAAC,UAAU,CAAC,CAAC;YACzD,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,eAAe,CAAC;gBAC7C,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ;oBACd,OAAO,EAAE,wBAAwB;oBACjC,MAAM,EAAE,QAAQ;oBAChB,YAAY,EAAE,cAAc;iBAC7B;aACF,CAAC,CAAC;YAEH,8CAA8C;YAC9C,MAAM,yBAAyB,GAAG,EAAE,CAAC,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnE,EAAE,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC;gBAChC,GAAG,CAAC,MAAM,EAAE,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;gBACrC,qBAAqB,EAAE,yBAAyB;aACjD,CAAC,CAAC,CAAC;YAEJ,MAAM,EAAE,oBAAoB,EAAE,0BAA0B,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,CAAC,CAAC;YACrF,MAAM,QAAQ,GAAG,0BAA0B,EAAE,CAAC;YAE9C,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,SAAS,GAAG;gBAChB,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,MAAM,EAAE;gBAC9C,EAAE,GAAG,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,SAAS,EAAE;gBACpD,EAAE,GAAG,EAAE,QAAQ,EAAE,QAAQ,EAAE,UAAU,CAAC,MAAM,EAAE;gBAC9C,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,GAAG,EAAE;aACzC,CAAC;YAEF,SAAS,CAAC,OAAO,CAAC,CAAC,EAAE,GAAG,EAAE,QAAQ,EAAE,EAAE,EAAE;gBACtC,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,GAAG,CAAC;gBACtC,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;gBACxC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/index.d.ts

@@ -1,11 +1,8 @@
+import { AIProvider } from './constants';
+export { AIProvider } from './constants';
 import { z } from 'zod';
-export declare enum AIProvider {
-    OpenAI = "openai",
-    Anthropic = "anthropic",
-    Google = "google",
-    XAI = "xai"
-}
 export interface AIOptions {
+    provider?: AIProvider;
     model?: string;
     temperature?: number;
     maxTokens?: number;
@@ -20,22 +17,24 @@ export interface StructuredAIOptions<T> extends Omit<AIOptions, 'streamCallback'
 export interface StreamStructuredAIOptions<T> extends StructuredAIOptions<T> {
     onPartialObject?: (partialObject: unknown) => void;
 }
-export declare function generateTextWithAI(prompt: string, provider: AIProvider, options?: AIOptions): Promise<string>;
-export declare function streamTextWithAI(prompt: string, provider: AIProvider, options?: AIOptions): AsyncGenerator<string>;
+export declare function getDefaultAIProvider(): AIProvider;
+export declare function getDefaultModel(provider: AIProvider): string;
+export declare function generateTextWithAI(prompt: string, options?: AIOptions): Promise<string>;
+export declare function streamTextWithAI(prompt: string, options?: AIOptions): AsyncGenerator<string>;
 /**
  * Generates text using streaming and collects the full response.
  * Optionally calls a stream callback for each token if provided.
  * Always returns the complete collected response.
  */
-export declare function generateTextStreamingWithAI(prompt: string, provider: AIProvider, options?: AIOptions): Promise<string>;
-export declare function generateTextWithToolsAI(prompt: string, provider: AIProvider, options?: AIOptions): Promise<{
+export declare function generateTextStreamingWithAI(prompt: string, options?: AIOptions): Promise<string>;
+export declare function generateTextWithToolsAI(prompt: string, options?: AIOptions): Promise<{
     text: string;
     toolCalls?: unknown[];
 }>;
-export declare function generateTextWithImageAI(text: string, imageBase64: string, provider: AIProvider, options?: AIOptions): Promise<string>;
+export declare function generateTextWithImageAI(text: string, imageBase64: string, options?: AIOptions): Promise<string>;
 export declare function getAvailableProviders(): AIProvider[];
-export declare function generateStructuredDataWithAI<T>(prompt: string, provider: AIProvider, options: StructuredAIOptions<T>): Promise<T>;
-export declare function streamStructuredDataWithAI<T>(prompt: string, provider: AIProvider, options: StreamStructuredAIOptions<T>): Promise<T>;
+export declare function generateStructuredDataWithAI<T>(prompt: string, options: StructuredAIOptions<T>): Promise<T>;
+export declare function streamStructuredDataWithAI<T>(prompt: string, options: StreamStructuredAIOptions<T>): Promise<T>;
 export { z };
 export { server as mcpServer, registerTool, registerTools, startServer, isServerStarted, getRegisteredTools, getRegisteredToolsForAI, getToolHandler, getSchemaByName, executeRegisteredTool, type ToolHandler, type RegisteredTool, } from './mcp-server';
 //# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAqIxB,oBAAY,UAAU;IACpB,MAAM,WAAW;IACjB,SAAS,cAAc;IACvB,MAAM,WAAW;IACjB,GAAG,QAAQ;CACZ;AAED,MAAM,WAAW,SAAS;IACxB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,mBAAmB,CAAC,CAAC,CAAE,SAAQ,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC;IAC/E,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,yBAAyB,CAAC,CAAC,CAAE,SAAQ,mBAAmB,CAAC,CAAC,CAAC;IAC1E,eAAe,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD;AA6CD,wBAAsB,kBAAkB,CACtC,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC,MAAM,CAAC,CA8BjB;AAED,wBAAuB,gBAAgB,CACrC,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,GAAE,SAAc,GACtB,cAAc,CAAC,MAAM,CAAC,CA2BxB;AAED;;;;GAIG;AACH,wBAAsB,2BAA2B,CAC/C,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC,MAAM,CAAC,CAqBjB;AAgFD,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,CAAC,CA2BlD;AAgCD,wBAAsB,uBAAuB,CAC3C,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,EACnB,QAAQ,EAAE,UAAU,EACpB,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC,MAAM,CAAC,CAuCjB;AAED,wBAAgB,qBAAqB,IAAI,UAAU,EAAE,CASpD;AA4HD,wBAAsB,4BAA4B,CAAC,CAAC,EAClD,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAC9B,OAAO,CAAC,CAAC,CAAC,CAYZ;AAED,wBAAsB,0BAA0B,CAAC,CAAC,EAChD,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,UAAU,EACpB,OAAO,EAAE,yBAAyB,CAAC,CAAC,CAAC,GACpC,OAAO,CAAC,CAAC,CAAC,CAiEZ;AAGD,OAAO,EAAE,CAAC,EAAE,CAAC;AAGb,OAAO,EACL,MAAM,IAAI,SAAS,EACnB,YAAY,EACZ,aAAa,EACb,WAAW,EACX,eAAe,EACf,kBAAkB,EAClB,uBAAuB,EACvB,cAAc,EACd,eAAe,EACf,qBAAqB,EACrB,KAAK,WAAW,EAChB,KAAK,cAAc,GACpB,MAAM,cAAc,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAMA,OAAO,EAAkB,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzD,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAqIxB,MAAM,WAAW,SAAS;IACxB,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,cAAc,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IACzC,YAAY,CAAC,EAAE,OAAO,CAAC;CACxB;AAED,MAAM,WAAW,mBAAmB,CAAC,CAAC,CAAE,SAAQ,IAAI,CAAC,SAAS,EAAE,gBAAgB,CAAC;IAC/E,MAAM,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IACvB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;CAC5B;AAED,MAAM,WAAW,yBAAyB,CAAC,CAAC,CAAE,SAAQ,mBAAmB,CAAC,CAAC,CAAC;IAC1E,eAAe,CAAC,EAAE,CAAC,aAAa,EAAE,OAAO,KAAK,IAAI,CAAC;CACpD;AAOD,wBAAgB,oBAAoB,IAAI,UAAU,CAiBjD;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,UAAU,GAAG,MAAM,CAwB5D;AA4BD,wBAAsB,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,SAAc,GAAG,OAAO,CAAC,MAAM,CAAC,CA+BjG;AAED,wBAAuB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,SAAc,GAAG,cAAc,CAAC,MAAM,CAAC,CA4BvG;AAED;;;;GAIG;AACH,wBAAsB,2BAA2B,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,GAAE,SAAc,GAAG,OAAO,CAAC,MAAM,CAAC,CAsB1G;AAgFD,wBAAsB,uBAAuB,CAC3C,MAAM,EAAE,MAAM,EACd,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,OAAO,EAAE,CAAA;CAAE,CAAC,CA4BlD;AAgCD,wBAAsB,uBAAuB,CAC3C,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,MAAM,EACnB,OAAO,GAAE,SAAc,GACtB,OAAO,CAAC,MAAM,CAAC,CAwCjB;AAED,wBAAgB,qBAAqB,IAAI,UAAU,EAAE,CAUpD;AA4HD,wBAAsB,4BAA4B,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,mBAAmB,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAiBjH;AAED,wBAAsB,0BAA0B,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,yBAAyB,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAkErH;AAGD,OAAO,EAAE,CAAC,EAAE,CAAC;AAGb,OAAO,EACL,MAAM,IAAI,SAAS,EACnB,YAAY,EACZ,aAAa,EACb,WAAW,EACX,eAAe,EACf,kBAAkB,EAClB,uBAAuB,EACvB,cAAc,EACd,eAAe,EACf,qBAAqB,EACrB,KAAK,WAAW,EAChB,KAAK,cAAc,GACpB,MAAM,cAAc,CAAC"}

package/dist/index.js

@@ -4,6 +4,9 @@ import { anthropic } from '@ai-sdk/anthropic';
 import { google } from '@ai-sdk/google';
 import { xai } from '@ai-sdk/xai';
 import { configureAIProvider } from './config.js';
+import { DEFAULT_MODELS, AIProvider } from './constants.js';
+import { createCustomProvider } from './providers/custom.js';
+export { AIProvider } from './constants.js';
 import { z } from 'zod';
 import { getRegisteredToolsForAI } from './mcp-server.js';
 import { startServer } from './mcp-server.js';
@@ -104,33 +107,47 @@ function extractAndLogError(error, provider, operation) {
         debugError('Unknown error type: %O', error);
     }
 }
-export var AIProvider;
-(function (AIProvider) {
-    AIProvider["OpenAI"] = "openai";
-    AIProvider["Anthropic"] = "anthropic";
-    AIProvider["Google"] = "google";
-    AIProvider["XAI"] = "xai";
-})(AIProvider || (AIProvider = {}));
 const defaultOptions = {
     temperature: 0.7,
     maxTokens: 1000,
 };
-function getDefaultModel(provider) {
-    const model = (() => {
-        switch (provider) {
-            case AIProvider.OpenAI:
-                return 'gpt-4o-mini'; // maybe 5
-            case AIProvider.Anthropic:
-                return 'claude-sonnet-4-20250514'; // 4
-            case AIProvider.Google:
-                return 'gemini-2.5-pro';
-            case AIProvider.XAI:
-                return 'grok-4';
-            default:
-                // eslint-disable-next-line @typescript-eslint/restrict-template-expressions
-                throw new Error(`Unknown provider: ${provider}`);
+export function getDefaultAIProvider() {
+    const envProvider = process.env.DEFAULT_AI_PROVIDER?.toLowerCase();
+    switch (envProvider) {
+        case 'openai':
+            return AIProvider.OpenAI;
+        case 'anthropic':
+            return AIProvider.Anthropic;
+        case 'google':
+            return AIProvider.Google;
+        case 'xai':
+            return AIProvider.XAI;
+        default: {
+            // Fallback to the first available provider
+            const available = getAvailableProviders();
+            return available.length > 0 ? available[0] : AIProvider.Anthropic;
         }
-    })();
+    }
+}
+export function getDefaultModel(provider) {
+    // Check if DEFAULT_AI_MODEL is set in environment
+    const envModel = process.env.DEFAULT_AI_MODEL;
+    if (envModel !== undefined && envModel !== null && envModel.trim().length > 0) {
+        debugConfig('Using DEFAULT_AI_MODEL from environment: %s for provider %s', envModel, provider);
+        return envModel.trim();
+    }
+    if (provider === AIProvider.Custom) {
+        const config = configureAIProvider();
+        if (config.custom != null) {
+            debugConfig('Selected custom provider default model %s for provider %s', config.custom.defaultModel, provider);
+            return config.custom.defaultModel;
+        }
+        throw new Error('Custom provider not configured');
+    }
+    const model = DEFAULT_MODELS[provider] ??
+        (() => {
+            throw new Error(`Unknown provider: ${provider}`);
+        })();
     debugConfig('Selected default model %s for provider %s', model, provider);
     return model;
 }
@@ -146,23 +163,32 @@ function getModel(provider, model) {
             return google(modelName);
         case AIProvider.XAI:
             return xai(modelName);
+        case AIProvider.Custom: {
+            const config = configureAIProvider();
+            if (config.custom == null) {
+                throw new Error('Custom provider not configured');
+            }
+            const customProvider = createCustomProvider(config.custom);
+            return customProvider.languageModel(modelName);
+        }
         default:
             throw new Error(`Unknown provider: ${provider}`);
     }
 }
-export async function generateTextWithAI(prompt, provider, options = {}) {
-    debugAPI('generateTextWithAI called - provider: %s, promptLength: %d', provider, prompt.length);
+export async function generateTextWithAI(prompt, options = {}) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugAPI('generateTextWithAI called - provider: %s, promptLength: %d', resolvedProvider, prompt.length);
     const finalOptions = { ...defaultOptions, ...options };
-    const model = finalOptions.model ?? getDefaultModel(provider);
-    const modelInstance = getModel(provider, model);
+    const model = finalOptions.model ?? getDefaultModel(resolvedProvider);
+    const modelInstance = getModel(resolvedProvider, model);
     if (finalOptions.includeTools === true) {
         debugTools('Tools requested, starting MCP server');
         await startServer();
-        const result = await generateTextWithToolsAI(prompt, provider, options);
+        const result = await generateTextWithToolsAI(prompt, options);
         return result.text;
     }
     try {
-        debugAPI('Making API call to %s with model %s', provider, model);
+        debugAPI('Making API call to %s with model %s', resolvedProvider, model);
         debugAPI('Request params - temperature: %d, maxTokens: %d', finalOptions.temperature, finalOptions.maxTokens);
         const result = await generateText({
             model: modelInstance,
@@ -174,16 +200,17 @@ export async function generateTextWithAI(prompt, provider, options = {}) {
         return result.text;
     }
     catch (error) {
-        extractAndLogError(error, provider, 'generateTextWithAI');
+        extractAndLogError(error, resolvedProvider, 'generateTextWithAI');
         throw error;
     }
 }
-export async function* streamTextWithAI(prompt, provider, options = {}) {
-    debugStream('streamTextWithAI called - provider: %s, promptLength: %d', provider, prompt.length);
+export async function* streamTextWithAI(prompt, options = {}) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugStream('streamTextWithAI called - provider: %s, promptLength: %d', resolvedProvider, prompt.length);
     const finalOptions = { ...defaultOptions, ...options };
-    const model = getModel(provider, finalOptions.model);
+    const model = getModel(resolvedProvider, finalOptions.model);
     try {
-        debugStream('Starting stream from %s', provider);
+        debugStream('Starting stream from %s', resolvedProvider);
         const stream = await streamText({
             model,
             prompt,
@@ -201,7 +228,7 @@ export async function* streamTextWithAI(prompt, provider, options = {}) {
         debugStream('Stream completed - total chunks: %d, total length: %d', totalChunks, totalLength);
     }
     catch (error) {
-        extractAndLogError(error, provider, 'streamTextWithAI');
+        extractAndLogError(error, resolvedProvider, 'streamTextWithAI');
         throw error;
     }
 }
@@ -210,11 +237,12 @@ export async function* streamTextWithAI(prompt, provider, options = {}) {
  * Optionally calls a stream callback for each token if provided.
  * Always returns the complete collected response.
  */
-export async function generateTextStreamingWithAI(prompt, provider, options = {}) {
-    debugStream('generateTextStreamingWithAI called - provider: %s', provider);
+export async function generateTextStreamingWithAI(prompt, options = {}) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugStream('generateTextStreamingWithAI called - provider: %s', resolvedProvider);
     const finalOptions = { ...defaultOptions, ...options };
     let collectedResult = '';
-    const stream = streamTextWithAI(prompt, provider, finalOptions);
+    const stream = streamTextWithAI(prompt, finalOptions);
     let tokenCount = 0;
     for await (const token of stream) {
         tokenCount++;
@@ -282,18 +310,19 @@ async function executeToolConversation(modelInstance, messages, registeredTools,
     }
     return { finalResult, allToolCalls };
 }
-export async function generateTextWithToolsAI(prompt, provider, options = {}) {
-    debugTools('generateTextWithToolsAI called - provider: %s', provider);
+export async function generateTextWithToolsAI(prompt, options = {}) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugTools('generateTextWithToolsAI called - provider: %s', resolvedProvider);
     const finalOptions = { ...defaultOptions, ...options };
-    const model = finalOptions.model ?? getDefaultModel(provider);
-    const modelInstance = getModel(provider, model);
+    const model = finalOptions.model ?? getDefaultModel(resolvedProvider);
+    const modelInstance = getModel(resolvedProvider, model);
     const registeredTools = finalOptions.includeTools === true ? getRegisteredToolsForAI() : {};
     debugTools('Registered tools: %o', Object.keys(registeredTools));
     const hasTools = Object.keys(registeredTools).length > 0;
     debugTools('Has tools available: %s', hasTools);
     // Build conversation messages
     const messages = [{ role: 'user', content: prompt }];
-    const { finalResult, allToolCalls } = await executeToolConversation(modelInstance, messages, registeredTools, hasTools, finalOptions, provider);
+    const { finalResult, allToolCalls } = await executeToolConversation(modelInstance, messages, registeredTools, hasTools, finalOptions, resolvedProvider);
     return {
         text: finalResult,
         toolCalls: allToolCalls,
@@ -325,17 +354,18 @@ async function executeToolCalls(toolCalls, registeredTools) {
     }
     return toolResults;
 }
-export async function generateTextWithImageAI(text, imageBase64, provider, options = {}) {
-    debugAPI('generateTextWithImageAI called - provider: %s, textLength: %d, imageSize: %d', provider, text.length, imageBase64.length);
+export async function generateTextWithImageAI(text, imageBase64, options = {}) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugAPI('generateTextWithImageAI called - provider: %s, textLength: %d, imageSize: %d', resolvedProvider, text.length, imageBase64.length);
     const finalOptions = { ...defaultOptions, ...options };
-    const model = finalOptions.model ?? getDefaultModel(provider);
-    const modelInstance = getModel(provider, model);
-    if (provider !== AIProvider.OpenAI && provider !== AIProvider.XAI) {
-        debugError('Provider %s does not support image inputs', provider);
-        throw new Error(`Provider ${provider} does not support image inputs`);
+    const model = finalOptions.model ?? getDefaultModel(resolvedProvider);
+    const modelInstance = getModel(resolvedProvider, model);
+    if (resolvedProvider !== AIProvider.OpenAI && resolvedProvider !== AIProvider.XAI) {
+        debugError('Provider %s does not support image inputs', resolvedProvider);
+        throw new Error(`Provider ${resolvedProvider} does not support image inputs`);
     }
     try {
-        debugAPI('Sending image+text to %s', provider);
+        debugAPI('Sending image+text to %s', resolvedProvider);
         const result = await generateText({
             model: modelInstance,
             messages: [
@@ -354,21 +384,23 @@ export async function generateTextWithImageAI(text, imageBase64, provider, optio
         return result.text;
     }
     catch (error) {
-        extractAndLogError(error, provider, 'generateTextWithImageAI');
+        extractAndLogError(error, resolvedProvider, 'generateTextWithImageAI');
         throw error;
     }
 }
 export function getAvailableProviders() {
     const config = configureAIProvider();
     const providers = [];
-    if (config.openai != null)
-        providers.push(AIProvider.OpenAI);
     if (config.anthropic != null)
         providers.push(AIProvider.Anthropic);
+    if (config.openai != null)
+        providers.push(AIProvider.OpenAI);
     if (config.google != null)
         providers.push(AIProvider.Google);
     if (config.xai != null)
         providers.push(AIProvider.XAI);
+    if (config.custom != null)
+        providers.push(AIProvider.Custom);
     debugConfig('Available providers: %o', providers);
     return providers;
 }
@@ -459,8 +491,9 @@ async function attemptStructuredGeneration(prompt, provider, options, registered
     // If we exhausted all retries, throw the last error
     throw lastError;
 }
-export async function generateStructuredDataWithAI(prompt, provider, options) {
-    debugAPI('generateStructuredDataWithAI called - provider: %s, schema: %s', provider, options.schemaName ?? 'unnamed');
+export async function generateStructuredDataWithAI(prompt, options) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugAPI('generateStructuredDataWithAI called - provider: %s, schema: %s', resolvedProvider, options.schemaName ?? 'unnamed');
     if (options.includeTools === true) {
         debugTools('Tools requested, starting MCP server');
         await startServer();
@@ -468,16 +501,17 @@ export async function generateStructuredDataWithAI(prompt, provider, options) {
     const registeredTools = options.includeTools === true ? getRegisteredToolsForAI() : {};
     debugTools('Registered tools for structured data: %o', Object.keys(registeredTools));
     const hasTools = Object.keys(registeredTools).length > 0;
-    return attemptStructuredGeneration(prompt, provider, options, registeredTools, hasTools);
+    return attemptStructuredGeneration(prompt, resolvedProvider, options, registeredTools, hasTools);
 }
-export async function streamStructuredDataWithAI(prompt, provider, options) {
-    debugStream('streamStructuredDataWithAI called - provider: %s, schema: %s', provider, options.schemaName ?? 'unnamed');
+export async function streamStructuredDataWithAI(prompt, options) {
+    const resolvedProvider = options.provider ?? getDefaultAIProvider();
+    debugStream('streamStructuredDataWithAI called - provider: %s, schema: %s', resolvedProvider, options.schemaName ?? 'unnamed');
     const maxSchemaRetries = 3;
     let lastError;
     for (let attempt = 0; attempt < maxSchemaRetries; attempt++) {
         try {
             debugValidation('Stream structured data attempt %d/%d', attempt + 1, maxSchemaRetries);
-            const model = getModel(provider, options.model);
+            const model = getModel(resolvedProvider, options.model);
             const enhancedPrompt = attempt > 0 && lastError ? getEnhancedPrompt(prompt, lastError) : prompt;
             const result = streamObject({
                 model,