test fix

Author: aadesh18

apps/backend/src/app/api/latest/ai/query/[mode]/route.ts

@@ -34,7 +34,7 @@ export const POST = createSmartRouteHandler({
 
 
     if (apiKey === "FORWARD_TO_PRODUCTION") {
-      const prodResponse = await forwardToProduction(fullReq.headers, mode, body);
+      const prodResponse = await forwardToProduction(mode, body);
       return {
         statusCode: prodResponse.status,
         bodyType: "response" as const,

apps/backend/src/lib/ai/forward.ts

@@ -1,24 +1,11 @@
 import { type RequestBody } from "@/lib/ai/schema";
 
 export async function forwardToProduction(
-  requestHeaders: Record<string, string[] | undefined>,
   mode: "stream" | "generate",
   body: RequestBody,
 ): Promise<Response> {
   const productionUrl = `https://api.stack-auth.com/api/latest/ai/query/${mode}`;
-  const allowedHeaders = new Set([
-    "x-stack-access-type",
-    "x-stack-access-token",
-    "x-stack-publishable-client-key",
-  ]);
-
   const forwardHeaders = new Headers();
-  for (const [key, values] of Object.entries(requestHeaders)) {
-    if (values == null) continue;
-    const lowerKey = key.toLowerCase();
-    if (!allowedHeaders.has(lowerKey)) continue;
-    forwardHeaders.set(lowerKey, values[0] ?? "");
-  }
 
   forwardHeaders.set("content-type", "application/json");
 

apps/e2e/tests/backend/endpoints/api/v1/ai-query.test.ts

@@ -1,7 +1,7 @@
+import { getEnvVariable } from "@stackframe/stack-shared/dist/utils/env";
 import { describe } from "vitest";
 import { it } from "../../../../helpers";
 import { niceBackendFetch, Project } from "../../../backend-helpers";
-import { getEnvVariable } from "@stackframe/stack-shared/dist/utils/env";
 
 const hasRealAiKey = (() => {
   const key = getEnvVariable("STACK_OPENROUTER_API_KEY", "");
@@ -210,7 +210,7 @@ describe("AI Query Endpoint - Validation", () => {
       },
     });
 
-    expect(response.status).not.toBe(400);
+    expect(response.body).not.toMatchObject({ code: "SCHEMA_ERROR" });
 
   }, 10000); // 60 seconds for AI API call
 });