@gitkraken/core-gitlens 0.1.0

package/dist/plus/ai/providers/ollamaProvider.js

@@ -0,0 +1,155 @@
+import { uuid } from '../../../utils/crypto.js';
+import { ollamaProviderDescriptor as provider } from '../constants.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+const defaultBaseUrl = 'http://localhost:11434';
+export class OllamaProvider extends OpenAICompatibleProviderBase {
+    id = provider.id;
+    name = provider.name;
+    descriptor = provider;
+    config = {
+        keyUrl: 'https://ollama.com/download',
+    };
+    async configured(silent) {
+        const url = await this.getOrPromptBaseUrl(silent);
+        if (url === undefined) {
+            return false;
+        }
+        // Ollama doesn't require an API key, but we'll check if the base URL is reachable
+        return this.validateUrl(url);
+    }
+    getApiKey(_silent) {
+        // Ollama doesn't require an API key — account enrollment is handled by the context's getApiKey
+        return Promise.resolve('<not applicable>');
+    }
+    async getModels() {
+        try {
+            const url = this.getBaseUrl();
+            const rsp = await this.context.fetch(`${url}/api/tags`, {
+                headers: {
+                    Accept: 'application/json',
+                    'Content-Type': 'application/json',
+                },
+                method: 'GET',
+            });
+            if (!rsp.ok) {
+                throw new Error(`Getting models failed: ${rsp.status} (${rsp.statusText})`);
+            }
+            const result = (await rsp.json());
+            // If there are models installed on the user's Ollama instance, use those
+            if (result.models?.length) {
+                return result.models.map(m => ({
+                    id: m.name,
+                    name: m.name,
+                    maxTokens: { input: 8192, output: 8192 },
+                    provider: provider,
+                    default: m.name === 'llama3',
+                }));
+            }
+        }
+        catch { }
+        return [];
+    }
+    async getOrPromptBaseUrl(silent) {
+        const cfg = this.context.getProviderConfig(this.id);
+        if (!cfg.enabled)
+            return undefined;
+        if (cfg.url)
+            return cfg.url;
+        const url = await this.context.getOrPromptUrl(this.id, {
+            currentUrl: defaultBaseUrl,
+            title: 'Connect to Ollama',
+            placeholder: 'Please enter your Ollama server URL to use this feature',
+            validator: async (u) => {
+                const valid = await this.validateUrl(u);
+                return valid
+                    ? undefined
+                    : 'Could not connect to Ollama server. Make sure Ollama is installed and running locally.';
+            },
+        }, silent);
+        return url ?? defaultBaseUrl;
+    }
+    async validateUrl(url) {
+        try {
+            const rsp = await this.context.fetch(`${url}/api/tags`, {
+                headers: { Accept: 'application/json', 'Content-Type': 'application/json' },
+                method: 'GET',
+            });
+            return rsp.ok;
+        }
+        catch {
+            return false;
+        }
+    }
+    getBaseUrl() {
+        const orgConf = this.context.getProviderConfig(this.id);
+        if (!orgConf.enabled)
+            return undefined;
+        return orgConf.url || defaultBaseUrl;
+    }
+    getUrl(_model) {
+        const url = this.getBaseUrl();
+        return url ? `${url}/api/chat` : undefined;
+    }
+    getHeaders(_action, _apiKey, _model, _url) {
+        return {
+            Accept: 'application/json',
+            'Content-Type': 'application/json',
+        };
+    }
+    async fetch(action, model, apiKey, messages, modelOptions, signal) {
+        let retries = 0;
+        let maxInputTokens = model.maxTokens.input;
+        while (true) {
+            // Get messages and prepare request payload for Ollama
+            const chatMessages = await messages(maxInputTokens, retries);
+            // Convert to the format expected by Ollama
+            const ollamaMessages = chatMessages.map(msg => ({
+                role: msg.role,
+                content: msg.content,
+            }));
+            // Ensure temperature is within valid range for Ollama (0.0-1.0)
+            const temperature = Math.min(Math.max(modelOptions?.temperature ?? 0.7, 0), 1);
+            const request = {
+                model: model.id,
+                messages: ollamaMessages,
+                stream: false,
+                options: {
+                    temperature: temperature,
+                    // Add num_predict if outputTokens is specified
+                    ...(modelOptions?.outputTokens ? { num_predict: modelOptions.outputTokens } : {}),
+                },
+            };
+            const rsp = await this.fetchCore(action, model, apiKey, request, signal);
+            if (!rsp.ok) {
+                const result = await this.handleFetchFailure(rsp, action, model, retries, maxInputTokens);
+                if (result.retry) {
+                    maxInputTokens = result.maxInputTokens;
+                    retries++;
+                    continue;
+                }
+            }
+            try {
+                // Parse response from Ollama
+                const data = (await rsp.json());
+                if (!data.message?.content) {
+                    throw new Error(`Empty response from Ollama model: ${model.id}`);
+                }
+                return {
+                    id: uuid(),
+                    content: data.message.content,
+                    model: model,
+                    usage: {
+                        promptTokens: data.prompt_eval_count ?? 0,
+                        completionTokens: data.eval_count ?? 0,
+                        totalTokens: (data.prompt_eval_count ?? 0) + (data.eval_count ?? 0),
+                    },
+                    result: undefined,
+                };
+            }
+            catch (err) {
+                throw new Error(`Failed to parse Ollama response: ${err instanceof Error ? err.message : String(err)}`, { cause: err });
+            }
+        }
+    }
+}
+//# sourceMappingURL=ollamaProvider.js.map

package/dist/plus/ai/providers/ollamaProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollamaProvider.js","sourceRoot":"","sources":["../../src/providers/ollamaProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAChD,OAAO,EAAE,wBAAwB,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAGvE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAIjF,MAAM,cAAc,GAAG,wBAAwB,CAAC;AAEhD,MAAM,OAAO,cAAe,SAAQ,4BAAgD;IAC1E,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;IACjB,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IACX,UAAU,GAAG,QAAQ,CAAC;IACtB,MAAM,GAAG;QAC3B,MAAM,EAAE,6BAA6B;KACrC,CAAC;IAEO,KAAK,CAAC,UAAU,CAAC,MAAe;QACxC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAClD,IAAI,GAAG,KAAK,SAAS,EAAE,CAAC;YACvB,OAAO,KAAK,CAAC;QACd,CAAC;QACD,kFAAkF;QAClF,OAAO,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC9B,CAAC;IAEQ,SAAS,CAAC,OAAgB;QAClC,+FAA+F;QAC/F,OAAO,OAAO,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,SAAS;QACd,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YAC9B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,WAAW,EAAE;gBACvD,OAAO,EAAE;oBACR,MAAM,EAAE,kBAAkB;oBAC1B,cAAc,EAAE,kBAAkB;iBAClC;gBACD,MAAM,EAAE,KAAK;aACb,CAAC,CAAC;YAEH,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,IAAI,KAAK,CAAC,0BAA0B,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC;YAC7E,CAAC;YAeD,MAAM,MAAM,GAAyB,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAyB,CAAC;YAEhF,yEAAyE;YACzE,IAAI,MAAM,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC;gBAC3B,OAAO,MAAM,CAAC,MAAM,CAAC,GAAG,CAAc,CAAC,CAAC,EAAE,CAAC,CAAC;oBAC3C,EAAE,EAAE,CAAC,CAAC,IAAI;oBACV,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,SAAS,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;oBACxC,QAAQ,EAAE,QAAQ;oBAClB,OAAO,EAAE,CAAC,CAAC,IAAI,KAAK,QAAQ;iBAC5B,CAAC,CAAC,CAAC;YACL,CAAC;QACF,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QAEV,OAAO,EAAE,CAAC;IACX,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,MAAe;QAC/C,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,GAAG,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAEnC,IAAI,GAAG,CAAC,GAAG;YAAE,OAAO,GAAG,CAAC,GAAG,CAAC;QAE5B,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,cAAc,CAC5C,IAAI,CAAC,EAAE,EACP;YACC,UAAU,EAAE,cAAc;YAC1B,KAAK,EAAE,mBAAmB;YAC1B,WAAW,EAAE,yDAAyD;YACtE,SAAS,EAAE,KAAK,EAAE,CAAS,EAAE,EAAE;gBAC9B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;gBACxC,OAAO,KAAK;oBACX,CAAC,CAAC,SAAS;oBACX,CAAC,CAAC,wFAAwF,CAAC;YAC7F,CAAC;SACD,EACD,MAAM,CACN,CAAC;QACF,OAAO,GAAG,IAAI,cAAc,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,GAAW;QACpC,IAAI,CAAC;YACJ,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,GAAG,WAAW,EAAE;gBACvD,OAAO,EAAE,EAAE,MAAM,EAAE,kBAAkB,EAAE,cAAc,EAAE,kBAAkB,EAAE;gBAC3E,MAAM,EAAE,KAAK;aACb,CAAC,CAAC;YACH,OAAO,GAAG,CAAC,EAAE,CAAC;QACf,CAAC;QAAC,MAAM,CAAC;YACR,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IAEO,UAAU;QACjB,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QACvC,OAAO,OAAO,CAAC,GAAG,IAAI,cAAc,CAAC;IACtC,CAAC;IAES,MAAM,CAAC,MAAmC;QACnD,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAC9B,OAAO,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC;IAC5C,CAAC;IAEkB,UAAU,CAC5B,OAAgB,EAChB,OAAe,EACf,MAAmC,EACnC,IAAY;QAEZ,OAAO;YACN,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;SAClC,CAAC;IACH,CAAC;IAEkB,KAAK,CAAC,KAAK,CAC7B,MAAe,EACf,KAAkC,EAClC,MAAc,EACd,QAA+E,EAC/E,YAA8D,EAC9D,MAAoB;QAEpB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC;QAE3C,OAAO,IAAI,EAAE,CAAC;YACb,sDAAsD;YACtD,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;YAE7D,2CAA2C;YAC3C,MAAM,cAAc,GAAG,YAAY,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/C,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,OAAO,EAAE,GAAG,CAAC,OAAO;aACpB,CAAC,CAAC,CAAC;YAEJ,gEAAgE;YAChE,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,WAAW,IAAI,GAAG,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAE/E,MAAM,OAAO,GAAsB;gBAClC,KAAK,EAAE,KAAK,CAAC,EAAE;gBACf,QAAQ,EAAE,cAAc;gBACxB,MAAM,EAAE,KAAK;gBACb,OAAO,EAAE;oBACR,WAAW,EAAE,WAAW;oBACxB,+CAA+C;oBAC/C,GAAG,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,YAAY,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;iBACjF;aACD,CAAC;YAEF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACzE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;gBAC1F,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAClB,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;oBACvC,OAAO,EAAE,CAAC;oBACV,SAAS;gBACV,CAAC;YACF,CAAC;YAED,IAAI,CAAC;gBACJ,6BAA6B;gBAC7B,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAuB,CAAC;gBAEtD,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,qCAAqC,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;gBAClE,CAAC;gBAED,OAAO;oBACN,EAAE,EAAE,IAAI,EAAE;oBACV,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;oBAC7B,KAAK,EAAE,KAAK;oBACZ,KAAK,EAAE;wBACN,YAAY,EAAE,IAAI,CAAC,iBAAiB,IAAI,CAAC;wBACzC,gBAAgB,EAAE,IAAI,CAAC,UAAU,IAAI,CAAC;wBACtC,WAAW,EAAE,CAAC,IAAI,CAAC,iBAAiB,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,CAAC;qBACnE;oBACD,MAAM,EAAE,SAAS;iBACjB,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CACd,oCAAoC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EACtF,EAAE,KAAK,EAAE,GAAG,EAAE,CACd,CAAC;YACH,CAAC;QACF,CAAC;IACF,CAAC;CACD"}

package/dist/plus/ai/providers/openAICompatibleProvider.d.ts

@@ -0,0 +1,17 @@
+import { openAICompatibleProviderDescriptor as provider } from '../constants.js';
+import type { AIModel } from '../models/model.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+export declare class OpenAICompatibleProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+    readonly id: "openaicompatible";
+    readonly name: string;
+    protected readonly descriptor: import("../models/model.js").AIProviderDescriptor<"openaicompatible">;
+    protected readonly config: {
+        keyUrl: undefined;
+        keyValidator: RegExp;
+    };
+    getModels(): Promise<readonly AIModel<typeof provider.id>[]>;
+    protected getUrl(_model?: AIModel<typeof provider.id>): string | undefined;
+    private getOrPromptBaseUrl;
+    configured(silent: boolean): Promise<boolean>;
+}
+//# sourceMappingURL=openAICompatibleProvider.d.ts.map

package/dist/plus/ai/providers/openAICompatibleProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openAICompatibleProvider.d.ts","sourceRoot":"","sources":["../../src/providers/openAICompatibleProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kCAAkC,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AACjF,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAGlD,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAKjF,qBAAa,wBAAyB,SAAQ,4BAA4B,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC;IAC7F,QAAQ,CAAC,EAAE,qBAAe;IAC1B,QAAQ,CAAC,IAAI,SAAiB;IAC9B,SAAS,CAAC,QAAQ,CAAC,UAAU,wEAAY;IACzC,SAAS,CAAC,QAAQ,CAAC,MAAM;;;MAGvB;IAEF,SAAS,IAAI,OAAO,CAAC,SAAS,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IAI5D,SAAS,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,GAAG,MAAM,GAAG,SAAS;YAO5D,kBAAkB;IAqBjB,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;CAQ5D"}

package/dist/plus/ai/providers/openAICompatibleProvider.js

@@ -0,0 +1,47 @@
+import { openAICompatibleProviderDescriptor as provider } from '../constants.js';
+import { openAIModels } from '../models/model.js';
+import { isAzureUrl } from '../utils/ai.utils.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+const models = openAIModels(provider);
+export class OpenAICompatibleProvider extends OpenAICompatibleProviderBase {
+    id = provider.id;
+    name = provider.name;
+    descriptor = provider;
+    config = {
+        keyUrl: undefined,
+        keyValidator: /(?:sk-)?[a-zA-Z0-9]{32,}/,
+    };
+    getModels() {
+        return Promise.resolve(models);
+    }
+    getUrl(_model) {
+        const cfg = this.context.getProviderConfig(this.id);
+        if (!cfg.enabled)
+            return undefined;
+        return cfg.url || undefined;
+    }
+    async getOrPromptBaseUrl(silent, hasApiKey) {
+        const orgConf = this.context.getProviderConfig(this.id);
+        if (!orgConf.enabled)
+            return undefined;
+        if (orgConf.url)
+            return orgConf.url;
+        const url = this.getUrl();
+        if (silent || (url != null && hasApiKey))
+            return url;
+        return this.context.getOrPromptUrl(this.id, {
+            currentUrl: url,
+            title: 'Connect to OpenAI-Compatible Provider',
+            placeholder: `Please enter your provider's URL to use this feature`,
+            validator: u => (isAzureUrl(u) ? 'Use the Azure OpenAI provider instead' : undefined),
+        }, silent);
+    }
+    async configured(silent) {
+        const hasApiKey = await super.configured(true);
+        const url = await this.getOrPromptBaseUrl(silent, hasApiKey);
+        if (!url || isAzureUrl(url))
+            return false;
+        return silent ? hasApiKey : super.configured(silent);
+    }
+}
+//# sourceMappingURL=openAICompatibleProvider.js.map

package/dist/plus/ai/providers/openAICompatibleProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openAICompatibleProvider.js","sourceRoot":"","sources":["../../src/providers/openAICompatibleProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kCAAkC,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAEjF,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AAClD,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAGjF,MAAM,MAAM,GAA4B,YAAY,CAAC,QAAQ,CAAC,CAAC;AAE/D,MAAM,OAAO,wBAAyB,SAAQ,4BAAgD;IACpF,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;IACjB,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IACX,UAAU,GAAG,QAAQ,CAAC;IACtB,MAAM,GAAG;QAC3B,MAAM,EAAE,SAAS;QACjB,YAAY,EAAE,0BAA0B;KACxC,CAAC;IAEF,SAAS;QACR,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAES,MAAM,CAAC,MAAoC;QACpD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,GAAG,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QAEnC,OAAO,GAAG,CAAC,GAAG,IAAI,SAAS,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAAC,MAAe,EAAE,SAAkB;QACnE,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QACvC,IAAI,OAAO,CAAC,GAAG;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC;QAEpC,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAE1B,IAAI,MAAM,IAAI,CAAC,GAAG,IAAI,IAAI,IAAI,SAAS,CAAC;YAAE,OAAO,GAAG,CAAC;QAErD,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CACjC,IAAI,CAAC,EAAE,EACP;YACC,UAAU,EAAE,GAAG;YACf,KAAK,EAAE,uCAAuC;YAC9C,WAAW,EAAE,sDAAsD;YACnE,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,uCAAuC,CAAC,CAAC,CAAC,SAAS,CAAC;SACrF,EACD,MAAM,CACN,CAAC;IACH,CAAC;IAEQ,KAAK,CAAC,UAAU,CAAC,MAAe;QACxC,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAE/C,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,MAAM,EAAE,SAAS,CAAC,CAAC;QAC7D,IAAI,CAAC,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAE1C,OAAO,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;CACD"}

package/dist/plus/ai/providers/openAICompatibleProviderBase.d.ts

@@ -0,0 +1,39 @@
+import type { AIProviders } from '../constants.js';
+import type { AIActionType, AIModel, AIProviderDescriptor } from '../models/model.js';
+import type { AIChatMessage, AIProvider, AIProviderResponse } from '../models/provider.js';
+import type { AIProviderContext } from './context.js';
+export declare abstract class OpenAICompatibleProviderBase<T extends AIProviders> implements AIProvider<T> {
+    protected readonly context: AIProviderContext;
+    constructor(context: AIProviderContext);
+    dispose(): void;
+    [Symbol.dispose](): void;
+    abstract readonly id: T;
+    abstract readonly name: string;
+    protected abstract readonly descriptor: AIProviderDescriptor<T>;
+    protected abstract readonly config: {
+        keyUrl?: string;
+        keyValidator?: RegExp;
+    };
+    configured(silent: boolean): Promise<boolean>;
+    getApiKey(silent: boolean): Promise<string | undefined>;
+    abstract getModels(): Promise<readonly AIModel<T>[]>;
+    protected abstract getUrl(_model: AIModel<T>): string | undefined;
+    protected getHeaders<TAction extends AIActionType>(_action: TAction, apiKey: string, _model: AIModel<T>, _url: string): Record<string, string> | Promise<Record<string, string>>;
+    sendRequest<TAction extends AIActionType>(action: TAction, model: AIModel<T>, apiKey: string, getMessages: (maxInputTokens: number, retries: number) => Promise<AIChatMessage[]>, options: {
+        signal: AbortSignal;
+        modelOptions?: {
+            outputTokens?: number;
+            temperature?: number;
+        };
+    }): Promise<AIProviderResponse<void> | undefined>;
+    protected fetch<TAction extends AIActionType>(action: TAction, model: AIModel<T>, apiKey: string, messages: (maxInputTokens: number, retries: number) => Promise<AIChatMessage[]>, modelOptions?: {
+        outputTokens?: number;
+        temperature?: number;
+    }, signal?: AbortSignal): Promise<AIProviderResponse<void>>;
+    protected handleFetchFailure<TAction extends AIActionType>(rsp: Response, _action: TAction, model: AIModel<T>, retries: number, maxInputTokens: number): Promise<{
+        retry: true;
+        maxInputTokens: number;
+    }>;
+    protected fetchCore<TAction extends AIActionType>(action: TAction, model: AIModel<T>, apiKey: string, request: object, signal: AbortSignal | undefined): Promise<Response>;
+}
+//# sourceMappingURL=openAICompatibleProviderBase.d.ts.map

package/dist/plus/ai/providers/openAICompatibleProviderBase.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openAICompatibleProviderBase.d.ts","sourceRoot":"","sources":["../../src/providers/openAICompatibleProviderBase.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AAEnD,OAAO,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,oBAAoB,CAAC;AACtF,OAAO,KAAK,EAAE,aAAa,EAAqB,UAAU,EAAE,kBAAkB,EAAE,MAAM,uBAAuB,CAAC;AAE9G,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEtD,8BAAsB,4BAA4B,CAAC,CAAC,SAAS,WAAW,CAAE,YAAW,UAAU,CAAC,CAAC,CAAC;IACrF,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,iBAAiB;gBAA1B,OAAO,EAAE,iBAAiB;IAEzD,OAAO,IAAI,IAAI;IACf,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,IAAI;IAIxB,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC,CAAC;IACxB,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAC/B,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,UAAU,EAAE,oBAAoB,CAAC,CAAC,CAAC,CAAC;IAChE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,EAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,YAAY,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC;IAEzE,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAW7C,SAAS,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC;IAmB7D,QAAQ,CAAC,SAAS,IAAI,OAAO,CAAC,SAAS,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IAEpD,SAAS,CAAC,QAAQ,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,MAAM,GAAG,SAAS;IAEjE,SAAS,CAAC,UAAU,CAAC,OAAO,SAAS,YAAY,EAChD,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAClB,IAAI,EAAE,MAAM,GACV,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IAQrD,WAAW,CAAC,OAAO,SAAS,YAAY,EAC7C,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,EAAE,MAAM,EACd,WAAW,EAAE,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,EAClF,OAAO,EAAE;QAAE,MAAM,EAAE,WAAW,CAAC;QAAC,YAAY,CAAC,EAAE;YAAE,YAAY,CAAC,EAAE,MAAM,CAAC;YAAC,WAAW,CAAC,EAAE,MAAM,CAAA;SAAE,CAAA;KAAE,GAC9F,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;cAsBhC,KAAK,CAAC,OAAO,SAAS,YAAY,EACjD,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,EAAE,MAAM,EACd,QAAQ,EAAE,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,OAAO,CAAC,aAAa,EAAE,CAAC,EAC/E,YAAY,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,EAC9D,MAAM,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,CAAC;cAqDpB,kBAAkB,CAAC,OAAO,SAAS,YAAY,EAC9D,GAAG,EAAE,QAAQ,EACb,OAAO,EAAE,OAAO,EAChB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,OAAO,EAAE,MAAM,EACf,cAAc,EAAE,MAAM,GACpB,OAAO,CAAC;QAAE,KAAK,EAAE,IAAI,CAAC;QAAC,cAAc,EAAE,MAAM,CAAA;KAAE,CAAC;cAsCnC,SAAS,CAAC,OAAO,SAAS,YAAY,EACrD,MAAM,EAAE,OAAO,EACf,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,MAAM,EACf,MAAM,EAAE,WAAW,GAAG,SAAS,GAC7B,OAAO,CAAC,QAAQ,CAAC;CAkBpB"}

package/dist/plus/ai/providers/openAICompatibleProviderBase.js

@@ -0,0 +1,218 @@
+var __addDisposableResource = (this && this.__addDisposableResource) || function (env, value, async) {
+    if (value !== null && value !== void 0) {
+        if (typeof value !== "object" && typeof value !== "function") throw new TypeError("Object expected.");
+        var dispose, inner;
+        if (async) {
+            if (!Symbol.asyncDispose) throw new TypeError("Symbol.asyncDispose is not defined.");
+            dispose = value[Symbol.asyncDispose];
+        }
+        if (dispose === void 0) {
+            if (!Symbol.dispose) throw new TypeError("Symbol.dispose is not defined.");
+            dispose = value[Symbol.dispose];
+            if (async) inner = dispose;
+        }
+        if (typeof dispose !== "function") throw new TypeError("Object not disposable.");
+        if (inner) dispose = function() { try { inner.call(this); } catch (e) { return Promise.reject(e); } };
+        env.stack.push({ value: value, dispose: dispose, async: async });
+    }
+    else if (async) {
+        env.stack.push({ async: true });
+    }
+    return value;
+};
+var __disposeResources = (this && this.__disposeResources) || (function (SuppressedError) {
+    return function (env) {
+        function fail(e) {
+            env.error = env.hasError ? new SuppressedError(e, env.error, "An error was suppressed during disposal.") : e;
+            env.hasError = true;
+        }
+        var r, s = 0;
+        function next() {
+            while (r = env.stack.pop()) {
+                try {
+                    if (!r.async && s === 1) return s = 0, env.stack.push(r), Promise.resolve().then(next);
+                    if (r.dispose) {
+                        var result = r.dispose.call(r.value);
+                        if (r.async) return s |= 2, Promise.resolve(result).then(next, function(e) { fail(e); return next(); });
+                    }
+                    else s |= 1;
+                }
+                catch (e) {
+                    fail(e);
+                }
+            }
+            if (s === 1) return env.hasError ? Promise.reject(env.error) : Promise.resolve();
+            if (env.hasError) throw env.error;
+        }
+        return next();
+    };
+})(typeof SuppressedError === "function" ? SuppressedError : function (error, suppressed, message) {
+    var e = new Error(message);
+    return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
+});
+import { CancellationError, isCancellationError } from '../../../utils/cancellation.js';
+import { uuid } from '../../../utils/crypto.js';
+import { getLoggableName } from '../../../utils/logger.js';
+import { maybeStartScopedLogger } from '../../../utils/logger.scoped.js';
+import { AIError, AIErrorReason } from '../errors.js';
+import { getActionName, getReducedMaxInputTokens, getValidatedTemperature } from '../utils/ai.utils.js';
+export class OpenAICompatibleProviderBase {
+    context;
+    constructor(context) {
+        this.context = context;
+    }
+    dispose() { }
+    [Symbol.dispose]() {
+        this.dispose();
+    }
+    async configured(silent) {
+        try {
+            const apiKey = await this.getApiKey(silent);
+            return apiKey != null;
+        }
+        catch (ex) {
+            if (isCancellationError(ex))
+                return false;
+            throw ex;
+        }
+    }
+    async getApiKey(silent) {
+        const orgConf = this.context.getProviderConfig(this.id);
+        if (!orgConf.enabled)
+            return undefined;
+        if (orgConf.key)
+            return orgConf.key;
+        const { keyUrl, keyValidator } = this.config;
+        return this.context.getApiKey({
+            id: this.id,
+            name: this.name,
+            requiresAccount: this.descriptor.requiresAccount,
+            validator: keyValidator != null ? v => keyValidator.test(v) : () => true,
+            url: keyUrl,
+        }, silent);
+    }
+    getHeaders(_action, apiKey, _model, _url) {
+        return {
+            Accept: 'application/json',
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${apiKey}`,
+        };
+    }
+    async sendRequest(action, model, apiKey, getMessages, options) {
+        const env_1 = { stack: [], error: void 0, hasError: false };
+        try {
+            const scope = __addDisposableResource(env_1, maybeStartScopedLogger(`${getLoggableName(this)}.sendRequest`), false);
+            try {
+                const result = await this.fetch(action, model, apiKey, getMessages, options.modelOptions, options.signal);
+                return result;
+            }
+            catch (ex) {
+                if (isCancellationError(ex)) {
+                    scope?.error(ex, `Cancelled request to ${getActionName(action)}: (${model.provider.name})`);
+                    throw ex;
+                }
+                scope?.error(ex, `Unable to ${getActionName(action)}: (${model.provider.name})`);
+                if (ex instanceof AIError)
+                    throw ex;
+                debugger;
+                throw new Error(`Unable to ${getActionName(action)}: (${model.provider.name}) ${ex.message}`, {
+                    cause: ex,
+                });
+            }
+        }
+        catch (e_1) {
+            env_1.error = e_1;
+            env_1.hasError = true;
+        }
+        finally {
+            __disposeResources(env_1);
+        }
+    }
+    async fetch(action, model, apiKey, messages, modelOptions, signal) {
+        let retries = 0;
+        let maxInputTokens = model.maxTokens.input;
+        while (true) {
+            const request = {
+                model: model.id,
+                messages: await messages(maxInputTokens, retries),
+                stream: false,
+                max_completion_tokens: model.maxTokens.output
+                    ? Math.min(modelOptions?.outputTokens ?? Infinity, model.maxTokens.output)
+                    : modelOptions?.outputTokens,
+                temperature: getValidatedTemperature(model, modelOptions?.temperature ?? model.temperature, this.context.defaultTemperature),
+            };
+            const rsp = await this.fetchCore(action, model, apiKey, request, signal);
+            if (!rsp.ok) {
+                const result = await this.handleFetchFailure(rsp, action, model, retries, maxInputTokens);
+                if (result.retry) {
+                    maxInputTokens = result.maxInputTokens;
+                    retries++;
+                    continue;
+                }
+            }
+            const data = (await rsp.json());
+            const result = {
+                id: data.id ?? uuid(),
+                content: data.choices?.[0].message.content?.trim() ?? data.content?.[0]?.text?.trim() ?? '',
+                model: model,
+                usage: {
+                    promptTokens: data.usage?.prompt_tokens ?? data.usage?.input_tokens,
+                    completionTokens: data.usage?.completion_tokens ?? data.usage?.output_tokens,
+                    totalTokens: data.usage?.total_tokens,
+                    limits: data?.usage?.gk != null
+                        ? {
+                            used: data.usage.gk.used,
+                            limit: data.usage.gk.limit,
+                            resetsOn: new Date(data.usage.gk.resets_on),
+                        }
+                        : undefined,
+                },
+                result: undefined,
+            };
+            return result;
+        }
+    }
+    async handleFetchFailure(rsp, _action, model, retries, maxInputTokens) {
+        if (rsp.status === 404) {
+            throw new AIError(AIErrorReason.Unauthorized, new Error(`Your API key doesn't seem to have access to the selected '${model.id}' model`));
+        }
+        if (rsp.status === 429) {
+            throw new AIError(AIErrorReason.RateLimitOrFundsExceeded, new Error(`(${this.name}) ${rsp.status}: Too many requests (rate limit exceeded) or your account is out of funds`));
+        }
+        let json;
+        try {
+            json = (await rsp.json());
+        }
+        catch { }
+        if (Array.isArray(json)) {
+            json = json[0];
+        }
+        if (json?.error?.code === 'context_length_exceeded') {
+            if (retries < 3) {
+                return { retry: true, maxInputTokens: getReducedMaxInputTokens(maxInputTokens, retries + 1) };
+            }
+            throw new AIError(AIErrorReason.RequestTooLarge, new Error(`(${this.name}) ${rsp.status}: ${json?.error?.message || rsp.statusText}`));
+        }
+        throw new Error(`(${this.name}) ${rsp.status}: ${json?.error?.message || rsp.statusText}`);
+    }
+    async fetchCore(action, model, apiKey, request, signal) {
+        const url = this.getUrl(model);
+        if (!url) {
+            throw new Error(`(${this.name}) ${getActionName(action)}: No URL configured`);
+        }
+        try {
+            return await this.context.fetch(url, {
+                headers: await this.getHeaders(action, apiKey, model, url),
+                method: 'POST',
+                body: JSON.stringify(request),
+                signal: signal,
+            });
+        }
+        catch (ex) {
+            if (ex.name === 'AbortError')
+                throw new CancellationError(ex);
+            throw ex;
+        }
+    }
+}
+//# sourceMappingURL=openAICompatibleProviderBase.js.map

package/dist/plus/ai/providers/openAICompatibleProviderBase.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openAICompatibleProviderBase.js","sourceRoot":"","sources":["../../src/providers/openAICompatibleProviderBase.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,OAAO,EAAE,iBAAiB,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AACxF,OAAO,EAAE,IAAI,EAAE,MAAM,0BAA0B,CAAC;AAChD,OAAO,EAAE,eAAe,EAAE,MAAM,0BAA0B,CAAC;AAC3D,OAAO,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAEzE,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAGtD,OAAO,EAAE,aAAa,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,MAAM,sBAAsB,CAAC;AAGxG,MAAM,OAAgB,4BAA4B;IAClB;IAA/B,YAA+B,OAA0B;QAA1B,YAAO,GAAP,OAAO,CAAmB;IAAG,CAAC;IAE7D,OAAO,KAAU,CAAC;IAClB,CAAC,MAAM,CAAC,OAAO,CAAC;QACf,IAAI,CAAC,OAAO,EAAE,CAAC;IAChB,CAAC;IAOD,KAAK,CAAC,UAAU,CAAC,MAAe;QAC/B,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAC5C,OAAO,MAAM,IAAI,IAAI,CAAC;QACvB,CAAC;QAAC,OAAO,EAAE,EAAE,CAAC;YACb,IAAI,mBAAmB,CAAC,EAAE,CAAC;gBAAE,OAAO,KAAK,CAAC;YAE1C,MAAM,EAAE,CAAC;QACV,CAAC;IACF,CAAC;IAED,KAAK,CAAC,SAAS,CAAC,MAAe;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxD,IAAI,CAAC,OAAO,CAAC,OAAO;YAAE,OAAO,SAAS,CAAC;QACvC,IAAI,OAAO,CAAC,GAAG;YAAE,OAAO,OAAO,CAAC,GAAG,CAAC;QAEpC,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAE7C,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAC5B;YACC,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,eAAe,EAAE,IAAI,CAAC,UAAU,CAAC,eAAe;YAChD,SAAS,EAAE,YAAY,IAAI,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC,IAAI;YACxE,GAAG,EAAE,MAAM;SACX,EACD,MAAM,CACN,CAAC;IACH,CAAC;IAMS,UAAU,CACnB,OAAgB,EAChB,MAAc,EACd,MAAkB,EAClB,IAAY;QAEZ,OAAO;YACN,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,MAAM,EAAE;SACjC,CAAC;IACH,CAAC;IAED,KAAK,CAAC,WAAW,CAChB,MAAe,EACf,KAAiB,EACjB,MAAc,EACd,WAAkF,EAClF,OAAgG;;;YAEhG,MAAM,KAAK,kCAAG,sBAAsB,CAAC,GAAG,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,QAAA,CAAC;YAE7E,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,CAAC,YAAY,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC1G,OAAO,MAAM,CAAC;YACf,CAAC;YAAC,OAAO,EAAE,EAAE,CAAC;gBACb,IAAI,mBAAmB,CAAC,EAAE,CAAC,EAAE,CAAC;oBAC7B,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,wBAAwB,aAAa,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;oBAC5F,MAAM,EAAE,CAAC;gBACV,CAAC;gBAED,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,aAAa,aAAa,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,IAAI,GAAG,CAAC,CAAC;gBACjF,IAAI,EAAE,YAAY,OAAO;oBAAE,MAAM,EAAE,CAAC;gBAEpC,QAAQ,CAAC;gBACT,MAAM,IAAI,KAAK,CAAC,aAAa,aAAa,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,EAAE,CAAC,OAAO,EAAE,EAAE;oBAC7F,KAAK,EAAE,EAAE;iBACT,CAAC,CAAC;YACJ,CAAC;;;;;;;;;KACD;IAES,KAAK,CAAC,KAAK,CACpB,MAAe,EACf,KAAiB,EACjB,MAAc,EACd,QAA+E,EAC/E,YAA8D,EAC9D,MAAoB;QAEpB,IAAI,OAAO,GAAG,CAAC,CAAC;QAChB,IAAI,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,KAAK,CAAC;QAE3C,OAAO,IAAI,EAAE,CAAC;YACb,MAAM,OAAO,GAA0B;gBACtC,KAAK,EAAE,KAAK,CAAC,EAAE;gBACf,QAAQ,EAAE,MAAM,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC;gBACjD,MAAM,EAAE,KAAK;gBACb,qBAAqB,EAAE,KAAK,CAAC,SAAS,CAAC,MAAM;oBAC5C,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,YAAY,IAAI,QAAQ,EAAE,KAAK,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC1E,CAAC,CAAC,YAAY,EAAE,YAAY;gBAC7B,WAAW,EAAE,uBAAuB,CACnC,KAAK,EACL,YAAY,EAAE,WAAW,IAAI,KAAK,CAAC,WAAW,EAC9C,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAC/B;aACD,CAAC;YAEF,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACzE,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;gBACb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,cAAc,CAAC,CAAC;gBAC1F,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;oBAClB,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;oBACvC,OAAO,EAAE,CAAC;oBACV,SAAS;gBACV,CAAC;YACF,CAAC;YAED,MAAM,IAAI,GAA2B,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA2B,CAAC;YAClF,MAAM,MAAM,GAA6B;gBACxC,EAAE,EAAE,IAAI,CAAC,EAAE,IAAI,IAAI,EAAE;gBACrB,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE;gBAC3F,KAAK,EAAE,KAAK;gBACZ,KAAK,EAAE;oBACN,YAAY,EAAE,IAAI,CAAC,KAAK,EAAE,aAAa,IAAI,IAAI,CAAC,KAAK,EAAE,YAAY;oBACnE,gBAAgB,EAAE,IAAI,CAAC,KAAK,EAAE,iBAAiB,IAAI,IAAI,CAAC,KAAK,EAAE,aAAa;oBAC5E,WAAW,EAAE,IAAI,CAAC,KAAK,EAAE,YAAY;oBACrC,MAAM,EACL,IAAI,EAAE,KAAK,EAAE,EAAE,IAAI,IAAI;wBACtB,CAAC,CAAC;4BACA,IAAI,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI;4BACxB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK;4BAC1B,QAAQ,EAAE,IAAI,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,SAAS,CAAC;yBAC3C;wBACF,CAAC,CAAC,SAAS;iBACb;gBACD,MAAM,EAAE,SAAS;aACjB,CAAC;YACF,OAAO,MAAM,CAAC;QACf,CAAC;IACF,CAAC;IAES,KAAK,CAAC,kBAAkB,CACjC,GAAa,EACb,OAAgB,EAChB,KAAiB,EACjB,OAAe,EACf,cAAsB;QAEtB,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACxB,MAAM,IAAI,OAAO,CAChB,aAAa,CAAC,YAAY,EAC1B,IAAI,KAAK,CAAC,6DAA6D,KAAK,CAAC,EAAE,SAAS,CAAC,CACzF,CAAC;QACH,CAAC;QACD,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;YACxB,MAAM,IAAI,OAAO,CAChB,aAAa,CAAC,wBAAwB,EACtC,IAAI,KAAK,CACR,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,2EAA2E,CACvG,CACD,CAAC;QACH,CAAC;QAED,IAAI,IAAI,CAAC;QACT,IAAI,CAAC;YACJ,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8D,CAAC;QACxF,CAAC;QAAC,MAAM,CAAC,CAAA,CAAC;QAEV,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,CAAC;YACzB,IAAI,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAChB,CAAC;QACD,IAAI,IAAI,EAAE,KAAK,EAAE,IAAI,KAAK,yBAAyB,EAAE,CAAC;YACrD,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;gBACjB,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,cAAc,EAAE,wBAAwB,CAAC,cAAc,EAAE,OAAO,GAAG,CAAC,CAAC,EAAE,CAAC;YAC/F,CAAC;YAED,MAAM,IAAI,OAAO,CAChB,aAAa,CAAC,eAAe,EAC7B,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,KAAK,EAAE,OAAO,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC,CACpF,CAAC;QACH,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,KAAK,IAAI,EAAE,KAAK,EAAE,OAAO,IAAI,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;IAC5F,CAAC;IAES,KAAK,CAAC,SAAS,CACxB,MAAe,EACf,KAAiB,EACjB,MAAc,EACd,OAAe,EACf,MAA+B;QAE/B,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC/B,IAAI,CAAC,GAAG,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,aAAa,CAAC,MAAM,CAAC,qBAAqB,CAAC,CAAC;QAC/E,CAAC;QAED,IAAI,CAAC;YACJ,OAAO,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE;gBACpC,OAAO,EAAE,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,GAAG,CAAC;gBAC1D,MAAM,EAAE,MAAM;gBACd,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;gBAC7B,MAAM,EAAE,MAAM;aACd,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,EAAE,EAAE,CAAC;YACb,IAAI,EAAE,CAAC,IAAI,KAAK,YAAY;gBAAE,MAAM,IAAI,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAC9D,MAAM,EAAE,CAAC;QACV,CAAC;IACF,CAAC;CACD"}

package/dist/plus/ai/providers/openRouterProvider.d.ts

@@ -0,0 +1,17 @@
+import { openRouterProviderDescriptor as provider } from '../constants.js';
+import type { AIActionType, AIModel } from '../models/model.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+export declare class OpenRouterProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+    readonly id: "openrouter";
+    readonly name: string;
+    protected readonly descriptor: import("../models/model.js").AIProviderDescriptor<"openrouter">;
+    protected readonly config: {
+        keyUrl: string;
+        keyValidator: RegExp;
+    };
+    getModels(): Promise<readonly AIModel<typeof provider.id>[]>;
+    protected getUrl(_model: AIModel<typeof provider.id>): string;
+    protected getHeaders<TAction extends AIActionType>(_action: TAction, apiKey: string, _model: AIModel<typeof provider.id>, _url: string): Record<string, string>;
+    private getHeadersCore;
+}
+//# sourceMappingURL=openRouterProvider.d.ts.map

package/dist/plus/ai/providers/openRouterProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openRouterProvider.d.ts","sourceRoot":"","sources":["../../src/providers/openRouterProvider.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,4BAA4B,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3E,OAAO,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAIjF,qBAAa,kBAAmB,SAAQ,4BAA4B,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC;IACvF,QAAQ,CAAC,EAAE,eAAe;IAC1B,QAAQ,CAAC,IAAI,SAAiB;IAC9B,SAAS,CAAC,QAAQ,CAAC,UAAU,kEAAY;IACzC,SAAS,CAAC,QAAQ,CAAC,MAAM;;;MAGvB;IAEI,SAAS,IAAI,OAAO,CAAC,SAAS,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IA6ClE,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,GAAG,MAAM;cAI1C,UAAU,CAAC,OAAO,SAAS,YAAY,EACzD,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,EACnC,IAAI,EAAE,MAAM,GACV,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;IAIzB,OAAO,CAAC,cAAc;CAStB"}

package/dist/plus/ai/providers/openRouterProvider.js

@@ -0,0 +1,57 @@
+import { isCancellationError } from '../../../utils/cancellation.js';
+import { openRouterProviderDescriptor as provider } from '../constants.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+export class OpenRouterProvider extends OpenAICompatibleProviderBase {
+    id = provider.id;
+    name = provider.name;
+    descriptor = provider;
+    config = {
+        keyUrl: 'https://openrouter.ai/keys',
+        keyValidator: /(?:sk-)?\w{24,128}/,
+    };
+    async getModels() {
+        let apiKey;
+        try {
+            apiKey = await this.getApiKey(true);
+        }
+        catch (ex) {
+            if (isCancellationError(ex))
+                return [];
+            throw ex;
+        }
+        if (!apiKey)
+            return [];
+        const url = 'https://openrouter.ai/api/v1/models';
+        const rsp = await this.context.fetch(url, { headers: this.getHeadersCore(apiKey) });
+        if (!rsp.ok) {
+            throw new Error(`Getting models (${url}) failed: ${rsp.status} (${rsp.statusText})`);
+        }
+        const results = (await rsp.json());
+        return results.data.map(m => ({
+            id: m.id,
+            name: m.name,
+            maxTokens: {
+                input: m.context_length,
+                output: m.top_provider?.max_completion_tokens ?? Math.floor(m.context_length / 2),
+            },
+            provider: provider,
+            temperature: null,
+        }));
+    }
+    getUrl(_model) {
+        return 'https://openrouter.ai/api/v1/chat/completions';
+    }
+    getHeaders(_action, apiKey, _model, _url) {
+        return this.getHeadersCore(apiKey);
+    }
+    getHeadersCore(apiKey) {
+        return {
+            Accept: 'application/json',
+            'Content-Type': 'application/json',
+            'HTTP-Referer': 'https://gitkraken.com/',
+            'X-Title': 'GitKraken',
+            Authorization: `Bearer ${apiKey}`,
+        };
+    }
+}
+//# sourceMappingURL=openRouterProvider.js.map

package/dist/plus/ai/providers/openRouterProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openRouterProvider.js","sourceRoot":"","sources":["../../src/providers/openRouterProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,gCAAgC,CAAC;AACrE,OAAO,EAAE,4BAA4B,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAE3E,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAIjF,MAAM,OAAO,kBAAmB,SAAQ,4BAAgD;IAC9E,EAAE,GAAG,QAAQ,CAAC,EAAE,CAAC;IACjB,IAAI,GAAG,QAAQ,CAAC,IAAI,CAAC;IACX,UAAU,GAAG,QAAQ,CAAC;IACtB,MAAM,GAAG;QAC3B,MAAM,EAAE,4BAA4B;QACpC,YAAY,EAAE,oBAAoB;KAClC,CAAC;IAEF,KAAK,CAAC,SAAS;QACd,IAAI,MAA0B,CAAC;QAC/B,IAAI,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,EAAE,EAAE,CAAC;YACb,IAAI,mBAAmB,CAAC,EAAE,CAAC;gBAAE,OAAO,EAAE,CAAC;YAEvC,MAAM,EAAE,CAAC;QACV,CAAC;QAED,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC;QAEvB,MAAM,GAAG,GAAG,qCAAqC,CAAC;QAClD,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QACpF,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,mBAAmB,GAAG,aAAa,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,GAAG,CAAC,CAAC;QACtF,CAAC;QAaD,MAAM,OAAO,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAAmB,CAAC;QACrD,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,CACtB,CAAC,CAAC,EAAE,CACH,CAAC;YACA,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,SAAS,EAAE;gBACV,KAAK,EAAE,CAAC,CAAC,cAAc;gBACvB,MAAM,EAAE,CAAC,CAAC,YAAY,EAAE,qBAAqB,IAAI,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC;aACjF;YACD,QAAQ,EAAE,QAAQ;YAClB,WAAW,EAAE,IAAI;SACjB,CAA2B,CAC7B,CAAC;IACH,CAAC;IAES,MAAM,CAAC,MAAmC;QACnD,OAAO,+CAA+C,CAAC;IACxD,CAAC;IAEkB,UAAU,CAC5B,OAAgB,EAChB,MAAc,EACd,MAAmC,EACnC,IAAY;QAEZ,OAAO,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;IACpC,CAAC;IAEO,cAAc,CAAC,MAAc;QACpC,OAAO;YACN,MAAM,EAAE,kBAAkB;YAC1B,cAAc,EAAE,kBAAkB;YAClC,cAAc,EAAE,wBAAwB;YACxC,SAAS,EAAE,WAAW;YACtB,aAAa,EAAE,UAAU,MAAM,EAAE;SACjC,CAAC;IACH,CAAC;CACD"}

package/dist/plus/ai/providers/openaiProvider.d.ts

@@ -0,0 +1,16 @@
+import { openAIProviderDescriptor as provider } from '../constants.js';
+import type { AIActionType, AIModel } from '../models/model.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+export declare class OpenAIProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+    readonly id: "openai";
+    readonly name: string;
+    protected readonly descriptor: import("../models/model.js").AIProviderDescriptor<"openai">;
+    protected readonly config: {
+        keyUrl: string;
+        keyValidator: RegExp;
+    };
+    getModels(): Promise<readonly AIModel<typeof provider.id>[]>;
+    protected getUrl(_model: AIModel<typeof provider.id>): string | undefined;
+    protected getHeaders<TAction extends AIActionType>(action: TAction, apiKey: string, model: AIModel<typeof provider.id>, url: string): Record<string, string> | Promise<Record<string, string>>;
+}
+//# sourceMappingURL=openaiProvider.d.ts.map

package/dist/plus/ai/providers/openaiProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openaiProvider.d.ts","sourceRoot":"","sources":["../../src/providers/openaiProvider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,wBAAwB,IAAI,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AACvE,OAAO,KAAK,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,oBAAoB,CAAC;AAGhE,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AAKjF,qBAAa,cAAe,SAAQ,4BAA4B,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC;IACnF,QAAQ,CAAC,EAAE,WAAe;IAC1B,QAAQ,CAAC,IAAI,SAAiB;IAC9B,SAAS,CAAC,QAAQ,CAAC,UAAU,8DAAY;IACzC,SAAS,CAAC,QAAQ,CAAC,MAAM;;;MAGvB;IAEF,SAAS,IAAI,OAAO,CAAC,SAAS,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,EAAE,CAAC;IAI5D,SAAS,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,GAAG,MAAM,GAAG,SAAS;cAOtD,UAAU,CAAC,OAAO,SAAS,YAAY,EACzD,MAAM,EAAE,OAAO,EACf,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,OAAO,CAAC,OAAO,QAAQ,CAAC,EAAE,CAAC,EAClC,GAAG,EAAE,MAAM,GACT,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;CAW3D"}