@tonyclaw/agent-inspector 2.1.17 → 2.1.19

package/src/mcp/server.ts

@@ -52,7 +52,9 @@ import {
 import { callApi } from "./loopback";
 import {
   addGroupSessionImpl,
+  addGroupProcessImpl,
   addProviderImpl,
+  diagnoseCaptureImpl,
   diagnoseProviderTestImpl,
   exportEvidenceBundleImpl,
   createGroupImpl,
@@ -66,8 +68,12 @@ import {
   exportGroupEvidenceBundleImpl,
   getGroupEvidenceFilesImpl,
   getGroupImpl,
+  getLogHeadersImpl,
   getRecentFailuresImpl,
+  getProcessImpl,
+  getProcessTimelineImpl,
   getProjectContextImpl,
+  getRuntimeConfigImpl,
   getRunImpl,
   getSessionImpl,
   getSessionTimelineImpl,
@@ -78,6 +84,7 @@ import {
   listKnowledgeCandidatesImpl,
   listLogsImpl,
   listModelsImpl,
+  listProcessesImpl,
   listProvidersImpl,
   listSessionsImpl,
   promoteKnowledgeCandidateImpl,
@@ -265,6 +272,10 @@ const LogResourceListResponseSchema = z.object({
   logs: z.array(z.object({ id: z.number().int().positive() })),
 });
 
+const ProcessResourceListResponseSchema = z.object({
+  logs: z.array(z.object({ clientPid: z.number().int().positive().nullable().optional() })),
+});
+
 type ResourceDescriptor = {
   uri: string;
   name: string;
@@ -317,6 +328,7 @@ const WRITE_TOOL_NAMES = new Set<string>([
   "inspector_update_group",
   "inspector_delete_group",
   "inspector_add_group_session",
+  "inspector_add_group_process",
   "inspector_export_group_evidence",
   "inspector_export_group_evidence_bundle",
   "inspector_set_current_context",
@@ -514,6 +526,44 @@ async function listLogResources(): Promise<ListResourcesResult> {
   }
 }
 
+async function listProcessResources(): Promise<ListResourcesResult> {
+  try {
+    const response = await callApi("/api/logs?q=&offset=0&limit=50&scanLimit=200");
+    if (!response.ok) return listResult([]);
+    const parsed = ProcessResourceListResponseSchema.safeParse(await response.json());
+    if (!parsed.success) return listResult([]);
+    const pids = [
+      ...new Set(
+        parsed.data.logs
+          .map((log) => log.clientPid ?? null)
+          .filter((pid): pid is number => pid !== null),
+      ),
+    ].sort((left, right) => left - right);
+    return listResult(
+      pids.map((pid) => ({
+        uri: `inspector://processes/${String(pid)}`,
+        name: `process:${String(pid)}`,
+        title: `Inspector process ${String(pid)}`,
+        mimeType: "application/json",
+      })),
+    );
+  } catch {
+    return listResult([]);
+  }
+}
+
+async function listProcessTimelineResources(): Promise<ListResourcesResult> {
+  const processResources = await listProcessResources();
+  return listResult(
+    processResources.resources.map((resource) => ({
+      uri: `${resource.uri}/timeline`,
+      name: `${resource.name}:timeline`,
+      title: `${resource.title ?? resource.name} timeline`,
+      mimeType: "application/json",
+    })),
+  );
+}
+
 function listRunResources(): ListResourcesResult {
   return listResult(
     listInspectorRuns().runs.map((run) => ({
@@ -639,6 +689,12 @@ function registerTools(server: McpServer): void {
           .describe("How many summaries to return. Hard-capped at 5."),
         sessionId: z.string().optional().describe("Filter by session/affinity id."),
         model: z.string().optional().describe("Filter by model name."),
+        clientPid: z
+          .number()
+          .int()
+          .positive()
+          .optional()
+          .describe("Filter by originating client process PID."),
       }),
     },
     (args) => safeCall(() => listLogsImpl(callApi, args)),
@@ -671,6 +727,12 @@ function registerTools(server: McpServer): void {
           .describe("How many recent compact logs to scan before filtering. Default 200."),
         sessionId: z.string().optional().describe("Optional session/affinity id filter."),
         model: z.string().optional().describe("Optional model name filter."),
+        clientPid: z
+          .number()
+          .int()
+          .positive()
+          .optional()
+          .describe("Optional originating client process PID filter."),
       }),
     },
     (args) => safeCall(() => searchLogsImpl(callApi, args)),
@@ -702,6 +764,87 @@ function registerTools(server: McpServer): void {
     ({ id }) => safeCall(() => getLogChunksImpl(callApi, id)),
   );
 
+  server.registerTool(
+    "inspector_get_log_headers",
+    {
+      title: "Get captured log headers",
+      description:
+        "Returns header-focused evidence for one log without large request/response bodies: raw headers, upstream headers, user-agent, origin, client PID/port, session id, model, and API path.",
+      inputSchema: z.object({
+        id: z.number().int().positive().describe("The log id."),
+      }),
+    },
+    ({ id }) => safeCall(() => getLogHeadersImpl(callApi, id)),
+  );
+
+  server.registerTool(
+    "inspector_list_processes",
+    {
+      title: "List observed client processes",
+      description:
+        "Groups recent compact logs by originating client PID and returns sessions, agents, user-agents, models, providers, ports, error counts, and latest logs. Useful for IDE -> process -> session navigation.",
+      inputSchema: z.object({
+        scanLimit: z
+          .number()
+          .int()
+          .positive()
+          .max(200)
+          .optional()
+          .describe("How many recent logs to scan. Default 50, max 200."),
+        limit: z
+          .number()
+          .int()
+          .positive()
+          .max(200)
+          .optional()
+          .describe("Alias for scanLimit for simpler MCP clients."),
+      }),
+    },
+    (args) => safeCall(() => listProcessesImpl(callApi, args)),
+  );
+
+  server.registerTool(
+    "inspector_get_process",
+    {
+      title: "Get process summary",
+      description:
+        "Returns a bounded summary for one originating client PID across sessions, including user-agents, ports, models, providers, errors, and latest compact logs.",
+      inputSchema: z.object({
+        clientPid: z.number().int().positive().describe("Originating client process PID."),
+        offset: z.number().int().nonnegative().optional().describe("Skip this many matching logs."),
+        limit: z
+          .number()
+          .int()
+          .positive()
+          .max(200)
+          .optional()
+          .describe("Maximum logs to summarize. Default 50, max 200."),
+      }),
+    },
+    (args) => safeCall(() => getProcessImpl(callApi, args)),
+  );
+
+  server.registerTool(
+    "inspector_get_process_timeline",
+    {
+      title: "Get process timeline",
+      description:
+        "Returns a bounded chronological timeline for one originating client PID, spanning all sessions observed for that process.",
+      inputSchema: z.object({
+        clientPid: z.number().int().positive().describe("Originating client process PID."),
+        offset: z.number().int().nonnegative().optional().describe("Skip this many matching logs."),
+        limit: z
+          .number()
+          .int()
+          .positive()
+          .max(50)
+          .optional()
+          .describe("Maximum timeline events to return. Default 20, max 50."),
+      }),
+    },
+    (args) => safeCall(() => getProcessTimelineImpl(callApi, args)),
+  );
+
   server.registerTool(
     "inspector_list_sessions",
     {
@@ -790,6 +933,30 @@ function registerTools(server: McpServer): void {
     (args) => safeWriteCall("inspector_set_current_context", () => setCurrentContextImpl(args)),
   );
 
+  server.registerTool(
+    "inspector_get_runtime_config",
+    {
+      title: "Get Inspector runtime config",
+      description:
+        "Read-only runtime diagnostics: current proxy config plus log storage stats, including captureMode and log/chunk directories. Does not expose provider secrets.",
+      inputSchema: z.object({}),
+    },
+    () => safeCall(() => getRuntimeConfigImpl(callApi)),
+  );
+
+  server.registerTool(
+    "inspector_diagnose_capture",
+    {
+      title: "Diagnose capture detail availability",
+      description:
+        "Explains why headers or bodies may be missing for the current runtime or a specific log, including captureMode, compact body mode, rawHeaders presence, and next-step recommendations.",
+      inputSchema: z.object({
+        id: z.number().int().positive().optional().describe("Optional log id to inspect."),
+      }),
+    },
+    (args) => safeCall(() => diagnoseCaptureImpl(callApi, args)),
+  );
+
   server.registerTool(
     "inspector_create_run",
     {
@@ -1009,6 +1176,26 @@ function registerTools(server: McpServer): void {
     (args) => safeWriteCall("inspector_add_group_session", () => addGroupSessionImpl(args)),
   );
 
+  server.registerTool(
+    "inspector_add_group_process",
+    {
+      title: "Attach a process to an Inspector group",
+      description:
+        "Finds sessions observed for one client PID and attaches each session to a group with inferred agent/model/provider metadata. Useful when a process has both process fallback and explicit sessions.",
+      inputSchema: z.object({
+        groupId: z.string().min(1).describe("The Inspector group id."),
+        clientPid: z.number().int().positive().describe("Originating client process PID."),
+        labelPrefix: z
+          .string()
+          .min(1)
+          .optional()
+          .describe("Optional label prefix. Defaults to PID <clientPid>."),
+      }),
+    },
+    (args) =>
+      safeWriteCall("inspector_add_group_process", () => addGroupProcessImpl(callApi, args)),
+  );
+
   server.registerTool(
     "inspector_export_group_evidence",
     {
@@ -1281,6 +1468,22 @@ function registerResources(server: McpServer): void {
     (uri) => jsonResource(uri, getCurrentContext()),
   );
 
+  server.registerResource(
+    "inspector_runtime_config",
+    "inspector://runtime/config",
+    {
+      title: "Inspector runtime config",
+      description: "Read-only runtime config and log storage stats.",
+      mimeType: "application/json",
+    },
+    async (uri) => {
+      const result = await getRuntimeConfigImpl(callApi);
+      const first = result.content[0];
+      const text = first?.text ?? "";
+      return textResource(uri, result.isError === true ? "text/plain" : "application/json", text);
+    },
+  );
+
   server.registerResource(
     "inspector_sessions",
     "inspector://sessions",
@@ -1351,6 +1554,67 @@ function registerResources(server: McpServer): void {
     },
   );
 
+  server.registerResource(
+    "inspector_processes",
+    "inspector://processes",
+    {
+      title: "Inspector client processes",
+      description: "Observed originating client process PIDs grouped from recent logs.",
+      mimeType: "application/json",
+    },
+    async (uri) => {
+      const result = await listProcessesImpl(callApi, { scanLimit: 200 });
+      const first = result.content[0];
+      const text = first?.text ?? "";
+      return textResource(uri, result.isError === true ? "text/plain" : "application/json", text);
+    },
+  );
+
+  server.registerResource(
+    "inspector_process",
+    new ResourceTemplate("inspector://processes/{clientPid}", { list: listProcessResources }),
+    {
+      title: "Inspector process",
+      description:
+        "One originating client process summary with sessions, agents, user-agents, ports, models, providers, errors, and latest logs.",
+      mimeType: "application/json",
+    },
+    async (uri, variables) => {
+      const clientPid = variableString(variables, "clientPid");
+      const parsed = clientPid === null ? Number.NaN : Number(clientPid);
+      if (!Number.isInteger(parsed) || parsed <= 0) {
+        return textResource(uri, "text/plain", "Invalid clientPid");
+      }
+      const result = await getProcessImpl(callApi, { clientPid: parsed });
+      const first = result.content[0];
+      const text = first?.text ?? "";
+      return textResource(uri, result.isError === true ? "text/plain" : "application/json", text);
+    },
+  );
+
+  server.registerResource(
+    "inspector_process_timeline",
+    new ResourceTemplate("inspector://processes/{clientPid}/timeline", {
+      list: listProcessTimelineResources,
+    }),
+    {
+      title: "Inspector process timeline",
+      description: "Bounded chronological timeline for one originating client process PID.",
+      mimeType: "application/json",
+    },
+    async (uri, variables) => {
+      const clientPid = variableString(variables, "clientPid");
+      const parsed = clientPid === null ? Number.NaN : Number(clientPid);
+      if (!Number.isInteger(parsed) || parsed <= 0) {
+        return textResource(uri, "text/plain", "Invalid clientPid");
+      }
+      const result = await getProcessTimelineImpl(callApi, { clientPid: parsed, limit: 50 });
+      const first = result.content[0];
+      const text = first?.text ?? "";
+      return textResource(uri, result.isError === true ? "text/plain" : "application/json", text);
+    },
+  );
+
   server.registerResource(
     "inspector_runs",
     "inspector://runs",
@@ -1661,11 +1925,17 @@ export const TOOL_NAMES = [
   "inspector_search_logs",
   "inspector_get_log",
   "inspector_get_log_chunks",
+  "inspector_get_log_headers",
+  "inspector_list_processes",
+  "inspector_get_process",
+  "inspector_get_process_timeline",
   "inspector_list_sessions",
   "inspector_get_session",
   "inspector_get_session_timeline",
   "inspector_get_current_context",
   "inspector_set_current_context",
+  "inspector_get_runtime_config",
+  "inspector_diagnose_capture",
   "inspector_create_run",
   "inspector_get_run",
   "inspector_update_run",
@@ -1679,6 +1949,7 @@ export const TOOL_NAMES = [
   "inspector_update_group",
   "inspector_delete_group",
   "inspector_add_group_session",
+  "inspector_add_group_process",
   "inspector_export_group_evidence",
   "inspector_get_group_evidence_files",
   "inspector_export_group_evidence_bundle",
@@ -1702,10 +1973,14 @@ export const RESOURCE_NAMES = [
   "inspector_mcp_capabilities",
   "inspector_mcp_config",
   "inspector_current_context",
+  "inspector_runtime_config",
   "inspector_sessions",
   "inspector_session",
   "inspector_session_timeline",
   "inspector_log",
+  "inspector_processes",
+  "inspector_process",
+  "inspector_process_timeline",
   "inspector_runs",
   "inspector_recent_failures",
   "inspector_run",