Преглед изворни кода

fix: adjust max tokens configuration in test request builder

- Update max tokens default value to 10
1808837298@qq.com пре 1 година
родитељ
комит
cf0ff0371b
1 измењених фајлова са 1 додато и 3 уклоњено
  1. 1 3
      controller/channel-test.go

+ 1 - 3
controller/channel-test.go

@@ -158,10 +158,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 	}
 	}
 	if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
 	if strings.HasPrefix(model, "o1") || strings.HasPrefix(model, "o3") {
 		testRequest.MaxCompletionTokens = 10
 		testRequest.MaxCompletionTokens = 10
-	} else if strings.HasPrefix(model, "gemini-2.0-flash-thinking") {
-		testRequest.MaxTokens = 10
 	} else {
 	} else {
-		testRequest.MaxTokens = 1
+		testRequest.MaxTokens = 10
 	}
 	}
 	content, _ := json.Marshal("hi")
 	content, _ := json.Marshal("hi")
 	testMessage := dto.Message{
 	testMessage := dto.Message{