fix(server): compatible with gateway.ai.cloudflare.com (#2029)

Chanzhaoyu · Mar 6, 2024 · c520580 · c520580
1 parent 15903fa
commit c520580
Showing 1 changed file with 7 additions and 2 deletions.
diff --git 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)