feat: support gemini-2.0-flash-thinking #639 #637

2024-12-20 13:20:07 +08:00
parent 2d865eb735
commit 9a54b345c5
4 changed files with 16 additions and 4 deletions
--- a/controller/channel-test.go
+++ b/controller/channel-test.go
@@ -153,6 +153,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 	}
 	if strings.HasPrefix(model, "o1-") {
 		testRequest.MaxCompletionTokens = 1
+	} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
+		testRequest.MaxTokens = 2
 	} else {
 		testRequest.MaxTokens = 1
 	}