@productbrain/mcp 0.0.1-beta.10898 → 0.0.1-beta.10908

package/dist/{chunk-GX5RQVW2.js → chunk-KKEKGR2H.js}

@@ -821,14 +821,12 @@ function kernelCallErrorFrom(fn, status, body) {
 }
 var MAX_RATE_LIMIT_RETRY_WAIT_SECONDS = 60;
 var NO_RATE_LIMIT_RETRY_ROUTES = /* @__PURE__ */ new Set([
-  "agent.touchSession",
   "agent.recordActivity",
   "gaps.record",
   "gaps.resolve",
   "chain.recordBriefRun",
   "agent.reportOrientMetric",
-  "chain.recordSessionSignal",
-  "chain.recordGroundingOutcome"
+  "chain.recordSessionSignal"
 ]);
 async function withOneRateLimitRetry(fn, attempt) {
   if (NO_RATE_LIMIT_RETRY_ROUTES.has(fn)) return attempt();
@@ -1548,17 +1546,14 @@ async function orphanAgentSession() {
 }
 var _lastTouchAtBySession = /* @__PURE__ */ new Map();
 var TOUCH_THROTTLE_MS = 5e3;
-function touchSessionActivity() {
+function sessionTouchHeader(fn) {
   const sessionId = sessionLifecycle().agentSessionId;
-  if (!sessionId) return;
-  const now = Date.now();
-  const lastTouchAt = _lastTouchAtBySession.get(sessionId) ?? 0;
-  if (now - lastTouchAt < TOUCH_THROTTLE_MS) return;
-  _lastTouchAtBySession.set(sessionId, now);
-  kernelCall("agent.touchSession", {
-    sessionId
-  }).catch(() => {
-  });
+  const prior = sessionId ? _lastTouchAtBySession.get(sessionId) : void 0;
+  if (!sessionId || TOUCH_EXCLUDED.has(fn) || Date.now() - (prior ?? 0) < TOUCH_THROTTLE_MS) return { headers: {}, release: () => {
+  } };
+  _lastTouchAtBySession.set(sessionId, Date.now());
+  const release = () => prior === void 0 ? _lastTouchAtBySession.delete(sessionId) : _lastTouchAtBySession.set(sessionId, prior);
+  return { headers: { "x-pb-session": sessionId }, release };
 }
 function resetTouchThrottle(sessionId) {
   if (sessionId) {
@@ -1646,6 +1641,7 @@ async function callGateway(fn, args) {
   const apiKey = getActiveApiKey();
   const budgetMs = latencyBudgetMsForRoute(fn);
   const start = Date.now();
+  const touch = sessionTouchHeader(fn);
   let res;
   try {
     res = await fetch(`${siteUrl}/api/aki`, {
@@ -1657,8 +1653,10 @@ async function callGateway(fn, args) {
         // DEC-1207: attribute this connector for the gateway's context.served seam. MCP is
         // the dominant agent surface (INS-1706); without this header it was emitting nothing.
         "x-pb-source": MCP_TELEMETRY_SOURCE,
-        ...commandHeader()
+        ...commandHeader(),
         // TASK-893: x-pb-command, the tool call this request serves
+        ...touch.headers
+        // TASK-894: the heartbeat rides the request
       },
       body: JSON.stringify({ fn, args })
     });
@@ -1691,6 +1689,7 @@ async function callGateway(fn, args) {
     const errJson = json;
     const msg = errJson.error ?? errJson.message ?? "unknown error";
     audit(fn, "error", Date.now() - start, errJson.code ? `${msg} [${errJson.code}]` : msg, { budgetMs });
+    if (res.status === 429) touch.release();
     throw kernelCallErrorFrom(fn, res.status, errJson);
   }
   audit(fn, "ok", Date.now() - start, void 0, { budgetMs });
@@ -1713,16 +1712,10 @@ async function kernelCall(fn, args = {}) {
   } else {
     setCached(fn, args, data);
   }
-  if (getAgentSessionId() && !TOUCH_EXCLUDED.has(fn)) {
-    touchSessionActivity();
-  }
   return data;
 }
 async function kernelCallEnvelope(fn, args = {}) {
   const { data, summary, next, _meta } = await withOneRateLimitRetry(fn, () => callGateway(fn, args));
-  if (getAgentSessionId() && !TOUCH_EXCLUDED.has(fn)) {
-    touchSessionActivity();
-  }
   return { ok: true, summary, data, next, _meta };
 }
 var resolveInFlightMap = /* @__PURE__ */ new Map();
@@ -1946,4 +1939,4 @@ export {
   appUrl,
   apiKeysUrl
 };
-//# sourceMappingURL=chunk-GX5RQVW2.js.map
+//# sourceMappingURL=chunk-KKEKGR2H.js.map