@@ -7,9 +7,9 @@ import { prefetchCodexPricing } from './_macro.ts' with { type: 'macro' };
 import { logger } from './logger.ts';
 
 const CODEX_PROVIDER_PREFIXES = ['openai/', 'azure/', 'openrouter/openai/'];
-const CODEX_MODEL_ALIASES: Record<string, string> = {
-	'gpt-5-codex': 'gpt-5',
-};
+const CODEX_MODEL_ALIASES_MAP = new Map<string, string>([
+	['gpt-5-codex', 'gpt-5'],
+]);
 
 function toPerMillion(value: number | undefined, fallback?: number): number {
 	const perToken = value ?? fallback ?? 0;
@@ -40,26 +40,32 @@ export class CodexPricingSource implements PricingSource, Disposable {
 	}
 
 	async getPricing(model: string): Promise<ModelPricing> {
-		const alias = CODEX_MODEL_ALIASES[model];
-		const lookupCandidates = Array.from(new Set([model, alias].filter((candidate): candidate is string => candidate != null)));
+		const directLookup = await this.fetcher.getModelPricing(model);
+		if (Result.isFailure(directLookup)) {
+			throw directLookup.error;
+		}
 
-		for (const candidate of lookupCandidates) {
-			const pricingResult = await this.fetcher.getModelPricing(candidate);
-			if (Result.isFailure(pricingResult)) {
-				throw pricingResult.error;
+		let pricing = directLookup.value;
+		if (pricing == null) {
+			const alias = CODEX_MODEL_ALIASES_MAP.get(model);
+			if (alias != null) {
+				const aliasLookup = await this.fetcher.getModelPricing(alias);
+				if (Result.isFailure(aliasLookup)) {
+					throw aliasLookup.error;
+				}
+				pricing = aliasLookup.value;
 			}
+		}
 
-			const pricing = pricingResult.value;
-			if (pricing != null) {
-				return {
-					inputCostPerMToken: toPerMillion(pricing.input_cost_per_token),
-					cachedInputCostPerMToken: toPerMillion(pricing.cache_read_input_token_cost, pricing.input_cost_per_token),
-					outputCostPerMToken: toPerMillion(pricing.output_cost_per_token),
-				};
-			}
+		if (pricing == null) {
+			throw new Error(`Pricing not found for model ${model}`);
 		}
 
-		throw new Error(`Pricing not found for model ${model}`);
+		return {
+			inputCostPerMToken: toPerMillion(pricing.input_cost_per_token),
+			cachedInputCostPerMToken: toPerMillion(pricing.cache_read_input_token_cost, pricing.input_cost_per_token),
+			outputCostPerMToken: toPerMillion(pricing.output_cost_per_token),
+		};
 	}
 }