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] 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
 	}