@unson/brainbase-mcp 0.23.0 → 0.23.1

package/dist/knowledge-continuation.d.ts

@@ -97,6 +97,8 @@ export interface KnowledgeLookupState {
     revision: number;
     status: KnowledgeStatus;
     started_at: number;
+    /** Set when the first validated retrieval action is reserved. Old states may omit it. */
+    retrieval_started_at?: number | null;
     limits: KnowledgeLookupLimits;
     attempts: KnowledgeAttempt[];
     required_fields: string[] | null;

package/dist/knowledge-continuation.js

@@ -20,6 +20,7 @@ export function createKnowledgeLookup({ lookupId, question, now = Date.now(), li
         revision: 0,
         status: 'resolving',
         started_at: now,
+        retrieval_started_at: null,
         limits: {
             attempts: bounded(limits.attempts, 8, 16),
             replans: bounded(limits.replans, 4, 8),
@@ -42,8 +43,51 @@ function terminate(state, reason, status = 'unresolved') {
     state.pending = null;
     return state;
 }
+const hasPendingRetrieval = (state) => state.pending !== null && state.pending.kind !== 'finish';
+const retrievalStartedAt = (state) => {
+    if (typeof state.retrieval_started_at === 'number' && Number.isFinite(state.retrieval_started_at)) {
+        return state.retrieval_started_at;
+    }
+    // States written before retrieval_started_at was introduced can still be
+    // bounded safely once they have evidence of a retrieval reservation.
+    if (state.attempts.length > 0 || hasPendingRetrieval(state))
+        return state.started_at;
+    return null;
+};
+const retrievalPhase = (state) => {
+    if (hasPendingRetrieval(state))
+        return 'in_flight';
+    if (state.attempts.length > 0)
+        return 'attempted';
+    return 'not_started';
+};
+function recordTimedOutPendingRetrieval(state) {
+    const pending = state.pending;
+    if (!pending)
+        return;
+    state.pending = null;
+    if (pending.kind === 'finish')
+        return;
+    if (state.attempts.some((attempt) => attempt.attempt_id === pending.tool_use_id))
+        return;
+    // The host cannot tell whether a timed-out reservation reached the tool.
+    // Keep the reservation as an unknown transport attempt rather than treating
+    // a late or missing result as a successful read.
+    state.attempts.push({
+        attempt_id: pending.tool_use_id,
+        kind: pending.kind,
+        fingerprint: pending.fingerprint,
+        outcome: 'transport_error',
+        references: [],
+        searched_scope: {},
+        missing_fields: state.required_fields ?? []
+    });
+    state.revision += 1;
+}
 function expire(state, now) {
-    if (!isKnowledgeLookupTerminal(state) && now - state.started_at >= state.limits.milliseconds) {
+    const startedAt = retrievalStartedAt(state);
+    if (startedAt !== null && !isKnowledgeLookupTerminal(state) && now - startedAt >= state.limits.milliseconds) {
+        recordTimedOutPendingRetrieval(state);
         terminate(state, 'time_budget_exhausted');
     }
     return state;
@@ -196,8 +240,10 @@ function validFinish(state, action) {
 }
 export function prepareKnowledgeAction(previous, input, toolUseId, now = Date.now()) {
     const state = expire(structuredClone(previous), now);
-    if (isKnowledgeLookupTerminal(state))
+    if (isKnowledgeLookupTerminal(state)) {
+        recordTimedOutPendingRetrieval(state);
         return { state, allowed: false, reason: state.termination_reason ?? state.status };
+    }
     if (state.pending)
         return { state, allowed: false, reason: 'attempt_in_flight' };
     if (!record(input))
@@ -312,6 +358,9 @@ export function prepareKnowledgeAction(previous, input, toolUseId, now = Date.no
         }
         state.replans += 1;
     }
+    if (state.retrieval_started_at == null && state.attempts.length === 0 && !hasPendingRetrieval(state)) {
+        state.retrieval_started_at = now;
+    }
     state.pending = {
         tool_use_id: toolUseId,
         kind: action.kind,
@@ -323,6 +372,14 @@ export function prepareKnowledgeAction(previous, input, toolUseId, now = Date.no
 }
 export function recordKnowledgeResult(previous, input, result, toolUseId, now = Date.now()) {
     const state = structuredClone(previous);
+    if (isKnowledgeLookupTerminal(state)) {
+        // A late PostToolUse must never reopen a finished lookup. Preserve a stale
+        // retrieval reservation as unknown while keeping the terminal state.
+        recordTimedOutPendingRetrieval(state);
+        return state;
+    }
+    if (isKnowledgeLookupTerminal(expire(state, now)))
+        return state;
     if (typeof toolUseId === 'string' && state.finished_calls.includes(toolUseId))
         return state;
     if (!state.pending ||
@@ -411,8 +468,10 @@ export function recordKnowledgeResult(previous, input, result, toolUseId, now =
 }
 export function stopKnowledgeLookup(previous, now = Date.now()) {
     const state = expire(structuredClone(previous), now);
-    if (isKnowledgeLookupTerminal(state))
+    if (isKnowledgeLookupTerminal(state)) {
+        recordTimedOutPendingRetrieval(state);
         return { state, block: false, reason: knowledgeLookupContext(state) };
+    }
     state.stop_requests += 1;
     if (state.stop_requests > state.limits.replans) {
         terminate(state, 'model_did_not_produce_executable_plan');
@@ -444,6 +503,7 @@ export function knowledgeLookupContext(state) {
         revision: state.revision,
         question: state.question,
         status: state.status,
+        retrieval_phase: retrievalPhase(state),
         required_fields: state.required_fields,
         attempts: state.attempts,
         remaining_attempts: Math.max(0, state.limits.attempts - state.attempts.length),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@unson/brainbase-mcp",
-  "version": "0.23.0",
+  "version": "0.23.1",
   "description": "Local-first Brainbase judgment OS that lets supported AI agents reuse a user's approved goals, priorities, decisions, and judgment criteria.",
   "type": "module",
   "license": "MIT",
@@ -407,5 +407,5 @@
   "engines": {
     "node": ">=20"
   },
-  "gitHead": "25fb5aa910ba704d704768d3df5f271b085df5a2"
+  "gitHead": "da34b87b5890ae3bc4c89bffe6faa54a302e3276"
 }