From ed4d26fc9ef9b0b4b1dc86b25779b324f9e71759 Mon Sep 17 00:00:00 2001
From: CalciumIon <1808837298@qq.com>
Date: Wed, 25 Dec 2024 17:55:20 +0800
Subject: [PATCH] fix: update MaxCompletionTokens for model prefix handling in
 buildTestRequest function

---
 controller/channel-test.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/controller/channel-test.go b/controller/channel-test.go
index 53c56aab..73f9bb6d 100644
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -152,8 +152,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 		Model:  "", // this will be set later
 		Stream: false,
 	}
-	if strings.HasPrefix(model, "o1-") {
-		testRequest.MaxCompletionTokens = 1
+	if strings.HasPrefix(model, "o1") {
+		testRequest.MaxCompletionTokens = 10
 	} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
 		testRequest.MaxTokens = 2
 	} else {