Set MaxTokens to 50 for gemini

950572b0 · Calcium-Ion · GitHub · 93d9f141 · 950572b0
Commit 950572b0 authored Apr 08, 2025 by Calcium-Ion Committed by GitHub Apr 08, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

controller/channel-test.go
+1 -1

No files found.
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -193,7 +193,7 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 			testRequest.MaxTokens = 50
 		}
 	} else if strings.Contains(model, "gemini") {
-		testRequest.MaxTokens = 500
+		testRequest.MaxTokens = 50
 	} else {
 		testRequest.MaxTokens = 10
 	}