diff --git a/src/service-info.ts b/src/service-info.ts index 1b7a7ad..fed18d7 100644 --- a/src/service-info.ts +++ b/src/service-info.ts @@ -67,6 +67,8 @@ type ModelAggregates = { awsSonnet3_5?: number; awsSonnet3_7?: number; awsSonnet4?: number; + awsOpus3?: number; + awsOpus4?: number; awsHaiku: number; gcpSonnet?: number; gcpSonnet35?: number; @@ -400,6 +402,10 @@ function addKeyToAggregates(k: KeyPoolKey) { addToFamily(`aws-claude__awsHaiku`, 1); } else if (id.includes("claude-sonnet-4")) { addToFamily(`aws-claude__awsSonnet4`, 1); + } else if (id.includes("claude-3-opus")) { + addToFamily(`aws-claude__awsOpus3`, 1); + } else if (id.includes("claude-opus-4")) { + addToFamily(`aws-claude__awsOpus4`, 1); } else if (id.includes("claude-v2")) { addToFamily(`aws-claude__awsClaude2`, 1); } @@ -537,6 +543,10 @@ function getInfoForFamily(family: ModelFamily): BaseFamilyInfo { variants.add("haiku"); if (familyStats.get(`${family}__awsSonnet4`) || 0) variants.add("sonnet4"); + if (familyStats.get(`${family}__awsOpus3`) || 0) + variants.add("opus3"); + if (familyStats.get(`${family}__awsOpus4`) || 0) + variants.add("opus4"); info.enabledVariants = variants.size ? `${Array.from(variants).join(",")}` : undefined;