From 9afcbf556e45e3019b486e293de5db3bea4b750a Mon Sep 17 00:00:00 2001
From: JustSong <songquanpeng@foxmail.com>
Date: Fri, 17 Nov 2023 20:01:39 +0800
Subject: [PATCH] chore: update impl

---
 common/model-ratio.go     | 2 +-
 controller/relay-image.go | 3 +--
 controller/relay.go       | 2 ++
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/common/model-ratio.go b/common/model-ratio.go
index 954ac3e4..4ae92eeb 100644
--- a/common/model-ratio.go
+++ b/common/model-ratio.go
@@ -65,7 +65,7 @@ var ModelRatio = map[string]float64{
 	"text-moderation-stable":    0.1,
 	"text-moderation-latest":    0.1,
 	"dall-e-2":                  8,      // $0.016 - $0.020 / image
-	"dall-e-3":                  20,     // ￥0.040 - ￥0.120 / image
+	"dall-e-3":                  20,     // $0.040 - $0.120 / image
 	"claude-instant-1":          0.815,  // $1.63 / 1M tokens
 	"claude-2":                  5.51,   // $11.02 / 1M tokens
 	"ERNIE-Bot":                 0.8572, // ￥0.012 / 1k tokens
diff --git a/controller/relay-image.go b/controller/relay-image.go
index 8ee07670..1d1b71ba 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -37,7 +37,6 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 	group := c.GetString("group")
 
 	var imageRequest ImageRequest
-
 	if consumeQuota {
 		err := common.UnmarshalBodyReusable(c, &imageRequest)
 		if err != nil {
@@ -82,7 +81,7 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 
 	// Number of generated images validation
 	if isWithinRange(imageModel, imageRequest.N) == false {
-		return errorWrapper(errors.New("invalud value of n"), "n_not_within_range", http.StatusBadRequest)
+		return errorWrapper(errors.New("invalid value of n"), "n_not_within_range", http.StatusBadRequest)
 	}
 
 	// map model name
diff --git a/controller/relay.go b/controller/relay.go
index 99d61379..9cff887b 100644
--- a/controller/relay.go
+++ b/controller/relay.go
@@ -77,6 +77,7 @@ type TextRequest struct {
 	//Stream   bool      `json:"stream"`
 }
 
+// ImageRequest docs: https://platform.openai.com/docs/api-reference/images/create
 type ImageRequest struct {
 	Model          string `json:"model"`
 	Prompt         string `json:"prompt" binding:"required"`
@@ -85,6 +86,7 @@ type ImageRequest struct {
 	Quality        string `json:"quality"`
 	ResponseFormat string `json:"response_format"`
 	Style          string `json:"style"`
+	User           string `json:"user"`
 }
 
 type AudioResponse struct {