From fdba7cd7e443f38cc6dd3df46df87942f4710e04 Mon Sep 17 00:00:00 2001
From: reanon <>
Date: Fri, 23 May 2025 08:44:37 +0200
Subject: [PATCH] I hate aws

---
 src/service-info.ts | 50 +++++++++++++++++++++++++--------------------
 1 file changed, 28 insertions(+), 22 deletions(-)

diff --git a/src/service-info.ts b/src/service-info.ts
index b3dec08..4b9b6dc 100644
--- a/src/service-info.ts
+++ b/src/service-info.ts
@@ -528,34 +528,40 @@ function getInfoForFamily(family: ModelFamily): BaseFamilyInfo {
         info.prefilledKeys = familyStats.get(`${family}__pozzed`) || 0;
         break;
       case "aws":
+        // Common AWS logging stat. Logging stats are always collected in aws-claude family
+        const awsLoggedStat = familyStats.get(`aws-claude__awsLogged`) || 0;
+
         if (family === "aws-claude") {
-          const logged = familyStats.get(`${family}__awsLogged`) || 0;
           const variants = new Set<string>();
-          if (familyStats.get(`${family}__awsClaude2`) || 0)
-            variants.add("claude2");
-          if (familyStats.get(`${family}__awsSonnet3`) || 0)
-            variants.add("sonnet3");
-          if (familyStats.get(`${family}__awsSonnet3_5`) || 0)
-            variants.add("sonnet3.5");
-          if (familyStats.get(`${family}__awsSonnet3_7`) || 0)
-            variants.add("sonnet3.7");
-          if (familyStats.get(`${family}__awsHaiku`) || 0)
-            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;
-          if (logged > 0) {
+          if (familyStats.get(`${family}__awsClaude2`) || 0) variants.add("claude2");
+          if (familyStats.get(`${family}__awsSonnet3`) || 0) variants.add("sonnet3");
+          if (familyStats.get(`${family}__awsSonnet3_5`) || 0) variants.add("sonnet3.5");
+          if (familyStats.get(`${family}__awsSonnet3_7`) || 0) 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;
+
+          if (awsLoggedStat > 0) {
             info.privacy = config.allowAwsLogging
               ? `AWS logging verification inactive. Prompts could be logged.`
-              : `${logged} active keys are potentially logged and can't be used. Set ALLOW_AWS_LOGGING=true to override.`;
+              : `${awsLoggedStat} active keys are potentially logged and can't be used. Set ALLOW_AWS_LOGGING=true to override.`;
+          }
+        } else if (family === "aws-claude-opus") {
+          const variants = new Set<string>();
+          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;
+
+          // Show privacy warning for Opus if there are active Opus keys AND some Claude keys are logged
+          if (awsLoggedStat > 0 && info.activeKeys > 0) {
+             info.privacy = config.allowAwsLogging
+              ? `AWS logging verification inactive. Prompts could be logged.`
+              : `Some AWS Claude keys are potentially logged. Set ALLOW_AWS_LOGGING=true to override.`;
           }
         }
+        // TODO: Consider if aws-mistral-* families need similar enabledVariant listings
         break;
       case "gcp":
         if (family === "gcp-claude") {