From ce9c8ec8b6e0f031b84ad74980b1027aa1ee96f3 Mon Sep 17 00:00:00 2001
From: reanon <>
Date: Fri, 23 May 2025 00:16:53 +0200
Subject: [PATCH] aws claude show variant sonnet

---
 src/service-info.ts | 5 +++++
 1 file changed, 5 insertions(+)

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;