@gitkraken/core-gitlens 0.1.0

package/src/plus/ai/providers/anthropicProvider.ts

@@ -0,0 +1,308 @@
+import { anthropicProviderDescriptor as provider } from '../constants.js';
+import { AIError, AIErrorReason } from '../errors.js';
+import type { AIActionType, AIModel } from '../models/model.js';
+import { getReducedMaxInputTokens } from '../utils/ai.utils.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+
+type AnthropicModel = AIModel<typeof provider.id>;
+const models: AnthropicModel[] = [
+	{
+		id: 'claude-haiku-4-5',
+		name: 'Claude Haiku 4.5',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		default: true,
+	},
+	{
+		id: 'claude-haiku-4-5-20251001',
+		name: 'Claude Haiku 4.5',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-sonnet-4-6',
+		name: 'Claude Sonnet 4.6',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+	},
+	{
+		id: 'claude-opus-4-6',
+		name: 'Claude Opus 4.6',
+		maxTokens: { input: 204800, output: 128000 },
+		provider: provider,
+	},
+	{
+		id: 'claude-sonnet-4-5',
+		name: 'Claude Sonnet 4.5',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+	},
+	{
+		id: 'claude-sonnet-4-5-20250929',
+		name: 'Claude Sonnet 4.5',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-opus-4-5',
+		name: 'Claude Opus 4.5',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+	},
+	{
+		id: 'claude-opus-4-5-20251101',
+		name: 'Claude Opus 4.5',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-opus-4-1',
+		name: 'Claude Opus 4.1',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-opus-4-1-20250805',
+		name: 'Claude Opus 4.1',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-opus-4-0',
+		name: 'Claude Opus 4',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-opus-4-20250514',
+		name: 'Claude Opus 4',
+		maxTokens: { input: 204800, output: 32000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-sonnet-4-0',
+		name: 'Claude Sonnet 4',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+	},
+	{
+		id: 'claude-sonnet-4-20250514',
+		name: 'Claude Sonnet 4',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-7-sonnet-latest',
+		name: 'Claude Sonnet 3.7',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-7-sonnet-20250219',
+		name: 'Claude Sonnet 3.7',
+		maxTokens: { input: 204800, output: 64000 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-5-sonnet-latest',
+		name: 'Claude Sonnet 3.5',
+		maxTokens: { input: 204800, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-5-sonnet-20241022',
+		name: 'Claude Sonnet 3.5',
+		maxTokens: { input: 204800, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-5-sonnet-20240620',
+		name: 'Claude Sonnet 3.5',
+		maxTokens: { input: 204800, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-5-haiku-latest',
+		name: 'Claude Haiku 3.5',
+		maxTokens: { input: 204800, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-5-haiku-20241022',
+		name: 'Claude Haiku 3.5',
+		maxTokens: { input: 204800, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-opus-latest',
+		name: 'Claude Opus 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-opus-20240229',
+		name: 'Claude Opus 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-sonnet-latest',
+		name: 'Claude Sonnet 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-sonnet-20240229',
+		name: 'Claude Sonnet 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-haiku-latest',
+		name: 'Claude Haiku 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-3-haiku-20240307',
+		name: 'Claude Haiku 3',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'claude-2.1',
+		name: 'Claude 2.1',
+		maxTokens: { input: 204800, output: 4096 },
+		provider: provider,
+		hidden: true,
+	},
+];
+
+export class AnthropicProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+	readonly id = provider.id;
+	readonly name = provider.name;
+	protected readonly descriptor = provider;
+	protected readonly config = {
+		keyUrl: 'https://console.anthropic.com/account/keys',
+		keyValidator: /(?:sk-)?[a-zA-Z0-9-_]{32,}/,
+	};
+
+	getModels(): Promise<readonly AIModel<typeof provider.id>[]> {
+		return Promise.resolve(models);
+	}
+
+	protected getUrl(_model: AIModel<typeof provider.id>): string {
+		return 'https://api.anthropic.com/v1/messages';
+	}
+
+	protected override getHeaders<TAction extends AIActionType>(
+		_action: TAction,
+		apiKey: string,
+		_model: AIModel<typeof provider.id>,
+		_url: string,
+	): Record<string, string> {
+		return {
+			Accept: 'application/json',
+			'Content-Type': 'application/json',
+			Authorization: `Bearer ${apiKey}`,
+			'x-api-key': apiKey,
+			'anthropic-version': '2023-06-01',
+		};
+	}
+
+	protected override fetchCore<TAction extends AIActionType>(
+		action: TAction,
+		model: AIModel<typeof provider.id>,
+		apiKey: string,
+		request: object,
+		signal: AbortSignal | undefined,
+	): Promise<Response> {
+		if ('max_completion_tokens' in request) {
+			const { max_completion_tokens: max, ...rest } = request;
+			request = max ? { max_tokens: max, ...rest } : rest;
+		}
+		return super.fetchCore(action, model, apiKey, request, signal);
+	}
+
+	protected override async handleFetchFailure<TAction extends AIActionType>(
+		rsp: Response,
+		action: TAction,
+		model: AIModel<typeof provider.id>,
+		retries: number,
+		maxInputTokens: number,
+	): Promise<{ retry: true; maxInputTokens: number }> {
+		if (rsp.status !== 404 && rsp.status !== 429) {
+			let json;
+			try {
+				json = (await rsp.json()) as AnthropicError | undefined;
+			} catch {}
+
+			debugger;
+
+			if (json?.error?.type === 'invalid_request_error') {
+				if (json?.error?.message?.includes('prompt is too long')) {
+					if (retries < 3) {
+						// Extract actual token count from error to calculate smarter reduction
+						const match = /prompt is too long: (\d+) tokens/.exec(json?.error?.message);
+						const estimatedTokens = match?.[1] != null ? parseInt(match[1], 10) : undefined;
+
+						return {
+							retry: true,
+							maxInputTokens: getReducedMaxInputTokens(maxInputTokens, retries + 1, estimatedTokens),
+						};
+					}
+
+					throw new AIError(
+						AIErrorReason.RequestTooLarge,
+						new Error(`(${this.name}) ${rsp.status}: ${json?.error?.message || rsp.statusText}`),
+					);
+				}
+
+				if (json?.error?.message?.includes('balance is too low')) {
+					throw new AIError(
+						AIErrorReason.RateLimitOrFundsExceeded,
+						new Error(`(${this.name}) ${rsp.status}: ${json?.error?.message || rsp.statusText}`),
+					);
+				}
+			}
+		}
+
+		return super.handleFetchFailure(rsp, action, model, retries, maxInputTokens);
+	}
+}
+
+interface AnthropicError {
+	type: 'error';
+	error: {
+		type:
+			| 'invalid_request_error'
+			| 'authentication_error'
+			| 'permission_error'
+			| 'not_found_error'
+			| 'rate_limit_error'
+			| 'api_error'
+			| 'overloaded_error';
+		message: string;
+	};
+}

package/src/plus/ai/providers/azureProvider.ts

@@ -0,0 +1,74 @@
+import { azureProviderDescriptor as provider } from '../constants.js';
+import type { AIActionType, AIModel } from '../models/model.js';
+import { openAIModels } from '../models/model.js';
+import { isAzureUrl } from '../utils/ai.utils.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+
+type AzureModel = AIModel<typeof provider.id>;
+const models: AzureModel[] = openAIModels(provider);
+
+export class AzureProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+	readonly id = provider.id;
+	readonly name = provider.name;
+	protected readonly descriptor = provider;
+	protected readonly config = {
+		keyUrl: undefined,
+		keyValidator: /(?:sk-)?[a-zA-Z0-9]{32,}/,
+	};
+
+	getModels(): Promise<readonly AIModel<typeof provider.id>[]> {
+		return Promise.resolve(models);
+	}
+
+	protected getUrl(_model?: AIModel<typeof provider.id>): string | undefined {
+		const cfg = this.context.getProviderConfig(this.id);
+		if (!cfg.enabled) return undefined;
+
+		return cfg.url || undefined;
+	}
+
+	private async getOrPromptBaseUrl(silent: boolean, hasApiKey: boolean): Promise<string | undefined> {
+		const cfg = this.context.getProviderConfig(this.id);
+		if (!cfg.enabled) return undefined;
+
+		if (cfg.url) return cfg.url;
+
+		const url = this.getUrl();
+		if (silent || (url != null && hasApiKey)) return url;
+
+		return (
+			(await this.context.getOrPromptUrl(
+				this.id,
+				{
+					currentUrl: url,
+					title: 'Connect to Azure OpenAI Provider',
+					placeholder: `Please enter your provider's URL to use this feature`,
+					validator: u => (isAzureUrl(u) ? undefined : 'Please enter a valid Azure OpenAI URL'),
+				},
+				silent,
+			)) ?? url
+		);
+	}
+
+	override async configured(silent: boolean): Promise<boolean> {
+		const hasApiKey = await super.configured(true);
+
+		const url = await this.getOrPromptBaseUrl(silent, hasApiKey);
+		if (!url) return false;
+
+		return silent ? hasApiKey : super.configured(silent);
+	}
+
+	protected override getHeaders<TAction extends AIActionType>(
+		_action: TAction,
+		apiKey: string,
+		_model: AIModel<typeof provider.id>,
+		_url: string,
+	): Record<string, string> | Promise<Record<string, string>> {
+		return {
+			Accept: 'application/json',
+			'Content-Type': 'application/json',
+			'api-key': apiKey,
+		};
+	}
+}

package/src/plus/ai/providers/context.ts

@@ -0,0 +1,29 @@
+export interface AIProviderContext {
+	defaultTemperature?: number;
+
+	fetch(url: string | URL, init?: RequestInit): Promise<Response>;
+
+	getApiKey(
+		config: {
+			id: string;
+			name: string;
+			requiresAccount: boolean;
+			validator?: (value: string) => boolean;
+			url?: string;
+		},
+		silent: boolean,
+	): Promise<string | undefined>;
+
+	getProviderConfig(type: string): { enabled: boolean; key?: string; url?: string };
+
+	getOrPromptUrl(
+		providerId: string,
+		options: {
+			currentUrl: string | undefined;
+			title: string;
+			placeholder: string;
+			validator?: (url: string) => string | undefined | Promise<string | undefined>;
+		},
+		silent: boolean,
+	): Promise<string | undefined>;
+}

package/src/plus/ai/providers/deepSeekProvider.ts

@@ -0,0 +1,40 @@
+import { deepSeekProviderDescriptor as provider } from '../constants.js';
+import type { AIModel } from '../models/model.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+
+type DeepSeekModel = AIModel<typeof provider.id>;
+const models: DeepSeekModel[] = [
+	{
+		id: 'deepseek-chat',
+		name: 'DeepSeek-V3',
+		maxTokens: { input: 65536, output: 8192 },
+		provider: provider,
+		default: true,
+		temperature: 0.0, // Recommended for Coding/Math
+	},
+	{
+		id: 'deepseek-reasoner',
+		name: 'DeepSeek-R1',
+		maxTokens: { input: 65536, output: 8192 },
+		provider: provider,
+		temperature: 0.0, // Recommended for Coding/Math
+	},
+];
+
+export class DeepSeekProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+	readonly id = provider.id;
+	readonly name = provider.name;
+	protected readonly descriptor = provider;
+	protected readonly config = {
+		keyUrl: 'https://platform.deepseek.com/api_keys',
+		keyValidator: /(?:sk-)?[a-zA-Z0-9]{32,}/,
+	};
+
+	getModels(): Promise<readonly AIModel<typeof provider.id>[]> {
+		return Promise.resolve(models);
+	}
+
+	protected getUrl(_model: AIModel<typeof provider.id>): string {
+		return 'https://api.deepseek.com/v1/chat/completions';
+	}
+}

package/src/plus/ai/providers/geminiProvider.ts

@@ -0,0 +1,207 @@
+import { geminiProviderDescriptor as provider } from '../constants.js';
+import type { AIActionType, AIModel } from '../models/model.js';
+import { OpenAICompatibleProviderBase } from './openAICompatibleProviderBase.js';
+
+type GeminiModel = AIModel<typeof provider.id>;
+const models: GeminiModel[] = [
+	{
+		id: 'gemini-3.1-pro-preview',
+		name: 'Gemini 3.1 Pro Preview',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-3.1-flash-lite-preview',
+		name: 'Gemini 3.1 Flash-Lite Preview',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-3-pro-preview',
+		name: 'Gemini 3 Pro Preview',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-3-flash-preview',
+		name: 'Gemini 3 Flash Preview',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		default: true,
+	},
+	{
+		id: 'gemini-2.5-pro',
+		name: 'Gemini 2.5 Pro',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-2.5-flash',
+		name: 'Gemini 2.5 Flash',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-2.5-flash-lite',
+		name: 'Gemini 2.5 Flash-Lite',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+	},
+	{
+		id: 'gemini-2.5-flash-preview-05-20',
+		name: 'Gemini 2.5 Flash (Preview)',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.5-flash-preview-04-17',
+		name: 'Gemini 2.5 Flash (Preview)',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.5-pro-preview-06-05',
+		name: 'Gemini 2.5 Pro (Preview)',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.5-pro-preview-03-25',
+		name: 'Gemini 2.5 Pro (Preview)',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.5-pro-exp-03-25',
+		name: 'Gemini 2.5 Pro (Experimental)',
+		maxTokens: { input: 1048576, output: 65536 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash',
+		name: 'Gemini 2.0 Flash',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-001',
+		name: 'Gemini 2.0 Flash',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-lite',
+		name: 'Gemini 2.0 Flash-Lite',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-lite-001',
+		name: 'Gemini 2.0 Flash-Lite',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-lite-preview-02-05',
+		name: 'Gemini 2.0 Flash-Lite (Preview)',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-pro-exp-02-05',
+		name: 'Gemini 2.0 Pro (Experimental)',
+		maxTokens: { input: 2097152, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-thinking-exp-01-21',
+		name: 'Gemini 2.0 Flash Thinking (Experimental)',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-2.0-flash-exp',
+		name: 'Gemini 2.0 Flash (Experimental)',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-exp-1206',
+		name: 'Gemini Experimental 1206',
+		maxTokens: { input: 2097152, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-exp-1121',
+		name: 'Gemini Experimental 1121',
+		maxTokens: { input: 2097152, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-1.5-pro',
+		name: 'Gemini 1.5 Pro',
+		maxTokens: { input: 2097152, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-1.5-flash',
+		name: 'Gemini 1.5 Flash',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+	{
+		id: 'gemini-1.5-flash-8b',
+		name: 'Gemini 1.5 Flash 8B',
+		maxTokens: { input: 1048576, output: 8192 },
+		provider: provider,
+		hidden: true,
+	},
+];
+
+export class GeminiProvider extends OpenAICompatibleProviderBase<typeof provider.id> {
+	readonly id = provider.id;
+	readonly name = provider.name;
+	protected readonly descriptor = provider;
+	protected readonly config = {
+		keyUrl: 'https://aistudio.google.com/app/apikey',
+	};
+
+	getModels(): Promise<readonly AIModel<typeof provider.id>[]> {
+		return Promise.resolve(models);
+	}
+
+	protected getUrl(_model: AIModel<typeof provider.id>): string {
+		return `https://generativelanguage.googleapis.com/v1beta/chat/completions`;
+	}
+
+	protected override fetchCore(
+		action: AIActionType,
+		model: AIModel<typeof provider.id>,
+		apiKey: string,
+		request: object,
+		signal: AbortSignal | undefined,
+	): Promise<Response> {
+		if ('max_completion_tokens' in request) {
+			const { max_completion_tokens: max, ...rest } = request;
+			request = max ? { max_tokens: max, ...rest } : rest;
+		}
+		return super.fetchCore(action, model, apiKey, request, signal);
+	}
+}