Просмотр исходного кода

fix: update MaxCompletionTokens for model prefix handling in buildTestRequest function

CalciumIon 1 год назад
Родитель
Сommit
ed4d26fc9e
1 измененных файлов с 2 добавлено и 2 удалено
  1. 2 2
      controller/channel-test.go

+ 2 - 2
controller/channel-test.go

@@ -152,8 +152,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 		Model:  "", // this will be set later
 		Model:  "", // this will be set later
 		Stream: false,
 		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") {
 	} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
 		testRequest.MaxTokens = 2
 		testRequest.MaxTokens = 2
 	} else {
 	} else {