@auto-engineer/ai-gateway 0.6.3 → 0.8.1

package/.turbo/turbo-build.log

@@ -1,5 +1,5 @@
 
-> @auto-engineer/ai-gateway@0.6.3 build /Users/sam/WebstormProjects/top/auto-engineer/packages/ai-gateway
+> @auto-engineer/ai-gateway@0.8.1 build /home/runner/work/auto-engineer/auto-engineer/packages/ai-gateway
 > tsc && tsx ../../scripts/fix-esm-imports.ts
 
 Fixed ESM imports in dist/

package/.turbo/turbo-test.log

@@ -1,35 +1,17 @@
-
-
-> @auto-engineer/ai-gateway@0.6.3 test /Users/sam/WebstormProjects/top/auto-engineer/packages/ai-gateway
-> vitest run
-
-[?25l
- RUN  v3.2.4 /Users/sam/WebstormProjects/top/auto-engineer/packages/ai-gateway
-
-[?2026h
- ❯ src/index.specs.ts [queued]
-
- Test Files 0 passed (1)
-      Tests 0 passed (0)
-   Start at 23:05:11
-   Duration 129ms
-[?2026l[?2026h
- ❯ src/index.specs.ts 0/2
-
- Test Files 0 passed (1)
-      Tests 0 passed (2)
-   Start at 23:05:11
-   Duration 328ms
-[?2026l ✓ src/index.specs.ts (2 tests) 1ms
-   ✓ AI Integration > should export the correct types 1ms
-   ✓ AI Integration > should have valid AIProvider type 0ms
-stdout | cleanup (/Users/sam/WebstormProjects/top/auto-engineer/packages/ai-gateway/src/mcp-server.ts:58:11)
-Cleaning up...
-
-
- Test Files  1 passed (1)
-      Tests  2 passed (2)
-   Start at  23:05:11
-   Duration  407ms (transform 50ms, setup 0ms, collect 172ms, tests 1ms, environment 0ms, prepare 38ms)
-
-[?25h
+
+> @auto-engineer/ai-gateway@0.8.1 test /home/runner/work/auto-engineer/auto-engineer/packages/ai-gateway
+> vitest run --reporter=dot
+
+
+ RUN  v3.2.4 /home/runner/work/auto-engineer/auto-engineer/packages/ai-gateway
+
+························stdout | cleanup (/home/runner/work/auto-engineer/auto-engineer/packages/ai-gateway/src/mcp-server.ts:58:11)
+Cleaning up...
+
+···················
+
+ Test Files  4 passed (4)
+      Tests  43 passed (43)
+   Start at  00:27:31
+   Duration  3.20s (transform 1.35s, setup 0ms, collect 3.94s, tests 202ms, environment 1ms, prepare 1.78s)
+

package/.turbo/turbo-type-check.log

@@ -1,5 +1,4 @@
-
-
-> @auto-engineer/ai-gateway@0.6.3 type-check /Users/sam/WebstormProjects/top/auto-engineer/packages/ai-gateway
-> tsc --noEmit
-
+
+> @auto-engineer/ai-gateway@0.8.1 type-check /home/runner/work/auto-engineer/auto-engineer/packages/ai-gateway
+> tsc --noEmit
+

package/CHANGELOG.md

@@ -1,5 +1,11 @@
 # @auto-engineer/ai-gateway
 
+## 0.7.0
+
+### Minor Changes
+
+- add command details in dashboard
+
 ## 0.6.3
 
 ### Patch Changes

package/README.md

@@ -0,0 +1,365 @@
+# @auto-engineer/ai-gateway
+
+AI Gateway plugin for the Auto Engineer CLI that provides a unified interface for interacting with multiple AI providers and managing AI-driven workflows. This plugin enables seamless integration with various AI models for text generation, structured data generation, and tool execution in event-driven architectures.
+
+## Installation
+
+This is a plugin for the Auto Engineer CLI. Install both the CLI and this plugin:
+
+```bash
+npm install -g @auto-engineer/cli
+npm install @auto-engineer/ai-gateway
+```
+
+## Configuration
+
+Add this plugin to your `auto.config.ts`:
+
+```typescript
+export default {
+  plugins: [
+    '@auto-engineer/ai-gateway',
+    // ... other plugins
+  ],
+};
+```
+
+### Environment Variables
+
+Configure AI providers by setting environment variables in a `.env` file or your environment:
+
+```bash
+# At least one of these is required
+OPENAI_API_KEY=your-openai-key
+ANTHROPIC_API_KEY=your-anthropic-key
+GEMINI_API_KEY=your-google-key
+XAI_API_KEY=your-xai-key
+
+# Custom Provider Configuration (optional)
+# Use this to connect to any OpenAI-compatible API endpoint
+CUSTOM_PROVIDER_NAME=litellm
+CUSTOM_PROVIDER_BASE_URL=https://api.litellm.ai
+CUSTOM_PROVIDER_API_KEY=your-custom-api-key
+CUSTOM_PROVIDER_DEFAULT_MODEL=claude-3-sonnet
+
+# Optional: Set default provider and model
+DEFAULT_AI_PROVIDER=openai
+DEFAULT_AI_MODEL=gpt-4o-mini
+```
+
+## Commands
+
+This plugin provides the following commands:
+
+- `ai:generate-text` - Generate text using AI models
+- `ai:stream-text` - Stream text output from AI models
+- `ai:generate-structured` - Generate structured data with schema validation
+- `ai:stream-structured` - Stream structured data with schema validation
+
+## What does this plugin do?
+
+The AI Gateway plugin provides a unified interface for interacting with multiple AI providers (OpenAI, Anthropic, Google, XAI, and Custom providers) and integrates with the Auto Engineer ecosystem for AI-driven code generation and tool execution. It supports text generation, structured data generation, and streaming capabilities with robust error handling and debugging.
+
+The plugin now includes custom provider support, enabling integration with any OpenAI-compatible API endpoint, including LiteLLM proxies, local AI models, and corporate AI services.
+
+## Key Features
+
+### Multi-Provider Support
+
+- Supports OpenAI, Anthropic, Google, XAI, and Custom providers
+- Automatic provider selection based on environment configuration
+- Fallback to available providers if default is not configured
+- Configurable default models per provider
+
+### Custom Provider Support
+
+The AI Gateway now supports custom providers, allowing you to connect to any OpenAI-compatible API endpoint. This is particularly useful for:
+
+- **LiteLLM Proxy**: Access 100+ AI models through a single interface
+- **Local AI models**: Connect to locally hosted models (Ollama, local OpenAI servers)
+- **Corporate AI endpoints**: Use company-hosted AI services
+- **Custom AI proxies**: Route through custom authentication or processing layers
+
+#### Configuration
+
+To configure a custom provider, set all four environment variables:
+
+```bash
+CUSTOM_PROVIDER_NAME=your-provider-name
+CUSTOM_PROVIDER_BASE_URL=https://your-api-endpoint.com
+CUSTOM_PROVIDER_API_KEY=your-api-key
+CUSTOM_PROVIDER_DEFAULT_MODEL=your-default-model
+```
+
+#### Common Use Cases
+
+**LiteLLM Proxy:**
+
+```bash
+CUSTOM_PROVIDER_NAME=litellm
+CUSTOM_PROVIDER_BASE_URL=https://api.litellm.ai
+CUSTOM_PROVIDER_API_KEY=sk-litellm-your-key
+CUSTOM_PROVIDER_DEFAULT_MODEL=claude-3-sonnet
+```
+
+**Local Ollama:**
+
+```bash
+CUSTOM_PROVIDER_NAME=ollama
+CUSTOM_PROVIDER_BASE_URL=http://localhost:11434/v1
+CUSTOM_PROVIDER_API_KEY=ollama
+CUSTOM_PROVIDER_DEFAULT_MODEL=llama3.1:8b
+```
+
+**Azure OpenAI:**
+
+```bash
+CUSTOM_PROVIDER_NAME=azure
+CUSTOM_PROVIDER_BASE_URL=https://your-resource.openai.azure.com/openai/deployments
+CUSTOM_PROVIDER_API_KEY=your-azure-key
+CUSTOM_PROVIDER_DEFAULT_MODEL=gpt-4
+```
+
+The custom provider leverages the same robust OpenAI-compatible interface, ensuring full compatibility with all AI Gateway features including streaming, structured generation, and tool integration.
+
+### Text Generation
+
+- Generate text with customizable parameters (temperature, max tokens)
+- Supports both synchronous and streaming text generation
+- Integrates with registered tools for enhanced functionality
+- Image-based text generation for supported providers (OpenAI, XAI)
+
+### Structured Data Generation
+
+- Generates structured data with Zod schema validation
+- Retry logic for schema validation failures
+- Enhanced error prompts for iterative refinement
+- Streaming support for partial object updates
+
+### Tool Integration
+
+- Registers and executes custom tools via the Model Context Protocol (MCP) server
+- Supports batch tool registration
+- Validates tool inputs with Zod schemas
+- Integrates tools with AI-driven workflows
+
+### Debugging Support
+
+Comprehensive debug logging with namespaces:
+
+- `ai-gateway`: General operations
+- `ai-gateway:call`: AI call operations
+- `ai-gateway:provider`: Provider selection and initialization
+- `ai-gateway:error`: Error handling
+- `ai-gateway:stream`: Streaming operations
+- `ai-gateway:result`: Result processing
+
+Enable debugging:
+
+```bash
+DEBUG=ai-gateway:* npm run dev
+```
+
+See [DEBUG.md](./DEBUG.md) for detailed debugging instructions.
+
+## Usage
+
+### Generating Text
+
+```typescript
+import { generateTextWithAI } from '@auto-engineer/ai-gateway';
+
+const result = await generateTextWithAI('Write a poem about the stars', {
+  provider: 'openai',
+  model: 'gpt-4o-mini',
+  temperature: 0.7,
+  maxTokens: 500,
+});
+
+console.log(result);
+```
+
+### Streaming Text
+
+```typescript
+import { generateTextStreamingWithAI } from '@auto-engineer/ai-gateway';
+
+const result = await generateTextStreamingWithAI('Explain quantum computing', {
+  provider: 'anthropic',
+  model: 'claude-sonnet-4-20250514',
+  streamCallback: (token) => process.stdout.write(token),
+});
+
+console.log(result); // Full collected result
+```
+
+### Generating Structured Data
+
+```typescript
+import { generateStructuredDataWithAI, z } from '@auto-engineer/ai-gateway';
+
+const schema = z.object({
+  title: z.string(),
+  description: z.string(),
+  completed: z.boolean(),
+});
+
+const result = await generateStructuredDataWithAI('Generate a todo item', {
+  provider: 'xai',
+  schema,
+  schemaName: 'TodoItem',
+  schemaDescription: 'A todo item with title, description, and completion status',
+});
+
+console.log(result); // { title: string, description: string, completed: boolean }
+```
+
+### Tool Registration and Execution
+
+```typescript
+import { registerTool, startServer } from '@auto-engineer/ai-gateway';
+import { z } from 'zod';
+
+registerTool(
+  'greet',
+  {
+    title: 'Greeting Tool',
+    description: 'Greets users in different languages',
+    inputSchema: {
+      name: z.string().min(1, 'Name is required'),
+      language: z.enum(['en', 'es', 'fr', 'de']).optional().default('en'),
+    },
+  },
+  async ({ name, language = 'en' }) => {
+    const greetings = {
+      en: `Hello, ${name}!`,
+      es: `¡Hola, ${name}!`,
+      fr: `Bonjour, ${name}!`,
+      de: `Hallo, ${name}!`,
+    };
+    return { content: [{ type: 'text', text: greetings[language] }] };
+  },
+);
+
+await startServer();
+```
+
+## Configuration Options
+
+Customize behavior through `auto.config.ts`:
+
+```typescript
+export default {
+  plugins: [
+    [
+      '@auto-engineer/ai-gateway',
+      {
+        // Default AI provider
+        defaultProvider: 'openai',
+
+        // Default model per provider
+        defaultModels: {
+          openai: 'gpt-4o-mini',
+          anthropic: 'claude-sonnet-4-20250514',
+          google: 'gemini-2.5-pro',
+          xai: 'grok-4',
+        },
+
+        // Generation parameters
+        temperature: 0.7,
+        maxTokens: 1000,
+
+        // Tool integration
+        includeToolsByDefault: true,
+      },
+    ],
+  ],
+};
+```
+
+## Integration with Auto Engineer Ecosystem
+
+Works with other Auto Engineer plugins:
+
+- **@auto-engineer/server-implementer**: Uses AI Gateway for AI-driven server code implementation
+- **@auto-engineer/frontend-implementer**: Powers AI-driven frontend code generation
+- **@auto-engineer/flow**: Integrates with Flow specifications for AI-driven workflows
+- **@auto-engineer/server-generator-apollo-emmett**: Enhances server generation with AI capabilities
+- **@auto-engineer/frontend-generator-react-graphql**: Supports AI-driven frontend scaffolding
+
+## Project Structure
+
+```
+ai-gateway/
+├── src/
+│   ├── config.ts          # AI provider configuration
+│   ├── index.ts           # Main API and provider logic
+│   ├── mcp-server.ts      # Model Context Protocol server for tool management
+│   └── example-use.ts     # Example tool implementations
+├── DEBUG.md               # Debugging instructions
+├── CHANGELOG.md           # Version history
+├── package.json
+└── tsconfig.json
+```
+
+## Quality Assurance
+
+- **Type Safety**: Full TypeScript support with Zod schema validation
+- **Error Handling**: Comprehensive error detection and logging
+- **Testing**: Unit tests with Vitest for core functionality
+- **Linting**: ESLint and Prettier for code quality
+- **Debugging**: Detailed logging with `debug` library
+
+## Advanced Features
+
+### Retry Logic
+
+- Automatic retries for schema validation failures
+- Enhanced error prompts for better AI responses
+- Configurable retry limits
+
+### Streaming Support
+
+- Real-time text streaming with callbacks
+- Partial object streaming for structured data
+- Efficient chunk handling for large responses
+
+### Provider Flexibility
+
+- Dynamic provider selection based on availability
+- Environment-based configuration
+- Support for provider-specific error handling
+
+## Getting Started
+
+1. Install the plugin (see Installation above)
+2. Configure environment variables for your AI providers
+3. Add the plugin to `auto.config.ts`
+4. Use the provided commands or import functions for AI operations
+
+Example workflow:
+
+```bash
+# Install dependencies
+npm install @auto-engineer/ai-gateway
+
+# Generate text
+auto ai:generate-text --prompt="Write a story" --provider=openai
+
+# Generate structured data
+auto ai:generate-structured --prompt="Create a user profile" --schema=userSchema.json
+```
+
+## Debugging
+
+Enable detailed logging for troubleshooting:
+
+```bash
+DEBUG=ai-gateway:* npm run dev
+```
+
+See [DEBUG.md](./DEBUG.md) for more details.
+
+## Changelog
+
+See [CHANGELOG.md](./CHANGELOG.md) for version history and updates.

package/dist/config.d.ts

@@ -1,3 +1,4 @@
+import { CustomProviderConfig } from './constants';
 export interface AIConfig {
     openai?: {
         apiKey: string;
@@ -11,6 +12,7 @@ export interface AIConfig {
     xai?: {
         apiKey: string;
     };
+    custom?: CustomProviderConfig;
 }
 export declare function configureAIProvider(): AIConfig;
 //# sourceMappingURL=config.d.ts.map

package/dist/config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAcA,MAAM,WAAW,QAAQ;IACvB,MAAM,CAAC,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,SAAS,CAAC,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,MAAM,CAAC,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,GAAG,CAAC,EAAE;QACJ,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;CACH;AAmBD,wBAAgB,mBAAmB,IAAI,QAAQ,CA6B9C"}
+{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAIA,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAWnD,MAAM,WAAW,QAAQ;IACvB,MAAM,CAAC,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,SAAS,CAAC,EAAE;QACV,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,MAAM,CAAC,EAAE;QACP,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,GAAG,CAAC,EAAE;QACJ,MAAM,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,MAAM,CAAC,EAAE,oBAAoB,CAAC;CAC/B;AAgDD,wBAAgB,mBAAmB,IAAI,QAAQ,CA0C9C"}

package/dist/config.js

@@ -15,6 +15,29 @@ function logProviderConfig(providerName, apiKey) {
         debug('%s provider configured with API key ending in: ...%s', providerName, apiKey.slice(-4));
     }
 }
+// Helper to build custom provider config
+function buildCustomProviderConfig() {
+    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() {
     return {
@@ -22,6 +45,7 @@ function buildProviderConfig() {
         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(),
     };
 }
 export function configureAIProvider() {
@@ -32,15 +56,20 @@ export function configureAIProvider() {
     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);
     logProviderConfig('Anthropic', config.anthropic?.apiKey);
     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.');
+        throw new Error('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.');
     }
     debug('AI configuration complete - %d provider(s) available', configuredProviders.length);
     return config;

package/dist/config.js.map

@@ -1 +1 @@
-{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACxC,OAAO,WAAW,MAAM,OAAO,CAAC;AAEhC,MAAM,KAAK,GAAG,WAAW,CAAC,mBAAmB,CAAC,CAAC;AAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,uBAAuB,CAAC,CAAC;AAEtD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AACtC,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;AACpD,KAAK,CAAC,8BAA8B,EAAE,OAAO,CAAC,CAAC;AAC/C,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;AAiBjC,uCAAuC;AACvC,SAAS,iBAAiB,CAAC,YAAoB,EAAE,MAA0B;IACzE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,KAAK,CAAC,sDAAsD,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChG,CAAC;AACH,CAAC;AAED,kCAAkC;AAClC,SAAS,mBAAmB;IAC1B,OAAO;QACL,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS;QAC/F,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,SAAS;QACxG,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS;QAC/F,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;KACvF,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,QAAQ,CAAC,iDAAiD,CAAC,CAAC;IAE5D,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;IAErC,6DAA6D;IAC7D,QAAQ,CAAC,uBAAuB,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IACzD,QAAQ,CAAC,0BAA0B,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;IAC/D,QAAQ,CAAC,uBAAuB,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IACzD,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;IAEnD,8BAA8B;IAC9B,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnD,iBAAiB,CAAC,WAAW,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACzD,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnD,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAE7C,MAAM,mBAAmB,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC;IAElH,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC3C,MAAM,IAAI,KAAK,CACb,kJAAkJ,CACnJ,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sDAAsD,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAE1F,OAAO,MAAM,CAAC;AAChB,CAAC"}
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AACpC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACxC,OAAO,WAAW,MAAM,OAAO,CAAC;AAGhC,MAAM,KAAK,GAAG,WAAW,CAAC,mBAAmB,CAAC,CAAC;AAC/C,MAAM,QAAQ,GAAG,WAAW,CAAC,uBAAuB,CAAC,CAAC;AAEtD,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AACtC,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;AACpD,KAAK,CAAC,8BAA8B,EAAE,OAAO,CAAC,CAAC;AAC/C,MAAM,CAAC,MAAM,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;AAkBjC,uCAAuC;AACvC,SAAS,iBAAiB,CAAC,YAAoB,EAAE,MAA0B;IACzE,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QACzB,KAAK,CAAC,sDAAsD,EAAE,YAAY,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAChG,CAAC;AACH,CAAC;AAED,yCAAyC;AACzC,SAAS,yBAAyB;IAChC,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC;IAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC;IACrD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC;IACnD,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,CAAC,6BAA6B,CAAC;IAE/D,IACE,IAAI,IAAI,IAAI;QACZ,IAAI,CAAC,MAAM,GAAG,CAAC;QACf,OAAO,IAAI,IAAI;QACf,OAAO,CAAC,MAAM,GAAG,CAAC;QAClB,MAAM,IAAI,IAAI;QACd,MAAM,CAAC,MAAM,GAAG,CAAC;QACjB,YAAY,IAAI,IAAI;QACpB,YAAY,CAAC,MAAM,GAAG,CAAC,EACvB,CAAC;QACD,OAAO;YACL,IAAI;YACJ,OAAO;YACP,MAAM;YACN,YAAY;SACb,CAAC;IACJ,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,kCAAkC;AAClC,SAAS,mBAAmB;IAC1B,OAAO;QACL,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS;QAC/F,SAAS,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,iBAAiB,EAAE,CAAC,CAAC,CAAC,SAAS;QACxG,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS;QAC/F,GAAG,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;QACtF,MAAM,EAAE,yBAAyB,EAAE;KACpC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,QAAQ,CAAC,iDAAiD,CAAC,CAAC;IAE5D,MAAM,MAAM,GAAG,mBAAmB,EAAE,CAAC;IAErC,6DAA6D;IAC7D,QAAQ,CAAC,uBAAuB,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IACzD,QAAQ,CAAC,0BAA0B,EAAE,MAAM,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC;IAC/D,QAAQ,CAAC,uBAAuB,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IACzD,QAAQ,CAAC,oBAAoB,EAAE,MAAM,CAAC,GAAG,IAAI,IAAI,CAAC,CAAC;IACnD,QAAQ,CAAC,uBAAuB,EAAE,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,CAAC;IAEzD,8BAA8B;IAC9B,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnD,iBAAiB,CAAC,WAAW,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IACzD,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACnD,iBAAiB,CAAC,KAAK,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAE7C,IAAI,MAAM,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;QAC1B,KAAK,CACH,oDAAoD,EACpD,MAAM,CAAC,MAAM,CAAC,IAAI,EAClB,MAAM,CAAC,MAAM,CAAC,OAAO,EACrB,MAAM,CAAC,MAAM,CAAC,YAAY,CAC3B,CAAC;QACF,iBAAiB,CAAC,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC;IAED,MAAM,mBAAmB,GAAG,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,CAC5G,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,IAAI,CACjB,CAAC;IAEF,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACrC,KAAK,CAAC,mCAAmC,CAAC,CAAC;QAC3C,MAAM,IAAI,KAAK,CACb,+RAA+R,CAChS,CAAC;IACJ,CAAC;IAED,KAAK,CAAC,sDAAsD,EAAE,mBAAmB,CAAC,MAAM,CAAC,CAAC;IAE1F,OAAO,MAAM,CAAC;AAChB,CAAC"}

package/dist/config.specs.d.ts

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

package/dist/config.specs.d.ts.map

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

package/dist/config.specs.js

@@ -0,0 +1,123 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+import { configureAIProvider } from './config.js';
+// 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 ');
+        });
+    });
+});
+//# sourceMappingURL=config.specs.js.map