diff --git a/relay/channel/ollama/main.go b/relay/channel/ollama/main.go index b422ab9a..821a335b 100644 --- a/relay/channel/ollama/main.go +++ b/relay/channel/ollama/main.go @@ -171,7 +171,7 @@ func EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStat }, nil } - fullTextResponse := embeddingResponseAli2OpenAI(&ollamaResponse) + fullTextResponse := embeddingResponseOllama2OpenAI(&ollamaResponse) jsonResponse, err := json.Marshal(fullTextResponse) if err != nil { return openai.ErrorWrapper(err, "marshal_response_body_failed", http.StatusInternalServerError), nil @@ -182,7 +182,7 @@ func EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStat return nil, &fullTextResponse.Usage } -func embeddingResponseAli2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse { +func embeddingResponseOllama2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse { openAIEmbeddingResponse := openai.EmbeddingResponse{ Object: "list", Data: make([]openai.EmbeddingResponseItem, 0, 1),