fix async log auth check (#1916)

Helicone · May 14, 2024 · c16e6b7 · c16e6b7
1 parent 83a0c27
commit c16e6b7
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/worker/src/lib/managers/AsyncLogManager.ts b/worker/src/lib/managers/AsyncLogManager.ts
@@ -13,6 +13,16 @@ import { S3Client } from "../clients/S3Client";
 import { RequestResponseManager } from "./RequestResponseManager";
 import { KafkaProducer } from "../clients/KafkaProducer";
 
+function mergeHeaders(x: Headers, y: Headers) {
+  const merged = new Headers();
+  for (const [key, value] of x.entries()) {
+    merged.set(key, value);
+  }
+  for (const [key, value] of y.entries()) {
+    merged.set(key, value);
+  }
+  return merged;
+}
 export async function logAsync(
   requestWrapper: RequestWrapper,
   env: Env,
@@ -37,7 +47,9 @@ export async function logAsync(
 
   const requestHeaders = new Headers(asyncLogModel.providerRequest.meta);
   const responseHeaders = new Headers(asyncLogModel.providerResponse.headers);
-  const heliconeHeaders = new HeliconeHeaders(requestHeaders);
+  const heliconeHeaders = new HeliconeHeaders(
+    mergeHeaders(requestHeaders, requestWrapper.headers)
+  );
 
   const loggable = await dbLoggableRequestFromAsyncLogModel({
     requestWrapper,