brains-mcp 1.75.0 → 1.76.0

package/dist/httpServer.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"httpServer.d.ts","sourceRoot":"","sources":["../src/httpServer.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAQxB,OAAO,EAAsD,KAAK,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAkH7G,wBAAgB,8BAA8B,IAAI,IAAI,CAErD;AAi6BD;;;;;;;;;;;;GAYG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAI9D;AAgsBD,wBAAsB,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAqpF9F"}
+{"version":3,"file":"httpServer.d.ts","sourceRoot":"","sources":["../src/httpServer.ts"],"names":[],"mappings":"AAAA,OAAO,IAAI,MAAM,MAAM,CAAC;AAQxB,OAAO,EAAsD,KAAK,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAkH7G,wBAAgB,8BAA8B,IAAI,IAAI,CAErD;AAu7BD;;;;;;;;;;;;GAYG;AACH,wBAAgB,sBAAsB,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAI9D;AAgsBD,wBAAsB,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAkrF9F"}

package/dist/httpServer.js

@@ -229,6 +229,23 @@ function raceWithTimeout(promise, ms) {
     });
     return Promise.race([promise, timeoutPromise]).finally(() => clearTimeout(timer));
 }
+// Shared enrichment for any request_events row produced by a raw
+// executeToolCall() result — used by both the /mcp tools/call path and the
+// /tool passthrough path, mirroring what the standalone REST /api/v1/search
+// handler has always set for itself (GH #411: neither path set this before).
+function toolResultLogExtras(result) {
+    const r = result;
+    if (!r || typeof r !== "object")
+        return {};
+    return {
+        backend: r.backend,
+        queryMs: r.timings?.queryMs,
+        syncMs: r.timings?.syncMs,
+        resultCount: r.count,
+        zeroResults: typeof r.count === "number" ? r.count === 0 : undefined,
+        pageCount: r.pageCount,
+    };
+}
 function emitRequestLog(method, reqPath, status, durationMs, extras = {}) {
     const logFormat = (process.env.LOG_FORMAT ?? "").toLowerCase();
     if (logFormat === "json") {
@@ -1898,7 +1915,8 @@ export async function startHttpServer(drive, port) {
                                     // Re-resolve the drive on every tool call rather than baking in a
                                     // snapshot here — otherwise a project-share grant revoked mid-session
                                     // keeps being honored for the lifetime of this MCP session (BRA-377).
-                                    const mcpServer = createBrainsServer(() => resolveDriveForAuthorization(authorization), mcpUserId, mcpUserAccessLevel);
+                                    const toolCallHook = {};
+                                    const mcpServer = createBrainsServer(() => resolveDriveForAuthorization(authorization), mcpUserId, mcpUserAccessLevel, toolCallHook);
                                     const transport = new StreamableHTTPServerTransport({
                                         sessionIdGenerator: () => staleSid,
                                     });
@@ -1910,7 +1928,7 @@ export async function startHttpServer(drive, port) {
                                     if (!ok) {
                                         return null;
                                     }
-                                    const entry = { server: mcpServer, transport };
+                                    const entry = { server: mcpServer, transport, toolCallHook };
                                     mcpSessions.set(staleSid, entry);
                                     return entry;
                                 })().finally(() => {
@@ -1938,7 +1956,8 @@ export async function startHttpServer(drive, port) {
                             // Re-resolve the drive on every tool call rather than baking in a
                             // snapshot here — otherwise a project-share grant revoked mid-session
                             // keeps being honored for the lifetime of this MCP session (BRA-377).
-                            const mcpServer = createBrainsServer(() => resolveDriveForAuthorization(authorization), mcpUserId, mcpUserAccessLevel);
+                            const toolCallHook = {};
+                            const mcpServer = createBrainsServer(() => resolveDriveForAuthorization(authorization), mcpUserId, mcpUserAccessLevel, toolCallHook);
                             const transport = new StreamableHTTPServerTransport({
                                 sessionIdGenerator: () => randomUUID(),
                             });
@@ -1949,7 +1968,7 @@ export async function startHttpServer(drive, port) {
                                 }
                             };
                             await mcpServer.connect(transport);
-                            existing = { server: mcpServer, transport };
+                            existing = { server: mcpServer, transport, toolCallHook };
                         }
                     }
                     const { transport } = existing;
@@ -1957,6 +1976,21 @@ export async function startHttpServer(drive, port) {
                     // Inject timing headers before the transport writes its response headers.
                     res.setHeader("X-Brains-Auth-Ms", String(mcpAuthMs));
                     res.setHeader("X-Brains-Handler-Start-Ms", String(mcpHandlerStart));
+                    // Wire this one request's hook so query_ms/backend/etc. reach
+                    // res._logExtras once the tool call the MCP SDK dispatches inside
+                    // transport.handleRequest() actually resolves (GH #411: this
+                    // enrichment previously only existed on the standalone REST
+                    // /api/v1/search handler, never on the shared /mcp path).
+                    if (rpcToolName) {
+                        existing.toolCallHook.current = (name, result) => {
+                            const priorExtras = res._logExtras ?? {};
+                            res._logExtras = {
+                                ...priorExtras,
+                                tool: name,
+                                ...toolResultLogExtras(result),
+                            };
+                        };
+                    }
                     try {
                         await transport.handleRequest(req, res, body);
                     }
@@ -1964,6 +1998,9 @@ export async function startHttpServer(drive, port) {
                         console.error("[brains] MCP transport error:", mcpErr instanceof Error ? mcpErr.stack : String(mcpErr));
                         throw mcpErr;
                     }
+                    finally {
+                        existing.toolCallHook.current = undefined;
+                    }
                     if (!res.headersSent) {
                         res.setHeader("X-Brains-Handler-End-Ms", String(Date.now()));
                     }
@@ -3625,6 +3662,12 @@ export async function startHttpServer(drive, port) {
                 const body = await readJsonBody(req);
                 const parsed = toolCallSchema.parse(body);
                 const result = await executeToolCall(requestDrive, parsed.name, parsed.arguments ?? {}, requestUserId, requestAccessCtx);
+                // GH #411: this passthrough endpoint never set _logExtras at all, so
+                // its request_events rows had a null `tool` even before query_ms.
+                res._logExtras = {
+                    tool: parsed.name,
+                    ...toolResultLogExtras(result),
+                };
                 json(req, res, 200, { ok: true, result });
                 return;
             }