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

feat: 支持ollama embedding数组传参

CaIon 1 год назад
Родитель
Сommit
6b97842f78
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      relay/channel/ollama/relay-ollama.go

+ 2 - 1
relay/channel/ollama/relay-ollama.go

@@ -9,6 +9,7 @@ import (
 	"net/http"
 	"one-api/dto"
 	"one-api/service"
+	"strings"
 )
 
 func requestOpenAI2Ollama(request dto.GeneralOpenAIRequest) *OllamaRequest {
@@ -41,7 +42,7 @@ func requestOpenAI2Ollama(request dto.GeneralOpenAIRequest) *OllamaRequest {
 func requestOpenAI2Embeddings(request dto.GeneralOpenAIRequest) *OllamaEmbeddingRequest {
 	return &OllamaEmbeddingRequest{
 		Model:  request.Model,
-		Prompt: request.Input,
+		Prompt: strings.Join(request.ParseInput(), " "),
 	}
 }