@tonyclaw/agent-inspector 2.0.17 → 2.0.19

package/src/routes/api/logs.$id.replay.ts

@@ -1,6 +1,6 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { z } from "zod";
-import { getLogById } from "../../proxy/store";
+import { addTestLogEntry, getLogById } from "../../proxy/store";
 import { findProviderByModel } from "../../proxy/providers";
 import { registry } from "../../proxy/formats";
 import { formatForPath } from "../../proxy/formats";
@@ -39,6 +39,7 @@ type ReplayResponse = {
   outputTokens?: number;
   elapsedMs?: number;
   streaming?: boolean;
+  replayLogId?: number;
 };
 
 export const Route = createFileRoute("/api/logs/$id/replay")({
@@ -161,6 +162,37 @@ export const Route = createFileRoute("/api/logs/$id/replay")({
           const mockLog: typeof log = { ...log };
           const responseText = formatHandler.extractStream(fullResponse, mockLog, model, false);
           const tokens = formatHandler.extractTokens(responseText);
+          const savedReplayLog = await addTestLogEntry({
+            timestamp: new Date().toISOString(),
+            method: log.method,
+            path: log.path,
+            model,
+            sessionId: log.sessionId,
+            rawRequestBody: modifiedBody,
+            responseStatus: upstreamRes.status,
+            responseText,
+            inputTokens: tokens.inputTokens ?? null,
+            outputTokens: tokens.outputTokens ?? null,
+            cacheCreationInputTokens: tokens.cacheCreationInputTokens ?? null,
+            cacheReadInputTokens: tokens.cacheReadInputTokens ?? null,
+            elapsedMs,
+            streaming: true,
+            userAgent: log.userAgent,
+            origin: log.origin,
+            rawHeaders: log.rawHeaders,
+            headers: log.headers,
+            apiFormat: log.apiFormat,
+            isTest: true,
+            replayOfLogId: log.id,
+            providerName: log.providerName,
+            clientPort: log.clientPort,
+            clientPid: log.clientPid,
+            clientCwd: log.clientCwd,
+            clientProjectFolder: log.clientProjectFolder,
+            streamingChunks: undefined,
+            streamingChunksPath: null,
+            error: null,
+          });
 
           return Response.json({
             success: true,
@@ -170,10 +202,42 @@ export const Route = createFileRoute("/api/logs/$id/replay")({
             outputTokens: tokens.outputTokens ?? undefined,
             elapsedMs,
             streaming: true,
+            replayLogId: savedReplayLog.id,
           } satisfies ReplayResponse);
         } else {
           const responseText = await upstreamRes.text();
           const tokens = formatHandler.extractTokens(responseText);
+          const savedReplayLog = await addTestLogEntry({
+            timestamp: new Date().toISOString(),
+            method: log.method,
+            path: log.path,
+            model,
+            sessionId: log.sessionId,
+            rawRequestBody: modifiedBody,
+            responseStatus: upstreamRes.status,
+            responseText,
+            inputTokens: tokens.inputTokens ?? null,
+            outputTokens: tokens.outputTokens ?? null,
+            cacheCreationInputTokens: tokens.cacheCreationInputTokens ?? null,
+            cacheReadInputTokens: tokens.cacheReadInputTokens ?? null,
+            elapsedMs,
+            streaming: false,
+            userAgent: log.userAgent,
+            origin: log.origin,
+            rawHeaders: log.rawHeaders,
+            headers: log.headers,
+            apiFormat: log.apiFormat,
+            isTest: true,
+            replayOfLogId: log.id,
+            providerName: log.providerName,
+            clientPort: log.clientPort,
+            clientPid: log.clientPid,
+            clientCwd: log.clientCwd,
+            clientProjectFolder: log.clientProjectFolder,
+            streamingChunks: undefined,
+            streamingChunksPath: null,
+            error: null,
+          });
 
           return Response.json({
             success: true,
@@ -183,6 +247,7 @@ export const Route = createFileRoute("/api/logs/$id/replay")({
             outputTokens: tokens.outputTokens ?? undefined,
             elapsedMs,
             streaming: false,
+            replayLogId: savedReplayLog.id,
           } satisfies ReplayResponse);
         }
       },

package/src/routes/api/logs.stream.ts

@@ -1,7 +1,9 @@
 import { createFileRoute } from "@tanstack/react-router";
-import { onLogUpdate, getFilteredLogs, getLogSessionId } from "../../proxy/store";
+import { onLogUpdate, getLogSessionId, listLogsPage } from "../../proxy/store";
 import { type CapturedLog } from "../../proxy/schemas";
 
+const INITIAL_STREAM_LOG_LIMIT = 500;
+
 export const Route = createFileRoute("/api/logs/stream")({
   server: {
     handlers: {
@@ -49,10 +51,16 @@ export const Route = createFileRoute("/api/logs/stream")({
         request.signal.addEventListener("abort", cleanup, { once: true });
 
         const stream = new ReadableStream<Uint8Array>({
-          start(controller) {
+          async start(controller) {
             controllerRef = controller;
             // Send initial batch of logs immediately
-            const logs = getFilteredLogs(sessionId, model);
+            const result = await listLogsPage({
+              sessionId,
+              model,
+              offset: 0,
+              limit: INITIAL_STREAM_LOG_LIMIT,
+            });
+            const logs = result.logs;
             const initData = `data: ${JSON.stringify({ type: "init", logs })}\n\n`;
             controller.enqueue(new TextEncoder().encode(initData));
           },

package/src/routes/api/logs.ts

@@ -1,6 +1,12 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { z } from "zod";
-import { clearAllLogs, clearLogsByIds, getFilteredLogs } from "../../proxy/store";
+import {
+  clearPersistedLogStorage,
+  clearPersistedLogsByIds,
+  getLogStorageStats,
+  getReplayLogsForSource,
+  listLogsPage,
+} from "../../proxy/store";
 
 const DeleteBodySchema = z
   .object({
@@ -8,25 +14,43 @@ const DeleteBodySchema = z
   })
   .optional();
 
+function parseNonNegativeInt(value: string | null, fallback: number): number {
+  if (value === null) return fallback;
+  const parsed = Number(value);
+  if (!Number.isInteger(parsed) || parsed < 0) return fallback;
+  return parsed;
+}
+
+function parsePositiveInt(value: string | null, fallback: number): number {
+  if (value === null) return fallback;
+  const parsed = Number(value);
+  if (!Number.isInteger(parsed) || parsed < 1) return fallback;
+  return parsed;
+}
+
 export const Route = createFileRoute("/api/logs")({
   server: {
     handlers: {
-      GET: ({ request }: { request: Request }) => {
+      GET: async ({ request }: { request: Request }) => {
         const url = new URL(request.url);
+        if (url.searchParams.get("stats") === "1") {
+          return Response.json(getLogStorageStats());
+        }
+        const replayOf = url.searchParams.get("replayOf");
+        if (replayOf !== null) {
+          const id = Number(replayOf);
+          if (!Number.isInteger(id) || id <= 0) {
+            return Response.json({ error: "Invalid replayOf log ID" }, { status: 400 });
+          }
+          const logs = getReplayLogsForSource(id);
+          return Response.json({ logs, total: logs.length, offset: 0, limit: logs.length });
+        }
         const sessionId = url.searchParams.get("sessionId") ?? undefined;
         const model = url.searchParams.get("model") ?? undefined;
-        const offset = Number(url.searchParams.get("offset") ?? 0);
-        const limit = Number(url.searchParams.get("limit") ?? 50);
+        const offset = parseNonNegativeInt(url.searchParams.get("offset"), 0);
+        const limit = parsePositiveInt(url.searchParams.get("limit"), 50);
 
-        const allLogs = getFilteredLogs(sessionId, model);
-        const paginatedLogs = allLogs.slice(offset, offset + limit);
-
-        return Response.json({
-          logs: paginatedLogs,
-          total: allLogs.length,
-          offset,
-          limit,
-        });
+        return Response.json(await listLogsPage({ sessionId, model, offset, limit }));
       },
       DELETE: async ({ request }: { request: Request }) => {
         let body: z.infer<typeof DeleteBodySchema> = undefined;
@@ -48,11 +72,21 @@ export const Route = createFileRoute("/api/logs")({
         }
 
         if (body?.ids !== undefined && body.ids.length > 0) {
-          const result = clearLogsByIds(body.ids);
-          return Response.json({ success: true, cleared: result.cleared });
+          const result = await clearPersistedLogsByIds(body.ids);
+          return Response.json({
+            success: true,
+            cleared: result.cleared,
+            logFilesRewritten: result.logFilesRewritten,
+            chunkFilesDeleted: result.chunkFilesDeleted,
+          });
         }
-        const result = clearAllLogs();
-        return Response.json({ success: true, cleared: result.cleared });
+        const result = await clearPersistedLogStorage();
+        return Response.json({
+          success: true,
+          cleared: result.cleared,
+          logFilesDeleted: result.logFilesDeleted,
+          chunkFilesDeleted: result.chunkFilesDeleted,
+        });
       },
     },
   },

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

@@ -1,6 +1,11 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { z } from "zod";
-import { getProviders, updateProvider, deleteProvider } from "../../proxy/providers";
+import {
+  getProviders,
+  updateProvider,
+  deleteProvider,
+  redactProvider,
+} from "../../proxy/providers";
 import { ProviderModelMetadataSchema } from "../../lib/providerContract";
 
 const ProviderUpdateSchema = z.object({
@@ -22,13 +27,15 @@ const ProviderUpdateSchema = z.object({
 export const Route = createFileRoute("/api/providers/$providerId")({
   server: {
     handlers: {
-      GET: ({ params }: { params: { providerId: string } }) => {
+      GET: ({ params, request }: { params: { providerId: string }; request: Request }) => {
+        const url = new URL(request.url);
+        const includeSecrets = url.searchParams.get("includeSecrets") === "1";
         const providers = getProviders();
         const provider = providers.find((p) => p.id === params.providerId);
         if (!provider) {
           return Response.json({ error: "Provider not found" }, { status: 404 });
         }
-        return Response.json(provider);
+        return Response.json(includeSecrets ? provider : redactProvider(provider));
       },
       PUT: async ({ params, request }: { params: { providerId: string }; request: Request }) => {
         const parsed = ProviderUpdateSchema.safeParse(await request.json());

package/src/routes/api/providers.ts

@@ -1,6 +1,6 @@
 import { createFileRoute } from "@tanstack/react-router";
 import { z } from "zod";
-import { getProviders, addProvider } from "../../proxy/providers";
+import { getProviders, addProvider, redactProviders } from "../../proxy/providers";
 import { ProviderModelMetadataSchema } from "../../lib/providerContract";
 
 const ProviderInputSchema = z.object({
@@ -21,8 +21,11 @@ const ProviderInputSchema = z.object({
 export const Route = createFileRoute("/api/providers")({
   server: {
     handlers: {
-      GET: () => {
-        return Response.json(getProviders());
+      GET: ({ request }: { request: Request }) => {
+        const url = new URL(request.url);
+        const includeSecrets = url.searchParams.get("includeSecrets") === "1";
+        const providers = getProviders();
+        return Response.json(includeSecrets ? providers : redactProviders(providers));
       },
       POST: async ({ request }: { request: Request }) => {
         const parsed = ProviderInputSchema.safeParse(await request.json());