centaurus-cli 2.0.0

package/dist/providers/GoogleProvider.js

@@ -0,0 +1,313 @@
+/**
+ * Google AI Provider (Gemini Models)
+ * Supports Gemini 2.5 Flash and Pro models with function calling
+ */
+import { GoogleGenerativeAI, FunctionCallingMode } from '@google/generative-ai';
+import { AIProvider } from './Provider.js';
+const GEMINI_MODELS = [
+    {
+        id: 'gemini-2.0-flash-exp',
+        name: 'Gemini 2.0 Flash (Experimental)',
+        description: 'Fastest model, best for real-time tasks and code generation',
+        contextWindow: 1048576,
+        supportsFunctions: true,
+        supportsStreaming: true,
+        costPer1kInput: 0.00015,
+        costPer1kOutput: 0.0006
+    },
+    {
+        id: 'gemini-1.5-flash',
+        name: 'Gemini 1.5 Flash',
+        description: 'Fast and efficient for most tasks',
+        contextWindow: 1048576,
+        supportsFunctions: true,
+        supportsStreaming: true,
+        costPer1kInput: 0.00035,
+        costPer1kOutput: 0.0014
+    },
+    {
+        id: 'gemini-1.5-pro',
+        name: 'Gemini 1.5 Pro',
+        description: 'Most capable model for complex tasks',
+        contextWindow: 2097152,
+        supportsFunctions: true,
+        supportsStreaming: true,
+        costPer1kInput: 0.00125,
+        costPer1kOutput: 0.005
+    }
+];
+export class GoogleProvider extends AIProvider {
+    client = null;
+    model = null;
+    retryCount = 0;
+    maxRetries = 3;
+    constructor(config) {
+        super(config);
+        this.maxRetries = config.maxRetries || 3;
+    }
+    async initialize() {
+        if (!this.config.apiKey) {
+            throw new Error('Google API key is required');
+        }
+        this.client = new GoogleGenerativeAI(this.config.apiKey);
+        this.model = this.client.getGenerativeModel({
+            model: this.currentModel,
+        });
+    }
+    listModels() {
+        return GEMINI_MODELS;
+    }
+    getDefaultModel() {
+        return 'gemini-2.0-flash-exp';
+    }
+    async validateConnection() {
+        try {
+            if (!this.model) {
+                await this.initialize();
+            }
+            // Test with a simple prompt
+            const result = await this.model.generateContent('Say "OK" if you are working.');
+            const response = result.response;
+            return response.text().toLowerCase().includes('ok');
+        }
+        catch (error) {
+            console.error('Connection validation failed:', error);
+            return false;
+        }
+    }
+    convertToGeminiContent(messages) {
+        return messages.map(msg => {
+            if (msg.role === 'tool') {
+                return {
+                    role: 'function',
+                    parts: [{
+                            functionResponse: {
+                                name: msg.toolCallId || 'tool',
+                                response: { result: msg.content }
+                            }
+                        }]
+                };
+            }
+            return {
+                role: msg.role === 'user' ? 'user' : 'model',
+                parts: [{ text: msg.content }]
+            };
+        });
+    }
+    convertToolsToGemini(tools) {
+        if (!tools || tools.length === 0)
+            return undefined;
+        return [{
+                functionDeclarations: tools.map(tool => ({
+                    name: tool.function.name,
+                    description: tool.function.description,
+                    parameters: tool.function.parameters || {
+                        type: 'object',
+                        properties: {}
+                    }
+                }))
+            }];
+    }
+    extractFunctionCalls(response) {
+        const functionCalls = [];
+        for (const candidate of response.candidates || []) {
+            for (const part of candidate.content.parts) {
+                if ('functionCall' in part && part.functionCall) {
+                    functionCalls.push({
+                        name: part.functionCall.name,
+                        arguments: part.functionCall.args
+                    });
+                }
+            }
+        }
+        return functionCalls.length > 0 ? functionCalls : undefined;
+    }
+    async generate(messages, options = {}) {
+        if (!this.model) {
+            await this.initialize();
+        }
+        // Apply system prompt if provided
+        let finalMessages = messages;
+        if (options.systemPrompt) {
+            finalMessages = [
+                { role: 'system', content: options.systemPrompt },
+                ...messages
+            ];
+        }
+        const geminiContent = this.convertToGeminiContent(finalMessages);
+        const geminiTools = this.convertToolsToGemini(options.tools);
+        // Configure generation parameters
+        const generationConfig = {
+            temperature: options.temperature || 0.7,
+            topK: options.topK || 40,
+            topP: options.topP || 0.95,
+            maxOutputTokens: options.maxTokens || 8192,
+            stopSequences: options.stopSequences
+        };
+        // Configure the model with tools if provided
+        let model = this.model;
+        if (geminiTools) {
+            model = this.client.getGenerativeModel({
+                model: this.currentModel,
+                tools: geminiTools,
+                toolConfig: {
+                    functionCallingConfig: {
+                        mode: FunctionCallingMode.AUTO
+                    }
+                }
+            });
+        }
+        try {
+            const result = await this.retryWithExponentialBackoff(async () => {
+                return await model.generateContent({
+                    contents: geminiContent,
+                    generationConfig
+                });
+            });
+            const response = result.response;
+            const text = response.text();
+            const functionCalls = this.extractFunctionCalls(response);
+            return {
+                text,
+                functionCalls,
+                usage: response.usageMetadata ? {
+                    promptTokens: response.usageMetadata.promptTokenCount || 0,
+                    completionTokens: response.usageMetadata.candidatesTokenCount || 0,
+                    totalTokens: response.usageMetadata.totalTokenCount || 0
+                } : undefined,
+                finishReason: response.candidates?.[0]?.finishReason
+            };
+        }
+        catch (error) {
+            throw this.handleError(error);
+        }
+    }
+    async streamGenerate(messages, onChunk, options = {}) {
+        if (!this.model) {
+            await this.initialize();
+        }
+        // Apply system prompt if provided
+        let finalMessages = messages;
+        if (options.systemPrompt) {
+            finalMessages = [
+                { role: 'system', content: options.systemPrompt },
+                ...messages
+            ];
+        }
+        const geminiContent = this.convertToGeminiContent(finalMessages);
+        const geminiTools = this.convertToolsToGemini(options.tools);
+        // Configure generation parameters
+        const generationConfig = {
+            temperature: options.temperature || 0.7,
+            topK: options.topK || 40,
+            topP: options.topP || 0.95,
+            maxOutputTokens: options.maxTokens || 8192,
+            stopSequences: options.stopSequences
+        };
+        // Configure the model with tools if provided
+        let model = this.model;
+        if (geminiTools) {
+            model = this.client.getGenerativeModel({
+                model: this.currentModel,
+                tools: geminiTools,
+                toolConfig: {
+                    functionCallingConfig: {
+                        mode: FunctionCallingMode.AUTO
+                    }
+                }
+            });
+        }
+        try {
+            const result = await this.retryWithExponentialBackoff(async () => {
+                return await model.generateContentStream({
+                    contents: geminiContent,
+                    generationConfig
+                });
+            });
+            let fullText = '';
+            let functionCalls;
+            let usage;
+            for await (const chunk of result.stream) {
+                const chunkText = chunk.text();
+                fullText += chunkText;
+                // Check for function calls in this chunk
+                const chunkFunctionCalls = this.extractFunctionCalls(chunk);
+                if (chunkFunctionCalls) {
+                    functionCalls = chunkFunctionCalls;
+                }
+                onChunk({
+                    text: chunkText,
+                    isComplete: false,
+                    functionCall: chunkFunctionCalls?.[0]
+                });
+            }
+            // Get final response for metadata
+            const response = await result.response;
+            // Send final chunk
+            onChunk({
+                text: '',
+                isComplete: true
+            });
+            return {
+                text: fullText,
+                functionCalls,
+                usage: response.usageMetadata ? {
+                    promptTokens: response.usageMetadata.promptTokenCount || 0,
+                    completionTokens: response.usageMetadata.candidatesTokenCount || 0,
+                    totalTokens: response.usageMetadata.totalTokenCount || 0
+                } : undefined,
+                finishReason: response.candidates?.[0]?.finishReason
+            };
+        }
+        catch (error) {
+            throw this.handleError(error);
+        }
+    }
+    async retryWithExponentialBackoff(fn, retries = this.maxRetries) {
+        try {
+            this.retryCount = 0;
+            return await fn();
+        }
+        catch (error) {
+            if (this.retryCount >= retries) {
+                throw error;
+            }
+            // Check if error is retryable
+            if (this.isRetryableError(error)) {
+                this.retryCount++;
+                const delay = Math.min(1000 * Math.pow(2, this.retryCount), 10000);
+                await new Promise(resolve => setTimeout(resolve, delay));
+                return this.retryWithExponentialBackoff(fn, retries);
+            }
+            throw error;
+        }
+    }
+    isRetryableError(error) {
+        // Retry on rate limit or temporary errors
+        if (error.status === 429 || error.status === 503) {
+            return true;
+        }
+        // Retry on network errors
+        if (error.code === 'ECONNRESET' || error.code === 'ETIMEDOUT') {
+            return true;
+        }
+        return false;
+    }
+    handleError(error) {
+        if (error.status === 401) {
+            return new Error('Invalid API key. Please check your Google AI Studio API key.');
+        }
+        if (error.status === 429) {
+            return new Error('Rate limit exceeded. Please try again later.');
+        }
+        if (error.status === 400) {
+            return new Error(`Bad request: ${error.message || 'Invalid parameters'}`);
+        }
+        if (error.message) {
+            return new Error(`Google AI Error: ${error.message}`);
+        }
+        return new Error('An unexpected error occurred with the Google AI provider');
+    }
+}
+export default GoogleProvider;
+//# sourceMappingURL=GoogleProvider.js.map

package/dist/providers/GoogleProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"GoogleProvider.js","sourceRoot":"","sources":["../../src/providers/GoogleProvider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EACL,kBAAkB,EAGlB,mBAAmB,EAIpB,MAAM,uBAAuB,CAAC;AAE/B,OAAO,EACL,UAAU,EASX,MAAM,eAAe,CAAC;AAEvB,MAAM,aAAa,GAAgB;IACjC;QACE,EAAE,EAAE,sBAAsB;QAC1B,IAAI,EAAE,iCAAiC;QACvC,WAAW,EAAE,6DAA6D;QAC1E,aAAa,EAAE,OAAO;QACtB,iBAAiB,EAAE,IAAI;QACvB,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,OAAO;QACvB,eAAe,EAAE,MAAM;KACxB;IACD;QACE,EAAE,EAAE,kBAAkB;QACtB,IAAI,EAAE,kBAAkB;QACxB,WAAW,EAAE,mCAAmC;QAChD,aAAa,EAAE,OAAO;QACtB,iBAAiB,EAAE,IAAI;QACvB,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,OAAO;QACvB,eAAe,EAAE,MAAM;KACxB;IACD;QACE,EAAE,EAAE,gBAAgB;QACpB,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,sCAAsC;QACnD,aAAa,EAAE,OAAO;QACtB,iBAAiB,EAAE,IAAI;QACvB,iBAAiB,EAAE,IAAI;QACvB,cAAc,EAAE,OAAO;QACvB,eAAe,EAAE,KAAK;KACvB;CACF,CAAC;AAEF,MAAM,OAAO,cAAe,SAAQ,UAAU;IACpC,MAAM,GAA8B,IAAI,CAAC;IACzC,KAAK,GAA2B,IAAI,CAAC;IACrC,UAAU,GAAG,CAAC,CAAC;IACf,UAAU,GAAG,CAAC,CAAC;IAEvB,YAAY,MAAsB;QAChC,KAAK,CAAC,MAAM,CAAC,CAAC;QACd,IAAI,CAAC,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,UAAU;QACd,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QAChD,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;YAC1C,KAAK,EAAE,IAAI,CAAC,YAAY;SACzB,CAAC,CAAC;IACL,CAAC;IAED,UAAU;QACR,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,eAAe;QACb,OAAO,sBAAsB,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,kBAAkB;QACtB,IAAI,CAAC;YACH,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBAChB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;YAC1B,CAAC;YAED,4BAA4B;YAC5B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAM,CAAC,eAAe,CAAC,8BAA8B,CAAC,CAAC;YACjF,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YACjC,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;QACtD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAEO,sBAAsB,CAAC,QAAmB;QAChD,OAAO,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;YACxB,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACxB,OAAO;oBACL,IAAI,EAAE,UAAU;oBAChB,KAAK,EAAE,CAAC;4BACN,gBAAgB,EAAE;gCAChB,IAAI,EAAE,GAAG,CAAC,UAAU,IAAI,MAAM;gCAC9B,QAAQ,EAAE,EAAE,MAAM,EAAE,GAAG,CAAC,OAAO,EAAE;6BAClC;yBACF,CAAC;iBACQ,CAAC;YACf,CAAC;YAED,OAAO;gBACL,IAAI,EAAE,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO;gBAC5C,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;aACpB,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,oBAAoB,CAAC,KAAc;QACzC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAEnD,OAAO,CAAC;gBACN,oBAAoB,EAAE,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;oBACvC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI;oBACxB,WAAW,EAAE,IAAI,CAAC,QAAQ,CAAC,WAAW;oBACtC,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAiB,IAAI;wBAC7C,IAAI,EAAE,QAAiB;wBACvB,UAAU,EAAE,EAAE;qBACf;iBACF,CAAC,CAAC;aACJ,CAAC,CAAC;IACL,CAAC;IAEO,oBAAoB,CAAC,QAAyC;QACpE,MAAM,aAAa,GAAmB,EAAE,CAAC;QAEzC,KAAK,MAAM,SAAS,IAAI,QAAQ,CAAC,UAAU,IAAI,EAAE,EAAE,CAAC;YAClD,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;gBAC3C,IAAI,cAAc,IAAI,IAAI,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;oBAChD,aAAa,CAAC,IAAI,CAAC;wBACjB,IAAI,EAAE,IAAI,CAAC,YAAY,CAAC,IAAI;wBAC5B,SAAS,EAAE,IAAI,CAAC,YAAY,CAAC,IAA2B;qBACzD,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC;IAC9D,CAAC;IAED,KAAK,CAAC,QAAQ,CACZ,QAAmB,EACnB,UAA2B,EAAE;QAE7B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,kCAAkC;QAClC,IAAI,aAAa,GAAG,QAAQ,CAAC;QAC7B,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACzB,aAAa,GAAG;gBACd,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE;gBACjD,GAAG,QAAQ;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;QACjE,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAE7D,kCAAkC;QAClC,MAAM,gBAAgB,GAAG;YACvB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;YACvC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;YAC1B,eAAe,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YAC1C,aAAa,EAAE,OAAO,CAAC,aAAa;SACrC,CAAC;QAEF,6CAA6C;QAC7C,IAAI,KAAK,GAAG,IAAI,CAAC,KAAM,CAAC;QACxB,IAAI,WAAW,EAAE,CAAC;YAChB,KAAK,GAAG,IAAI,CAAC,MAAO,CAAC,kBAAkB,CAAC;gBACtC,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,KAAK,EAAE,WAAW;gBAClB,UAAU,EAAE;oBACV,qBAAqB,EAAE;wBACrB,IAAI,EAAE,mBAAmB,CAAC,IAAI;qBAC/B;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,KAAK,IAAI,EAAE;gBAC/D,OAAO,MAAM,KAAK,CAAC,eAAe,CAAC;oBACjC,QAAQ,EAAE,aAAa;oBACvB,gBAAgB;iBACjB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;YACjC,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;YAC7B,MAAM,aAAa,GAAG,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YAE1D,OAAO;gBACL,IAAI;gBACJ,aAAa;gBACb,KAAK,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;oBAC9B,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;oBAC1D,gBAAgB,EAAE,QAAQ,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;oBAClE,WAAW,EAAE,QAAQ,CAAC,aAAa,CAAC,eAAe,IAAI,CAAC;iBACzD,CAAC,CAAC,CAAC,SAAS;gBACb,YAAY,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY;aACrD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,cAAc,CAClB,QAAmB,EACnB,OAAqC,EACrC,UAA2B,EAAE;QAE7B,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YAChB,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1B,CAAC;QAED,kCAAkC;QAClC,IAAI,aAAa,GAAG,QAAQ,CAAC;QAC7B,IAAI,OAAO,CAAC,YAAY,EAAE,CAAC;YACzB,aAAa,GAAG;gBACd,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,YAAY,EAAE;gBACjD,GAAG,QAAQ;aACZ,CAAC;QACJ,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;QACjE,MAAM,WAAW,GAAG,IAAI,CAAC,oBAAoB,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;QAE7D,kCAAkC;QAClC,MAAM,gBAAgB,GAAG;YACvB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,GAAG;YACvC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,EAAE;YACxB,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI;YAC1B,eAAe,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI;YAC1C,aAAa,EAAE,OAAO,CAAC,aAAa;SACrC,CAAC;QAEF,6CAA6C;QAC7C,IAAI,KAAK,GAAG,IAAI,CAAC,KAAM,CAAC;QACxB,IAAI,WAAW,EAAE,CAAC;YAChB,KAAK,GAAG,IAAI,CAAC,MAAO,CAAC,kBAAkB,CAAC;gBACtC,KAAK,EAAE,IAAI,CAAC,YAAY;gBACxB,KAAK,EAAE,WAAW;gBAClB,UAAU,EAAE;oBACV,qBAAqB,EAAE;wBACrB,IAAI,EAAE,mBAAmB,CAAC,IAAI;qBAC/B;iBACF;aACF,CAAC,CAAC;QACL,CAAC;QAED,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,2BAA2B,CAAC,KAAK,IAAI,EAAE;gBAC/D,OAAO,MAAM,KAAK,CAAC,qBAAqB,CAAC;oBACvC,QAAQ,EAAE,aAAa;oBACvB,gBAAgB;iBACjB,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,IAAI,QAAQ,GAAG,EAAE,CAAC;YAClB,IAAI,aAAyC,CAAC;YAC9C,IAAI,KAAgC,CAAC;YAErC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,EAAE,CAAC;gBACxC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;gBAC/B,QAAQ,IAAI,SAAS,CAAC;gBAEtB,yCAAyC;gBACzC,MAAM,kBAAkB,GAAG,IAAI,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;gBAC5D,IAAI,kBAAkB,EAAE,CAAC;oBACvB,aAAa,GAAG,kBAAkB,CAAC;gBACrC,CAAC;gBAED,OAAO,CAAC;oBACN,IAAI,EAAE,SAAS;oBACf,UAAU,EAAE,KAAK;oBACjB,YAAY,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;iBACtC,CAAC,CAAC;YACL,CAAC;YAED,kCAAkC;YAClC,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC;YAEvC,mBAAmB;YACnB,OAAO,CAAC;gBACN,IAAI,EAAE,EAAE;gBACR,UAAU,EAAE,IAAI;aACjB,CAAC,CAAC;YAEH,OAAO;gBACL,IAAI,EAAE,QAAQ;gBACd,aAAa;gBACb,KAAK,EAAE,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;oBAC9B,YAAY,EAAE,QAAQ,CAAC,aAAa,CAAC,gBAAgB,IAAI,CAAC;oBAC1D,gBAAgB,EAAE,QAAQ,CAAC,aAAa,CAAC,oBAAoB,IAAI,CAAC;oBAClE,WAAW,EAAE,QAAQ,CAAC,aAAa,CAAC,eAAe,IAAI,CAAC;iBACzD,CAAC,CAAC,CAAC,SAAS;gBACb,YAAY,EAAE,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,YAAY;aACrD,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,2BAA2B,CACvC,EAAoB,EACpB,OAAO,GAAG,IAAI,CAAC,UAAU;QAEzB,IAAI,CAAC;YACH,IAAI,CAAC,UAAU,GAAG,CAAC,CAAC;YACpB,OAAO,MAAM,EAAE,EAAE,CAAC;QACpB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,EAAE,CAAC;gBAC/B,MAAM,KAAK,CAAC;YACd,CAAC;YAED,8BAA8B;YAC9B,IAAI,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,IAAI,CAAC,UAAU,EAAE,CAAC;gBAClB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,EAAE,KAAK,CAAC,CAAC;gBACnE,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBACzD,OAAO,IAAI,CAAC,2BAA2B,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;YACvD,CAAC;YAED,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAEO,gBAAgB,CAAC,KAAU;QACjC,0CAA0C;QAC1C,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACjD,OAAO,IAAI,CAAC;QACd,CAAC;QAED,0BAA0B;QAC1B,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC9D,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,WAAW,CAAC,KAAU;QAC5B,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACzB,OAAO,IAAI,KAAK,CAAC,8DAA8D,CAAC,CAAC;QACnF,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACzB,OAAO,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,KAAK,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACzB,OAAO,IAAI,KAAK,CAAC,gBAAgB,KAAK,CAAC,OAAO,IAAI,oBAAoB,EAAE,CAAC,CAAC;QAC5E,CAAC;QAED,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YAClB,OAAO,IAAI,KAAK,CAAC,oBAAoB,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;QACxD,CAAC;QAED,OAAO,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC/E,CAAC;CACF;AAED,eAAe,cAAc,CAAC"}

package/dist/providers/Provider.d.ts

@@ -0,0 +1,114 @@
+/**
+ * Base Provider Interface
+ * Defines the contract for all AI providers (Google, OpenAI, Anthropic, etc.)
+ */
+export interface ModelInfo {
+    id: string;
+    name: string;
+    description: string;
+    contextWindow: number;
+    supportsFunctions: boolean;
+    supportsStreaming: boolean;
+    costPer1kInput?: number;
+    costPer1kOutput?: number;
+}
+export interface ProviderConfig {
+    apiKey: string;
+    baseUrl?: string;
+    maxRetries?: number;
+    timeout?: number;
+    headers?: Record<string, string>;
+}
+export interface StreamChunk {
+    text: string;
+    isComplete: boolean;
+    functionCall?: FunctionCall;
+}
+export interface FunctionCall {
+    name: string;
+    arguments: Record<string, any>;
+}
+export interface Tool {
+    type: 'function';
+    function: {
+        name: string;
+        description: string;
+        parameters?: {
+            type: string;
+            properties: Record<string, any>;
+            required?: string[];
+        };
+    };
+}
+export interface Message {
+    role: 'user' | 'assistant' | 'system' | 'tool';
+    content: string;
+    toolCallId?: string;
+    toolCalls?: FunctionCall[];
+}
+export interface GenerateOptions {
+    model?: string;
+    temperature?: number;
+    maxTokens?: number;
+    topP?: number;
+    topK?: number;
+    stopSequences?: string[];
+    systemPrompt?: string;
+    tools?: Tool[];
+    stream?: boolean;
+}
+export interface GenerateResponse {
+    text: string;
+    functionCalls?: FunctionCall[];
+    usage?: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+    finishReason?: string;
+}
+/**
+ * Abstract base class for AI providers
+ */
+export declare abstract class AIProvider {
+    protected config: ProviderConfig;
+    protected currentModel: string;
+    constructor(config: ProviderConfig);
+    /**
+     * Initialize the provider connection
+     */
+    abstract initialize(): Promise<void>;
+    /**
+     * List available models for this provider
+     */
+    abstract listModels(): ModelInfo[];
+    /**
+     * Get the default model for this provider
+     */
+    abstract getDefaultModel(): string;
+    /**
+     * Generate a response from the AI
+     */
+    abstract generate(messages: Message[], options?: GenerateOptions): Promise<GenerateResponse>;
+    /**
+     * Stream a response from the AI
+     */
+    abstract streamGenerate(messages: Message[], onChunk: (chunk: StreamChunk) => void, options?: GenerateOptions): Promise<GenerateResponse>;
+    /**
+     * Validate the API key and connection
+     */
+    abstract validateConnection(): Promise<boolean>;
+    /**
+     * Get the current model
+     */
+    getCurrentModel(): string;
+    /**
+     * Set the current model
+     */
+    setCurrentModel(model: string): void;
+    /**
+     * Calculate cost for token usage
+     */
+    calculateCost(inputTokens: number, outputTokens: number, model?: string): number;
+}
+//# sourceMappingURL=Provider.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"Provider.d.ts","sourceRoot":"","sources":["../../src/providers/Provider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,OAAO,CAAC;IAC3B,iBAAiB,EAAE,OAAO,CAAC;IAC3B,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,eAAe,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,cAAc;IAC7B,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAClC;AAED,MAAM,WAAW,WAAW;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,UAAU,EAAE,OAAO,CAAC;IACpB,YAAY,CAAC,EAAE,YAAY,CAAC;CAC7B;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,IAAI;IACnB,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE;QACR,IAAI,EAAE,MAAM,CAAC;QACb,WAAW,EAAE,MAAM,CAAC;QACpB,UAAU,CAAC,EAAE;YACX,IAAI,EAAE,MAAM,CAAC;YACb,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAChC,QAAQ,CAAC,EAAE,MAAM,EAAE,CAAC;SACrB,CAAC;KACH,CAAC;CACH;AAED,MAAM,WAAW,OAAO;IACtB,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,SAAS,CAAC,EAAE,YAAY,EAAE,CAAC;CAC5B;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,CAAC,EAAE,IAAI,EAAE,CAAC;IACf,MAAM,CAAC,EAAE,OAAO,CAAC;CAClB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,aAAa,CAAC,EAAE,YAAY,EAAE,CAAC;IAC/B,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED;;GAEG;AACH,8BAAsB,UAAU;IAC9B,SAAS,CAAC,MAAM,EAAE,cAAc,CAAC;IACjC,SAAS,CAAC,YAAY,EAAE,MAAM,CAAC;gBAEnB,MAAM,EAAE,cAAc;IAKlC;;OAEG;IACH,QAAQ,CAAC,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAEpC;;OAEG;IACH,QAAQ,CAAC,UAAU,IAAI,SAAS,EAAE;IAElC;;OAEG;IACH,QAAQ,CAAC,eAAe,IAAI,MAAM;IAElC;;OAEG;IACH,QAAQ,CAAC,QAAQ,CACf,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,gBAAgB,CAAC;IAE5B;;OAEG;IACH,QAAQ,CAAC,cAAc,CACrB,QAAQ,EAAE,OAAO,EAAE,EACnB,OAAO,EAAE,CAAC,KAAK,EAAE,WAAW,KAAK,IAAI,EACrC,OAAO,CAAC,EAAE,eAAe,GACxB,OAAO,CAAC,gBAAgB,CAAC;IAE5B;;OAEG;IACH,QAAQ,CAAC,kBAAkB,IAAI,OAAO,CAAC,OAAO,CAAC;IAE/C;;OAEG;IACH,eAAe,IAAI,MAAM;IAIzB;;OAEG;IACH,eAAe,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAQpC;;OAEG;IACH,aAAa,CAAC,WAAW,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM;CAWjF"}

package/dist/providers/Provider.js

@@ -0,0 +1,44 @@
+/**
+ * Base Provider Interface
+ * Defines the contract for all AI providers (Google, OpenAI, Anthropic, etc.)
+ */
+/**
+ * Abstract base class for AI providers
+ */
+export class AIProvider {
+    config;
+    currentModel;
+    constructor(config) {
+        this.config = config;
+        this.currentModel = this.getDefaultModel();
+    }
+    /**
+     * Get the current model
+     */
+    getCurrentModel() {
+        return this.currentModel;
+    }
+    /**
+     * Set the current model
+     */
+    setCurrentModel(model) {
+        const availableModels = this.listModels();
+        if (!availableModels.find(m => m.id === model)) {
+            throw new Error(`Model ${model} not available for this provider`);
+        }
+        this.currentModel = model;
+    }
+    /**
+     * Calculate cost for token usage
+     */
+    calculateCost(inputTokens, outputTokens, model) {
+        const modelId = model || this.currentModel;
+        const modelInfo = this.listModels().find(m => m.id === modelId);
+        if (!modelInfo || !modelInfo.costPer1kInput || !modelInfo.costPer1kOutput) {
+            return 0;
+        }
+        return (inputTokens / 1000) * modelInfo.costPer1kInput +
+            (outputTokens / 1000) * modelInfo.costPer1kOutput;
+    }
+}
+//# sourceMappingURL=Provider.js.map

package/dist/providers/Provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Provider.js","sourceRoot":"","sources":["../../src/providers/Provider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AA2EH;;GAEG;AACH,MAAM,OAAgB,UAAU;IACpB,MAAM,CAAiB;IACvB,YAAY,CAAS;IAE/B,YAAY,MAAsB;QAChC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,eAAe,EAAE,CAAC;IAC7C,CAAC;IAuCD;;OAEG;IACH,eAAe;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,KAAa;QAC3B,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC1C,IAAI,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC;YAC/C,MAAM,IAAI,KAAK,CAAC,SAAS,KAAK,kCAAkC,CAAC,CAAC;QACpE,CAAC;QACD,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC;IAC5B,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,WAAmB,EAAE,YAAoB,EAAE,KAAc;QACrE,MAAM,OAAO,GAAG,KAAK,IAAI,IAAI,CAAC,YAAY,CAAC;QAC3C,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;QAEhE,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,CAAC,cAAc,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CAAC;YAC1E,OAAO,CAAC,CAAC;QACX,CAAC;QAED,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,GAAG,SAAS,CAAC,cAAc;YAC/C,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,SAAS,CAAC,eAAe,CAAC;IAC3D,CAAC;CACF"}