🔖 chore: update Gemini retry/disable rules

MartialBE · Dec 28, 2024 · faa5090 · faa5090
1 parent a8860a9
commit faa5090
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/controller/common.go b/controller/common.go
@@ -31,6 +31,7 @@ var containsKeywords = []string{
 	"You exceeded your current quota",                       // openai
 	"Permission denied",                                     // gcp
 	"Quota exceeded for quota metric",                       // gemini
+	"API key not valid",                                     // gemini
 	"The security token included in the request is invalid", //AWS
 	"Operation not allowed",                                 //AWS
 	"Your account is not authorized",                        //AWS

diff --git a/relay/common.go b/relay/common.go
@@ -386,6 +386,10 @@ func shouldRetryBadRequest(channelType int, apiErr *types.OpenAIErrorWithStatusC
 	case config.ChannelTypeBedrock:
 		return strings.Contains(apiErr.OpenAIError.Message, "Operation not allowed")
 	default:
+		// gemini
+		if apiErr.OpenAIError.Param == "INVALID_ARGUMENT" && strings.Contains(apiErr.OpenAIError.Message, "API key not valid") {
+			return true
+		}
 		return false
 	}
 }