diff --git a/relay/adaptor/cloudflare/main.go b/relay/adaptor/cloudflare/main.go index e164d473..980a2891 100644 --- a/relay/adaptor/cloudflare/main.go +++ b/relay/adaptor/cloudflare/main.go @@ -19,8 +19,9 @@ import ( ) func ConvertCompletionsRequest(textRequest model.GeneralOpenAIRequest) *Request { + p, _ := textRequest.Prompt.(string) return &Request{ - Prompt: textRequest.Prompt, + Prompt: p, MaxTokens: textRequest.MaxTokens, Stream: textRequest.Stream, Temperature: textRequest.Temperature, diff --git a/relay/billing/ratio/model.go b/relay/billing/ratio/model.go index a5428057..6ef2a457 100644 --- a/relay/billing/ratio/model.go +++ b/relay/billing/ratio/model.go @@ -337,7 +337,7 @@ var ModelRatio = map[string]float64{ "command-r": 0.5 / 1000 * USD, "command-r-plus": 3.0 / 1000 * USD, // https://platform.deepseek.com/api-docs/pricing/ - "deepseek-chat": 0.27 * MILLI_USD, + "deepseek-chat": 0.14 * MILLI_USD, "deepseek-reasoner": 0.55 * MILLI_USD, // https://www.deepl.com/pro?cta=header-prices "deepl-zh": 25.0 / 1000 * USD, @@ -630,7 +630,7 @@ var CompletionRatio = map[string]float64{ // whisper "whisper-1": 0, // only count input tokens // deepseek - "deepseek-chat": 1.1 / 0.27, + "deepseek-chat": 0.28 / 0.14, "deepseek-reasoner": 2.19 / 0.55, } @@ -784,6 +784,9 @@ func GetCompletionRatio(name string, channelType int) float64 { if strings.HasPrefix(name, "gemini-") { return 3 } + if strings.HasPrefix(name, "deepseek-") { + return 2 + } switch name { case "llama2-70b-4096": diff --git a/relay/controller/text.go b/relay/controller/text.go index 72082629..f912498a 100644 --- a/relay/controller/text.go +++ b/relay/controller/text.go @@ -33,10 +33,6 @@ func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode { } meta.IsStream = textRequest.Stream - if reqBody, ok := c.Get(ctxkey.KeyRequestBody); ok { - logger.Debugf(c.Request.Context(), "get text request: %s\n", string(reqBody.([]byte))) - } - // map model name meta.OriginModelName = textRequest.Model textRequest.Model, _ = getMappedModelName(textRequest.Model, meta.ModelMapping) diff --git a/relay/model/general.go b/relay/model/general.go index 1b3e3189..a87928bd 100644 --- a/relay/model/general.go +++ b/relay/model/general.go @@ -61,7 +61,7 @@ type GeneralOpenAIRequest struct { EncodingFormat string `json:"encoding_format,omitempty"` Dimensions int `json:"dimensions,omitempty"` // https://platform.openai.com/docs/api-reference/images/create - Prompt string `json:"prompt,omitempty"` + Prompt any `json:"prompt,omitempty"` Quality *string `json:"quality,omitempty"` Size string `json:"size,omitempty"` Style *string `json:"style,omitempty"`