| | package relay
|
| |
|
| | import (
|
| | "bytes"
|
| | "encoding/json"
|
| | "fmt"
|
| | "net/http"
|
| |
|
| | "github.com/QuantumNous/new-api/common"
|
| | "github.com/QuantumNous/new-api/dto"
|
| | "github.com/QuantumNous/new-api/logger"
|
| | relaycommon "github.com/QuantumNous/new-api/relay/common"
|
| | "github.com/QuantumNous/new-api/relay/helper"
|
| | "github.com/QuantumNous/new-api/service"
|
| | "github.com/QuantumNous/new-api/types"
|
| |
|
| | "github.com/gin-gonic/gin"
|
| | )
|
| |
|
| | func EmbeddingHelper(c *gin.Context, info *relaycommon.RelayInfo) (newAPIError *types.NewAPIError) {
|
| | info.InitChannelMeta(c)
|
| |
|
| | embeddingReq, ok := info.Request.(*dto.EmbeddingRequest)
|
| | if !ok {
|
| | return types.NewErrorWithStatusCode(fmt.Errorf("invalid request type, expected *dto.EmbeddingRequest, got %T", info.Request), types.ErrorCodeInvalidRequest, http.StatusBadRequest, types.ErrOptionWithSkipRetry())
|
| | }
|
| |
|
| | request, err := common.DeepCopy(embeddingReq)
|
| | if err != nil {
|
| | return types.NewError(fmt.Errorf("failed to copy request to EmbeddingRequest: %w", err), types.ErrorCodeInvalidRequest, types.ErrOptionWithSkipRetry())
|
| | }
|
| |
|
| | err = helper.ModelMappedHelper(c, info, request)
|
| | if err != nil {
|
| | return types.NewError(err, types.ErrorCodeChannelModelMappedError, types.ErrOptionWithSkipRetry())
|
| | }
|
| |
|
| | adaptor := GetAdaptor(info.ApiType)
|
| | if adaptor == nil {
|
| | return types.NewError(fmt.Errorf("invalid api type: %d", info.ApiType), types.ErrorCodeInvalidApiType, types.ErrOptionWithSkipRetry())
|
| | }
|
| | adaptor.Init(info)
|
| |
|
| | convertedRequest, err := adaptor.ConvertEmbeddingRequest(c, info, *request)
|
| | if err != nil {
|
| | return types.NewError(err, types.ErrorCodeConvertRequestFailed, types.ErrOptionWithSkipRetry())
|
| | }
|
| | relaycommon.AppendRequestConversionFromRequest(info, convertedRequest)
|
| | jsonData, err := json.Marshal(convertedRequest)
|
| | if err != nil {
|
| | return types.NewError(err, types.ErrorCodeConvertRequestFailed, types.ErrOptionWithSkipRetry())
|
| | }
|
| |
|
| | if len(info.ParamOverride) > 0 {
|
| | jsonData, err = relaycommon.ApplyParamOverride(jsonData, info.ParamOverride, relaycommon.BuildParamOverrideContext(info))
|
| | if err != nil {
|
| | return types.NewError(err, types.ErrorCodeChannelParamOverrideInvalid, types.ErrOptionWithSkipRetry())
|
| | }
|
| | }
|
| |
|
| | logger.LogDebug(c, fmt.Sprintf("converted embedding request body: %s", string(jsonData)))
|
| | requestBody := bytes.NewBuffer(jsonData)
|
| | statusCodeMappingStr := c.GetString("status_code_mapping")
|
| | resp, err := adaptor.DoRequest(c, info, requestBody)
|
| | if err != nil {
|
| | return types.NewOpenAIError(err, types.ErrorCodeDoRequestFailed, http.StatusInternalServerError)
|
| | }
|
| |
|
| | var httpResp *http.Response
|
| | if resp != nil {
|
| | httpResp = resp.(*http.Response)
|
| | if httpResp.StatusCode != http.StatusOK {
|
| | newAPIError = service.RelayErrorHandler(c.Request.Context(), httpResp, false)
|
| |
|
| | service.ResetStatusCode(newAPIError, statusCodeMappingStr)
|
| | return newAPIError
|
| | }
|
| | }
|
| |
|
| | usage, newAPIError := adaptor.DoResponse(c, httpResp, info)
|
| | if newAPIError != nil {
|
| |
|
| | service.ResetStatusCode(newAPIError, statusCodeMappingStr)
|
| | return newAPIError
|
| | }
|
| | postConsumeQuota(c, info, usage.(*dto.Usage))
|
| | return nil
|
| | }
|
| |
|
