fix: nextjs handler x-fal header passthrough

fal-ai · Sep 27, 2023 · a2166cd · a2166cd
1 parent 2e7a4d0
commit a2166cd
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/libs/nextjs/package.json b/libs/nextjs/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@fal-ai/serverless-nextjs",
   "description": "The fal-serverless Next.js integration",
-  "version": "0.2.1",
+  "version": "0.2.2",
   "license": "MIT",
   "repository": {
     "type": "git",

diff --git a/libs/nextjs/src/handler.ts b/libs/nextjs/src/handler.ts
@@ -71,9 +71,18 @@ export const handler: NextApiHandler = async (request, response) => {
     return;
   }
 
+  // pass over headers prefixed with x-fal-*
+  const headers: Record<string, string | string[] | undefined> = {};
+  Object.keys(request.headers).forEach((key) => {
+    if (key.toLowerCase().startsWith('x-fal-')) {
+      headers[key.toLowerCase()] = request.headers[key];
+    }
+  });
+
   const res = await fetch(targetUrl, {
     method: request.method,
     headers: {
+      ...headers,
       authorization: `Key ${falKey}`,
       accept: 'application/json',
       'content-type': 'application/json',