From 93d9f141fe3a98218b0a52438199380fde9abb89 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 950572b02ebb5764a58e771243adf40b871e3fa9 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 }