@tonyclaw/agent-inspector 3.1.19 → 3.1.20

package/src/proxy/sqliteLogIndex.ts

@@ -1,7 +1,12 @@
 import { existsSync } from "node:fs";
 import { dirname, join } from "node:path";
 import { logger, resolveLogDir } from "./logger";
-import { ToolTraceEventSchema, type ApiFormat, type CaptureIncompleteReason } from "./schemas";
+import {
+  ToolTraceEventSchema,
+  type ApiFormat,
+  type CaptureIncompleteReason,
+  type StopReason,
+} from "./schemas";
 import type { LogIndexEntry, LogIndexSummary } from "./logIndex";
 import { ensurePrivateDirectorySync, securePrivateFileSync } from "./privateDataPath";
 
@@ -52,6 +57,7 @@ CREATE TABLE IF NOT EXISTS log_index (
   dropped_bytes INTEGER NOT NULL DEFAULT 0,
   capture_incomplete_reason TEXT,
   warnings_json TEXT,
+  stop_reason TEXT,
   tool_trace_events_json TEXT,
   error TEXT
 );
@@ -99,10 +105,11 @@ INSERT INTO log_index (
   dropped_bytes,
   capture_incomplete_reason,
   warnings_json,
+  stop_reason,
   tool_trace_events_json,
   error
 ) VALUES (
-  ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
+  ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?
 )
 ON CONFLICT(id) DO UPDATE SET
   file = excluded.file,
@@ -141,6 +148,7 @@ ON CONFLICT(id) DO UPDATE SET
   dropped_bytes = excluded.dropped_bytes,
   capture_incomplete_reason = excluded.capture_incomplete_reason,
   warnings_json = excluded.warnings_json,
+  stop_reason = excluded.stop_reason,
   tool_trace_events_json = excluded.tool_trace_events_json,
   error = excluded.error
 `;
@@ -235,6 +243,7 @@ const REQUIRED_SCHEMA_COLUMNS = [
   { name: "dropped_chunks", definition: "INTEGER NOT NULL DEFAULT 0" },
   { name: "dropped_bytes", definition: "INTEGER NOT NULL DEFAULT 0" },
   { name: "capture_incomplete_reason", definition: "TEXT" },
+  { name: "stop_reason", definition: "TEXT" },
   { name: "tool_trace_events_json", definition: "TEXT" },
 ];
 
@@ -446,6 +455,7 @@ function entryParams(entry: LogIndexEntry): readonly unknown[] {
     summary?.droppedBytes ?? 0,
     summary?.captureIncompleteReason ?? null,
     warningJson(summary?.warnings),
+    summary?.stopReason ?? null,
     toolTraceEventsJson(summary),
     summary?.error ?? null,
   ];
@@ -496,6 +506,20 @@ function readCaptureIncompleteReason(row: unknown): CaptureIncompleteReason | nu
   }
 }
 
+function readStopReason(row: unknown): StopReason | null {
+  const value = readString(row, "stop_reason");
+  switch (value) {
+    case "end_turn":
+    case "tool_use":
+    case "stop":
+    case "length":
+      return value;
+    case null:
+    default:
+      return null;
+  }
+}
+
 function readWarnings(row: unknown): string[] | undefined {
   const value = readString(row, "warnings_json");
   if (value === null) return undefined;
@@ -566,6 +590,7 @@ function summaryFromRow(row: unknown, id: number): LogIndexSummary {
     droppedBytes: readNumber(row, "dropped_bytes") ?? 0,
     captureIncompleteReason: readCaptureIncompleteReason(row),
     warnings: readWarnings(row),
+    stopReason: readStopReason(row),
     toolTraceEvents: readToolTraceEvents(row),
     error: readString(row, "error"),
   };

package/src/proxy/store.ts

@@ -37,6 +37,7 @@ import { writeChunks } from "./chunkStorage";
 import { closeSqliteLogIndex } from "./sqliteLogIndex";
 import type { SessionInfo, SessionLogSummary } from "../lib/sessionInfoContract";
 import { extractToolTraceEvents } from "../lib/toolTrace";
+import { extractStopReason } from "../lib/stopReason";
 import { textByteLength } from "../lib/unknown";
 import { buildSessionInfo, buildSessionLogSummary } from "./sessionInfo";
 import type { CapturedLog } from "./schemas";
@@ -277,13 +278,15 @@ function normalizeLog(log: CapturedLog): CapturedLog {
   const droppedChunks = log.droppedChunks ?? 0;
   const droppedBytes = log.droppedBytes ?? 0;
   const captureIncompleteReason = log.captureIncompleteReason ?? null;
+  const stopReason = log.stopReason ?? extractStopReason(log);
   if (
     sessionId === log.sessionId &&
     apiFormat === log.apiFormat &&
     log.captureIncomplete !== undefined &&
     log.droppedChunks !== undefined &&
     log.droppedBytes !== undefined &&
-    log.captureIncompleteReason !== undefined
+    log.captureIncompleteReason !== undefined &&
+    log.stopReason !== undefined
   ) {
     return log;
   }
@@ -295,6 +298,7 @@ function normalizeLog(log: CapturedLog): CapturedLog {
     droppedChunks,
     droppedBytes,
     captureIncompleteReason,
+    stopReason,
   };
 }
 
@@ -339,6 +343,7 @@ export function compactLogForList(log: CapturedLog): CapturedLog {
     rawHeaders: undefined,
     headers: undefined,
     streamingChunks: undefined,
+    stopReason: log.stopReason ?? extractStopReason(log),
     toolTraceEvents: log.toolTraceEvents ?? extractToolTraceEvents(log),
     rawRequestBodyBytes: log.rawRequestBodyBytes ?? textByteLength(log.rawRequestBody),
     responseTextBytes: log.responseTextBytes ?? textByteLength(log.responseText),
@@ -385,6 +390,7 @@ function compactLogFromIndexSummary(summary: LogIndexSummary): CapturedLog {
     droppedBytes: summary.droppedBytes,
     captureIncompleteReason: summary.captureIncompleteReason,
     warnings: summary.warnings,
+    stopReason: summary.stopReason,
     toolTraceEvents: summary.toolTraceEvents,
     error: summary.error,
   });