Update proxy.ts

littson · Nov 11, 2024 · 18a6571 · 18a6571
1 parent 2b0f2e5
commit 18a6571
Showing 1 changed file with 10 additions and 3 deletions.
diff --git 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 = {