handles 'invalid subscription' 403 errors from Mistral API

src/admin/web/manage.ts

@@ -268,7 +268,13 @@ router.post("/maintenance", (req, res) => {
   let flash = { type: "", message: "" };
   switch (action) {
     case "recheck": {
-      const checkable: LLMService[] = ["openai", "anthropic", "aws", "gcp","azure"];
+      const checkable: LLMService[] = [
+        "openai",
+        "anthropic",
+        "aws",
+        "gcp",
+        "azure",
+      ];
       checkable.forEach((s) => keyPool.recheck(s));
       const keyCount = keyPool
         .list()
@@ -348,14 +354,8 @@ router.post("/maintenance", (req, res) => {
       const tempUsers = userStore
         .getUsers()
         .filter((u) => u.type === "temporary");
-      const ipv4RangeMap: Map<string, Set<string>> = new Map<
-        string,
-        Set<string>
-      >();
-      const ipv6RangeMap: Map<string, Set<string>> = new Map<
-        string,
-        Set<string>
-      >();
+      const ipv4RangeMap = new Map<string, Set<string>>();
+      const ipv6RangeMap = new Map<string, Set<string>>();
 
       tempUsers.forEach((u) => {
         u.ip.forEach((ip) => {
@@ -365,14 +365,14 @@ router.post("/maintenance", (req, res) => {
               const subnet =
                 parsed.toNormalizedString().split(".").slice(0, 3).join(".") +
                 ".0/24";
-              const userSet = ipv4RangeMap.get(subnet) || new Set<string>();
+              const userSet = ipv4RangeMap.get(subnet) || new Set();
               userSet.add(u.token);
               ipv4RangeMap.set(subnet, userSet);
             } else if (parsed.kind() === "ipv6") {
               const subnet =
                 parsed.toNormalizedString().split(":").slice(0, 4).join(":") +
                 "::/48";
-              const userSet = ipv6RangeMap.get(subnet) || new Set<string>();
+              const userSet = ipv6RangeMap.get(subnet) || new Set();
               userSet.add(u.token);
               ipv6RangeMap.set(subnet, userSet);
             }

src/proxy/middleware/response/index.ts

@@ -293,7 +293,8 @@ const handleUpstreamErrors: ProxyResHandlerWithBody = async (
             errorPayload.proxy_note = `Received 403 error. Key may be invalid.`;
         }
         return;
-      case "gcp":
+        case "mistral-ai":
+        case "gcp":
         keyPool.disable(req.key!, "revoked");
         errorPayload.proxy_note = `Assigned API key is invalid or revoked, please try again.`;
         return;

src/shared/key-management/mistral-ai/checker.ts

@@ -69,9 +69,9 @@ export class MistralAIKeyChecker extends KeyCheckerBase<MistralAIKey> {
   protected handleAxiosError(key: MistralAIKey, error: AxiosError) {
     if (error.response && MistralAIKeyChecker.errorIsMistralAIError(error)) {
       const { status, data } = error.response;
-      if (status === 401) {
+      if ([401, 403].includes(status)) {
         this.log.warn(
-          { key: key.hash, error: data },
+          { key: key.hash, error: data, status },
           "Key is invalid or revoked. Disabling key."
         );
         this.updateKey(key.hash, {