feat: add client availability tracking and error handling improvements

- Introduced `IsAvailable` and `SetUnavailable` methods to clients for availability tracking.
- Integrated availability checks in client selection logic to skip unavailable clients.
- Enhanced error handling by marking clients unavailable on specific error codes (e.g., 401, 402).
- Removed redundant quota verification logs in client reordering logic.

internal/api/handlers/claude/code_handlers.go

@@ -205,9 +205,13 @@ outLoop:
 						err := cliClient.RefreshTokens(cliCtx)
 						if err != nil {
 							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
 						}
 						retryCount++
 						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(errInfo.StatusCode)

internal/api/handlers/gemini/gemini-cli_handlers.go

@@ -221,6 +221,18 @@ outLoop:
 						log.Debugf("http status code %d, switch client", err.StatusCode)
 						retryCount++
 						continue outLoop
+					case 401:
+						log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+						errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+						if errRefreshTokens != nil {
+							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
+						}
+						retryCount++
+						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(err.StatusCode)
@@ -293,9 +305,13 @@ func (h *GeminiCLIAPIHandler) handleInternalGenerateContent(c *gin.Context, rawJ
 				errRefreshTokens := cliClient.RefreshTokens(cliCtx)
 				if errRefreshTokens != nil {
 					log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+					cliClient.SetUnavailable()
 				}
 				retryCount++
 				continue
+			case 402:
+				cliClient.SetUnavailable()
+				continue
 			default:
 				// Forward other errors directly to the client
 				c.Status(err.StatusCode)

internal/api/handlers/gemini/gemini_handlers.go

@@ -276,6 +276,18 @@ outLoop:
 						log.Debugf("http status code %d, switch client", err.StatusCode)
 						retryCount++
 						continue outLoop
+					case 401:
+						log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+						errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+						if errRefreshTokens != nil {
+							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
+						}
+						retryCount++
+						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(err.StatusCode)
@@ -406,9 +418,13 @@ func (h *GeminiAPIHandler) handleGenerateContent(c *gin.Context, modelName strin
 				errRefreshTokens := cliClient.RefreshTokens(cliCtx)
 				if errRefreshTokens != nil {
 					log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+					cliClient.SetUnavailable()
 				}
 				retryCount++
 				continue
+			case 402:
+				cliClient.SetUnavailable()
+				continue
 			default:
 				// Forward other errors directly to the client
 				c.Status(err.StatusCode)

internal/api/handlers/handlers.go

@@ -8,11 +8,9 @@ import (
 	"sync"
 
 	"github.com/gin-gonic/gin"
-	"github.com/luispater/CLIProxyAPI/v5/internal/client"
 	"github.com/luispater/CLIProxyAPI/v5/internal/config"
 	"github.com/luispater/CLIProxyAPI/v5/internal/interfaces"
 	"github.com/luispater/CLIProxyAPI/v5/internal/util"
-	log "github.com/sirupsen/logrus"
 	"golang.org/x/net/context"
 )
 
@@ -97,7 +95,7 @@ func (h *BaseAPIHandler) UpdateClients(clients []interfaces.Client, cfg *config.
 func (h *BaseAPIHandler) GetClient(modelName string, isGenerateContent ...bool) (interfaces.Client, *interfaces.ErrorMessage) {
 	clients := make([]interfaces.Client, 0)
 	for i := 0; i < len(h.CliClients); i++ {
-		if h.CliClients[i].CanProvideModel(modelName) {
+		if h.CliClients[i].CanProvideModel(modelName) && h.CliClients[i].IsAvailable() && !h.CliClients[i].IsModelQuotaExceeded(modelName) {
 			clients = append(clients, h.CliClients[i])
 		}
 	}
@@ -126,24 +124,6 @@ func (h *BaseAPIHandler) GetClient(modelName string, isGenerateContent ...bool)
 	reorderedClients := make([]interfaces.Client, 0)
 	for i := 0; i < len(clients); i++ {
 		cliClient = clients[(startIndex+1+i)%len(clients)]
-		if cliClient.IsModelQuotaExceeded(modelName) {
-			if cliClient.Provider() == "gemini-cli" {
-				log.Debugf("Gemini Model %s is quota exceeded for account %s, project id: %s", modelName, cliClient.GetEmail(), cliClient.(*client.GeminiCLIClient).GetProjectID())
-			} else if cliClient.Provider() == "gemini" {
-				log.Debugf("Gemini Model %s is quota exceeded for account %s", modelName, cliClient.GetEmail())
-			} else if cliClient.Provider() == "codex" {
-				log.Debugf("Codex Model %s is quota exceeded for account %s", modelName, cliClient.GetEmail())
-			} else if cliClient.Provider() == "claude" {
-				log.Debugf("Claude Model %s is quota exceeded for account %s", modelName, cliClient.GetEmail())
-			} else if cliClient.Provider() == "qwen" {
-				log.Debugf("Qwen Model %s is quota exceeded for account %s", modelName, cliClient.GetEmail())
-			} else if cliClient.Type() == "openai-compatibility" {
-				log.Debugf("OpenAI Compatibility Model %s is quota exceeded for provider %s", modelName, cliClient.Provider())
-			}
-			cliClient = nil
-			continue
-
-		}
 		reorderedClients = append(reorderedClients, cliClient)
 	}
 

internal/api/handlers/openai/openai_handlers.go

@@ -442,9 +442,13 @@ func (h *OpenAIAPIHandler) handleNonStreamingResponse(c *gin.Context, rawJSON []
 				errRefreshTokens := cliClient.RefreshTokens(cliCtx)
 				if errRefreshTokens != nil {
 					log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+					cliClient.SetUnavailable()
 				}
 				retryCount++
 				continue
+			case 402:
+				cliClient.SetUnavailable()
+				continue
 			default:
 				// Forward other errors directly to the client
 				c.Status(err.StatusCode)
@@ -557,6 +561,18 @@ outLoop:
 						log.Debugf("http status code %d, switch client", err.StatusCode)
 						retryCount++
 						continue outLoop
+					case 401:
+						log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+						errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+						if errRefreshTokens != nil {
+							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
+						}
+						retryCount++
+						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(err.StatusCode)
@@ -632,6 +648,18 @@ func (h *OpenAIAPIHandler) handleCompletionsNonStreamingResponse(c *gin.Context,
 				log.Debugf("http status code %d, switch client", err.StatusCode)
 				retryCount++
 				continue
+			case 401:
+				log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+				errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+				if errRefreshTokens != nil {
+					log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+					cliClient.SetUnavailable()
+				}
+				retryCount++
+				continue
+			case 402:
+				cliClient.SetUnavailable()
+				continue
 			default:
 				// Forward other errors directly to the client
 				c.Status(err.StatusCode)
@@ -755,6 +783,18 @@ outLoop:
 						log.Debugf("http status code %d, switch client", err.StatusCode)
 						retryCount++
 						continue outLoop
+					case 401:
+						log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+						errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+						if errRefreshTokens != nil {
+							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
+						}
+						retryCount++
+						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(err.StatusCode)

internal/api/handlers/openai/openai_responses_handlers.go

@@ -146,9 +146,13 @@ func (h *OpenAIResponsesAPIHandler) handleNonStreamingResponse(c *gin.Context, r
 				errRefreshTokens := cliClient.RefreshTokens(cliCtx)
 				if errRefreshTokens != nil {
 					log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+					cliClient.SetUnavailable()
 				}
 				retryCount++
 				continue
+			case 402:
+				cliClient.SetUnavailable()
+				continue
 			default:
 				// Forward other errors directly to the client
 				c.Status(err.StatusCode)
@@ -260,6 +264,18 @@ outLoop:
 						log.Debugf("http status code %d, switch client", err.StatusCode)
 						retryCount++
 						continue outLoop
+					case 401:
+						log.Debugf("unauthorized request, try to refresh token, %s", util.HideAPIKey(cliClient.GetEmail()))
+						errRefreshTokens := cliClient.RefreshTokens(cliCtx)
+						if errRefreshTokens != nil {
+							log.Debugf("refresh token failed, switch client, %s", util.HideAPIKey(cliClient.GetEmail()))
+							cliClient.SetUnavailable()
+						}
+						retryCount++
+						continue outLoop
+					case 402:
+						cliClient.SetUnavailable()
+						continue outLoop
 					default:
 						// Forward other errors directly to the client
 						c.Status(err.StatusCode)