f*** o3-mini

2025-02-01 14:11:34 +08:00
parent 030ffd5f2d
commit b80c1ee3a4
3 changed files with 7 additions and 4 deletions
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -156,10 +156,10 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 		Model:  "", // this will be set later
 		Stream: false,
 	}
-	if strings.HasPrefix(model, "o1") {
+	if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
 		testRequest.MaxCompletionTokens = 10
 	} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
-		testRequest.MaxTokens = 2
+		testRequest.MaxTokens = 10
 	} else {
 		testRequest.MaxTokens = 1
 	}