From 18a657188339ac96905f0eaadd0583fdd6001f65 Mon Sep 17 00:00:00 2001
From: JingSyue <144884872+JingSyue@users.noreply.github.com>
Date: Mon, 11 Nov 2024 12:59:29 +0800
Subject: [PATCH] Update proxy.ts

Update proxy.ts
---
 app/api/proxy.ts | 13 ++++++++++---
 1 file changed, 10 insertions(+), 3 deletions(-)

diff --git a/app/api/proxy.ts b/app/api/proxy.ts
index 40639fdc7..b3e5e7b7b 100644
--- a/app/api/proxy.ts
+++ b/app/api/proxy.ts
@@ -34,9 +34,16 @@ export async function handle(
     }),
   );
   // if dalle3 use openai api key
-  if (req.headers.get("x-base-url")?.includes("openai")) {
-    headers.set("Authorization", `Bearer ${serverConfig.apiKey}`);
-  }
+    const baseUrl = req.headers.get("x-base-url");
+    if (baseUrl?.includes("api.openai.com")) {
+      if (!serverConfig.apiKey) {
+        return NextResponse.json(
+          { error: "OpenAI API key not configured" },
+          { status: 500 },
+        );
+      }
+      headers.set("Authorization", `Bearer ${serverConfig.apiKey}`);
+    }
 
   const controller = new AbortController();
   const fetchOptions: RequestInit = {