@agentuity/coder-tui 2.0.15 → 2.0.17

package/dist/aigateway.js

@@ -1,173 +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 { createMinimalLogger, StructuredError } from '@agentuity/core';
-import { AIGatewayService, } from '@agentuity/core/aigateway';
-import { createServerFetchAdapter } from '@agentuity/server';
-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',
-]);
-const AIGatewayModelFetchError = StructuredError('AIGatewayModelFetchError')();
-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 (error) {
-                    throw new AIGatewayModelFetchError({
-                        message: 'Failed to fetch models from AI Gateway',
-                        cause: error,
-                    });
-                }
-            }
-        }
-        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;
-    }
-    try {
-        const service = new AIGatewayService(baseUrl, createServerFetchAdapter({ headers: {} }, createMinimalLogger()));
-        return await service.listModels();
-    }
-    catch (error) {
-        throw new AIGatewayModelFetchError({
-            message: 'Failed to fetch models from AI Gateway',
-            cause: error,
-        });
-    }
-}
-function sanitizeModalities(modalities) {
-    const sanitized = (modalities ?? []).filter((modality) => modality === 'text' || modality === 'image');
-    return sanitized.length > 0 ? sanitized : ['text'];
-}
-function toPiModel(m) {
-    return {
-        id: m.id,
-        name: m.name,
-        reasoning: m.reasoning ?? false,
-        input: sanitizeModalities(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;AAClD,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AACvE,OAAO,EACN,gBAAgB,GAGhB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EAAE,wBAAwB,EAAE,MAAM,mBAAmB,CAAC;AAe7D,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;AAExB,MAAM,wBAAwB,GAAG,eAAe,CAAC,0BAA0B,CAAC,EAExE,CAAC;AAEL,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,KAAK,EAAE,CAAC;oBAChB,MAAM,IAAI,wBAAwB,CAAC;wBAClC,OAAO,EAAE,wCAAwC;wBACjD,KAAK,EAAE,KAAK;qBACZ,CAAC,CAAC;gBACJ,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,IAAI,CAAC;QACJ,MAAM,OAAO,GAAG,IAAI,gBAAgB,CACnC,OAAO,EACP,wBAAwB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,mBAAmB,EAAE,CAAC,CAChE,CAAC;QACF,OAAO,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;IACnC,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,MAAM,IAAI,wBAAwB,CAAC;YAClC,OAAO,EAAE,wCAAwC;YACjD,KAAK,EAAE,KAAK;SACZ,CAAC,CAAC;IACJ,CAAC;AACF,CAAC;AAED,SAAS,kBAAkB,CAAC,UAAgC;IAC3D,MAAM,SAAS,GAAG,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,MAAM,CAC1C,CAAC,QAAQ,EAAgC,EAAE,CAAC,QAAQ,KAAK,MAAM,IAAI,QAAQ,KAAK,OAAO,CACvF,CAAC;IACF,OAAO,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;AACpD,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,IAAI,KAAK;QAC/B,KAAK,EAAE,kBAAkB,CAAC,CAAC,CAAC,gBAAgB,CAAC;QAC7C,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,227 +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 { createMinimalLogger, StructuredError } from '@agentuity/core';
-import {
-	AIGatewayService,
-	type AIGatewayModel,
-	type AIGatewayModels,
-} from '@agentuity/core/aigateway';
-import { createServerFetchAdapter } from '@agentuity/server';
-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 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[]);
-
-const AIGatewayModelFetchError = StructuredError('AIGatewayModelFetchError')<{
-	cause?: unknown;
-}>();
-
-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 (error) {
-					throw new AIGatewayModelFetchError({
-						message: 'Failed to fetch models from AI Gateway',
-						cause: error,
-					});
-				}
-			}
-		}
-		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;
-	}
-
-	try {
-		const service = new AIGatewayService(
-			baseUrl,
-			createServerFetchAdapter({ headers: {} }, createMinimalLogger())
-		);
-		return await service.listModels();
-	} catch (error) {
-		throw new AIGatewayModelFetchError({
-			message: 'Failed to fetch models from AI Gateway',
-			cause: error,
-		});
-	}
-}
-
-function sanitizeModalities(modalities: string[] | undefined): ('text' | 'image')[] {
-	const sanitized = (modalities ?? []).filter(
-		(modality): modality is 'text' | 'image' => modality === 'text' || modality === 'image'
-	);
-	return sanitized.length > 0 ? sanitized : ['text'];
-}
-
-function toPiModel(m: AIGatewayModel): ProviderModelConfig {
-	return {
-		id: m.id,
-		name: m.name,
-		reasoning: m.reasoning ?? false,
-		input: sanitizeModalities(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: 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,
-		});
-	}
-}