@zhafron/opencode-kiro-auth 1.4.1 → 1.4.2

package/dist/plugin/accounts.js

@@ -72,7 +72,7 @@ export class AccountManager {
         const now = Date.now();
         const available = this.accounts.filter((a) => {
             if (!a.isHealthy) {
-                if (a.failCount < 3 && a.recoveryTime && now >= a.recoveryTime) {
+                if (a.failCount < 10 && a.recoveryTime && now >= a.recoveryTime) {
                     a.isHealthy = true;
                     delete a.unhealthyReason;
                     delete a.recoveryTime;
@@ -97,7 +97,7 @@ export class AccountManager {
         }
         if (!selected) {
             const fallback = this.accounts
-                .filter((a) => !a.isHealthy && a.failCount < 3)
+                .filter((a) => !a.isHealthy && a.failCount < 10)
                 .sort((a, b) => (a.usedCount || 0) - (b.usedCount || 0) || (a.lastUsed || 0) - (b.lastUsed || 0))[0];
             if (fallback) {
                 fallback.isHealthy = true;
@@ -178,7 +178,7 @@ export class AccountManager {
             acc.failCount = (acc.failCount || 0) + 1;
             acc.unhealthyReason = reason;
             acc.lastUsed = Date.now();
-            if (acc.failCount >= 3) {
+            if (acc.failCount >= 10) {
                 acc.isHealthy = false;
                 acc.recoveryTime = recovery || Date.now() + 3600000;
             }

package/dist/plugin.js

@@ -11,6 +11,7 @@ import { transformToCodeWhisperer } from './plugin/request';
 import { parseEventStream } from './plugin/response';
 import { startIDCAuthServer } from './plugin/server';
 import { migrateJsonToSqlite } from './plugin/storage/migration';
+import { kiroDb } from './plugin/storage/sqlite';
 import { transformKiroStream } from './plugin/streaming';
 import { syncFromKiroCli } from './plugin/sync/kiro-cli';
 import { refreshAccessToken } from './plugin/token';
@@ -176,6 +177,10 @@ export const createKiroPlugin = (id) => async ({ client, directory }) => {
                                     }, apiTimestamp);
                                 }
                                 if (res.ok) {
+                                    if (acc.failCount && acc.failCount > 0) {
+                                        acc.failCount = 0;
+                                        kiroDb.upsertAccount(acc);
+                                    }
                                     if (config.usage_tracking_enabled) {
                                         const sync = async (att = 0) => {
                                             try {
@@ -261,9 +266,35 @@ export const createKiroPlugin = (id) => async ({ client, directory }) => {
                                     continue;
                                 }
                                 if ((res.status === 402 || res.status === 403) && count > 1) {
-                                    am.markUnhealthy(acc, res.status === 402 ? 'Quota' : 'Forbidden');
-                                    await am.saveToDisk();
-                                    continue;
+                                    let errorReason = res.status === 402 ? 'Quota' : 'Forbidden';
+                                    let isPermanent = false;
+                                    let shouldSkipAccount = false;
+                                    try {
+                                        const errorBody = await res.text();
+                                        const errorData = JSON.parse(errorBody);
+                                        if (errorData.reason === 'TEMPORARILY_SUSPENDED') {
+                                            errorReason = 'Account Suspended';
+                                            isPermanent = true;
+                                            shouldSkipAccount = true;
+                                        }
+                                        else if (errorData.reason === 'INVALID_MODEL_ID') {
+                                            logger.warn(`Invalid model ID for ${acc.email}: ${errorData.message}`);
+                                            throw new Error(`Invalid model: ${errorData.message}`);
+                                        }
+                                    }
+                                    catch (e) {
+                                        if (e instanceof Error && e.message.includes('Invalid model')) {
+                                            throw e;
+                                        }
+                                    }
+                                    if (shouldSkipAccount) {
+                                        if (isPermanent) {
+                                            acc.failCount = 10;
+                                        }
+                                        am.markUnhealthy(acc, errorReason);
+                                        await am.saveToDisk();
+                                        continue;
+                                    }
                                 }
                                 const h = {};
                                 res.headers.forEach((v, k) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@zhafron/opencode-kiro-auth",
-  "version": "1.4.1",
+  "version": "1.4.2",
   "description": "OpenCode plugin for AWS Kiro (CodeWhisperer) providing access to Claude models",
   "type": "module",
   "main": "dist/index.js",