From c520580cdabbe53acb20b94c81532a7be6c20cce Mon Sep 17 00:00:00 2001
From: Lukin <mylukin@gmail.com>
Date: Wed, 6 Mar 2024 13:17:10 +0800
Subject: [PATCH] fix(server): compatible with gateway.ai.cloudflare.com
 (#2029)

---
 service/src/chatgpt/index.ts | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/service/src/chatgpt/index.ts b/service/src/chatgpt/index.ts
index 9cb0313b61..7f50fd9bf3 100644
--- a/service/src/chatgpt/index.ts
+++ b/service/src/chatgpt/index.ts
@@ -70,8 +70,13 @@ let api: ChatGPTAPI | ChatGPTUnofficialProxyAPI
       }
     }
 
-    if (isNotEmptyString(OPENAI_API_BASE_URL))
-      options.apiBaseUrl = `${OPENAI_API_BASE_URL}/v1`
+    if (isNotEmptyString(OPENAI_API_BASE_URL)) {
+      // if find /v1 in OPENAI_API_BASE_URL then use it
+      if (OPENAI_API_BASE_URL.includes('/v1'))
+        options.apiBaseUrl = `${OPENAI_API_BASE_URL}`
+      else
+        options.apiBaseUrl = `${OPENAI_API_BASE_URL}/v1`
+    }
 
     setupProxy(options)