Merge pull request #936 from lamcodes/main

fix: gemini test MaxTokens

Merge pull request #936 from lamcodes/main
fix: gemini test MaxTokens
ba9cd931 · Calcium-Ion · GitHub · d85e25fc · 950572b0 · ba9cd931
Commit ba9cd931 authored Apr 08, 2025 by Calcium-Ion Committed by GitHub Apr 08, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

controller/channel-test.go
+2 -0

No files found.
--- 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 = 50
 	} else {
 		testRequest.MaxTokens = 10
 	}