@agentuity/coder-tui 3.0.0-beta.1 → 3.0.0-beta.3

package/dist/aigateway.js

@@ -1,178 +0,0 @@
-/**
- * Agentuity AI Gateway Custom Provider Extension
- *
- * Registers models from the Agentuity AI Gateway using the appropriate API type
- * based on model ID patterns. Models are loaded dynamically from the gateway's /models endpoint.
- *
- * Usage:
- *   Use /model to switch to aigateway models
- */
-import { delimiter, join } from 'node:path';
-import { existsSync } from 'node:fs';
-import { execFileSync } from 'node:child_process';
-const MODEL_CATALOG_TIMEOUT_MS = 5_000;
-const KNOWN_APIS = new Set([
-    'openai-completions',
-    'mistral-conversations',
-    'openai-responses',
-    'azure-openai-responses',
-    'openai-codex-responses',
-    'anthropic-messages',
-    'bedrock-converse-stream',
-    'google-generative-ai',
-    'google-gemini-cli',
-    'google-vertex',
-]);
-function getEnv(...keys) {
-    for (const key of keys) {
-        if (process.env[key]) {
-            return process.env[key];
-        }
-    }
-}
-function normalizeCredential(value) {
-    if (value === undefined || value === null) {
-        return undefined;
-    }
-    const normalized = String(value).trim();
-    return normalized.length > 0 ? normalized : undefined;
-}
-function isKnownApi(api) {
-    return typeof api === 'string' && KNOWN_APIS.has(api);
-}
-function getRegion() {
-    return getEnv('AGENTUITY_REGION') ?? 'usc';
-}
-function getBaseUrl() {
-    const region = getRegion();
-    return `https://aigateway-${region}.agentuity.cloud`;
-}
-async function fetchModels() {
-    const baseUrl = getBaseUrl();
-    let apiKey = normalizeCredential(getEnv('AGENTUITY_CODER_API_KEY', 'AGENTUITY_SDK_KEY', 'AGENTUITY_CLI_API_KEY', 'AGENTUITY_CLI_KEY'));
-    let orgId = normalizeCredential(getEnv('AGENTUITY_ORGID', 'AGENTUITY_CLOUD_ORG_ID', 'AGENTUITY_ORG_ID'));
-    if (!apiKey) {
-        let found = false;
-        const path = process.env.PATH?.split(delimiter) ?? [];
-        for (const dir of path) {
-            const fn = join(dir, 'agentuity');
-            if (existsSync(fn)) {
-                try {
-                    const res = execFileSync(fn, ['auth', 'apikey', '--json']);
-                    const apiKeyResult = JSON.parse(res.toString());
-                    apiKey = normalizeCredential(apiKeyResult.apiKey);
-                    found = true;
-                    if (!orgId) {
-                        const ores = execFileSync(fn, ['auth', 'org', 'current']);
-                        orgId = normalizeCredential(ores);
-                        if (!orgId) {
-                            console.warn('Cannot determine the org id. Use `agentuity auth org select` to select a default organization');
-                            return {};
-                        }
-                    }
-                    break;
-                }
-                catch (_ex) {
-                    //
-                }
-            }
-        }
-        if (!found) {
-            console.warn('AGENTUITY_SDK_KEY, AGENTUITY_CLI_API_KEY or AGENTUITY_CLI_KEY not set and cannot find the agentuit cli, cannot fetch models from AI Gateway');
-            return {};
-        }
-    }
-    if (!apiKey) {
-        console.warn('Cannot determine the API key, cannot fetch models from AI Gateway');
-        return {};
-    }
-    process.env.AGENTUITY_AIGATEWAY_KEY = apiKey;
-    if (orgId) {
-        process.env.AGENTUITY_AIGATEWAY_ORGID = orgId;
-    }
-    const controller = new AbortController();
-    const timeout = setTimeout(() => controller.abort(), MODEL_CATALOG_TIMEOUT_MS);
-    try {
-        const response = await fetch(`${baseUrl}/models`, { signal: controller.signal });
-        if (!response.ok) {
-            console.warn(`Failed to fetch models from AI Gateway: ${response.status} ${response.statusText}`);
-            return {};
-        }
-        const payload = (await response.json());
-        if (!payload.success) {
-            console.warn(`Failed to load models. ${payload.message} ${payload}`);
-        }
-        return payload.data;
-    }
-    catch (error) {
-        if (error instanceof Error && error.name === 'AbortError') {
-            console.warn(`Timed out fetching models from AI Gateway after ${MODEL_CATALOG_TIMEOUT_MS}ms`);
-            return {};
-        }
-        console.warn('Failed to fetch models from AI Gateway:', error);
-        return {};
-    }
-    finally {
-        clearTimeout(timeout);
-    }
-}
-function toPiModel(m) {
-    return {
-        id: m.id,
-        name: m.name,
-        reasoning: m.reasoning,
-        input: m.input_modalities,
-        contextWindow: m.context_window ?? 40000,
-        maxTokens: m.max_output_tokens ?? 64000,
-        cost: {
-            input: m.pricing?.input ?? 0,
-            output: m.pricing?.output ?? 0,
-            cacheRead: m.pricing?.cached_input ?? 0,
-            cacheWrite: 0,
-        },
-        compat: {
-            supportsDeveloperRole: false,
-        },
-    };
-}
-export async function setupAIGateway(pi) {
-    const models = await fetchModels();
-    const baseUrl = getBaseUrl();
-    const allModels = [];
-    for (const providerModels of Object.values(models)) {
-        if (providerModels) {
-            allModels.push(...providerModels);
-        }
-    }
-    if (allModels.length === 0) {
-        return;
-    }
-    const modelsByApi = new Map();
-    for (const m of allModels) {
-        const apiType = m.api;
-        if (!isKnownApi(apiType)) {
-            continue; // THIS SHOULD NEVER HAPPEN BUT JUST IN CASE
-        }
-        const existing = modelsByApi.get(apiType) ?? [];
-        existing.push(toPiModel(m));
-        modelsByApi.set(apiType, existing);
-    }
-    const headers = {};
-    if (process.env.AGENTUITY_AIGATEWAY_ORGID) {
-        headers['x-agentuity-orgid'] = process.env.AGENTUITY_AIGATEWAY_ORGID;
-    }
-    for (const [apiType, providerModels] of modelsByApi) {
-        const apitok = apiType.split('-');
-        const name = apitok.length >= 2 ? apitok.slice(0, 2).join('-') : apitok[0];
-        const providerName = `agentuity/${name}`;
-        pi.registerProvider(providerName, {
-            baseUrl,
-            apiKey: 'AGENTUITY_AIGATEWAY_KEY',
-            headers,
-            authHeader: true,
-            api: apiType,
-            models: providerModels,
-        });
-    }
-}
-//# sourceMappingURL=aigateway.js.map

package/dist/aigateway.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"aigateway.js","sourceRoot":"","sources":["../src/aigateway.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AACH,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC5C,OAAO,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AACrC,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAelD,MAAM,wBAAwB,GAAG,KAAK,CAAC;AAEvC,MAAM,UAAU,GAAG,IAAI,GAAG,CAAS;IAClC,oBAAoB;IACpB,uBAAuB;IACvB,kBAAkB;IAClB,wBAAwB;IACxB,wBAAwB;IACxB,oBAAoB;IACpB,yBAAyB;IACzB,sBAAsB;IACtB,mBAAmB;IACnB,eAAe;CACM,CAAC,CAAC;AA8BxB,SAAS,MAAM,CAAC,GAAG,IAAc;IAChC,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACxB,IAAI,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACtB,OAAO,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACzB,CAAC;IACF,CAAC;AACF,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IAC1C,IAAI,KAAK,KAAK,SAAS,IAAI,KAAK,KAAK,IAAI,EAAE,CAAC;QAC3C,OAAO,SAAS,CAAC;IAClB,CAAC;IACD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IACxC,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;AACvD,CAAC;AAED,SAAS,UAAU,CAAC,GAAY;IAC/B,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACvD,CAAC;AAED,SAAS,SAAS;IACjB,OAAO,MAAM,CAAC,kBAAkB,CAAC,IAAI,KAAK,CAAC;AAC5C,CAAC;AAED,SAAS,UAAU;IAClB,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,OAAO,qBAAqB,MAAM,kBAAkB,CAAC;AACtD,CAAC;AAED,KAAK,UAAU,WAAW;IACzB,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAC7B,IAAI,MAAM,GAAG,mBAAmB,CAC/B,MAAM,CACL,yBAAyB,EACzB,mBAAmB,EACnB,uBAAuB,EACvB,mBAAmB,CACnB,CACD,CAAC;IACF,IAAI,KAAK,GAAG,mBAAmB,CAC9B,MAAM,CAAC,iBAAiB,EAAE,wBAAwB,EAAE,kBAAkB,CAAC,CACvE,CAAC;IAEF,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,IAAI,KAAK,GAAG,KAAK,CAAC;QAClB,MAAM,IAAI,GAAG,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,KAAK,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;QACtD,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;YACxB,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAClC,IAAI,UAAU,CAAC,EAAE,CAAC,EAAE,CAAC;gBACpB,IAAI,CAAC;oBACJ,MAAM,GAAG,GAAG,YAAY,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC;oBAC3D,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAuB,CAAC;oBACtE,MAAM,GAAG,mBAAmB,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;oBAClD,KAAK,GAAG,IAAI,CAAC;oBACb,IAAI,CAAC,KAAK,EAAE,CAAC;wBACZ,MAAM,IAAI,GAAG,YAAY,CAAC,EAAE,EAAE,CAAC,MAAM,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC,CAAC;wBAC1D,KAAK,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;wBAClC,IAAI,CAAC,KAAK,EAAE,CAAC;4BACZ,OAAO,CAAC,IAAI,CACX,+FAA+F,CAC/F,CAAC;4BACF,OAAO,EAAE,CAAC;wBACX,CAAC;oBACF,CAAC;oBACD,MAAM;gBACP,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACd,EAAE;gBACH,CAAC;YACF,CAAC;QACF,CAAC;QACD,IAAI,CAAC,KAAK,EAAE,CAAC;YACZ,OAAO,CAAC,IAAI,CACX,6IAA6I,CAC7I,CAAC;YACF,OAAO,EAAE,CAAC;QACX,CAAC;IACF,CAAC;IAED,IAAI,CAAC,MAAM,EAAE,CAAC;QACb,OAAO,CAAC,IAAI,CAAC,mEAAmE,CAAC,CAAC;QAClF,OAAO,EAAE,CAAC;IACX,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,uBAAuB,GAAG,MAAM,CAAC;IAC7C,IAAI,KAAK,EAAE,CAAC;QACX,OAAO,CAAC,GAAG,CAAC,yBAAyB,GAAG,KAAK,CAAC;IAC/C,CAAC;IAED,MAAM,UAAU,GAAG,IAAI,eAAe,EAAE,CAAC;IACzC,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,UAAU,CAAC,KAAK,EAAE,EAAE,wBAAwB,CAAC,CAAC;IAE/E,IAAI,CAAC;QACJ,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,OAAO,SAAS,EAAE,EAAE,MAAM,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;QAEjF,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YAClB,OAAO,CAAC,IAAI,CACX,2CAA2C,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,CACnF,CAAC;YACF,OAAO,EAAE,CAAC;QACX,CAAC;QAED,MAAM,OAAO,GAAG,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAA2B,CAAC;QAElE,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACtB,OAAO,CAAC,IAAI,CAAC,0BAA0B,OAAO,CAAC,OAAO,IAAI,OAAO,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,OAAO,OAAO,CAAC,IAAI,CAAC;IACrB,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,KAAK,YAAY,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3D,OAAO,CAAC,IAAI,CACX,mDAAmD,wBAAwB,IAAI,CAC/E,CAAC;YACF,OAAO,EAAE,CAAC;QACX,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,yCAAyC,EAAE,KAAK,CAAC,CAAC;QAC/D,OAAO,EAAE,CAAC;IACX,CAAC;YAAS,CAAC;QACV,YAAY,CAAC,OAAO,CAAC,CAAC;IACvB,CAAC;AACF,CAAC;AAED,SAAS,SAAS,CAAC,CAAiB;IACnC,OAAO;QACN,EAAE,EAAE,CAAC,CAAC,EAAE;QACR,IAAI,EAAE,CAAC,CAAC,IAAI;QACZ,SAAS,EAAE,CAAC,CAAC,SAAS;QACtB,KAAK,EAAE,CAAC,CAAC,gBAAwC;QACjD,aAAa,EAAE,CAAC,CAAC,cAAc,IAAI,KAAK;QACxC,SAAS,EAAE,CAAC,CAAC,iBAAiB,IAAI,KAAK;QACvC,IAAI,EAAE;YACL,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,KAAK,IAAI,CAAC;YAC5B,MAAM,EAAE,CAAC,CAAC,OAAO,EAAE,MAAM,IAAI,CAAC;YAC9B,SAAS,EAAE,CAAC,CAAC,OAAO,EAAE,YAAY,IAAI,CAAC;YACvC,UAAU,EAAE,CAAC;SACb;QACD,MAAM,EAAE;YACP,qBAAqB,EAAE,KAAK;SAC5B;KACD,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,EAAgB;IACpD,MAAM,MAAM,GAAG,MAAM,WAAW,EAAE,CAAC;IACnC,MAAM,OAAO,GAAG,UAAU,EAAE,CAAC;IAE7B,MAAM,SAAS,GAAqB,EAAE,CAAC;IACvC,KAAK,MAAM,cAAc,IAAI,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;QACpD,IAAI,cAAc,EAAE,CAAC;YACpB,SAAS,CAAC,IAAI,CAAC,GAAG,cAAc,CAAC,CAAC;QACnC,CAAC;IACF,CAAC;IACD,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC5B,OAAO;IACR,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,GAAG,EAAmC,CAAC;IAE/D,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1B,SAAS,CAAC,4CAA4C;QACvD,CAAC;QACD,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAChD,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;QAC5B,WAAW,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,IAAI,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,CAAC;QAC3C,OAAO,CAAC,mBAAmB,CAAC,GAAG,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC;IACtE,CAAC;IAED,KAAK,MAAM,CAAC,OAAO,EAAE,cAAc,CAAC,IAAI,WAAW,EAAE,CAAC;QACrD,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAClC,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;QAC3E,MAAM,YAAY,GAAG,aAAa,IAAI,EAAE,CAAC;QACzC,EAAE,CAAC,gBAAgB,CAAC,YAAY,EAAE;YACjC,OAAO;YACP,MAAM,EAAE,yBAAyB;YACjC,OAAO;YACP,UAAU,EAAE,IAAI;YAChB,GAAG,EAAE,OAAO;YACZ,MAAM,EAAE,cAAc;SACtB,CAAC,CAAC;IACJ,CAAC;AACF,CAAC"}

package/src/aigateway.ts

@@ -1,256 +0,0 @@
-/**
- * Agentuity AI Gateway Custom Provider Extension
- *
- * Registers models from the Agentuity AI Gateway using the appropriate API type
- * based on model ID patterns. Models are loaded dynamically from the gateway's /models endpoint.
- *
- * Usage:
- *   Use /model to switch to aigateway models
- */
-import { delimiter, join } from 'node:path';
-import { existsSync } from 'node:fs';
-import { execFileSync } from 'node:child_process';
-import type { ExtensionAPI, ProviderModelConfig } from '@mariozechner/pi-coding-agent';
-
-export type KnownApi =
-	| 'openai-completions'
-	| 'mistral-conversations'
-	| 'openai-responses'
-	| 'azure-openai-responses'
-	| 'openai-codex-responses'
-	| 'anthropic-messages'
-	| 'bedrock-converse-stream'
-	| 'google-generative-ai'
-	| 'google-gemini-cli'
-	| 'google-vertex';
-
-const MODEL_CATALOG_TIMEOUT_MS = 5_000;
-
-const KNOWN_APIS = new Set<string>([
-	'openai-completions',
-	'mistral-conversations',
-	'openai-responses',
-	'azure-openai-responses',
-	'openai-codex-responses',
-	'anthropic-messages',
-	'bedrock-converse-stream',
-	'google-generative-ai',
-	'google-gemini-cli',
-	'google-vertex',
-] satisfies KnownApi[]);
-
-interface AIGatewayModels {
-	[key: string]: AIGatewayModel[];
-}
-
-interface AIGatewayModelResponse {
-	success: boolean;
-	data: AIGatewayModels;
-	message?: string;
-	error?: string;
-}
-
-interface AIGatewayModel {
-	id: string;
-	name: string;
-	api: KnownApi;
-	reasoning: boolean;
-	input_modalities?: ('text' | 'image')[];
-	context_window?: number;
-	max_output_tokens?: number;
-	pricing?: {
-		input: number;
-		output: number;
-		cached_input: number;
-		unit: 'per_million_tokens';
-		currency: 'USD';
-	};
-}
-
-function getEnv(...keys: string[]): string | undefined {
-	for (const key of keys) {
-		if (process.env[key]) {
-			return process.env[key];
-		}
-	}
-}
-
-function normalizeCredential(value: unknown): string | undefined {
-	if (value === undefined || value === null) {
-		return undefined;
-	}
-	const normalized = String(value).trim();
-	return normalized.length > 0 ? normalized : undefined;
-}
-
-function isKnownApi(api: unknown): api is KnownApi {
-	return typeof api === 'string' && KNOWN_APIS.has(api);
-}
-
-function getRegion(): string {
-	return getEnv('AGENTUITY_REGION') ?? 'usc';
-}
-
-function getBaseUrl(): string {
-	const region = getRegion();
-	return `https://aigateway-${region}.agentuity.cloud`;
-}
-
-async function fetchModels(): Promise<AIGatewayModels> {
-	const baseUrl = getBaseUrl();
-	let apiKey = normalizeCredential(
-		getEnv(
-			'AGENTUITY_CODER_API_KEY',
-			'AGENTUITY_SDK_KEY',
-			'AGENTUITY_CLI_API_KEY',
-			'AGENTUITY_CLI_KEY'
-		)
-	);
-	let orgId = normalizeCredential(
-		getEnv('AGENTUITY_ORGID', 'AGENTUITY_CLOUD_ORG_ID', 'AGENTUITY_ORG_ID')
-	);
-
-	if (!apiKey) {
-		let found = false;
-		const path = process.env.PATH?.split(delimiter) ?? [];
-		for (const dir of path) {
-			const fn = join(dir, 'agentuity');
-			if (existsSync(fn)) {
-				try {
-					const res = execFileSync(fn, ['auth', 'apikey', '--json']);
-					const apiKeyResult = JSON.parse(res.toString()) as { apiKey: string };
-					apiKey = normalizeCredential(apiKeyResult.apiKey);
-					found = true;
-					if (!orgId) {
-						const ores = execFileSync(fn, ['auth', 'org', 'current']);
-						orgId = normalizeCredential(ores);
-						if (!orgId) {
-							console.warn(
-								'Cannot determine the org id. Use `agentuity auth org select` to select a default organization'
-							);
-							return {};
-						}
-					}
-					break;
-				} catch (_ex) {
-					//
-				}
-			}
-		}
-		if (!found) {
-			console.warn(
-				'AGENTUITY_SDK_KEY, AGENTUITY_CLI_API_KEY or AGENTUITY_CLI_KEY not set and cannot find the agentuit cli, cannot fetch models from AI Gateway'
-			);
-			return {};
-		}
-	}
-
-	if (!apiKey) {
-		console.warn('Cannot determine the API key, cannot fetch models from AI Gateway');
-		return {};
-	}
-
-	process.env.AGENTUITY_AIGATEWAY_KEY = apiKey;
-	if (orgId) {
-		process.env.AGENTUITY_AIGATEWAY_ORGID = orgId;
-	}
-
-	const controller = new AbortController();
-	const timeout = setTimeout(() => controller.abort(), MODEL_CATALOG_TIMEOUT_MS);
-
-	try {
-		const response = await fetch(`${baseUrl}/models`, { signal: controller.signal });
-
-		if (!response.ok) {
-			console.warn(
-				`Failed to fetch models from AI Gateway: ${response.status} ${response.statusText}`
-			);
-			return {};
-		}
-
-		const payload = (await response.json()) as AIGatewayModelResponse;
-
-		if (!payload.success) {
-			console.warn(`Failed to load models. ${payload.message} ${payload}`);
-		}
-
-		return payload.data;
-	} catch (error) {
-		if (error instanceof Error && error.name === 'AbortError') {
-			console.warn(
-				`Timed out fetching models from AI Gateway after ${MODEL_CATALOG_TIMEOUT_MS}ms`
-			);
-			return {};
-		}
-		console.warn('Failed to fetch models from AI Gateway:', error);
-		return {};
-	} finally {
-		clearTimeout(timeout);
-	}
-}
-
-function toPiModel(m: AIGatewayModel): ProviderModelConfig {
-	return {
-		id: m.id,
-		name: m.name,
-		reasoning: m.reasoning,
-		input: m.input_modalities as ('text' | 'image')[],
-		contextWindow: m.context_window ?? 40000,
-		maxTokens: m.max_output_tokens ?? 64000,
-		cost: {
-			input: m.pricing?.input ?? 0,
-			output: m.pricing?.output ?? 0,
-			cacheRead: m.pricing?.cached_input ?? 0,
-			cacheWrite: 0,
-		},
-		compat: {
-			supportsDeveloperRole: false,
-		},
-	};
-}
-
-export async function setupAIGateway(pi: ExtensionAPI) {
-	const models = await fetchModels();
-	const baseUrl = getBaseUrl();
-
-	const allModels: AIGatewayModel[] = [];
-	for (const providerModels of Object.values(models)) {
-		if (providerModels) {
-			allModels.push(...providerModels);
-		}
-	}
-	if (allModels.length === 0) {
-		return;
-	}
-
-	const modelsByApi = new Map<KnownApi, ProviderModelConfig[]>();
-
-	for (const m of allModels) {
-		const apiType = m.api;
-		if (!isKnownApi(apiType)) {
-			continue; // THIS SHOULD NEVER HAPPEN BUT JUST IN CASE
-		}
-		const existing = modelsByApi.get(apiType) ?? [];
-		existing.push(toPiModel(m));
-		modelsByApi.set(apiType, existing);
-	}
-
-	const headers: Record<string, string> = {};
-	if (process.env.AGENTUITY_AIGATEWAY_ORGID) {
-		headers['x-agentuity-orgid'] = process.env.AGENTUITY_AIGATEWAY_ORGID;
-	}
-
-	for (const [apiType, providerModels] of modelsByApi) {
-		const apitok = apiType.split('-');
-		const name = apitok.length >= 2 ? apitok.slice(0, 2).join('-') : apitok[0];
-		const providerName = `agentuity/${name}`;
-		pi.registerProvider(providerName, {
-			baseUrl,
-			apiKey: 'AGENTUITY_AIGATEWAY_KEY',
-			headers,
-			authHeader: true,
-			api: apiType,
-			models: providerModels,
-		});
-	}
-}