claude-flow 2.0.0-alpha.66 → 2.0.0-alpha.68

package/dist/providers/google-provider.js

@@ -0,0 +1,362 @@
+/**
+ * Google AI Provider Implementation
+ * Supports Gemini Pro, PaLM, and other Google models
+ */
+import { BaseProvider } from './base-provider.js';
+import { LLMProviderError, RateLimitError, AuthenticationError, } from './types.js';
+export class GoogleProvider extends BaseProvider {
+    name = 'google';
+    capabilities = {
+        supportedModels: [
+            'gemini-pro',
+            'gemini-pro-vision',
+            'palm-2',
+            'bison',
+        ],
+        maxContextLength: {
+            'gemini-pro': 32768,
+            'gemini-pro-vision': 16384,
+            'palm-2': 8192,
+            'bison': 4096,
+        },
+        maxOutputTokens: {
+            'gemini-pro': 2048,
+            'gemini-pro-vision': 2048,
+            'palm-2': 1024,
+            'bison': 1024,
+        },
+        supportsStreaming: true,
+        supportsFunctionCalling: true,
+        supportsSystemMessages: false, // Google AI doesn't have explicit system messages
+        supportsVision: true, // Gemini Pro Vision
+        supportsAudio: false,
+        supportsTools: true,
+        supportsFineTuning: false,
+        supportsEmbeddings: true,
+        supportsLogprobs: false,
+        supportsBatching: true,
+        rateLimit: {
+            requestsPerMinute: 60,
+            tokensPerMinute: 60000,
+            concurrentRequests: 10,
+        },
+        pricing: {
+            'gemini-pro': {
+                promptCostPer1k: 0.00025,
+                completionCostPer1k: 0.0005,
+                currency: 'USD',
+            },
+            'gemini-pro-vision': {
+                promptCostPer1k: 0.00025,
+                completionCostPer1k: 0.0005,
+                currency: 'USD',
+            },
+            'palm-2': {
+                promptCostPer1k: 0.0005,
+                completionCostPer1k: 0.001,
+                currency: 'USD',
+            },
+            'bison': {
+                promptCostPer1k: 0.0005,
+                completionCostPer1k: 0.001,
+                currency: 'USD',
+            },
+        },
+    };
+    baseUrl;
+    async doInitialize() {
+        if (!this.config.apiKey) {
+            throw new AuthenticationError('Google AI API key is required', 'google');
+        }
+        // Use Gemini API for newer models, PaLM API for older ones
+        const model = this.config.model;
+        if (model.startsWith('gemini')) {
+            this.baseUrl = 'https://generativelanguage.googleapis.com/v1beta';
+        }
+        else {
+            this.baseUrl = 'https://generativelanguage.googleapis.com/v1beta2';
+        }
+    }
+    async doComplete(request) {
+        const googleRequest = this.buildGoogleRequest(request);
+        const model = this.mapToGoogleModel(request.model || this.config.model);
+        const url = `${this.baseUrl}/models/${model}:generateContent?key=${this.config.apiKey}`;
+        const controller = new AbortController();
+        const timeout = setTimeout(() => controller.abort(), this.config.timeout || 60000);
+        try {
+            const response = await fetch(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(googleRequest),
+                signal: controller.signal,
+            });
+            clearTimeout(timeout);
+            if (!response.ok) {
+                await this.handleErrorResponse(response);
+            }
+            const data = await response.json();
+            if (!data.candidates || data.candidates.length === 0) {
+                throw new LLMProviderError('No response generated', 'NO_RESPONSE', 'google', undefined, false);
+            }
+            const candidate = data.candidates[0];
+            const content = candidate.content.parts.map(part => part.text).join('');
+            // Calculate cost
+            const usageData = data.usageMetadata || {
+                promptTokenCount: this.estimateTokens(JSON.stringify(request.messages)),
+                candidatesTokenCount: this.estimateTokens(content),
+                totalTokenCount: 0,
+            };
+            usageData.totalTokenCount = usageData.promptTokenCount + usageData.candidatesTokenCount;
+            const pricing = this.capabilities.pricing[request.model || this.config.model];
+            const promptCost = (usageData.promptTokenCount / 1000) * pricing.promptCostPer1k;
+            const completionCost = (usageData.candidatesTokenCount / 1000) * pricing.completionCostPer1k;
+            return {
+                id: `google-${Date.now()}`,
+                model: request.model || this.config.model,
+                provider: 'google',
+                content,
+                usage: {
+                    promptTokens: usageData.promptTokenCount,
+                    completionTokens: usageData.candidatesTokenCount,
+                    totalTokens: usageData.totalTokenCount,
+                },
+                cost: {
+                    promptCost,
+                    completionCost,
+                    totalCost: promptCost + completionCost,
+                    currency: 'USD',
+                },
+                finishReason: this.mapFinishReason(candidate.finishReason),
+            };
+        }
+        catch (error) {
+            clearTimeout(timeout);
+            throw this.transformError(error);
+        }
+    }
+    async *doStreamComplete(request) {
+        const googleRequest = this.buildGoogleRequest(request);
+        const model = this.mapToGoogleModel(request.model || this.config.model);
+        const url = `${this.baseUrl}/models/${model}:streamGenerateContent?key=${this.config.apiKey}`;
+        const controller = new AbortController();
+        const timeout = setTimeout(() => controller.abort(), (this.config.timeout || 60000) * 2);
+        try {
+            const response = await fetch(url, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify(googleRequest),
+                signal: controller.signal,
+            });
+            if (!response.ok) {
+                await this.handleErrorResponse(response);
+            }
+            const reader = response.body.getReader();
+            const decoder = new TextDecoder();
+            let buffer = '';
+            let totalContent = '';
+            let promptTokens = 0;
+            let completionTokens = 0;
+            while (true) {
+                const { done, value } = await reader.read();
+                if (done)
+                    break;
+                buffer += decoder.decode(value, { stream: true });
+                const lines = buffer.split('\n');
+                buffer = lines.pop() || '';
+                for (const line of lines) {
+                    if (line.trim() === '')
+                        continue;
+                    try {
+                        const data = JSON.parse(line);
+                        if (data.candidates && data.candidates.length > 0) {
+                            const candidate = data.candidates[0];
+                            const content = candidate.content.parts.map(part => part.text).join('');
+                            if (content) {
+                                totalContent += content;
+                                yield {
+                                    type: 'content',
+                                    delta: { content },
+                                };
+                            }
+                            if (data.usageMetadata) {
+                                promptTokens = data.usageMetadata.promptTokenCount;
+                                completionTokens = data.usageMetadata.candidatesTokenCount;
+                            }
+                        }
+                    }
+                    catch (e) {
+                        this.logger.warn('Failed to parse Google AI stream chunk', { line, error: e });
+                    }
+                }
+            }
+            // Final event with usage and cost
+            const pricing = this.capabilities.pricing[request.model || this.config.model];
+            const promptCost = (promptTokens / 1000) * pricing.promptCostPer1k;
+            const completionCost = (completionTokens / 1000) * pricing.completionCostPer1k;
+            yield {
+                type: 'done',
+                usage: {
+                    promptTokens,
+                    completionTokens,
+                    totalTokens: promptTokens + completionTokens,
+                },
+                cost: {
+                    promptCost,
+                    completionCost,
+                    totalCost: promptCost + completionCost,
+                    currency: 'USD',
+                },
+            };
+        }
+        catch (error) {
+            clearTimeout(timeout);
+            throw this.transformError(error);
+        }
+        finally {
+            clearTimeout(timeout);
+        }
+    }
+    async listModels() {
+        return this.capabilities.supportedModels;
+    }
+    async getModelInfo(model) {
+        return {
+            model,
+            name: model,
+            description: this.getModelDescription(model),
+            contextLength: this.capabilities.maxContextLength[model] || 4096,
+            maxOutputTokens: this.capabilities.maxOutputTokens[model] || 2048,
+            supportedFeatures: [
+                'chat',
+                'completion',
+                ...(model.includes('vision') ? ['vision'] : []),
+                ...(model.startsWith('gemini') ? ['function_calling'] : []),
+            ],
+            pricing: this.capabilities.pricing[model],
+        };
+    }
+    async doHealthCheck() {
+        try {
+            const url = `${this.baseUrl}/models?key=${this.config.apiKey}`;
+            const response = await fetch(url);
+            if (!response.ok) {
+                throw new Error(`Health check failed: ${response.status}`);
+            }
+            return {
+                healthy: true,
+                timestamp: new Date(),
+            };
+        }
+        catch (error) {
+            return {
+                healthy: false,
+                error: error instanceof Error ? error.message : 'Unknown error',
+                timestamp: new Date(),
+            };
+        }
+    }
+    buildGoogleRequest(request) {
+        // Convert messages to Google format
+        const contents = [];
+        for (const message of request.messages) {
+            // Skip system messages or prepend to first user message
+            if (message.role === 'system') {
+                if (contents.length === 0) {
+                    contents.push({
+                        role: 'user',
+                        parts: [{ text: `Instructions: ${message.content}` }],
+                    });
+                }
+                continue;
+            }
+            contents.push({
+                role: message.role === 'assistant' ? 'model' : 'user',
+                parts: [{ text: message.content }],
+            });
+        }
+        return {
+            contents,
+            generationConfig: {
+                temperature: request.temperature ?? this.config.temperature,
+                topK: request.topK ?? this.config.topK,
+                topP: request.topP ?? this.config.topP,
+                maxOutputTokens: request.maxTokens ?? this.config.maxTokens,
+                stopSequences: request.stopSequences ?? this.config.stopSequences,
+            },
+            safetySettings: [
+                {
+                    category: 'HARM_CATEGORY_HARASSMENT',
+                    threshold: 'BLOCK_NONE',
+                },
+                {
+                    category: 'HARM_CATEGORY_HATE_SPEECH',
+                    threshold: 'BLOCK_NONE',
+                },
+                {
+                    category: 'HARM_CATEGORY_SEXUALLY_EXPLICIT',
+                    threshold: 'BLOCK_NONE',
+                },
+                {
+                    category: 'HARM_CATEGORY_DANGEROUS_CONTENT',
+                    threshold: 'BLOCK_NONE',
+                },
+            ],
+        };
+    }
+    mapToGoogleModel(model) {
+        const modelMap = {
+            'gemini-pro': 'gemini-pro',
+            'gemini-pro-vision': 'gemini-pro-vision',
+            'palm-2': 'text-bison-001',
+            'bison': 'text-bison-001',
+        };
+        return modelMap[model] || model;
+    }
+    mapFinishReason(reason) {
+        switch (reason) {
+            case 'STOP':
+                return 'stop';
+            case 'MAX_TOKENS':
+                return 'length';
+            case 'SAFETY':
+            case 'RECITATION':
+                return 'content_filter';
+            default:
+                return 'stop';
+        }
+    }
+    getModelDescription(model) {
+        const descriptions = {
+            'gemini-pro': 'Google\'s most capable text model',
+            'gemini-pro-vision': 'Gemini Pro with vision capabilities',
+            'palm-2': 'Previous generation large language model',
+            'bison': 'Efficient model for various tasks',
+        };
+        return descriptions[model] || 'Google AI language model';
+    }
+    async handleErrorResponse(response) {
+        const errorText = await response.text();
+        let errorData;
+        try {
+            errorData = JSON.parse(errorText);
+        }
+        catch {
+            errorData = { error: { message: errorText } };
+        }
+        const message = errorData.error?.message || 'Unknown error';
+        switch (response.status) {
+            case 401:
+            case 403:
+                throw new AuthenticationError(message, 'google', errorData);
+            case 429:
+                throw new RateLimitError(message, 'google', undefined, errorData);
+            default:
+                throw new LLMProviderError(message, `GOOGLE_${response.status}`, 'google', response.status, response.status >= 500, errorData);
+        }
+    }
+}
+//# sourceMappingURL=google-provider.js.map

package/dist/providers/google-provider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-provider.js","sourceRoot":"","sources":["../../src/providers/google-provider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EASL,gBAAgB,EAChB,cAAc,EACd,mBAAmB,GACpB,MAAM,YAAY,CAAC;AAsDpB,MAAM,OAAO,cAAe,SAAQ,YAAY;IACrC,IAAI,GAAgB,QAAQ,CAAC;IAC7B,YAAY,GAAyB;QAC5C,eAAe,EAAE;YACf,YAAY;YACZ,mBAAmB;YACnB,QAAQ;YACR,OAAO;SACR;QACD,gBAAgB,EAAE;YAChB,YAAY,EAAE,KAAK;YACnB,mBAAmB,EAAE,KAAK;YAC1B,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;SACc;QAC7B,eAAe,EAAE;YACf,YAAY,EAAE,IAAI;YAClB,mBAAmB,EAAE,IAAI;YACzB,QAAQ,EAAE,IAAI;YACd,OAAO,EAAE,IAAI;SACc;QAC7B,iBAAiB,EAAE,IAAI;QACvB,uBAAuB,EAAE,IAAI;QAC7B,sBAAsB,EAAE,KAAK,EAAE,kDAAkD;QACjF,cAAc,EAAE,IAAI,EAAE,oBAAoB;QAC1C,aAAa,EAAE,KAAK;QACpB,aAAa,EAAE,IAAI;QACnB,kBAAkB,EAAE,KAAK;QACzB,kBAAkB,EAAE,IAAI;QACxB,gBAAgB,EAAE,KAAK;QACvB,gBAAgB,EAAE,IAAI;QACtB,SAAS,EAAE;YACT,iBAAiB,EAAE,EAAE;YACrB,eAAe,EAAE,KAAK;YACtB,kBAAkB,EAAE,EAAE;SACvB;QACD,OAAO,EAAE;YACP,YAAY,EAAE;gBACZ,eAAe,EAAE,OAAO;gBACxB,mBAAmB,EAAE,MAAM;gBAC3B,QAAQ,EAAE,KAAK;aAChB;YACD,mBAAmB,EAAE;gBACnB,eAAe,EAAE,OAAO;gBACxB,mBAAmB,EAAE,MAAM;gBAC3B,QAAQ,EAAE,KAAK;aAChB;YACD,QAAQ,EAAE;gBACR,eAAe,EAAE,MAAM;gBACvB,mBAAmB,EAAE,KAAK;gBAC1B,QAAQ,EAAE,KAAK;aAChB;YACD,OAAO,EAAE;gBACP,eAAe,EAAE,MAAM;gBACvB,mBAAmB,EAAE,KAAK;gBAC1B,QAAQ,EAAE,KAAK;aAChB;SACF;KACF,CAAC;IAEM,OAAO,CAAS;IAEd,KAAK,CAAC,YAAY;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YACxB,MAAM,IAAI,mBAAmB,CAAC,+BAA+B,EAAE,QAAQ,CAAC,CAAC;QAC3E,CAAC;QAED,2DAA2D;QAC3D,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAChC,IAAI,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC/B,IAAI,CAAC,OAAO,GAAG,kDAAkD,CAAC;QACpE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,OAAO,GAAG,mDAAmD,CAAC;QACrE,CAAC;IACH,CAAC;IAES,KAAK,CAAC,UAAU,CAAC,OAAmB;QAC5C,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAExE,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,WAAW,KAAK,wBAAwB,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAExF,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,KAAK,CAAC,CAAC;QAEnF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;gBACnC,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,YAAY,CAAC,OAAO,CAAC,CAAC;YAEtB,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,IAAI,GAAqB,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAErD,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrD,MAAM,IAAI,gBAAgB,CACxB,uBAAuB,EACvB,aAAa,EACb,QAAQ,EACR,SAAS,EACT,KAAK,CACN,CAAC;YACJ,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YACrC,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAExE,iBAAiB;YACjB,MAAM,SAAS,GAAG,IAAI,CAAC,aAAa,IAAI;gBACtC,gBAAgB,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;gBACvE,oBAAoB,EAAE,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC;gBAClD,eAAe,EAAE,CAAC;aACnB,CAAC;YACF,SAAS,CAAC,eAAe,GAAG,SAAS,CAAC,gBAAgB,GAAG,SAAS,CAAC,oBAAoB,CAAC;YAExF,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,OAAQ,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC/E,MAAM,UAAU,GAAG,CAAC,SAAS,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;YACjF,MAAM,cAAc,GAAG,CAAC,SAAS,CAAC,oBAAoB,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC;YAE7F,OAAO;gBACL,EAAE,EAAE,UAAU,IAAI,CAAC,GAAG,EAAE,EAAE;gBAC1B,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK;gBACzC,QAAQ,EAAE,QAAQ;gBAClB,OAAO;gBACP,KAAK,EAAE;oBACL,YAAY,EAAE,SAAS,CAAC,gBAAgB;oBACxC,gBAAgB,EAAE,SAAS,CAAC,oBAAoB;oBAChD,WAAW,EAAE,SAAS,CAAC,eAAe;iBACvC;gBACD,IAAI,EAAE;oBACJ,UAAU;oBACV,cAAc;oBACd,SAAS,EAAE,UAAU,GAAG,cAAc;oBACtC,QAAQ,EAAE,KAAK;iBAChB;gBACD,YAAY,EAAE,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,YAAY,CAAC;aAC3D,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;IACH,CAAC;IAES,KAAK,CAAC,CAAC,gBAAgB,CAAC,OAAmB;QACnD,MAAM,aAAa,GAAG,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAExE,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,WAAW,KAAK,8BAA8B,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;QAE9F,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;QACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;QAEzF,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;gBAChC,MAAM,EAAE,MAAM;gBACd,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;iBACnC;gBACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,aAAa,CAAC;gBACnC,MAAM,EAAE,UAAU,CAAC,MAAM;aAC1B,CAAC,CAAC;YAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,CAAC,mBAAmB,CAAC,QAAQ,CAAC,CAAC;YAC3C,CAAC;YAED,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAK,CAAC,SAAS,EAAE,CAAC;YAC1C,MAAM,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;YAClC,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,IAAI,YAAY,GAAG,EAAE,CAAC;YACtB,IAAI,YAAY,GAAG,CAAC,CAAC;YACrB,IAAI,gBAAgB,GAAG,CAAC,CAAC;YAEzB,OAAO,IAAI,EAAE,CAAC;gBACZ,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,IAAI;oBAAE,MAAM;gBAEhB,MAAM,IAAI,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAClD,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACjC,MAAM,GAAG,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC;gBAE3B,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;oBACzB,IAAI,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE;wBAAE,SAAS;oBAEjC,IAAI,CAAC;wBACH,MAAM,IAAI,GAAqB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAEhD,IAAI,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;4BAClD,MAAM,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;4BACrC,MAAM,OAAO,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;4BAExE,IAAI,OAAO,EAAE,CAAC;gCACZ,YAAY,IAAI,OAAO,CAAC;gCACxB,MAAM;oCACJ,IAAI,EAAE,SAAS;oCACf,KAAK,EAAE,EAAE,OAAO,EAAE;iCACnB,CAAC;4BACJ,CAAC;4BAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gCACvB,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC;gCACnD,gBAAgB,GAAG,IAAI,CAAC,aAAa,CAAC,oBAAoB,CAAC;4BAC7D,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,OAAO,CAAC,EAAE,CAAC;wBACX,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;oBACjF,CAAC;gBACH,CAAC;YACH,CAAC;YAED,kCAAkC;YAClC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,OAAQ,CAAC,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC/E,MAAM,UAAU,GAAG,CAAC,YAAY,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,eAAe,CAAC;YACnE,MAAM,cAAc,GAAG,CAAC,gBAAgB,GAAG,IAAI,CAAC,GAAG,OAAO,CAAC,mBAAmB,CAAC;YAE/E,MAAM;gBACJ,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE;oBACL,YAAY;oBACZ,gBAAgB;oBAChB,WAAW,EAAE,YAAY,GAAG,gBAAgB;iBAC7C;gBACD,IAAI,EAAE;oBACJ,UAAU;oBACV,cAAc;oBACd,SAAS,EAAE,UAAU,GAAG,cAAc;oBACtC,QAAQ,EAAE,KAAK;iBAChB;aACF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACnC,CAAC;gBAAS,CAAC;YACT,YAAY,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC;IAC3C,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,KAAe;QAChC,OAAO;YACL,KAAK;YACL,IAAI,EAAE,KAAK;YACX,WAAW,EAAE,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC;YAC5C,aAAa,EAAE,IAAI,CAAC,YAAY,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,IAAI;YAChE,eAAe,EAAE,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,IAAI;YACjE,iBAAiB,EAAE;gBACjB,MAAM;gBACN,YAAY;gBACZ,GAAG,CAAC,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC/C,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;aAC5D;YACD,OAAO,EAAE,IAAI,CAAC,YAAY,CAAC,OAAQ,CAAC,KAAK,CAAC;SAC3C,CAAC;IACJ,CAAC;IAES,KAAK,CAAC,aAAa;QAC3B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,eAAe,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YAC/D,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAAC,CAAC;YAElC,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACjB,MAAM,IAAI,KAAK,CAAC,wBAAwB,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,eAAe;gBAC/D,SAAS,EAAE,IAAI,IAAI,EAAE;aACtB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,kBAAkB,CAAC,OAAmB;QAC5C,oCAAoC;QACpC,MAAM,QAAQ,GAAgC,EAAE,CAAC;QAEjD,KAAK,MAAM,OAAO,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;YACvC,wDAAwD;YACxD,IAAI,OAAO,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;gBAC9B,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC1B,QAAQ,CAAC,IAAI,CAAC;wBACZ,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,iBAAiB,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;qBACtD,CAAC,CAAC;gBACL,CAAC;gBACD,SAAS;YACX,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC;gBACZ,IAAI,EAAE,OAAO,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM;gBACrD,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,OAAO,CAAC,OAAO,EAAE,CAAC;aACnC,CAAC,CAAC;QACL,CAAC;QAED,OAAO;YACL,QAAQ;YACR,gBAAgB,EAAE;gBAChB,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW;gBAC3D,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,IAAI;gBACtC,eAAe,EAAE,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;gBAC3D,aAAa,EAAE,OAAO,CAAC,aAAa,IAAI,IAAI,CAAC,MAAM,CAAC,aAAa;aAClE;YACD,cAAc,EAAE;gBACd;oBACE,QAAQ,EAAE,0BAA0B;oBACpC,SAAS,EAAE,YAAY;iBACxB;gBACD;oBACE,QAAQ,EAAE,2BAA2B;oBACrC,SAAS,EAAE,YAAY;iBACxB;gBACD;oBACE,QAAQ,EAAE,iCAAiC;oBAC3C,SAAS,EAAE,YAAY;iBACxB;gBACD;oBACE,QAAQ,EAAE,iCAAiC;oBAC3C,SAAS,EAAE,YAAY;iBACxB;aACF;SACF,CAAC;IACJ,CAAC;IAEO,gBAAgB,CAAC,KAAe;QACtC,MAAM,QAAQ,GAA2B;YACvC,YAAY,EAAE,YAAY;YAC1B,mBAAmB,EAAE,mBAAmB;YACxC,QAAQ,EAAE,gBAAgB;YAC1B,OAAO,EAAE,gBAAgB;SAC1B,CAAC;QACF,OAAO,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC;IAClC,CAAC;IAEO,eAAe,CAAC,MAAc;QACpC,QAAQ,MAAM,EAAE,CAAC;YACf,KAAK,MAAM;gBACT,OAAO,MAAM,CAAC;YAChB,KAAK,YAAY;gBACf,OAAO,QAAQ,CAAC;YAClB,KAAK,QAAQ,CAAC;YACd,KAAK,YAAY;gBACf,OAAO,gBAAgB,CAAC;YAC1B;gBACE,OAAO,MAAM,CAAC;QAClB,CAAC;IACH,CAAC;IAEO,mBAAmB,CAAC,KAAe;QACzC,MAAM,YAAY,GAA2B;YAC3C,YAAY,EAAE,mCAAmC;YACjD,mBAAmB,EAAE,qCAAqC;YAC1D,QAAQ,EAAE,0CAA0C;YACpD,OAAO,EAAE,mCAAmC;SAC7C,CAAC;QACF,OAAO,YAAY,CAAC,KAAK,CAAC,IAAI,0BAA0B,CAAC;IAC3D,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,QAAkB;QAClD,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACxC,IAAI,SAAc,CAAC;QAEnB,IAAI,CAAC;YACH,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QACpC,CAAC;QAAC,MAAM,CAAC;YACP,SAAS,GAAG,EAAE,KAAK,EAAE,EAAE,OAAO,EAAE,SAAS,EAAE,EAAE,CAAC;QAChD,CAAC;QAED,MAAM,OAAO,GAAG,SAAS,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,CAAC;QAE5D,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC;YACxB,KAAK,GAAG,CAAC;YACT,KAAK,GAAG;gBACN,MAAM,IAAI,mBAAmB,CAAC,OAAO,EAAE,QAAQ,EAAE,SAAS,CAAC,CAAC;YAC9D,KAAK,GAAG;gBACN,MAAM,IAAI,cAAc,CAAC,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YACpE;gBACE,MAAM,IAAI,gBAAgB,CACxB,OAAO,EACP,UAAU,QAAQ,CAAC,MAAM,EAAE,EAC3B,QAAQ,EACR,QAAQ,CAAC,MAAM,EACf,QAAQ,CAAC,MAAM,IAAI,GAAG,EACtB,SAAS,CACV,CAAC;QACN,CAAC;IACH,CAAC;CACF"}

package/dist/providers/index.d.ts

@@ -0,0 +1,14 @@
+/**
+ * Multi-LLM Provider System
+ * Export all provider types and implementations
+ */
+export * from './types.js';
+export { BaseProvider } from './base-provider.js';
+export { AnthropicProvider } from './anthropic-provider.js';
+export { OpenAIProvider } from './openai-provider.js';
+export { GoogleProvider } from './google-provider.js';
+export { CohereProvider } from './cohere-provider.js';
+export { OllamaProvider } from './ollama-provider.js';
+export { ProviderManager, ProviderManagerConfig } from './provider-manager.js';
+export { createProviderManager, getDefaultProviderConfig } from './utils.js';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAGH,cAAc,YAAY,CAAC;AAG3B,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EAAE,eAAe,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAG/E,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,YAAY,CAAC"}

package/dist/providers/index.js

@@ -0,0 +1,18 @@
+/**
+ * Multi-LLM Provider System
+ * Export all provider types and implementations
+ */
+// Export types
+export * from './types.js';
+// Export providers
+export { BaseProvider } from './base-provider.js';
+export { AnthropicProvider } from './anthropic-provider.js';
+export { OpenAIProvider } from './openai-provider.js';
+export { GoogleProvider } from './google-provider.js';
+export { CohereProvider } from './cohere-provider.js';
+export { OllamaProvider } from './ollama-provider.js';
+// Export manager
+export { ProviderManager } from './provider-manager.js';
+// Export utility functions
+export { createProviderManager, getDefaultProviderConfig } from './utils.js';
+//# sourceMappingURL=index.js.map

package/dist/providers/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/providers/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,eAAe;AACf,cAAc,YAAY,CAAC;AAE3B,mBAAmB;AACnB,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,iBAAiB,EAAE,MAAM,yBAAyB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AACtD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,iBAAiB;AACjB,OAAO,EAAE,eAAe,EAAyB,MAAM,uBAAuB,CAAC;AAE/E,2BAA2B;AAC3B,OAAO,EAAE,qBAAqB,EAAE,wBAAwB,EAAE,MAAM,YAAY,CAAC"}

package/dist/providers/ollama-provider.d.ts

@@ -0,0 +1,23 @@
+/**
+ * Ollama Provider Implementation
+ * Supports local models running via Ollama
+ */
+import { BaseProvider } from './base-provider.js';
+import { LLMProvider, LLMModel, LLMRequest, LLMResponse, LLMStreamEvent, ModelInfo, ProviderCapabilities, HealthCheckResult } from './types.js';
+export declare class OllamaProvider extends BaseProvider {
+    readonly name: LLMProvider;
+    readonly capabilities: ProviderCapabilities;
+    private baseUrl;
+    private availableModels;
+    protected doInitialize(): Promise<void>;
+    private fetchAvailableModels;
+    protected doComplete(request: LLMRequest): Promise<LLMResponse>;
+    protected doStreamComplete(request: LLMRequest): AsyncIterable<LLMStreamEvent>;
+    listModels(): Promise<LLMModel[]>;
+    getModelInfo(model: LLMModel): Promise<ModelInfo>;
+    protected doHealthCheck(): Promise<HealthCheckResult>;
+    private mapToOllamaModel;
+    private getModelDescription;
+    private handleErrorResponse;
+}
+//# sourceMappingURL=ollama-provider.d.ts.map

package/dist/providers/ollama-provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama-provider.d.ts","sourceRoot":"","sources":["../../src/providers/ollama-provider.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EACL,WAAW,EACX,QAAQ,EACR,UAAU,EACV,WAAW,EACX,cAAc,EACd,SAAS,EACT,oBAAoB,EACpB,iBAAiB,EAGlB,MAAM,YAAY,CAAC;AAiEpB,qBAAa,cAAe,SAAQ,YAAY;IAC9C,QAAQ,CAAC,IAAI,EAAE,WAAW,CAAY;IACtC,QAAQ,CAAC,YAAY,EAAE,oBAAoB,CA4CzC;IAEF,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,eAAe,CAA0B;cAEjC,YAAY,IAAI,OAAO,CAAC,IAAI,CAAC;YAW/B,oBAAoB;cAmClB,UAAU,CAAC,OAAO,EAAE,UAAU,GAAG,OAAO,CAAC,WAAW,CAAC;cAkFpD,gBAAgB,CAAC,OAAO,EAAE,UAAU,GAAG,aAAa,CAAC,cAAc,CAAC;IAsG/E,UAAU,IAAI,OAAO,CAAC,QAAQ,EAAE,CAAC;IAUjC,YAAY,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;cA8CvC,aAAa,IAAI,OAAO,CAAC,iBAAiB,CAAC;IAwB3D,OAAO,CAAC,gBAAgB;IAYxB,OAAO,CAAC,mBAAmB;YAYb,mBAAmB;CAqBlC"}