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

fix: Refine embedding model detection in channel test

1808837298@qq.com 1 год назад
Родитель
Сommit
4a8bb625b8
1 измененных файлов с 3 добавлено и 3 удалено
  1. 3 3
      controller/channel-test.go

+ 3 - 3
controller/channel-test.go

@@ -175,10 +175,10 @@ func buildTestRequest(model string) *dto.GeneralOpenAIRequest {
 	}
 
 	// 先判断是否为 Embedding 模型
-	if strings.Contains(strings.ToLower(model), "embedding") ||
+	if strings.Contains(strings.ToLower(model), "embedding") || // 其他 embedding 模型
 		strings.HasPrefix(model, "m3e") || // m3e 系列模型
-		strings.Contains(model, "bge-") || // bge 系列模型
-		model == "text-embedding-v1" { // 其他 embedding 模型
+		strings.Contains(model, "bge-") {
+		testRequest.Model = model
 		// Embedding 请求
 		testRequest.Input = []string{"hello world"}
 		return testRequest