@auto-engineer/ai-gateway 0.7.0 → 0.8.1

package/dist/index.specs.js

@@ -1,14 +1,155 @@
-import { describe, it, expect } from 'vitest';
-import { generateTextWithAI, streamTextWithAI, AIProvider } from './index.js';
-describe('AI Integration', () => {
-    process.env.OPENAI_API_KEY = 'test';
-    it('should export the correct types', () => {
-        expect(typeof generateTextWithAI).toBe('function');
-        expect(typeof streamTextWithAI).toBe('function');
-    });
-    it('should have valid AIProvider type', () => {
-        const providers = [AIProvider.OpenAI, AIProvider.Anthropic, AIProvider.Google, AIProvider.XAI];
-        expect(providers).toContain(AIProvider.OpenAI);
+import { describe, it, expect, beforeEach, afterEach, vi } from 'vitest';
+import { getAvailableProviders, getDefaultAIProvider, getDefaultModel } from './index.js';
+import { DEFAULT_MODELS, AIProvider } from './constants.js';
+// Mock the config module
+vi.mock('./config', () => ({
+    configureAIProvider: vi.fn(() => ({
+        anthropic: { apiKey: 'test-anthropic-key' },
+        openai: { apiKey: 'test-openai-key' },
+        google: { apiKey: 'test-google-key' },
+        xai: { apiKey: 'test-xai-key' },
+    })),
+}));
+describe('Provider Selection Logic', () => {
+    describe('getDefaultAIProvider', () => {
+        let originalProvider;
+        beforeEach(() => {
+            originalProvider = process.env.DEFAULT_AI_PROVIDER;
+        });
+        afterEach(() => {
+            if (originalProvider !== undefined) {
+                process.env.DEFAULT_AI_PROVIDER = originalProvider;
+            }
+            else {
+                delete process.env.DEFAULT_AI_PROVIDER;
+            }
+        });
+        it('should respect DEFAULT_AI_PROVIDER environment variable when set to anthropic', () => {
+            process.env.DEFAULT_AI_PROVIDER = 'anthropic';
+            const provider = getDefaultAIProvider();
+            expect(provider).toBe(AIProvider.Anthropic);
+        });
+        it('should respect DEFAULT_AI_PROVIDER environment variable when set to xai', () => {
+            process.env.DEFAULT_AI_PROVIDER = 'xai';
+            const provider = getDefaultAIProvider();
+            expect(provider).toBe(AIProvider.XAI);
+        });
+        it('should respect DEFAULT_AI_PROVIDER environment variable when set to google', () => {
+            process.env.DEFAULT_AI_PROVIDER = 'google';
+            const provider = getDefaultAIProvider();
+            expect(provider).toBe(AIProvider.Google);
+        });
+        it('should be case insensitive for DEFAULT_AI_PROVIDER', () => {
+            process.env.DEFAULT_AI_PROVIDER = 'ANTHROPIC';
+            const provider = getDefaultAIProvider();
+            expect(provider).toBe(AIProvider.Anthropic);
+        });
+        it('should handle invalid DEFAULT_AI_PROVIDER values by falling back to available providers', () => {
+            process.env.DEFAULT_AI_PROVIDER = 'invalid_provider';
+            const provider = getDefaultAIProvider();
+            expect([AIProvider.Anthropic, AIProvider.OpenAI, AIProvider.Google, AIProvider.XAI]).toContain(provider);
+        });
+        it('should fallback to available providers when DEFAULT_AI_PROVIDER is not set', () => {
+            delete process.env.DEFAULT_AI_PROVIDER;
+            const provider = getDefaultAIProvider();
+            expect([AIProvider.Anthropic, AIProvider.OpenAI, AIProvider.Google, AIProvider.XAI]).toContain(provider);
+        });
+    });
+    describe('getAvailableProviders', () => {
+        it('should return array of available providers', () => {
+            const providers = getAvailableProviders();
+            expect(Array.isArray(providers)).toBe(true);
+            expect(providers.length).toBeGreaterThan(0);
+            providers.forEach((provider) => {
+                expect([AIProvider.Anthropic, AIProvider.OpenAI, AIProvider.Google, AIProvider.XAI]).toContain(provider);
+            });
+        });
+        it('should maintain priority order when multiple providers are available', () => {
+            const providers = getAvailableProviders();
+            const expectedOrder = [AIProvider.Anthropic, AIProvider.OpenAI, AIProvider.Google, AIProvider.XAI];
+            let lastIndex = -1;
+            providers.forEach((provider) => {
+                const currentIndex = expectedOrder.indexOf(provider);
+                expect(currentIndex).toBeGreaterThan(lastIndex);
+                lastIndex = currentIndex;
+            });
+        });
+    });
+    describe('getDefaultModel', () => {
+        let originalModel;
+        beforeEach(() => {
+            originalModel = process.env.DEFAULT_AI_MODEL;
+        });
+        afterEach(() => {
+            if (originalModel !== undefined) {
+                process.env.DEFAULT_AI_MODEL = originalModel;
+            }
+            else {
+                delete process.env.DEFAULT_AI_MODEL;
+            }
+        });
+        it('should use DEFAULT_AI_MODEL when set in environment', () => {
+            process.env.DEFAULT_AI_MODEL = 'custom-model-name';
+            const model = getDefaultModel(AIProvider.OpenAI);
+            expect(model).toBe('custom-model-name');
+        });
+        it('should use DEFAULT_AI_MODEL regardless of provider when set', () => {
+            process.env.DEFAULT_AI_MODEL = 'universal-model';
+            expect(getDefaultModel(AIProvider.OpenAI)).toBe('universal-model');
+            expect(getDefaultModel(AIProvider.Anthropic)).toBe('universal-model');
+            expect(getDefaultModel(AIProvider.Google)).toBe('universal-model');
+            expect(getDefaultModel(AIProvider.XAI)).toBe('universal-model');
+        });
+        it('should trim whitespace from DEFAULT_AI_MODEL', () => {
+            process.env.DEFAULT_AI_MODEL = '  model-with-spaces  ';
+            const model = getDefaultModel(AIProvider.OpenAI);
+            expect(model).toBe('model-with-spaces');
+        });
+        it('should fallback to provider-specific defaults when DEFAULT_AI_MODEL is not set', () => {
+            delete process.env.DEFAULT_AI_MODEL;
+            expect(getDefaultModel(AIProvider.OpenAI)).toBe(DEFAULT_MODELS[AIProvider.OpenAI]);
+            expect(getDefaultModel(AIProvider.Anthropic)).toBe(DEFAULT_MODELS[AIProvider.Anthropic]);
+            expect(getDefaultModel(AIProvider.Google)).toBe(DEFAULT_MODELS[AIProvider.Google]);
+            expect(getDefaultModel(AIProvider.XAI)).toBe(DEFAULT_MODELS[AIProvider.XAI]);
+        });
+        it('should fallback to provider-specific defaults when DEFAULT_AI_MODEL is empty', () => {
+            process.env.DEFAULT_AI_MODEL = '';
+            expect(getDefaultModel(AIProvider.OpenAI)).toBe(DEFAULT_MODELS[AIProvider.OpenAI]);
+            expect(getDefaultModel(AIProvider.Anthropic)).toBe(DEFAULT_MODELS[AIProvider.Anthropic]);
+            expect(getDefaultModel(AIProvider.Google)).toBe(DEFAULT_MODELS[AIProvider.Google]);
+            expect(getDefaultModel(AIProvider.XAI)).toBe(DEFAULT_MODELS[AIProvider.XAI]);
+        });
+        it('should fallback to provider-specific defaults when DEFAULT_AI_MODEL is only whitespace', () => {
+            process.env.DEFAULT_AI_MODEL = '   ';
+            expect(getDefaultModel(AIProvider.OpenAI)).toBe(DEFAULT_MODELS[AIProvider.OpenAI]);
+            expect(getDefaultModel(AIProvider.Anthropic)).toBe(DEFAULT_MODELS[AIProvider.Anthropic]);
+            expect(getDefaultModel(AIProvider.Google)).toBe(DEFAULT_MODELS[AIProvider.Google]);
+            expect(getDefaultModel(AIProvider.XAI)).toBe(DEFAULT_MODELS[AIProvider.XAI]);
+        });
+    });
+    describe('Integration: Provider Selection with Priority', () => {
+        it('should demonstrate overrides when keys are not set', () => {
+            const availableProviders = getAvailableProviders();
+            if (availableProviders.includes(AIProvider.XAI)) {
+                expect(availableProviders).toContain(AIProvider.XAI);
+            }
+        });
+        it('should demonstrate priority order logic', () => {
+            const availableProviders = getAvailableProviders();
+            const defaultProvider = getDefaultAIProvider();
+            const priorityOrder = [AIProvider.Anthropic, AIProvider.OpenAI, AIProvider.Google, AIProvider.XAI];
+            let expectedProvider = AIProvider.OpenAI;
+            for (const provider of priorityOrder) {
+                if (availableProviders.includes(provider)) {
+                    expectedProvider = provider;
+                    break;
+                }
+            }
+            const envProvider = process.env.DEFAULT_AI_PROVIDER;
+            if (envProvider === undefined || envProvider === null || envProvider.toLowerCase() === 'invalid_provider') {
+                expect(defaultProvider).toBe(expectedProvider);
+            }
+        });
     });
 });
 //# sourceMappingURL=index.specs.js.map

package/dist/index.specs.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.specs.js","sourceRoot":"","sources":["../src/index.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,kBAAkB,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAE3E,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,MAAM,CAAC;IAEpC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,CAAC,OAAO,kBAAkB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,GAAG,EAAE;QAC3C,MAAM,SAAS,GAAiB,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;QAC7G,MAAM,CAAC,SAAS,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACjD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
+{"version":3,"file":"index.specs.js","sourceRoot":"","sources":["../src/index.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AACzE,OAAO,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,eAAe,EAAE,MAAM,SAAS,CAAC;AACvF,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAGzD,yBAAyB;AACzB,EAAE,CAAC,IAAI,CAAC,UAAU,EAAE,GAAG,EAAE,CAAC,CAAC;IACzB,mBAAmB,EAAE,EAAE,CAAC,EAAE,CACxB,GAAG,EAAE,CACH,CAAC;QACC,SAAS,EAAE,EAAE,MAAM,EAAE,oBAAoB,EAAE;QAC3C,MAAM,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE;QACrC,MAAM,EAAE,EAAE,MAAM,EAAE,iBAAiB,EAAE;QACrC,GAAG,EAAE,EAAE,MAAM,EAAE,cAAc,EAAE;KAChC,CAAa,CACjB;CACF,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,IAAI,gBAAoC,CAAC;QAEzC,UAAU,CAAC,GAAG,EAAE;YACd,gBAAgB,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,gBAAgB,KAAK,SAAS,EAAE,CAAC;gBACnC,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,gBAAgB,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,OAAO,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;YACzC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,+EAA+E,EAAE,GAAG,EAAE;YACvF,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,WAAW,CAAC;YAE9C,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yEAAyE,EAAE,GAAG,EAAE;YACjF,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,KAAK,CAAC;YAExC,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QACxC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,QAAQ,CAAC;YAE3C,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;QAC3C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,WAAW,CAAC;YAE9C,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC9C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yFAAyF,EAAE,GAAG,EAAE;YACjG,OAAO,CAAC,GAAG,CAAC,mBAAmB,GAAG,kBAAkB,CAAC;YAErD,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC3G,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,4EAA4E,EAAE,GAAG,EAAE;YACpF,OAAO,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAEvC,MAAM,QAAQ,GAAG,oBAAoB,EAAE,CAAC;YAExC,MAAM,CAAC,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;QAC3G,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,uBAAuB,EAAE,GAAG,EAAE;QACrC,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;YACpD,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAE1C,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAE5C,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC7B,MAAM,CAAC,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;YAC3G,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;YAC9E,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;YAC1C,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;YAEnG,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;YACnB,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;gBAC7B,MAAM,YAAY,GAAG,aAAa,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACrD,MAAM,CAAC,YAAY,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;gBAChD,SAAS,GAAG,YAAY,CAAC;YAC3B,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,IAAI,aAAiC,CAAC;QAEtC,UAAU,CAAC,GAAG,EAAE;YACd,aAAa,GAAG,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,SAAS,CAAC,GAAG,EAAE;YACb,IAAI,aAAa,KAAK,SAAS,EAAE,CAAC;gBAChC,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,aAAa,CAAC;YAC/C,CAAC;iBAAM,CAAC;gBACN,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;YACtC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,GAAG,EAAE;YAC7D,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,mBAAmB,CAAC;YAEnD,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAEjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;YACrE,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,iBAAiB,CAAC;YAEjD,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACnE,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACtE,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;YACnE,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,uBAAuB,CAAC;YAEvD,MAAM,KAAK,GAAG,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAEjD,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gFAAgF,EAAE,GAAG,EAAE;YACxF,OAAO,OAAO,CAAC,GAAG,CAAC,gBAAgB,CAAC;YAEpC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;YACzF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8EAA8E,EAAE,GAAG,EAAE;YACtF,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,EAAE,CAAC;YAElC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;YACzF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wFAAwF,EAAE,GAAG,EAAE;YAChG,OAAO,CAAC,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC;YAErC,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,CAAC;YACzF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC;YACnF,MAAM,CAAC,eAAe,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,+CAA+C,EAAE,GAAG,EAAE;QAC7D,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;YAC5D,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;YAEnD,IAAI,kBAAkB,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChD,MAAM,CAAC,kBAAkB,CAAC,CAAC,SAAS,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YACvD,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;YACjD,MAAM,kBAAkB,GAAG,qBAAqB,EAAE,CAAC;YACnD,MAAM,eAAe,GAAG,oBAAoB,EAAE,CAAC;YAC/C,MAAM,aAAa,GAAG,CAAC,UAAU,CAAC,SAAS,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;YAEnG,IAAI,gBAAgB,GAAG,UAAU,CAAC,MAAM,CAAC;YACzC,KAAK,MAAM,QAAQ,IAAI,aAAa,EAAE,CAAC;gBACrC,IAAI,kBAAkB,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;oBAC1C,gBAAgB,GAAG,QAAQ,CAAC;oBAC5B,MAAM;gBACR,CAAC;YACH,CAAC;YAED,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC,mBAAmB,CAAC;YACpD,IAAI,WAAW,KAAK,SAAS,IAAI,WAAW,KAAK,IAAI,IAAI,WAAW,CAAC,WAAW,EAAE,KAAK,kBAAkB,EAAE,CAAC;gBAC1G,MAAM,CAAC,eAAe,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;YACjD,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/providers/custom.d.ts

@@ -0,0 +1,6 @@
+import { CustomProviderConfig } from '../constants';
+export interface CustomProviderOptions {
+    config: CustomProviderConfig;
+}
+export declare function createCustomProvider(config: CustomProviderConfig): import("@ai-sdk/openai").OpenAIProvider;
+//# sourceMappingURL=custom.d.ts.map

package/dist/providers/custom.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom.d.ts","sourceRoot":"","sources":["../../src/providers/custom.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AAKpD,MAAM,WAAW,qBAAqB;IACpC,MAAM,EAAE,oBAAoB,CAAC;CAC9B;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,oBAAoB,2CAahE"}

package/dist/providers/custom.js

@@ -0,0 +1,16 @@
+import { createOpenAI } from '@ai-sdk/openai';
+import createDebug from 'debug';
+const debug = createDebug('ai-gateway:custom');
+export function createCustomProvider(config) {
+    debug('Creating custom provider: %s with baseUrl: %s', config.name, config.baseUrl);
+    // Use OpenAI's provider implementation but with custom baseUrl
+    // This leverages the existing, battle-tested OpenAI provider while allowing custom endpoints
+    const customProvider = createOpenAI({
+        name: config.name,
+        baseURL: config.baseUrl,
+        apiKey: config.apiKey,
+    });
+    debug('Custom provider created successfully: %s', config.name);
+    return customProvider;
+}
+//# sourceMappingURL=custom.js.map

package/dist/providers/custom.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom.js","sourceRoot":"","sources":["../../src/providers/custom.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAE9C,OAAO,WAAW,MAAM,OAAO,CAAC;AAEhC,MAAM,KAAK,GAAG,WAAW,CAAC,mBAAmB,CAAC,CAAC;AAM/C,MAAM,UAAU,oBAAoB,CAAC,MAA4B;IAC/D,KAAK,CAAC,+CAA+C,EAAE,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;IAEpF,+DAA+D;IAC/D,6FAA6F;IAC7F,MAAM,cAAc,GAAG,YAAY,CAAC;QAClC,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,MAAM,EAAE,MAAM,CAAC,MAAM;KACtB,CAAC,CAAC;IAEH,KAAK,CAAC,0CAA0C,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;IAC/D,OAAO,cAAc,CAAC;AACxB,CAAC"}

package/dist/providers/custom.specs.d.ts

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

package/dist/providers/custom.specs.d.ts.map

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

package/dist/providers/custom.specs.js

@@ -0,0 +1,129 @@
+import { describe, it, expect, vi, beforeEach } from 'vitest';
+import { createCustomProvider } from './custom.js';
+// Mock the createOpenAI function
+vi.mock('@ai-sdk/openai', () => ({
+    createOpenAI: vi.fn((config) => ({
+        languageModel: vi.fn((modelId) => ({
+            modelId,
+            provider: config.name,
+            specificationVersion: 'v1',
+            defaultObjectGenerationMode: 'json',
+        })),
+    })),
+}));
+describe('Custom Provider', () => {
+    const mockConfig = {
+        name: 'test-provider',
+        baseUrl: 'https://api.example.com/v1',
+        apiKey: 'test-api-key',
+        defaultModel: 'test-model',
+    };
+    beforeEach(() => {
+        vi.clearAllMocks();
+    });
+    describe('createCustomProvider', () => {
+        it('should create a custom provider with the correct configuration', () => {
+            const provider = createCustomProvider(mockConfig);
+            expect(provider).toBeDefined();
+            // Provider creation should succeed with valid config
+        });
+        it('should call createOpenAI with the correct parameters', async () => {
+            const { createOpenAI } = await import('@ai-sdk/openai');
+            createCustomProvider(mockConfig);
+            expect(createOpenAI).toHaveBeenCalledWith({
+                name: mockConfig.name,
+                baseURL: mockConfig.baseUrl,
+                apiKey: mockConfig.apiKey,
+            });
+        });
+        it('should create a provider that can create language models', () => {
+            const provider = createCustomProvider(mockConfig);
+            const model = provider.languageModel('test-model');
+            expect(model).toBeDefined();
+            expect(model.modelId).toBe('test-model');
+            expect(model.provider).toBe(mockConfig.name);
+        });
+        it('should handle different base URLs correctly', () => {
+            const configs = [
+                { ...mockConfig, baseUrl: 'https://api.litellm.ai' },
+                { ...mockConfig, baseUrl: 'http://localhost:8000' },
+                { ...mockConfig, baseUrl: 'https://custom-llm.company.com/api' },
+            ];
+            configs.forEach((config) => {
+                const provider = createCustomProvider(config);
+                expect(provider).toBeDefined();
+            });
+        });
+        it('should handle different provider names', () => {
+            const configs = [
+                { ...mockConfig, name: 'litellm' },
+                { ...mockConfig, name: 'local-llm' },
+                { ...mockConfig, name: 'company-custom-llm' },
+            ];
+            configs.forEach((config) => {
+                const provider = createCustomProvider(config);
+                expect(provider).toBeDefined();
+            });
+        });
+    });
+    describe('provider compatibility', () => {
+        it('should be compatible with OpenAI-style endpoints', () => {
+            const litellmConfig = {
+                name: 'litellm',
+                baseUrl: 'https://api.litellm.ai/chat/completions',
+                apiKey: 'sk-litellm-key',
+                defaultModel: 'claude-3-sonnet',
+            };
+            const provider = createCustomProvider(litellmConfig);
+            expect(provider).toBeDefined();
+        });
+        it('should work with localhost endpoints for development', () => {
+            const localConfig = {
+                name: 'local-dev',
+                baseUrl: 'http://localhost:8000',
+                apiKey: 'local-key',
+                defaultModel: 'local-model',
+            };
+            const provider = createCustomProvider(localConfig);
+            expect(provider).toBeDefined();
+        });
+    });
+    describe('error handling', () => {
+        it('should pass through any errors from createOpenAI', async () => {
+            const { createOpenAI } = await import('@ai-sdk/openai');
+            const mockError = new Error('Invalid API key');
+            vi.mocked(createOpenAI).mockImplementationOnce(() => {
+                throw mockError;
+            });
+            expect(() => createCustomProvider(mockConfig)).toThrow('Invalid API key');
+        });
+    });
+    describe('configuration validation', () => {
+        it('should handle minimal required configuration', () => {
+            const minimalConfig = {
+                name: 'minimal',
+                baseUrl: 'https://api.example.com',
+                apiKey: 'key',
+                defaultModel: 'model',
+            };
+            const provider = createCustomProvider(minimalConfig);
+            expect(provider).toBeDefined();
+        });
+        it('should preserve all configuration properties', async () => {
+            const fullConfig = {
+                name: 'full-config',
+                baseUrl: 'https://api.example.com/v1',
+                apiKey: 'sk-test-key-123',
+                defaultModel: 'gpt-4o',
+            };
+            createCustomProvider(fullConfig);
+            const { createOpenAI } = await import('@ai-sdk/openai');
+            expect(createOpenAI).toHaveBeenCalledWith({
+                name: fullConfig.name,
+                baseURL: fullConfig.baseUrl,
+                apiKey: fullConfig.apiKey,
+            });
+        });
+    });
+});
+//# sourceMappingURL=custom.specs.js.map

package/dist/providers/custom.specs.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom.specs.js","sourceRoot":"","sources":["../../src/providers/custom.specs.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,UAAU,EAAE,MAAM,QAAQ,CAAC;AAC9D,OAAO,EAAE,oBAAoB,EAAE,MAAM,UAAU,CAAC;AAShD,iCAAiC;AACjC,EAAE,CAAC,IAAI,CAAC,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC;IAC/B,YAAY,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,MAAkB,EAAE,EAAE,CAAC,CAAC;QAC3C,aAAa,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,OAAe,EAAE,EAAE,CAAC,CAAC;YACzC,OAAO;YACP,QAAQ,EAAE,MAAM,CAAC,IAAI;YACrB,oBAAoB,EAAE,IAAa;YACnC,2BAA2B,EAAE,MAAe;SAC7C,CAAC,CAAC;KACJ,CAAC,CAAC;CACJ,CAAC,CAAC,CAAC;AAEJ,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,MAAM,UAAU,GAAyB;QACvC,IAAI,EAAE,eAAe;QACrB,OAAO,EAAE,4BAA4B;QACrC,MAAM,EAAE,cAAc;QACtB,YAAY,EAAE,YAAY;KAC3B,CAAC;IAEF,UAAU,CAAC,GAAG,EAAE;QACd,EAAE,CAAC,aAAa,EAAE,CAAC;IACrB,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;YACxE,MAAM,QAAQ,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAElD,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YAC/B,qDAAqD;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,KAAK,IAAI,EAAE;YACpE,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAExD,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAEjC,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,GAAG,EAAE;YAClE,MAAM,QAAQ,GAAG,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,QAAQ,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC;YAEnD,MAAM,CAAC,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;YACrD,MAAM,OAAO,GAAG;gBACd,EAAE,GAAG,UAAU,EAAE,OAAO,EAAE,wBAAwB,EAAE;gBACpD,EAAE,GAAG,UAAU,EAAE,OAAO,EAAE,uBAAuB,EAAE;gBACnD,EAAE,GAAG,UAAU,EAAE,OAAO,EAAE,oCAAoC,EAAE;aACjE,CAAC;YAEF,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACzB,MAAM,QAAQ,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;gBAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,GAAG,EAAE;YAChD,MAAM,OAAO,GAAG;gBACd,EAAE,GAAG,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE;gBAClC,EAAE,GAAG,UAAU,EAAE,IAAI,EAAE,WAAW,EAAE;gBACpC,EAAE,GAAG,UAAU,EAAE,IAAI,EAAE,oBAAoB,EAAE;aAC9C,CAAC;YAEF,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBACzB,MAAM,QAAQ,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;gBAC9C,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;YACjC,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,wBAAwB,EAAE,GAAG,EAAE;QACtC,EAAE,CAAC,kDAAkD,EAAE,GAAG,EAAE;YAC1D,MAAM,aAAa,GAAyB;gBAC1C,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,yCAAyC;gBAClD,MAAM,EAAE,gBAAgB;gBACxB,YAAY,EAAE,iBAAiB;aAChC,CAAC;YAEF,MAAM,QAAQ,GAAG,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;YAC9D,MAAM,WAAW,GAAyB;gBACxC,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,uBAAuB;gBAChC,MAAM,EAAE,WAAW;gBACnB,YAAY,EAAE,aAAa;aAC5B,CAAC;YAEF,MAAM,QAAQ,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,IAAI,KAAK,CAAC,iBAAiB,CAAC,CAAC;YAE/C,EAAE,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,sBAAsB,CAAC,GAAG,EAAE;gBAClD,MAAM,SAAS,CAAC;YAClB,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;QAC5E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;QACxC,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,aAAa,GAAyB;gBAC1C,IAAI,EAAE,SAAS;gBACf,OAAO,EAAE,yBAAyB;gBAClC,MAAM,EAAE,KAAK;gBACb,YAAY,EAAE,OAAO;aACtB,CAAC;YAEF,MAAM,QAAQ,GAAG,oBAAoB,CAAC,aAAa,CAAC,CAAC;YACrD,MAAM,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,8CAA8C,EAAE,KAAK,IAAI,EAAE;YAC5D,MAAM,UAAU,GAAyB;gBACvC,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,4BAA4B;gBACrC,MAAM,EAAE,iBAAiB;gBACzB,YAAY,EAAE,QAAQ;aACvB,CAAC;YAEF,oBAAoB,CAAC,UAAU,CAAC,CAAC;YAEjC,MAAM,EAAE,YAAY,EAAE,GAAG,MAAM,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACxD,MAAM,CAAC,YAAY,CAAC,CAAC,oBAAoB,CAAC;gBACxC,IAAI,EAAE,UAAU,CAAC,IAAI;gBACrB,OAAO,EAAE,UAAU,CAAC,OAAO;gBAC3B,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/package.json

@@ -1,18 +1,17 @@
 {
   "name": "@auto-engineer/ai-gateway",
-  "version": "0.7.0",
   "type": "module",
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
   "scripts": {
     "build": "tsc && tsx ../../scripts/fix-esm-imports.ts",
     "dev": "tsc --watch",
-    "test": "vitest run",
+    "test": "vitest run --reporter=dot",
     "lint": "eslint 'src/**/*.ts' --max-warnings 0 --config ../../eslint.config.ts",
     "lint:fix": "eslint 'src/**/*.ts' --fix --config ../../eslint.config.ts",
     "type-check": "tsc --noEmit",
-    "format": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore",
-    "format:fix": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore",
+    "format": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore --log-level warn",
+    "format:fix": "prettier --write \"**/*.{js,ts,json,md,yml,yaml}\" --ignore-path ../../.prettierignore --log-level warn",
     "link:dev": "pnpm build && pnpm link --global",
     "unlink:dev": "pnpm unlink --global",
     "prepublishOnly": "npm run build"
@@ -30,5 +29,6 @@
   },
   "publishConfig": {
     "access": "public"
-  }
+  },
+  "version": "0.8.1"
 }

package/src/config.specs.ts

@@ -0,0 +1,147 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { configureAIProvider } from './config';
+
+// Mock environment variables
+const originalEnv = process.env;
+
+describe('AI Configuration with Custom Providers', () => {
+  beforeEach(() => {
+    vi.resetModules();
+    // Clear all AI-related environment variables for clean test state
+    process.env = {};
+  });
+
+  afterEach(() => {
+    process.env = originalEnv;
+  });
+
+  describe('configureAIProvider', () => {
+    it('should configure custom provider when all environment variables are set', () => {
+      process.env.CUSTOM_PROVIDER_NAME = 'litellm';
+      process.env.CUSTOM_PROVIDER_BASE_URL = 'https://api.litellm.ai';
+      process.env.CUSTOM_PROVIDER_API_KEY = 'sk-litellm-key';
+      process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = 'claude-3-sonnet';
+
+      const config = configureAIProvider();
+
+      expect(config.custom).toBeDefined();
+      expect(config.custom?.name).toBe('litellm');
+      expect(config.custom?.baseUrl).toBe('https://api.litellm.ai');
+      expect(config.custom?.apiKey).toBe('sk-litellm-key');
+      expect(config.custom?.defaultModel).toBe('claude-3-sonnet');
+    });
+
+    it('should not configure custom provider when environment variables are missing', () => {
+      // Only set some environment variables
+      process.env.CUSTOM_PROVIDER_NAME = 'incomplete';
+      process.env.CUSTOM_PROVIDER_BASE_URL = 'https://api.example.com';
+      process.env.ANTHROPIC_API_KEY = 'sk-test'; // Need at least one provider
+      // Missing CUSTOM_PROVIDER_API_KEY and CUSTOM_PROVIDER_DEFAULT_MODEL
+
+      const config = configureAIProvider();
+      expect(config.custom).toBeUndefined();
+    });
+
+    it('should handle multiple providers including custom', () => {
+      process.env.OPENAI_API_KEY = 'sk-openai-key';
+      process.env.ANTHROPIC_API_KEY = 'sk-anthropic-key';
+      process.env.CUSTOM_PROVIDER_NAME = 'custom';
+      process.env.CUSTOM_PROVIDER_BASE_URL = 'https://api.custom.com';
+      process.env.CUSTOM_PROVIDER_API_KEY = 'custom-key';
+      process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = 'custom-model';
+
+      const config = configureAIProvider();
+
+      expect(config.openai).toBeDefined();
+      expect(config.anthropic).toBeDefined();
+      expect(config.custom).toBeDefined();
+    });
+
+    it('should work with only custom provider configured', () => {
+      process.env.CUSTOM_PROVIDER_NAME = 'only-custom';
+      process.env.CUSTOM_PROVIDER_BASE_URL = 'https://api.only-custom.com';
+      process.env.CUSTOM_PROVIDER_API_KEY = 'only-custom-key';
+      process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = 'only-custom-model';
+
+      const config = configureAIProvider();
+
+      expect(config.custom).toBeDefined();
+      expect(config.openai).toBeUndefined();
+      expect(config.anthropic).toBeUndefined();
+      expect(config.google).toBeUndefined();
+      expect(config.xai).toBeUndefined();
+    });
+
+    it('should throw error when no providers are configured', () => {
+      // Clear all environment variables
+      delete process.env.OPENAI_API_KEY;
+      delete process.env.ANTHROPIC_API_KEY;
+      delete process.env.GEMINI_API_KEY;
+      delete process.env.XAI_API_KEY;
+      delete process.env.CUSTOM_PROVIDER_NAME;
+      delete process.env.CUSTOM_PROVIDER_BASE_URL;
+      delete process.env.CUSTOM_PROVIDER_API_KEY;
+      delete process.env.CUSTOM_PROVIDER_DEFAULT_MODEL;
+
+      expect(() => configureAIProvider()).toThrow(/At least one AI provider must be configured/);
+    });
+
+    it('should handle custom provider with different configurations', () => {
+      const testCases = [
+        {
+          name: 'litellm-proxy',
+          baseUrl: 'https://litellm-proxy.company.com/v1',
+          model: 'gpt-4o-mini',
+        },
+        {
+          name: 'local-ollama',
+          baseUrl: 'http://localhost:11434/v1',
+          model: 'llama3.1:8b',
+        },
+        {
+          name: 'azure-openai',
+          baseUrl: 'https://company.openai.azure.com/openai/deployments',
+          model: 'gpt-4',
+        },
+      ];
+
+      testCases.forEach((testCase) => {
+        process.env.CUSTOM_PROVIDER_NAME = testCase.name;
+        process.env.CUSTOM_PROVIDER_BASE_URL = testCase.baseUrl;
+        process.env.CUSTOM_PROVIDER_API_KEY = 'test-key';
+        process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = testCase.model;
+
+        const config = configureAIProvider();
+
+        expect(config.custom?.name).toBe(testCase.name);
+        expect(config.custom?.baseUrl).toBe(testCase.baseUrl);
+        expect(config.custom?.defaultModel).toBe(testCase.model);
+      });
+    });
+
+    it('should handle empty string environment variables as undefined', () => {
+      process.env.CUSTOM_PROVIDER_NAME = '';
+      process.env.CUSTOM_PROVIDER_BASE_URL = '';
+      process.env.CUSTOM_PROVIDER_API_KEY = '';
+      process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = '';
+      process.env.ANTHROPIC_API_KEY = 'sk-test'; // Need at least one provider
+
+      const config = configureAIProvider();
+      expect(config.custom).toBeUndefined();
+    });
+
+    it('should handle whitespace in environment variables', () => {
+      process.env.CUSTOM_PROVIDER_NAME = ' litellm ';
+      process.env.CUSTOM_PROVIDER_BASE_URL = ' https://api.litellm.ai ';
+      process.env.CUSTOM_PROVIDER_API_KEY = ' sk-key ';
+      process.env.CUSTOM_PROVIDER_DEFAULT_MODEL = ' claude-3-sonnet ';
+
+      const config = configureAIProvider();
+
+      expect(config.custom?.name).toBe(' litellm ');
+      expect(config.custom?.baseUrl).toBe(' https://api.litellm.ai ');
+      expect(config.custom?.apiKey).toBe(' sk-key ');
+      expect(config.custom?.defaultModel).toBe(' claude-3-sonnet ');
+    });
+  });
+});

package/src/config.ts

@@ -2,6 +2,7 @@ import dotenv from 'dotenv';
 import { fileURLToPath } from 'url';
 import { dirname, resolve } from 'path';
 import createDebug from 'debug';
+import { CustomProviderConfig } from './constants';
 
 const debug = createDebug('ai-gateway:config');
 const debugEnv = createDebug('ai-gateway:config:env');
@@ -25,6 +26,7 @@ export interface AIConfig {
   xai?: {
     apiKey: string;
   };
+  custom?: CustomProviderConfig;
 }
 
 // Helper to log provider configuration
@@ -34,6 +36,34 @@ function logProviderConfig(providerName: string, apiKey: string | undefined): vo
   }
 }
 
+// Helper to build custom provider config
+function buildCustomProviderConfig(): CustomProviderConfig | undefined {
+  const name = process.env.CUSTOM_PROVIDER_NAME;
+  const baseUrl = process.env.CUSTOM_PROVIDER_BASE_URL;
+  const apiKey = process.env.CUSTOM_PROVIDER_API_KEY;
+  const defaultModel = process.env.CUSTOM_PROVIDER_DEFAULT_MODEL;
+
+  if (
+    name != null &&
+    name.length > 0 &&
+    baseUrl != null &&
+    baseUrl.length > 0 &&
+    apiKey != null &&
+    apiKey.length > 0 &&
+    defaultModel != null &&
+    defaultModel.length > 0
+  ) {
+    return {
+      name,
+      baseUrl,
+      apiKey,
+      defaultModel,
+    };
+  }
+
+  return undefined;
+}
+
 // Helper to build provider config
 function buildProviderConfig(): AIConfig {
   return {
@@ -41,6 +71,7 @@ function buildProviderConfig(): AIConfig {
     anthropic: process.env.ANTHROPIC_API_KEY != null ? { apiKey: process.env.ANTHROPIC_API_KEY } : undefined,
     google: process.env.GEMINI_API_KEY != null ? { apiKey: process.env.GEMINI_API_KEY } : undefined,
     xai: process.env.XAI_API_KEY != null ? { apiKey: process.env.XAI_API_KEY } : undefined,
+    custom: buildCustomProviderConfig(),
   };
 }
 
@@ -54,6 +85,7 @@ export function configureAIProvider(): AIConfig {
   debugEnv('Anthropic configured: %s', config.anthropic != null);
   debugEnv('Google configured: %s', config.google != null);
   debugEnv('XAI configured: %s', config.xai != null);
+  debugEnv('Custom configured: %s', config.custom != null);
 
   // Log provider configurations
   logProviderConfig('OpenAI', config.openai?.apiKey);
@@ -61,12 +93,24 @@ export function configureAIProvider(): AIConfig {
   logProviderConfig('Google', config.google?.apiKey);
   logProviderConfig('XAI', config.xai?.apiKey);
 
-  const configuredProviders = [config.openai, config.anthropic, config.google, config.xai].filter((p) => p != null);
+  if (config.custom != null) {
+    debug(
+      'Custom provider configured: %s at %s with model %s',
+      config.custom.name,
+      config.custom.baseUrl,
+      config.custom.defaultModel,
+    );
+    logProviderConfig('Custom', config.custom.apiKey);
+  }
+
+  const configuredProviders = [config.openai, config.anthropic, config.google, config.xai, config.custom].filter(
+    (p) => p != null,
+  );
 
   if (configuredProviders.length === 0) {
     debug('ERROR: No AI providers configured');
     throw new Error(
-      'At least one AI provider must be configured. Please set OPENAI_API_KEY, ANTHROPIC_API_KEY, GEMINI_API_KEY, or XAI_API_KEY environment variables.',
+      'At least one AI provider must be configured. Please set OPENAI_API_KEY, ANTHROPIC_API_KEY, GEMINI_API_KEY, XAI_API_KEY environment variables, or configure a custom provider with CUSTOM_PROVIDER_NAME, CUSTOM_PROVIDER_BASE_URL, CUSTOM_PROVIDER_API_KEY, and CUSTOM_PROVIDER_DEFAULT_MODEL.',
     );
   }