From 1a6ce7ea046c5a9d4317208ae161e5778a8eebc5 Mon Sep 17 00:00:00 2001
From: reanon <>
Date: Fri, 23 May 2025 08:48:36 +0200
Subject: [PATCH] I hate aws v2

---
 src/service-info.ts | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

diff --git a/src/service-info.ts b/src/service-info.ts
index 4b9b6dc..a8de3b0 100644
--- a/src/service-info.ts
+++ b/src/service-info.ts
@@ -528,10 +528,9 @@ 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") {
+          // Original behavior: get logged count from the same family
+          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");
@@ -542,23 +541,25 @@ function getInfoForFamily(family: ModelFamily): BaseFamilyInfo {
           
           info.enabledVariants = variants.size ? Array.from(variants).join(",") : undefined;
 
-          if (awsLoggedStat > 0) {
+          if (logged > 0) {
             info.privacy = config.allowAwsLogging
               ? `AWS logging verification inactive. Prompts could be logged.`
-              : `${awsLoggedStat} active keys are potentially logged and can't be used. Set ALLOW_AWS_LOGGING=true to override.`;
+              : `${logged} active keys are potentially logged and can't be used. Set ALLOW_AWS_LOGGING=true to override.`;
           }
         } else if (family === "aws-claude-opus") {
+          // Get logging info from aws-claude family since that's where it's collected
+          const awsLogged = familyStats.get(`aws-claude__awsLogged`) || 0;
           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) {
+          // Show privacy warning for Opus if there are active Opus keys AND some AWS keys are logged
+          if (awsLogged > 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.`;
+              : `Some AWS keys are potentially logged. Set ALLOW_AWS_LOGGING=true to override.`;
           }
         }
         // TODO: Consider if aws-mistral-* families need similar enabledVariant listings