diff --git a/src/service-info.ts b/src/service-info.ts index 804dc69..1b7a7ad 100644 --- a/src/service-info.ts +++ b/src/service-info.ts @@ -66,6 +66,7 @@ type ModelAggregates = { awsSonnet3?: number; awsSonnet3_5?: number; awsSonnet3_7?: number; + awsSonnet4?: number; awsHaiku: number; gcpSonnet?: number; gcpSonnet35?: number; @@ -397,6 +398,8 @@ function addKeyToAggregates(k: KeyPoolKey) { addToFamily(`aws-claude__awsSonnet3_7`, 1); } else if (id.includes("claude-3-haiku")) { addToFamily(`aws-claude__awsHaiku`, 1); + } else if (id.includes("claude-sonnet-4")) { + addToFamily(`aws-claude__awsSonnet4`, 1); } else if (id.includes("claude-v2")) { addToFamily(`aws-claude__awsClaude2`, 1); } @@ -532,6 +535,8 @@ function getInfoForFamily(family: ModelFamily): BaseFamilyInfo { variants.add("sonnet3.7"); if (familyStats.get(`${family}__awsHaiku`) || 0) variants.add("haiku"); + if (familyStats.get(`${family}__awsSonnet4`) || 0) + variants.add("sonnet4"); info.enabledVariants = variants.size ? `${Array.from(variants).join(",")}` : undefined;