(c *gin.Context, resp *http.Response)
| 40 | } |
| 41 | |
| 42 | func aliEmbeddingHandler(c *gin.Context, resp *http.Response) (*types.NewAPIError, *dto.Usage) { |
| 43 | var fullTextResponse dto.FlexibleEmbeddingResponse |
| 44 | err := json.NewDecoder(resp.Body).Decode(&fullTextResponse) |
| 45 | if err != nil { |
| 46 | return types.NewError(err, types.ErrorCodeBadResponseBody), nil |
| 47 | } |
| 48 | |
| 49 | common.CloseResponseBodyGracefully(resp) |
| 50 | |
| 51 | model := c.GetString("model") |
| 52 | if model == "" { |
| 53 | model = "text-embedding-v4" |
| 54 | } |
| 55 | jsonResponse, err := json.Marshal(fullTextResponse) |
| 56 | if err != nil { |
| 57 | return types.NewError(err, types.ErrorCodeBadResponseBody), nil |
| 58 | } |
| 59 | c.Writer.Header().Set("Content-Type", "application/json") |
| 60 | c.Writer.WriteHeader(resp.StatusCode) |
| 61 | c.Writer.Write(jsonResponse) |
| 62 | return nil, &fullTextResponse.Usage |
| 63 | } |
| 64 | |
| 65 | func embeddingResponseAli2OpenAI(response *AliEmbeddingResponse, model string) *dto.OpenAIEmbeddingResponse { |
| 66 | openAIEmbeddingResponse := dto.OpenAIEmbeddingResponse{ |
no test coverage detected