From 5135c9f71170db6d4a301fbe4ad1af9f14c1fe8d Mon Sep 17 00:00:00 2001
From: vc <127632570+v1cc0@users.noreply.github.com>
Date: Wed, 18 Oct 2023 02:37:46 +0800
Subject: [PATCH] Update relay-image.go

---
 controller/relay-image.go | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/controller/relay-image.go b/controller/relay-image.go
index 998a7851..f297e0b7 100644
--- a/controller/relay-image.go
+++ b/controller/relay-image.go
@@ -10,6 +10,7 @@ import (
 	"net/http"
 	"one-api/common"
 	"one-api/model"
+	"strings"
 
 	"github.com/gin-gonic/gin"
 )
@@ -61,16 +62,17 @@ func relayImageHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode
 			isModelMapped = true
 		}
 	}
-
 	baseURL := common.ChannelBaseURLs[channelType]
 	requestURL := c.Request.URL.String()
-
 	if c.GetString("base_url") != "" {
 		baseURL = c.GetString("base_url")
 	}
-
 	fullRequestURL := fmt.Sprintf("%s%s", baseURL, requestURL)
-
+	if channelType == common.ChannelTypeOpenAI {
+		if strings.HasPrefix(baseURL, "https://gateway.ai.cloudflare.com") {
+			fullRequestURL = fmt.Sprintf("%s%s", baseURL, strings.TrimPrefix(requestURL, "/v1"))
+		}
+	}
 	var requestBody io.Reader
 	if isModelMapped {
 		jsonStr, err := json.Marshal(imageRequest)