@tonyclaw/agent-inspector 2.0.14 → 2.0.15

package/README.md

@@ -7,7 +7,7 @@ The npm package and CLI remain `agent-inspector`; the platform name is Agent Ins
 ## What It Does
 
 - Runs a local transparent proxy for Anthropic and OpenAI-compatible APIs.
-- Captures model requests, responses, streaming chunks, tool definitions, token usage, and errors.
+- Captures model requests, responses, tool definitions, token usage, and errors.
 - Provides a web UI for browsing sessions and logs in real time.
 - Includes a Chrome companion extension for quick capture status and log navigation.
 - Generates reviewable knowledge candidates from agent sessions.
@@ -71,6 +71,36 @@ ANTHROPIC_BASE_URL=http://localhost:25947/proxy <tool>
 
 The web UI runs at http://localhost:25947.
 
+## Capture Modes
+
+Agent Inspector starts in `simple` capture mode by default:
+
+```bash
+agent-inspector
+agent-inspector --mode simple
+```
+
+Simple mode keeps the proxy lightweight for everyday use. It records the data
+needed for the session timeline, request/response views, token usage, provider
+routing, errors, and knowledge workflows, but it does not retain raw request
+headers, upstream headers, or detailed streaming chunk artifacts.
+
+Use `full` mode when you need protocol-level diagnostics:
+
+```bash
+agent-inspector --mode full
+```
+
+Full mode captures the additional raw details used by the `Raw Headers`,
+`Headers`, and `Raw Response` views. It is useful for provider debugging,
+header transformation checks, and SSE stream inspection, but it uses more CPU,
+memory, and disk.
+
+The mode is decided at process start. To change modes, restart the server. You
+can also set `AGENT_INSPECTOR_MODE=full` or
+`AGENT_INSPECTOR_CAPTURE_MODE=full`; the CLI flag takes precedence over the
+environment.
+
 ## Jenkins Release Webhook
 
 The `agent-inspector-release` Jenkins pipeline can be triggered from GitCode by

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tonyclaw/agent-inspector",
-  "version": "2.0.14",
+  "version": "2.0.15",
   "type": "module",
   "description": "Agent observability and knowledge capture layer for AI coding tools.",
   "license": "MIT",

package/src/cli.ts

@@ -11,6 +11,9 @@ const __dirname = dirname(__filename);
 const DEFAULT_PORT = 25947;
 const LOCAL_PROBE_TIMEOUT_MS = 2000;
 const BRANDED_WINDOWS_RUNTIME_EXE = "agent-inspector.exe";
+const DEFAULT_CAPTURE_MODE = "simple";
+
+type CaptureMode = "simple" | "full";
 
 process.title = "Agent Inspector";
 
@@ -56,6 +59,24 @@ async function isInspectorHealthy(port: number): Promise<boolean> {
   }
 }
 
+async function getRunningCaptureMode(port: number): Promise<CaptureMode | null> {
+  const controller = new AbortController();
+  const timeout = setTimeout(() => controller.abort(), LOCAL_PROBE_TIMEOUT_MS);
+  try {
+    const response = await fetch(`http://127.0.0.1:${port}/api/config`, {
+      cache: "no-store",
+      signal: controller.signal,
+    });
+    if (!response.ok) return null;
+    const raw: unknown = await response.json();
+    return readCaptureMode(raw);
+  } catch {
+    return null;
+  } finally {
+    clearTimeout(timeout);
+  }
+}
+
 function isPortAcceptingConnections(port: number): Promise<boolean> {
   return new Promise((resolve) => {
     const socket = createConnection({ host: "127.0.0.1", port });
@@ -138,9 +159,31 @@ function resolveServerCommand(outputDir: string, serverPath: string): ServerComm
     : { command: process.execPath, args: [serverPath] };
 }
 
+function parseCaptureMode(value: string | undefined): CaptureMode | null {
+  switch (value) {
+    case undefined:
+      return null;
+    case "simple":
+      return "simple";
+    case "full":
+      return "full";
+    default:
+      return null;
+  }
+}
+
+function readCaptureMode(raw: unknown): CaptureMode | null {
+  if (typeof raw !== "object" || raw === null || Array.isArray(raw)) return null;
+  const descriptor = Object.getOwnPropertyDescriptor(raw, "captureMode");
+  const value: unknown = descriptor?.value;
+  return typeof value === "string" ? parseCaptureMode(value) : null;
+}
+
 async function runStart(args: string[]): Promise<void> {
   const envPort = process.env["PORT"];
   const portDefault = envPort !== undefined ? Number(envPort) : DEFAULT_PORT;
+  const envMode =
+    process.env["AGENT_INSPECTOR_CAPTURE_MODE"] ?? process.env["AGENT_INSPECTOR_MODE"];
 
   let port = portDefault;
   let open = true;
@@ -149,9 +192,34 @@ async function runStart(args: string[]): Promise<void> {
   let forceRestart = false;
   let configDir: string | undefined;
   let providersJson: string | undefined;
+  let captureMode = DEFAULT_CAPTURE_MODE;
+  let captureModeWasSpecified = false;
+
+  if (envMode !== undefined && envMode !== "") {
+    const parsedMode = parseCaptureMode(envMode);
+    if (parsedMode === null) {
+      console.error(`Invalid capture mode: ${envMode}. Use simple or full.`);
+      process.exitCode = 1;
+      return;
+    }
+    captureMode = parsedMode;
+    captureModeWasSpecified = true;
+  }
 
   for (let i = 0; i < args.length; i++) {
     const arg = args[i] ?? "";
+    if (arg.startsWith("--mode=") || arg.startsWith("--capture-mode=")) {
+      const value = arg.slice(arg.indexOf("=") + 1);
+      const parsedMode = parseCaptureMode(value);
+      if (parsedMode === null) {
+        console.error(`Invalid capture mode: ${value}. Use simple or full.`);
+        process.exitCode = 1;
+        return;
+      }
+      captureMode = parsedMode;
+      captureModeWasSpecified = true;
+      continue;
+    }
     switch (arg) {
       case "--port":
       case "-p":
@@ -181,6 +249,20 @@ async function runStart(args: string[]): Promise<void> {
         providersJson = args[i + 1];
         i++;
         break;
+      case "--mode":
+      case "--capture-mode": {
+        const value = args[i + 1];
+        const parsedMode = parseCaptureMode(value);
+        if (parsedMode === null) {
+          console.error(`Invalid capture mode: ${String(value)}. Use simple or full.`);
+          process.exitCode = 1;
+          return;
+        }
+        captureMode = parsedMode;
+        captureModeWasSpecified = true;
+        i++;
+        break;
+      }
       default:
         break;
     }
@@ -270,6 +352,13 @@ async function runStart(args: string[]): Promise<void> {
 
   if (!forceRestart && (await isInspectorHealthy(port))) {
     console.log(`agent-inspector is already running at ${url}`);
+    if (captureModeWasSpecified) {
+      const runningMode = await getRunningCaptureMode(port);
+      if (runningMode !== null && runningMode !== captureMode) {
+        console.log(`Existing instance capture mode is ${runningMode}; requested ${captureMode}.`);
+        console.log(`Use --force-restart to restart with ${captureMode} mode.`);
+      }
+    }
     console.log(`Use --force-restart to restart the existing instance.`);
     if (open && openWasSpecified) {
       openBrowser(url);
@@ -289,6 +378,7 @@ async function runStart(args: string[]): Promise<void> {
   }
 
   console.log(`Server running at ${url}`);
+  console.log(`   Capture mode: ${captureMode}`);
   console.log(`   Proxy: ${url}/proxy`);
   console.log(``);
   console.log(`Route AI coding tools through the proxy:`);
@@ -339,6 +429,7 @@ async function runStart(args: string[]): Promise<void> {
   if (providersJson !== undefined) {
     serverEnv["AGENT_INSPECTOR_PROVIDERS_JSON"] = providersJson;
   }
+  serverEnv["AGENT_INSPECTOR_CAPTURE_MODE"] = captureMode;
   const serverProcess = spawn(serverCommand.command, serverCommand.args, {
     stdio: background ? ["ignore", "ignore", "ignore"] : "inherit",
     detached: background,

package/src/components/ProxyViewer.tsx

@@ -3,7 +3,7 @@ import { ArrowDownRight, ArrowLeft, ArrowUpRight, Check, Copy, Download, Plus }
 
 import type { CapturedLog } from "../proxy/schemas";
 import { exportLogsAsZip } from "../lib/export-logs";
-import type { TimeDisplayFormat } from "../lib/runtimeConfig";
+import type { CaptureMode, TimeDisplayFormat } from "../lib/runtimeConfig";
 import { formatTimestampRange } from "../lib/timeDisplay";
 import { cn, formatContextWindowTokens, formatTokens } from "../lib/utils";
 import { useProviders } from "../lib/useProviders";
@@ -599,6 +599,7 @@ export type ProxyViewerProps = {
   onClearGroup: (ids: number[]) => void;
   viewMode: "simple" | "full";
   onViewModeChange: (mode: "simple" | "full") => void;
+  captureMode: CaptureMode;
   /** Live strip-Claude-Code-billing-header flag, sourced once at the container. */
   strip: boolean;
   /** Slow-response threshold in seconds. `0` disables the warning indicator. */
@@ -625,6 +626,7 @@ export function ProxyViewer({
   onClearGroup,
   viewMode,
   onViewModeChange,
+  captureMode,
   strip,
   slowResponseThresholdSeconds,
   timeDisplayFormat,
@@ -821,30 +823,36 @@ export function ProxyViewer({
               ))}
             </SelectContent>
           </Select>
-          <div className="flex items-center border border-border rounded-md overflow-hidden">
-            <button
-              type="button"
-              onClick={() => onViewModeChange("simple")}
-              className={`h-8 px-3 cursor-pointer transition-colors text-xs ${
-                viewMode === "simple"
-                  ? "bg-muted text-foreground"
-                  : "text-muted-foreground hover:bg-muted/50"
-              }`}
-            >
+          {captureMode === "full" ? (
+            <div className="flex items-center border border-border rounded-md overflow-hidden">
+              <button
+                type="button"
+                onClick={() => onViewModeChange("simple")}
+                className={`h-8 px-3 cursor-pointer transition-colors text-xs ${
+                  viewMode === "simple"
+                    ? "bg-muted text-foreground"
+                    : "text-muted-foreground hover:bg-muted/50"
+                }`}
+              >
+                Simple
+              </button>
+              <button
+                type="button"
+                onClick={() => onViewModeChange("full")}
+                className={`h-8 px-3 cursor-pointer transition-colors text-xs ${
+                  viewMode === "full"
+                    ? "bg-muted text-foreground"
+                    : "text-muted-foreground hover:bg-muted/50"
+                }`}
+              >
+                Full
+              </button>
+            </div>
+          ) : (
+            <div className="flex h-8 items-center rounded-md border border-border bg-muted px-3 text-xs text-foreground">
               Simple
-            </button>
-            <button
-              type="button"
-              onClick={() => onViewModeChange("full")}
-              className={`h-8 px-3 cursor-pointer transition-colors text-xs ${
-                viewMode === "full"
-                  ? "bg-muted text-foreground"
-                  : "text-muted-foreground hover:bg-muted/50"
-              }`}
-            >
-              Full
-            </button>
-          </div>
+            </div>
+          )}
           <div className="flex-1" />
           {!hasSessionContext && (
             <span className="text-muted-foreground text-xs font-mono">

package/src/components/ProxyViewerContainer.tsx

@@ -330,7 +330,7 @@ export function ProxyViewerContainer({
 
   // Read the strip config once at the container so the virtualized list does
   // not need N independent SWR subscriptions per row.
-  const { strip, slowResponseThresholdSeconds, timeDisplayFormat } = useStripConfig();
+  const { strip, captureMode, slowResponseThresholdSeconds, timeDisplayFormat } = useStripConfig();
 
   return (
     <>
@@ -352,6 +352,7 @@ export function ProxyViewerContainer({
         onClearGroup={handleClearGroup}
         viewMode={viewMode}
         onViewModeChange={setViewMode}
+        captureMode={captureMode}
         strip={strip}
         slowResponseThresholdSeconds={slowResponseThresholdSeconds}
         timeDisplayFormat={timeDisplayFormat}

package/src/components/providers/SettingsDialog.tsx

@@ -196,6 +196,7 @@ function OnboardingSettingsTab(): JSX.Element {
 function ProxySettingsTab(): JSX.Element {
   const {
     strip,
+    captureMode,
     slowResponseThresholdSeconds,
     timeDisplayFormat,
     isLoading,
@@ -205,6 +206,7 @@ function ProxySettingsTab(): JSX.Element {
   } = useStripConfig();
   const [error, setError] = useState<string | null>(null);
   const [pending, setPending] = useState(false);
+  const alternateCaptureMode = captureMode === "full" ? "simple" : "full";
 
   const handleToggle = useCallback(
     async (next: boolean) => {
@@ -253,6 +255,14 @@ function ProxySettingsTab(): JSX.Element {
 
   return (
     <div className="space-y-4">
+      <div className="space-y-1">
+        <h3 className="text-sm font-semibold">Capture mode</h3>
+        <p className="text-xs text-muted-foreground">
+          This process is running in <code>{captureMode}</code> mode. Restart with{" "}
+          <code>agent-inspector --mode {alternateCaptureMode}</code> to change it.
+        </p>
+      </div>
+
       <div className="space-y-1">
         <h3 className="text-sm font-semibold">Claude Code billing header</h3>
         <p className="text-xs text-muted-foreground">

package/src/lib/runtimeConfig.ts

@@ -3,10 +3,13 @@ import { z } from "zod";
 export const DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS = 10;
 export const MAX_SLOW_RESPONSE_THRESHOLD_SECONDS = 600;
 export const DEFAULT_TIME_DISPLAY_FORMAT = "time";
+export const DEFAULT_CAPTURE_MODE = "simple";
 
 export const TimeDisplayFormatSchema = z.enum(["time", "full"]);
+export const CaptureModeSchema = z.enum(["simple", "full"]);
 
 export type TimeDisplayFormat = z.infer<typeof TimeDisplayFormatSchema>;
+export type CaptureMode = z.infer<typeof CaptureModeSchema>;
 
 /**
  * Schema for the runtime proxy config. Shared between server
@@ -26,6 +29,7 @@ export const RuntimeConfigSchema = z.object({
     .max(MAX_SLOW_RESPONSE_THRESHOLD_SECONDS)
     .default(DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS),
   timeDisplayFormat: TimeDisplayFormatSchema.default(DEFAULT_TIME_DISPLAY_FORMAT),
+  captureMode: CaptureModeSchema.default(DEFAULT_CAPTURE_MODE),
 });
 
 export type RuntimeConfig = z.infer<typeof RuntimeConfigSchema>;

package/src/lib/useStripConfig.ts

@@ -1,8 +1,10 @@
 import useSWR, { type SWRResponse, useSWRConfig } from "swr";
 import { fetchJson } from "./apiClient";
 import {
+  DEFAULT_CAPTURE_MODE,
   DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS,
   DEFAULT_TIME_DISPLAY_FORMAT,
+  type CaptureMode,
   RuntimeConfigSchema,
   type TimeDisplayFormat,
   type RuntimeConfig,
@@ -22,7 +24,9 @@ async function fetcher(url: string): Promise<RuntimeConfig> {
 /**
  * PATCH the runtime config. Exported separately for testability.
  */
-export async function setRuntimeConfig(patch: Partial<RuntimeConfig>): Promise<RuntimeConfig> {
+type RuntimeConfigPatch = Partial<Omit<RuntimeConfig, "captureMode">>;
+
+export async function setRuntimeConfig(patch: RuntimeConfigPatch): Promise<RuntimeConfig> {
   return fetchJson(
     STRIP_CONFIG_SWR_KEY,
     RuntimeConfigSchema,
@@ -37,6 +41,7 @@ export async function setRuntimeConfig(patch: Partial<RuntimeConfig>): Promise<R
 
 export type UseStripConfig = {
   strip: boolean;
+  captureMode: CaptureMode;
   slowResponseThresholdSeconds: number;
   timeDisplayFormat: TimeDisplayFormat;
   isLoading: boolean;
@@ -66,16 +71,18 @@ export function useStripConfig(): UseStripConfig {
   const { mutate: globalMutate } = useSWRConfig();
 
   const strip = response.data?.stripClaudeCodeBillingHeader ?? false;
+  const captureMode = response.data?.captureMode ?? DEFAULT_CAPTURE_MODE;
   const slowResponseThresholdSeconds =
     response.data?.slowResponseThresholdSeconds ?? DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS;
   const timeDisplayFormat = response.data?.timeDisplayFormat ?? DEFAULT_TIME_DISPLAY_FORMAT;
 
-  const optimisticConfig = (patch: Partial<RuntimeConfig>): RuntimeConfig => ({
+  const optimisticConfig = (patch: RuntimeConfigPatch): RuntimeConfig => ({
     stripClaudeCodeBillingHeader: response.data?.stripClaudeCodeBillingHeader ?? false,
     hasSeenOnboarding: response.data?.hasSeenOnboarding ?? false,
     slowResponseThresholdSeconds:
       response.data?.slowResponseThresholdSeconds ?? DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS,
     timeDisplayFormat: response.data?.timeDisplayFormat ?? DEFAULT_TIME_DISPLAY_FORMAT,
+    captureMode,
     ...patch,
   });
 
@@ -113,6 +120,7 @@ export function useStripConfig(): UseStripConfig {
 
   return {
     strip,
+    captureMode,
     slowResponseThresholdSeconds,
     timeDisplayFormat,
     isLoading: response.isLoading,

package/src/proxy/config.ts

@@ -11,9 +11,12 @@ import { join } from "node:path";
 import { logger } from "./logger";
 import { getDataDir } from "./dataDir";
 import {
+  CaptureModeSchema,
+  DEFAULT_CAPTURE_MODE,
   DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS,
   DEFAULT_TIME_DISPLAY_FORMAT,
   RuntimeConfigSchema,
+  type CaptureMode,
   type RuntimeConfig,
 } from "../lib/runtimeConfig";
 
@@ -52,7 +55,22 @@ function getConfigTempPath(): string {
 // Internal: writeable singleton. Read via getConfig(), mutate via setConfig().
 let currentConfig: RuntimeConfig = resolveInitialConfig();
 
+type MutableRuntimeConfig = Omit<RuntimeConfig, "captureMode">;
+
+function resolveInitialCaptureMode(): CaptureMode {
+  const value = process.env["AGENT_INSPECTOR_CAPTURE_MODE"] ?? process.env["AGENT_INSPECTOR_MODE"];
+  if (value === undefined || value === "") return DEFAULT_CAPTURE_MODE;
+
+  const result = CaptureModeSchema.safeParse(value);
+  if (result.success) return result.data;
+
+  logger.warn(`[config] Invalid capture mode ${value}; falling back to ${DEFAULT_CAPTURE_MODE}`);
+  return DEFAULT_CAPTURE_MODE;
+}
+
 function resolveInitialConfig(): RuntimeConfig {
+  const captureMode = resolveInitialCaptureMode();
+
   // 1. Persisted file
   const filePath = getConfigFilePath();
   if (existsSync(filePath)) {
@@ -62,7 +80,7 @@ function resolveInitialConfig(): RuntimeConfig {
       const result = RuntimeConfigSchema.safeParse(parsed);
       if (result.success) {
         logger.info(`[config] Loaded persisted config from ${filePath}`);
-        return result.data;
+        return { ...result.data, captureMode };
       }
       logger.warn(
         `[config] Persisted config at ${filePath} is invalid; falling back to env/default`,
@@ -81,6 +99,7 @@ function resolveInitialConfig(): RuntimeConfig {
       hasSeenOnboarding: false,
       slowResponseThresholdSeconds: DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS,
       timeDisplayFormat: DEFAULT_TIME_DISPLAY_FORMAT,
+      captureMode,
     };
   }
 
@@ -90,6 +109,7 @@ function resolveInitialConfig(): RuntimeConfig {
     hasSeenOnboarding: false,
     slowResponseThresholdSeconds: DEFAULT_SLOW_RESPONSE_THRESHOLD_SECONDS,
     timeDisplayFormat: DEFAULT_TIME_DISPLAY_FORMAT,
+    captureMode,
   };
 }
 
@@ -106,10 +126,11 @@ export function getConfig(): RuntimeConfig {
  * do not throw — the in-memory value is always updated so the caller
  * sees a consistent response.
  */
-export function setConfig(patch: Partial<RuntimeConfig>): RuntimeConfig {
+export function setConfig(patch: Partial<MutableRuntimeConfig>): RuntimeConfig {
   const next: RuntimeConfig = {
     ...currentConfig,
     ...patch,
+    captureMode: currentConfig.captureMode,
   };
   currentConfig = next;
   persistConfig(next);
@@ -152,7 +173,7 @@ function persistConfig(value: RuntimeConfig): boolean {
   }
 
   try {
-    writeFileSync(tempPath, JSON.stringify(value), "utf-8");
+    writeFileSync(tempPath, JSON.stringify(toPersistedConfig(value)), "utf-8");
   } catch (err) {
     logger.error(`[config] Failed to write config temp file: ${String(err)}`);
     return false;
@@ -173,3 +194,12 @@ function persistConfig(value: RuntimeConfig): boolean {
     }
   }
 }
+
+function toPersistedConfig(value: RuntimeConfig): Record<string, unknown> {
+  return {
+    stripClaudeCodeBillingHeader: value.stripClaudeCodeBillingHeader,
+    hasSeenOnboarding: value.hasSeenOnboarding,
+    slowResponseThresholdSeconds: value.slowResponseThresholdSeconds,
+    timeDisplayFormat: value.timeDisplayFormat,
+  };
+}

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 {