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

Update MaxTokens for gemini model to 300 in test request

CaIon 11 месяцев назад
Родитель
Сommit
79b35e385f
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      controller/channel-test.go

+ 1 - 1
controller/channel-test.go

@@ -193,7 +193,7 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 			testRequest.MaxTokens = 50
 		}
 	} else if strings.Contains(model, "gemini") {
-		testRequest.MaxTokens = 50
+		testRequest.MaxTokens = 300
 	} else {
 		testRequest.MaxTokens = 10
 	}