From 6de338c6ace8096b433bfa3b020d7129bf6c2086 Mon Sep 17 00:00:00 2001
From: user <user@example.com>
Date: Sat, 25 Jan 2025 11:06:38 +0000
Subject: [PATCH] Properly separate deepseek keys from the generic ones in
 service info

---
 src/service-info.ts | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/service-info.ts b/src/service-info.ts
index 71126fe..261b4f0 100644
--- a/src/service-info.ts
+++ b/src/service-info.ts
@@ -382,10 +382,6 @@ function addKeyToAggregates(k: KeyPoolKey) {
       k.modelFamilies.forEach(incrementGenericFamilyStats);
       // TODO: add modelIds to GcpKey
       break;
-    // These services don't have any additional stats to track.
-    case "azure":
-    case "google-ai":
-    case "mistral-ai":
     case "deepseek":
       if (!keyIsDeepseekKey(k)) throw new Error("Invalid key type");
       k.modelFamilies.forEach((f) => {
@@ -393,6 +389,12 @@ function addKeyToAggregates(k: KeyPoolKey) {
         addToFamily(`${f}__overQuota`, k.isOverQuota ? 1 : 0);
       });
       break;
+    // These services don't have any additional stats to track.
+    case "azure":
+    case "google-ai":
+    case "mistral-ai":
+      k.modelFamilies.forEach(incrementGenericFamilyStats);
+      break;
     default:
       assertNever(k.service);
   }