@tonyclaw/agent-inspector 2.0.13 → 2.0.15

package/src/proxy/handler.ts

@@ -36,16 +36,21 @@ import {
  * Strips all custom/non-standard headers from the request and replaces with
  * unified proxy identity. Only preserves standard HTTP headers needed for API calls.
  */
-function buildProxyHeaders(originalHeaders: Headers): {
+function buildProxyHeaders(
+  originalHeaders: Headers,
+  captureRawHeaders: boolean,
+): {
   headers: Headers;
-  rawHeaders: Record<string, string>;
+  rawHeaders: Record<string, string> | undefined;
 } {
-  const rawHeaders: Record<string, string> = {};
+  const rawHeaders: Record<string, string> | undefined = captureRawHeaders ? {} : undefined;
   const headers = new Headers();
 
-  originalHeaders.forEach((value, key) => {
-    rawHeaders[key.toLowerCase()] = value;
-  });
+  if (rawHeaders !== undefined) {
+    originalHeaders.forEach((value, key) => {
+      rawHeaders[key.toLowerCase()] = value;
+    });
+  }
 
   headers.set(HEADER_USER_AGENT, PROXY_IDENTITY);
   headers.set(HEADER_X_PROXY_IDENTITY, PROXY_IDENTITY);
@@ -128,6 +133,7 @@ function handleStreamingResponse(
   startTime: number,
   upstreamUrl: string,
   log: CapturedLog,
+  collectStreamingChunks: boolean,
 ): Response {
   log.streaming = true;
   log.responseStatus = upstreamRes.status;
@@ -151,6 +157,7 @@ function handleStreamingResponse(
         elapsedMs,
         responseStatus: upstreamRes.status,
         rawStream: full,
+        collectStreamingChunks,
       });
     },
   });
@@ -175,6 +182,7 @@ function handleStreamingResponse(
         elapsedMs,
         rawStream: full,
         hasChunks,
+        collectStreamingChunks,
       });
     }
   });
@@ -208,8 +216,9 @@ export async function handleProxy(req: Request): Promise<Response> {
   // Important: the original `requestBody` must be preserved so the log stores
   // the unmodified request (the "Raw Request" tab needs the true bytes the
   // client sent). The rewritten body is only used for the upstream fetch.
+  const config = getConfig();
   let bodyToForward = requestBody;
-  if (getConfig().stripClaudeCodeBillingHeader && requestBody !== null && parsed.isMessages) {
+  if (config.stripClaudeCodeBillingHeader && requestBody !== null && parsed.isMessages) {
     const stripped = stripClaudeCodeBillingHeader(requestBody);
     if (stripped.removed > 0) {
       logger.info(
@@ -248,7 +257,11 @@ export async function handleProxy(req: Request): Promise<Response> {
   const upstreamUrl = buildUpstreamUrl(upstreamBase, parsed.normalizedPath);
   const startTime = Date.now();
 
-  const { headers: upstreamHeaders, rawHeaders } = buildProxyHeaders(req.headers);
+  const captureFullDetails = config.captureMode === "full";
+  const { headers: upstreamHeaders, rawHeaders } = buildProxyHeaders(
+    req.headers,
+    captureFullDetails,
+  );
   setUpstreamHost(upstreamHeaders, upstreamBase);
   injectProviderAuth(upstreamHeaders, matchedProviderConfig);
 
@@ -277,10 +290,14 @@ export async function handleProxy(req: Request): Promise<Response> {
   // full PowerShell round-trip on top of everything else before
   // `fetch` is even issued.
   const [clientInfo, preAcquiredId] = await Promise.all([getClientInfo(req), getNextLogId()]);
-  const upstreamHeadersObj: Record<string, string> = {};
-  upstreamHeaders.forEach((value, key) => {
-    upstreamHeadersObj[key.toLowerCase()] = value;
-  });
+  const upstreamHeadersObj: Record<string, string> | undefined = captureFullDetails
+    ? {}
+    : undefined;
+  if (upstreamHeadersObj !== undefined) {
+    upstreamHeaders.forEach((value, key) => {
+      upstreamHeadersObj[key.toLowerCase()] = value;
+    });
+  }
   const log = await createLog(
     req.method,
     parsed.apiPath,
@@ -330,5 +347,5 @@ export async function handleProxy(req: Request): Promise<Response> {
     return handleNonStreamingResponse(upstreamRes, responseBody, startTime, upstreamUrl, log);
   }
 
-  return handleStreamingResponse(upstreamRes, req, startTime, upstreamUrl, log);
+  return handleStreamingResponse(upstreamRes, req, startTime, upstreamUrl, log, captureFullDetails);
 }

package/src/proxy/logFinalizer.ts

@@ -22,12 +22,14 @@ export type FinalizeStreamingLogJob = BaseFinalizeLogJob & {
   type: "streaming";
   responseStatus: number;
   rawStream: string;
+  collectStreamingChunks: boolean;
 };
 
 export type FinalizeStreamAbortLogJob = BaseFinalizeLogJob & {
   type: "stream-abort";
   rawStream: string;
   hasChunks: boolean;
+  collectStreamingChunks: boolean;
 };
 
 export type FinalizeLogJob =
@@ -145,7 +147,7 @@ function finalizeStreaming(job: FinalizeStreamingLogJob, log: CapturedLog): Fina
       job.rawStream,
       log,
       log.model ?? undefined,
-      true,
+      job.collectStreamingChunks,
     );
     persistStreamingChunks(log);
 
@@ -168,7 +170,7 @@ function finalizeStreamAbort(job: FinalizeStreamAbortLogJob, log: CapturedLog):
         job.rawStream,
         log,
         log.model ?? undefined,
-        true,
+        job.collectStreamingChunks,
       );
       persistStreamingChunks(log);
     } else {

package/src/routes/api/providers.$providerId.test.log.ts

@@ -1,6 +1,7 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { getProvider, getModelUsageName } from "../../proxy/providers";
 import { addTestLogEntry } from "../../proxy/store";
+import { getConfig } from "../../proxy/config";
 import { buildProviderTestRequestBody } from "../../lib/providerTestPrompt";
 import {
   ProviderTestResultsSchema,
@@ -21,6 +22,7 @@ async function logModelResults(
     anthropic: { nonStreaming: ProviderTestState; streaming: ProviderTestState };
     openai: { nonStreaming: ProviderTestState; streaming: ProviderTestState };
   },
+  captureFullDetails: boolean,
 ): Promise<void> {
   const usageModel = getModelUsageName(displayName, providerName);
 
@@ -53,7 +55,7 @@ async function logModelResults(
         apiFormat: "anthropic",
         isTest: true,
         providerName,
-        headers: nonStreamingResult.requestHeaders ?? {},
+        headers: captureFullDetails ? (nonStreamingResult.requestHeaders ?? {}) : undefined,
       });
 
       const streamingRequestBody = JSON.stringify(
@@ -75,13 +77,13 @@ async function logModelResults(
         cacheReadInputTokens: streamingResult.cacheReadInputTokens ?? null,
         elapsedMs: streamingResult.latencyMs ?? 0,
         streaming: true,
-        streamingChunks: streamingResult.streamingChunks,
+        streamingChunks: captureFullDetails ? streamingResult.streamingChunks : undefined,
         userAgent: "provider-test",
         origin: null,
         apiFormat: "anthropic",
         isTest: true,
         providerName,
-        headers: streamingResult.requestHeaders ?? {},
+        headers: captureFullDetails ? (streamingResult.requestHeaders ?? {}) : undefined,
       });
     }
   }
@@ -115,7 +117,7 @@ async function logModelResults(
         apiFormat: "openai",
         isTest: true,
         providerName,
-        headers: nonStreamingResult.requestHeaders ?? {},
+        headers: captureFullDetails ? (nonStreamingResult.requestHeaders ?? {}) : undefined,
       });
 
       const streamingRequestBody = JSON.stringify(
@@ -137,13 +139,13 @@ async function logModelResults(
         cacheReadInputTokens: null,
         elapsedMs: streamingResult.latencyMs ?? 0,
         streaming: true,
-        streamingChunks: streamingResult.streamingChunks,
+        streamingChunks: captureFullDetails ? streamingResult.streamingChunks : undefined,
         userAgent: "provider-test",
         origin: null,
         apiFormat: "openai",
         isTest: true,
         providerName,
-        headers: streamingResult.requestHeaders ?? {},
+        headers: captureFullDetails ? (streamingResult.requestHeaders ?? {}) : undefined,
       });
     }
   }
@@ -174,6 +176,7 @@ export const Route = createFileRoute("/api/providers/$providerId/test/log")({
         }
 
         const results = parsed.data;
+        const captureFullDetails = getConfig().captureMode === "full";
         const anthropicUrl =
           provider.anthropicBaseUrl !== undefined && provider.anthropicBaseUrl.length > 0
             ? provider.anthropicBaseUrl
@@ -187,7 +190,14 @@ export const Route = createFileRoute("/api/providers/$providerId/test/log")({
         if (results.models !== undefined) {
           for (const [modelName, modelResult] of Object.entries(results.models)) {
             entries.push(
-              logModelResults(modelName, provider.name, anthropicUrl, openaiUrl, modelResult),
+              logModelResults(
+                modelName,
+                provider.name,
+                anthropicUrl,
+                openaiUrl,
+                modelResult,
+                captureFullDetails,
+              ),
             );
           }
         }

package/.output/public/assets/_sessionId-BNG_mnaH.js

@@ -1 +0,0 @@
-import{R as s,j as e}from"./main-DvFSJlOd.js";import{P as i}from"./ProxyViewerContainer-BZuo0px3.js";function t(){const{sessionId:o}=s.useParams();return e.jsx(i,{initialSessionId:o},o)}export{t as component};

package/.output/public/assets/index-BUxIwHhC.js

@@ -1 +0,0 @@
-import{P as o}from"./ProxyViewerContainer-BZuo0px3.js";import"./main-DvFSJlOd.js";const r=o;export{r as component};