use bodies

Author: colegottdank

worker/test/ai-gateway/pass-through.spec.ts

@@ -92,7 +92,7 @@ describe("Pass-Through Tests", () => {
                 expects: {
                   headers: {
                     // Without NO_MAPPING, Anthropic uses OpenAI compatibility mode with Authorization header
-                    "Authorization": /^Bearer /,
+                    Authorization: /^Bearer /,
                   },
                 },
               },
@@ -315,7 +315,8 @@ describe("Pass-Through Tests", () => {
 
       it("should route to Bedrock endpoint with ap-southeast prefix", () =>
         runGatewayTest({
-          model: "ap-southeast-2.anthropic.claude-3-7-sonnet-20250219-v1:0/bedrock",
+          model:
+            "ap-southeast-2.anthropic.claude-3-7-sonnet-20250219-v1:0/bedrock",
           request: {
             messages: [{ role: "user", content: "Test APAC Bedrock" }],
             maxTokens: 100,
@@ -326,7 +327,8 @@ describe("Pass-Through Tests", () => {
                 // Note: AP prefix in model doesn't change the region, still uses us-east-1
                 url: "https://bedrock-runtime.us-east-1.amazonaws.com/model/ap-southeast-2.anthropic.claude-3-7-sonnet-20250219-v1:0/invoke",
                 response: "success",
-                model: "ap-southeast-2.anthropic.claude-3-7-sonnet-20250219-v1:0",
+                model:
+                  "ap-southeast-2.anthropic.claude-3-7-sonnet-20250219-v1:0",
                 data: {
                   id: "msg_bedrock_apac",
                   type: "message",
@@ -350,7 +352,8 @@ describe("Pass-Through Tests", () => {
     describe("Mixed provider fallback scenarios", () => {
       it("should fallback from Groq to OpenAI to Anthropic", () =>
         runGatewayTest({
-          model: "llama-3.3-70b-versatile/groq,gpt-4-turbo/openai,claude-3-7-sonnet-20250219/anthropic",
+          model:
+            "llama-3.3-70b-versatile/groq,gpt-4-turbo/openai,claude-3-7-sonnet-20250219/anthropic",
           request: {
             messages: [{ role: "user", content: "Test triple fallback" }],
             maxTokens: 100,
@@ -387,7 +390,7 @@ describe("Pass-Through Tests", () => {
                 expects: {
                   headers: {
                     // Without NO_MAPPING, Anthropic uses OpenAI compatibility mode with Authorization header
-                    "Authorization": /^Bearer /,
+                    Authorization: /^Bearer /,
                   },
                 },
               },
@@ -422,7 +425,9 @@ describe("Pass-Through Tests", () => {
                   id: "msg_no_mapping_fallback",
                   type: "message",
                   role: "assistant",
-                  content: [{ type: "text", text: "NO_MAPPING fallback response" }],
+                  content: [
+                    { type: "text", text: "NO_MAPPING fallback response" },
+                  ],
                   model: "claude-3-7-sonnet-20250219",
                   usage: { input_tokens: 10, output_tokens: 5 },
                 },

worker/test/ai-gateway/registry-anthropic.spec.ts

@@ -682,4 +682,4 @@ describe("Anthropic Registry Tests", () => {
         }));
     });
   });
-});
+});

worker/test/ai-gateway/test-framework.ts

@@ -164,28 +164,13 @@ export async function runGatewayTest(
           expect(requestWrapper.getMethod()).toBe(method);
         }
 
-        // Check body contains
-        // Note: Commenting out body checks because getText() consumes the stream
-        // and breaks fallback tests where multiple providers need to read the body
-        // if (bodyContains && requestWrapper.getText) {
-        //   const body = await requestWrapper.getText();
-        //   for (const text of bodyContains) {
-        //     expect(body).toContain(text);
-        //   }
-        // }
+        if (bodyContains && requestWrapper.getText) {
+          const body = await requestWrapper.getText();
+          for (const text of bodyContains) {
+            expect(body).toContain(text);
+          }
+        }
       }
-
-      // Verify full URL with path
-      // if (expectation && requestWrapper.getUrl) {
-      //   const fullUrl = requestWrapper.getUrl();
-      //   const expectedPath = expectation.url.includes("/v1/")
-      //     ? expectation.url.substring(expectation.url.indexOf("/v1/"))
-      //     : "";
-      //   if (expectedPath) {
-      //     expect(fullUrl).toContain(expectedPath);
-      //   }
-      // }
-
       // Custom verification for complex cases
       if (expectation?.customVerify) {
         expectation.customVerify({ targetProps, requestWrapper, env, ctx });
@@ -245,24 +230,21 @@ export async function runGatewayTest(
   );
 
   // Build request with custom headers if provided
-  const requestOptions = createAIGatewayRequest(scenario.model, scenario.request);
-  
+  const requestOptions = createAIGatewayRequest(
+    scenario.model,
+    scenario.request
+  );
+
   // Add custom headers like NO_MAPPING if specified
   if (scenario.request?.bodyMapping) {
-    requestOptions.headers = {
-      ...requestOptions.headers,
-      "Helicone-Gateway-Body-Mapping": scenario.request.bodyMapping,
-    };
+    (requestOptions.headers as any)["Helicone-Gateway-Body-Mapping"] = scenario.request.bodyMapping;
   }
-  
+
   // Add any additional custom headers
   if (scenario.request?.headers) {
-    requestOptions.headers = {
-      ...requestOptions.headers,
-      ...scenario.request.headers,
-    };
+    Object.assign(requestOptions.headers, scenario.request.headers);
   }
-  
+
   const response = await SELF.fetch(
     "https://ai-gateway.helicone.ai/v1/chat/completions",
     requestOptions