@aryee337/aery-ai 0.1.148 → 0.2.10

package/src/providers/error-message.ts

@@ -0,0 +1,21 @@
+import type { Api, Model } from "../types";
+
+export function createProviderErrorMessage(model: Model<Api>, err: unknown) {
+	return {
+		role: "assistant" as const,
+		content: [{ type: "text" as const, text: err instanceof Error ? err.message : String(err) }],
+		api: model.api,
+		provider: model.provider,
+		model: model.id,
+		usage: {
+			input: 0,
+			output: 0,
+			cacheRead: 0,
+			cacheWrite: 0,
+			totalTokens: 0,
+			cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
+		},
+		stopReason: "error" as const,
+		timestamp: Date.now(),
+	};
+}

package/src/providers/github-copilot-headers.ts

@@ -0,0 +1,140 @@
+import type { Message } from "../types";
+import { getGitHubCopilotBaseUrl, parseGitHubCopilotApiKey } from "../utils/oauth/github-copilot";
+/**
+ * Infer whether the current request to Copilot is user-initiated or agent-initiated.
+ * Accepts `unknown[]` because providers may pass pre-converted message shapes.
+ */
+export type CopilotInitiator = "user" | "agent";
+export type CopilotPremiumRequests = number;
+export type CopilotDynamicHeaders = {
+	headers: Record<string, string>;
+	initiator: CopilotInitiator;
+	premiumRequests: CopilotPremiumRequests;
+};
+export function resolveGitHubCopilotBaseUrl(
+	baseUrl: string | undefined,
+	apiKey: string | undefined,
+): string | undefined {
+	if (!apiKey) return baseUrl;
+	const { enterpriseUrl } = parseGitHubCopilotApiKey(apiKey);
+	if (!enterpriseUrl) return baseUrl;
+	if (baseUrl && !baseUrl.includes("githubcopilot.com")) return baseUrl;
+	return getGitHubCopilotBaseUrl(enterpriseUrl);
+}
+export function inferCopilotInitiator(messages: unknown[]): CopilotInitiator {
+	if (messages.length === 0) return "user";
+
+	const last = messages[messages.length - 1] as Record<string, unknown>;
+	const attribution = last.attribution;
+	if (typeof attribution === "string") {
+		const normalizedAttribution = attribution.trim().toLowerCase();
+		if (normalizedAttribution === "user" || normalizedAttribution === "agent") {
+			return normalizedAttribution;
+		}
+	}
+
+	const role = last.role as string | undefined;
+	if (!role) return "user";
+
+	if (role !== "user") return "agent";
+
+	// Check if last content block is a tool_result (Anthropic-converted shape)
+	const content = last.content;
+	if (Array.isArray(content) && content.length > 0) {
+		const lastBlock = content[content.length - 1] as Record<string, unknown>;
+		if (lastBlock.type === "tool_result") {
+			return "agent";
+		}
+	}
+
+	return "user";
+}
+
+/** Check whether any message in the conversation contains image content. */
+export function hasCopilotVisionInput(messages: Message[]): boolean {
+	return messages.some(msg => {
+		if (msg.role === "user" && Array.isArray(msg.content)) {
+			return msg.content.some(c => c.type === "image");
+		}
+		if (msg.role === "toolResult" && Array.isArray(msg.content)) {
+			return msg.content.some(c => c.type === "image");
+		}
+		return false;
+	});
+}
+
+/**
+ * Resolve an explicitly configured Copilot initiator header, if present.
+ * Handles case-insensitive X-Initiator keys and returns the last valid value.
+ */
+export function getCopilotInitiatorOverride(headers: Record<string, string> | undefined): CopilotInitiator | undefined {
+	if (!headers) return undefined;
+
+	let override: CopilotInitiator | undefined;
+	for (const [key, value] of Object.entries(headers)) {
+		if (key.toLowerCase() !== "x-initiator") continue;
+		const normalized = value.trim().toLowerCase();
+		if (normalized === "user" || normalized === "agent") {
+			override = normalized;
+		}
+	}
+
+	return override;
+}
+
+export type CopilotPlanTier = "free" | "paid";
+
+function normalizeCopilotPlanTier(planTier: string | undefined): CopilotPlanTier {
+	if (planTier === "paid") return "paid";
+	return "free";
+}
+export function getCopilotPremiumMultiplier(premiumMultiplier: number | undefined, planTier?: string): number {
+	const normalizedMultiplier = premiumMultiplier ?? 1;
+	if (normalizeCopilotPlanTier(planTier) === "free" && normalizedMultiplier === 0) {
+		return 1;
+	}
+	return normalizedMultiplier;
+}
+
+export function getCopilotPremiumRequests(params: {
+	initiator: CopilotInitiator;
+	premiumMultiplier?: number;
+	planTier?: string;
+}): CopilotPremiumRequests {
+	if (params.initiator === "agent") return 0;
+	return getCopilotPremiumMultiplier(params.premiumMultiplier, params.planTier);
+}
+
+/**
+ * Build dynamic Copilot headers that vary per-request.
+ * Static headers (User-Agent, Editor-Version, etc.) come from model.headers.
+ */
+export function buildCopilotDynamicHeaders(params: {
+	messages: unknown[];
+	hasImages: boolean;
+	premiumMultiplier?: number;
+	headers?: Record<string, string>;
+	initiatorOverride?: CopilotInitiator;
+	planTier?: string;
+}): CopilotDynamicHeaders {
+	const initiator =
+		params.initiatorOverride ?? getCopilotInitiatorOverride(params.headers) ?? inferCopilotInitiator(params.messages);
+	const headers: Record<string, string> = {
+		"X-Initiator": initiator,
+		"Openai-Intent": "conversation-edits",
+	};
+
+	if (params.hasImages) {
+		headers["Copilot-Vision-Request"] = "true";
+	}
+
+	return {
+		headers,
+		initiator,
+		premiumRequests: getCopilotPremiumRequests({
+			initiator,
+			premiumMultiplier: params.premiumMultiplier,
+			planTier: params.planTier,
+		}),
+	};
+}

package/src/providers/gitlab-duo.ts

@@ -0,0 +1,372 @@
+import { ANTHROPIC_THINKING, mapAnthropicToolChoice } from "../stream";
+import type { Api, Context, FetchImpl, Model, SimpleStreamOptions } from "../types";
+import { AssistantMessageEventStream } from "../utils/event-stream";
+import { createProviderErrorMessage } from "./error-message";
+import type { OpenAICompletionsOptions } from "./openai-completions";
+import type { OpenAIResponsesOptions } from "./openai-responses";
+import { streamAnthropic, streamOpenAICompletions, streamOpenAIResponses } from "./register-builtins";
+
+const GITLAB_COM_URL = "https://gitlab.com";
+const AI_GATEWAY_URL = "https://cloud.gitlab.com";
+const ANTHROPIC_PROXY_URL = `${AI_GATEWAY_URL}/ai/v1/proxy/anthropic/`;
+const OPENAI_PROXY_URL = `${AI_GATEWAY_URL}/ai/v1/proxy/openai/v1`;
+const DIRECT_ACCESS_TTL_MS = 25 * 60 * 1000;
+
+type GitLabProvider = "anthropic" | "openai";
+type GitLabOpenAIApiType = "chat" | "responses";
+
+export type GitLabModelMapping = {
+	provider: GitLabProvider;
+	model: string;
+	openaiApiType?: GitLabOpenAIApiType;
+	name: string;
+	reasoning: boolean;
+	input: ("text" | "image")[];
+	cost: { input: number; output: number; cacheRead: number; cacheWrite: number };
+	contextWindow: number;
+	maxTokens: number;
+};
+
+export const MODEL_MAPPINGS: Record<string, GitLabModelMapping> = {
+	"duo-chat-opus-4-6": {
+		provider: "anthropic",
+		model: "claude-opus-4-6",
+		name: "Duo Chat Opus 4.6",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 200000,
+		maxTokens: 64000,
+	},
+	"duo-chat-sonnet-4-6": {
+		provider: "anthropic",
+		model: "claude-sonnet-4-6",
+		name: "Duo Chat Sonnet 4.6",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 200000,
+		maxTokens: 64000,
+	},
+	"duo-chat-opus-4-5": {
+		provider: "anthropic",
+		model: "claude-opus-4-5-20251101",
+		name: "Duo Chat Opus 4.5",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 15, output: 75, cacheRead: 1.5, cacheWrite: 18.75 },
+		contextWindow: 200000,
+		maxTokens: 64000,
+	},
+	"duo-chat-sonnet-4-5": {
+		provider: "anthropic",
+		model: "claude-sonnet-4-5-20250929",
+		name: "Duo Chat Sonnet 4.5",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 3, output: 15, cacheRead: 0.3, cacheWrite: 3.75 },
+		contextWindow: 200000,
+		maxTokens: 64000,
+	},
+	"duo-chat-haiku-4-5": {
+		provider: "anthropic",
+		model: "claude-haiku-4-5-20251001",
+		name: "Duo Chat Haiku 4.5",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 1, output: 5, cacheRead: 0.1, cacheWrite: 1.25 },
+		contextWindow: 200000,
+		maxTokens: 64000,
+	},
+	"duo-chat-gpt-5-1": {
+		provider: "openai",
+		model: "gpt-5.1-2025-11-13",
+		openaiApiType: "chat",
+		name: "Duo Chat GPT-5.1",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 2.5, output: 10, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+	"duo-chat-gpt-5-2": {
+		provider: "openai",
+		model: "gpt-5.2-2025-12-11",
+		openaiApiType: "chat",
+		name: "Duo Chat GPT-5.2",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+	"duo-chat-gpt-5-mini": {
+		provider: "openai",
+		model: "gpt-5-mini-2025-08-07",
+		openaiApiType: "chat",
+		name: "Duo Chat GPT-5 Mini",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0.15, output: 0.6, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 128000,
+		maxTokens: 16384,
+	},
+	"duo-chat-gpt-5-codex": {
+		provider: "openai",
+		model: "gpt-5-codex",
+		openaiApiType: "responses",
+		name: "Duo Chat GPT-5 Codex",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 2.5, output: 10, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 272000,
+		maxTokens: 128000,
+	},
+	"duo-chat-gpt-5-2-codex": {
+		provider: "openai",
+		model: "gpt-5.2-codex",
+		openaiApiType: "responses",
+		name: "Duo Chat GPT-5.2 Codex",
+		reasoning: true,
+		input: ["text", "image"],
+		cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+		contextWindow: 272000,
+		maxTokens: 128000,
+	},
+};
+
+export function getModelMapping(modelId: string): GitLabModelMapping | undefined {
+	const direct = MODEL_MAPPINGS[modelId];
+	if (direct) return direct;
+
+	// Support canonical model IDs (e.g. "gpt-5-codex", "claude-sonnet-4-5-20250929")
+	// in addition to Duo aliases (e.g. "duo-chat-gpt-5-codex").
+	return Object.values(MODEL_MAPPINGS).find(mapping => mapping.model === modelId);
+}
+
+export function getGitLabDuoModels(): Model<Api>[] {
+	return Object.entries(MODEL_MAPPINGS).map(([id, mapping]) => ({
+		id,
+		name: mapping.name,
+		api:
+			mapping.provider === "anthropic"
+				? "anthropic-messages"
+				: mapping.openaiApiType === "responses"
+					? "openai-responses"
+					: "openai-completions",
+		provider: "gitlab-duo",
+		baseUrl: mapping.provider === "anthropic" ? ANTHROPIC_PROXY_URL : OPENAI_PROXY_URL,
+		reasoning: mapping.reasoning,
+		input: [...mapping.input],
+		cost: { ...mapping.cost },
+		contextWindow: mapping.contextWindow,
+		maxTokens: mapping.maxTokens,
+	}));
+}
+
+interface DirectAccessToken {
+	token: string;
+	headers: Record<string, string>;
+	expiresAt: number;
+}
+
+const directAccessCache = new Map<string, DirectAccessToken>();
+
+async function getDirectAccessToken(
+	gitlabAccessToken: string,
+	fetchImpl: FetchImpl = fetch,
+): Promise<DirectAccessToken> {
+	const cached = directAccessCache.get(gitlabAccessToken);
+	if (cached && cached.expiresAt > Date.now()) {
+		return cached;
+	}
+
+	const response = await fetchImpl(`${GITLAB_COM_URL}/api/v4/ai/third_party_agents/direct_access`, {
+		method: "POST",
+		headers: {
+			Authorization: `Bearer ${gitlabAccessToken}`,
+			"Content-Type": "application/json",
+		},
+		body: JSON.stringify({
+			feature_flags: { DuoAgentPlatformNext: true },
+		}),
+	});
+
+	if (!response.ok) {
+		const detail = await response.text();
+		if (response.status === 403) {
+			throw new Error(`GitLab Duo access denied. Ensure Duo is enabled for this account. ${detail}`);
+		}
+		throw new Error(`Failed to get GitLab Duo direct access token: ${response.status} ${detail}`);
+	}
+
+	const payload = (await response.json()) as { token?: string; headers?: Record<string, string> };
+	if (!payload.token || typeof payload.token !== "string") {
+		throw new Error("GitLab Duo direct access response missing token");
+	}
+	if (!payload.headers || typeof payload.headers !== "object") {
+		throw new Error("GitLab Duo direct access response missing headers");
+	}
+
+	const token: DirectAccessToken = {
+		token: payload.token,
+		headers: payload.headers,
+		expiresAt: Date.now() + DIRECT_ACCESS_TTL_MS,
+	};
+	directAccessCache.set(gitlabAccessToken, token);
+	return token;
+}
+
+export function clearGitLabDuoDirectAccessCache(): void {
+	directAccessCache.clear();
+}
+
+export function isGitLabDuoModel(model: Model<Api>): boolean {
+	return model.provider === "gitlab-duo";
+}
+
+export function streamGitLabDuo(
+	model: Model<Api>,
+	context: Context,
+	options?: SimpleStreamOptions,
+): AssistantMessageEventStream {
+	const stream = new AssistantMessageEventStream();
+
+	(async () => {
+		try {
+			if (!options?.apiKey) {
+				throw new Error("Missing GitLab access token. Run /login gitlab-duo or set GITLAB_TOKEN.");
+			}
+
+			const mapping = getModelMapping(model.id);
+			if (!mapping) {
+				throw new Error(`Unsupported GitLab Duo model: ${model.id}`);
+			}
+
+			const directAccess = await getDirectAccessToken(options.apiKey, options.fetch);
+			const headers = {
+				...directAccess.headers,
+				...options.headers,
+			};
+
+			const reasoningEffort = options.reasoning;
+
+			const inner =
+				mapping.provider === "anthropic"
+					? streamAnthropic(
+							{
+								...model,
+								id: mapping.model,
+								api: "anthropic-messages",
+								baseUrl: ANTHROPIC_PROXY_URL,
+							} as Model<"anthropic-messages">,
+							context,
+							{
+								apiKey: directAccess.token,
+								isOAuth: true,
+								temperature: options.temperature,
+								topP: options.topP,
+								topK: options.topK,
+								minP: options.minP,
+								presencePenalty: options.presencePenalty,
+								repetitionPenalty: options.repetitionPenalty,
+								maxTokens: options.maxTokens ?? Math.min(model.maxTokens, 32000),
+								signal: options.signal,
+								cacheRetention: options.cacheRetention,
+								headers,
+								maxRetryDelayMs: options.maxRetryDelayMs,
+								metadata: options.metadata,
+								sessionId: options.sessionId,
+								providerSessionState: options.providerSessionState,
+								onPayload: options.onPayload,
+								onResponse: options.onResponse,
+								onSseEvent: options.onSseEvent,
+								fetch: options.fetch,
+								thinkingEnabled: Boolean(reasoningEffort) && model.reasoning,
+								thinkingBudgetTokens: reasoningEffort
+									? (options.thinkingBudgets?.[reasoningEffort] ?? ANTHROPIC_THINKING[reasoningEffort])
+									: undefined,
+								reasoning: reasoningEffort,
+								toolChoice: mapAnthropicToolChoice(options.toolChoice),
+							},
+						)
+					: mapping.openaiApiType === "responses"
+						? streamOpenAIResponses(
+								{
+									...model,
+									id: mapping.model,
+									api: "openai-responses",
+									baseUrl: OPENAI_PROXY_URL,
+								} as Model<"openai-responses">,
+								context,
+								{
+									apiKey: directAccess.token,
+									temperature: options.temperature,
+									topP: options.topP,
+									topK: options.topK,
+									minP: options.minP,
+									presencePenalty: options.presencePenalty,
+									repetitionPenalty: options.repetitionPenalty,
+									maxTokens: options.maxTokens ?? model.maxTokens,
+									signal: options.signal,
+									cacheRetention: options.cacheRetention,
+									headers,
+									maxRetryDelayMs: options.maxRetryDelayMs,
+									metadata: options.metadata,
+									sessionId: options.sessionId,
+									providerSessionState: options.providerSessionState,
+									onPayload: options.onPayload,
+									onResponse: options.onResponse,
+									onSseEvent: options.onSseEvent,
+									fetch: options.fetch,
+									reasoning: reasoningEffort,
+									toolChoice: options.toolChoice,
+								} satisfies OpenAIResponsesOptions,
+							)
+						: streamOpenAICompletions(
+								{
+									...model,
+									id: mapping.model,
+									api: "openai-completions",
+									baseUrl: OPENAI_PROXY_URL,
+								} as Model<"openai-completions">,
+								context,
+								{
+									apiKey: directAccess.token,
+									temperature: options.temperature,
+									topP: options.topP,
+									topK: options.topK,
+									minP: options.minP,
+									presencePenalty: options.presencePenalty,
+									repetitionPenalty: options.repetitionPenalty,
+									maxTokens: options.maxTokens ?? model.maxTokens,
+									signal: options.signal,
+									cacheRetention: options.cacheRetention,
+									headers,
+									maxRetryDelayMs: options.maxRetryDelayMs,
+									metadata: options.metadata,
+									sessionId: options.sessionId,
+									providerSessionState: options.providerSessionState,
+									onPayload: options.onPayload,
+									onResponse: options.onResponse,
+									onSseEvent: options.onSseEvent,
+									fetch: options.fetch,
+									reasoning: reasoningEffort,
+									toolChoice: options.toolChoice,
+								} satisfies OpenAICompletionsOptions,
+							);
+
+			for await (const event of inner) {
+				stream.push(event);
+			}
+		} catch (err) {
+			stream.push({
+				type: "error",
+				reason: "error",
+				error: createProviderErrorMessage(model, err),
+			});
+		}
+	})();
+
+	return stream;
+}