fix: gemini test MaxTokens

2025-04-06 23:23:53 +08:00 · 2025-04-06 23:23:53 +08:00 · 93d9f141fe
commit 93d9f141fe
parent 69d790b47a
1 changed files with 2 additions and 0 deletions
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@ -192,6 +192,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 		if !strings.Contains(model, "claude") {
 			testRequest.MaxTokens = 50
 		}
+	} else if strings.Contains(model, "gemini") {
+		testRequest.MaxTokens = 500
 	} else {
 		testRequest.MaxTokens = 10
 	}