Function buildPricingMap

packages/core/script/generate-digitalocean.ts:158–182

(entries: z.infer<typeof PricingEntry>[])

}

function buildPricingMap(entries: z.infer<typeof PricingEntry>[]): Map<string, ModelPricing> {
  const map = new Map<string, ModelPricing>();

  for (const entry of entries) {
    const displayName = normalizeDisplayName(entry.name);
    const modelId = PRICING_NAME_MAP[displayName];
    if (!modelId) continue;

    const isInput = entry.name.toLowerCase().includes("input tokens");
    const isOver200k = entry.prompt_tokens === ">200k";
    // Round to avoid float noise (e.g. 0.9900000000000001)
    const rate = Math.round(entry.price.rate * 10000) / 10000;

    const existing = map.get(modelId) ?? ({} as ModelPricing);

    if (isInput && isOver200k) existing.inputOver200k = rate;
    else if (!isInput && isOver200k) existing.outputOver200k = rate;
    else if (isInput) existing.input = rate;
    else existing.output = rate;

    map.set(modelId, existing);
  }

  return map;
}

// ---------------------------------------------------------------------------
// Existing TOML shape (fields we read and may preserve)