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

fix: gemini test MaxTokens

zkp пре 11 месеци
родитељ
комит
f24de65626
1 измењених фајлова са 2 додато и 0 уклоњено
  1. 2 0
      controller/channel-test.go

+ 2 - 0
controller/channel-test.go

@@ -192,6 +192,8 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 		if !strings.Contains(model, "claude") {
 		if !strings.Contains(model, "claude") {
 			testRequest.MaxTokens = 50
 			testRequest.MaxTokens = 50
 		}
 		}
+	} else if strings.Contains(model, "gemini") {
+		testRequest.MaxTokens = 500
 	} else {
 	} else {
 		testRequest.MaxTokens = 10
 		testRequest.MaxTokens = 10
 	}
 	}