From f24de65626c16e78b4edd2e60f226d92f56b38ee Mon Sep 17 00:00:00 2001 From: zkp <2470904132@qq.com> Date: Sun, 6 Apr 2025 23:23:53 +0800 Subject: [PATCH 1/2] fix: gemini test MaxTokens --- controller/channel-test.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/controller/channel-test.go b/controller/channel-test.go index 5bc733d4..a549480e 100644 --- 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 } From 30f32c6a6dac6916d62f0014b9a79e256fac79f8 Mon Sep 17 00:00:00 2001 From: Calcium-Ion <61247483+Calcium-Ion@users.noreply.github.com> Date: Tue, 8 Apr 2025 17:33:10 +0800 Subject: [PATCH 2/2] Set MaxTokens to 50 for gemini --- controller/channel-test.go | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/controller/channel-test.go b/controller/channel-test.go index a549480e..90556924 100644 --- 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 }