@mastra/core 1.6.0 → 1.7.0

package/dist/harness/index.d.ts

@@ -1,5 +1,6 @@
 export { Harness } from './harness.js';
 export { askUserTool, parseSubagentMeta, submitPlanTool, taskCheckTool, taskWriteTool } from './tools.js';
 export type { TaskItem } from './tools.js';
-export type { AvailableModel, HarnessConfig, HarnessEvent, HarnessEventListener, HarnessMessage, HarnessMessageContent, HarnessMode, HarnessOMConfig, HarnessRequestContext, HarnessSession, HarnessStateSchema, HarnessSubagent, HarnessThread, HeartbeatHandler, ModelAuthChecker, ModelAuthStatus, ModelUseCountProvider, PermissionPolicy, PermissionRules, ToolCategory, TokenUsage, } from './types.js';
+export { defaultDisplayState, defaultOMProgressState } from './types.js';
+export type { ActiveSubagentState, ActiveToolState, AvailableModel, HarnessConfig, HarnessDisplayState, HarnessEvent, HarnessEventListener, HarnessMessage, HarnessMessageContent, HarnessMode, HarnessOMConfig, HarnessRequestContext, HarnessSession, HarnessStateSchema, HarnessSubagent, HarnessThread, HeartbeatHandler, ModelAuthChecker, ModelAuthStatus, ModelUseCountProvider, OMBufferedStatus, OMProgressState, OMStatus, PermissionPolicy, PermissionRules, ToolCategory, TokenUsage, } from './types.js';
 //# sourceMappingURL=index.d.ts.map

package/dist/harness/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/harness/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACvG,YAAY,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACxC,YAAY,EACV,cAAc,EACd,aAAa,EACb,YAAY,EACZ,oBAAoB,EACpB,cAAc,EACd,qBAAqB,EACrB,WAAW,EACX,eAAe,EACf,qBAAqB,EACrB,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,gBAAgB,EAChB,eAAe,EACf,YAAY,EACZ,UAAU,GACX,MAAM,SAAS,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/harness/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,cAAc,EAAE,aAAa,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AACvG,YAAY,EAAE,QAAQ,EAAE,MAAM,SAAS,CAAC;AACxC,OAAO,EAAE,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AACtE,YAAY,EACV,mBAAmB,EACnB,eAAe,EACf,cAAc,EACd,aAAa,EACb,mBAAmB,EACnB,YAAY,EACZ,oBAAoB,EACpB,cAAc,EACd,qBAAqB,EACrB,WAAW,EACX,eAAe,EACf,qBAAqB,EACrB,cAAc,EACd,kBAAkB,EAClB,eAAe,EACf,aAAa,EACb,gBAAgB,EAChB,gBAAgB,EAChB,eAAe,EACf,qBAAqB,EACrB,gBAAgB,EAChB,eAAe,EACf,QAAQ,EACR,gBAAgB,EAChB,eAAe,EACf,YAAY,EACZ,UAAU,GACX,MAAM,SAAS,CAAC"}

package/dist/harness/index.js

@@ -1,5 +1,5 @@
-import { Agent } from '../chunk-IHDE4CJV.js';
-import { Workspace } from '../chunk-XB3DA67Q.js';
+import { Agent } from '../chunk-QSHV7GPT.js';
+import { Workspace } from '../chunk-ZSBM2SVU.js';
 import { createTool } from '../chunk-BQHWJLXU.js';
 import { RequestContext } from '../chunk-CCLV5CAA.js';
 import { z } from 'zod';
@@ -425,6 +425,55 @@ function parseSubagentMeta(content) {
   return { text, modelId, durationMs, toolCalls };
 }
 
+// src/harness/types.ts
+function defaultDisplayState() {
+  return {
+    isRunning: false,
+    currentMessage: null,
+    tokenUsage: { promptTokens: 0, completionTokens: 0, totalTokens: 0 },
+    activeTools: /* @__PURE__ */ new Map(),
+    toolInputBuffers: /* @__PURE__ */ new Map(),
+    pendingApproval: null,
+    pendingQuestion: null,
+    pendingPlanApproval: null,
+    activeSubagents: /* @__PURE__ */ new Map(),
+    omProgress: defaultOMProgressState(),
+    bufferingMessages: false,
+    bufferingObservations: false,
+    modifiedFiles: /* @__PURE__ */ new Map(),
+    tasks: [],
+    previousTasks: []
+  };
+}
+function defaultOMProgressState() {
+  return {
+    status: "idle",
+    pendingTokens: 0,
+    threshold: 3e4,
+    thresholdPercent: 0,
+    observationTokens: 0,
+    reflectionThreshold: 4e4,
+    reflectionThresholdPercent: 0,
+    buffered: {
+      observations: {
+        status: "idle",
+        chunks: 0,
+        messageTokens: 0,
+        projectedMessageRemoval: 0,
+        observationTokens: 0
+      },
+      reflection: {
+        status: "idle",
+        inputObservationTokens: 0,
+        observationTokens: 0
+      }
+    },
+    generationCount: 0,
+    stepNumber: 0,
+    preReflectionTokens: 0
+  };
+}
+
 // src/harness/harness.ts
 var Harness = class {
   id;
@@ -454,6 +503,7 @@ var Harness = class {
   };
   sessionGrantedCategories = /* @__PURE__ */ new Set();
   sessionGrantedTools = /* @__PURE__ */ new Set();
+  displayState = defaultDisplayState();
   constructor(config) {
     this.id = config.id;
     this.config = config;
@@ -1082,6 +1132,15 @@ var Harness = class {
     } catch {
     }
   }
+  async getObservationalMemoryRecord() {
+    if (!this.currentThreadId) return null;
+    try {
+      const memoryStorage = await this.getMemoryStorage();
+      return await memoryStorage.getObservationalMemory(this.currentThreadId, this.resourceId);
+    } catch {
+      return null;
+    }
+  }
   /**
    * Returns the observer model ID from state, falling back to omConfig defaults.
    */
@@ -1790,6 +1849,35 @@ var Harness = class {
   getCurrentRunId() {
     return this.currentRunId;
   }
+  // ===========================================================================
+  // Display State
+  // ===========================================================================
+  /**
+   * Returns a read-only snapshot of the canonical display state.
+   * UIs should use this to render instead of building up state from raw events.
+   */
+  getDisplayState() {
+    return this.displayState;
+  }
+  /**
+   * Reset display state fields that are scoped to a thread.
+   * Called on thread switch/creation.
+   */
+  resetThreadDisplayState() {
+    this.displayState.activeTools = /* @__PURE__ */ new Map();
+    this.displayState.toolInputBuffers = /* @__PURE__ */ new Map();
+    this.displayState.pendingApproval = null;
+    this.displayState.pendingQuestion = null;
+    this.displayState.pendingPlanApproval = null;
+    this.displayState.activeSubagents = /* @__PURE__ */ new Map();
+    this.displayState.currentMessage = null;
+    this.displayState.modifiedFiles = /* @__PURE__ */ new Map();
+    this.displayState.tasks = [];
+    this.displayState.previousTasks = [];
+    this.displayState.omProgress = defaultOMProgressState();
+    this.displayState.bufferingMessages = false;
+    this.displayState.bufferingObservations = false;
+  }
   /**
    * Respond to a pending tool approval from the UI.
    * "always_allow_category" grants the tool's category for the rest of the session, then approves.
@@ -1917,6 +2005,16 @@ var Harness = class {
     };
   }
   emit(event) {
+    this.applyDisplayStateUpdate(event);
+    this.dispatchToListeners(event);
+    if (event.type !== "display_state_changed") {
+      this.dispatchToListeners({
+        type: "display_state_changed",
+        displayState: this.displayState
+      });
+    }
+  }
+  dispatchToListeners(event) {
     for (const listener of this.listeners) {
       try {
         const result = listener(event);
@@ -1928,6 +2026,323 @@ var Harness = class {
       }
     }
   }
+  /**
+   * Apply a display state update based on an incoming event.
+   * This is the centralized state machine that keeps HarnessDisplayState in sync
+   * with every event the Harness emits.
+   */
+  applyDisplayStateUpdate(event) {
+    const ds = this.displayState;
+    switch (event.type) {
+      // ── Agent lifecycle ────────────────────────────────────────────────
+      case "agent_start":
+        ds.isRunning = true;
+        ds.activeTools = /* @__PURE__ */ new Map();
+        ds.toolInputBuffers = /* @__PURE__ */ new Map();
+        ds.currentMessage = null;
+        ds.pendingApproval = null;
+        break;
+      case "agent_end":
+        ds.isRunning = false;
+        ds.pendingApproval = null;
+        ds.pendingQuestion = null;
+        ds.pendingPlanApproval = null;
+        for (const [, tool] of ds.activeTools) {
+          if (tool.status === "running" || tool.status === "streaming_input") {
+            tool.status = "error";
+          }
+        }
+        ds.activeSubagents = /* @__PURE__ */ new Map();
+        break;
+      // ── Message streaming ──────────────────────────────────────────────
+      case "message_start":
+        ds.currentMessage = event.message;
+        break;
+      case "message_update":
+        ds.currentMessage = event.message;
+        break;
+      case "message_end":
+        ds.currentMessage = event.message;
+        break;
+      // ── Tool lifecycle ─────────────────────────────────────────────────
+      case "tool_input_start": {
+        ds.toolInputBuffers.set(event.toolCallId, { text: "", toolName: event.toolName });
+        const existing = ds.activeTools.get(event.toolCallId);
+        if (existing) {
+          existing.status = "streaming_input";
+        } else {
+          ds.activeTools.set(event.toolCallId, {
+            name: event.toolName,
+            args: {},
+            status: "streaming_input"
+          });
+        }
+        break;
+      }
+      case "tool_input_delta": {
+        const buf = ds.toolInputBuffers.get(event.toolCallId);
+        if (buf) {
+          buf.text += event.argsTextDelta;
+        }
+        break;
+      }
+      case "tool_input_end":
+        ds.toolInputBuffers.delete(event.toolCallId);
+        break;
+      case "tool_start": {
+        const existingTool = ds.activeTools.get(event.toolCallId);
+        if (existingTool) {
+          existingTool.name = event.toolName;
+          existingTool.args = event.args;
+          existingTool.status = "running";
+        } else {
+          ds.activeTools.set(event.toolCallId, {
+            name: event.toolName,
+            args: event.args,
+            status: "running"
+          });
+        }
+        break;
+      }
+      case "tool_update": {
+        const tool = ds.activeTools.get(event.toolCallId);
+        if (tool) {
+          tool.partialResult = typeof event.partialResult === "string" ? event.partialResult : JSON.stringify(event.partialResult);
+        }
+        break;
+      }
+      case "tool_end": {
+        const endedTool = ds.activeTools.get(event.toolCallId);
+        if (endedTool) {
+          endedTool.status = event.isError ? "error" : "completed";
+          endedTool.result = event.result;
+          endedTool.isError = event.isError;
+        }
+        if (!event.isError) {
+          const FILE_TOOLS = ["string_replace_lsp", "write_file", "ast_smart_edit"];
+          const toolState = ds.activeTools.get(event.toolCallId);
+          if (toolState && FILE_TOOLS.includes(toolState.name)) {
+            const toolArgs = toolState.args;
+            const filePath = toolArgs?.path;
+            if (filePath) {
+              const existing = ds.modifiedFiles.get(filePath);
+              if (existing) {
+                existing.operations.push(toolState.name);
+              } else {
+                ds.modifiedFiles.set(filePath, {
+                  operations: [toolState.name],
+                  firstModified: /* @__PURE__ */ new Date()
+                });
+              }
+            }
+          }
+        }
+        break;
+      }
+      case "shell_output": {
+        const shellTool = ds.activeTools.get(event.toolCallId);
+        if (shellTool) {
+          shellTool.shellOutput = (shellTool.shellOutput ?? "") + event.output;
+        }
+        break;
+      }
+      case "tool_approval_required":
+        ds.pendingApproval = {
+          toolCallId: event.toolCallId,
+          toolName: event.toolName,
+          args: event.args
+        };
+        break;
+      // ── Interactive prompts ────────────────────────────────────────────
+      case "ask_question":
+        ds.pendingQuestion = {
+          questionId: event.questionId,
+          question: event.question,
+          options: event.options
+        };
+        break;
+      case "plan_approval_required":
+        ds.pendingPlanApproval = {
+          planId: event.planId,
+          title: event.title,
+          plan: event.plan
+        };
+        break;
+      case "plan_approved":
+        ds.pendingPlanApproval = null;
+        break;
+      // ── Subagent tracking ──────────────────────────────────────────────
+      case "subagent_start":
+        ds.activeSubagents.set(event.toolCallId, {
+          agentType: event.agentType,
+          task: event.task,
+          modelId: event.modelId,
+          toolCalls: [],
+          textDelta: "",
+          status: "running"
+        });
+        break;
+      case "subagent_text_delta": {
+        const sub = ds.activeSubagents.get(event.toolCallId);
+        if (sub) {
+          sub.textDelta += event.textDelta;
+        }
+        break;
+      }
+      case "subagent_tool_start": {
+        const subAgent = ds.activeSubagents.get(event.toolCallId);
+        if (subAgent) {
+          subAgent.toolCalls.push({ name: event.subToolName, isError: false });
+        }
+        break;
+      }
+      case "subagent_tool_end": {
+        const subTool = ds.activeSubagents.get(event.toolCallId);
+        if (subTool) {
+          const tc = subTool.toolCalls.find((t) => t.name === event.subToolName && !t.isError);
+          if (tc) {
+            tc.isError = event.isError;
+          }
+        }
+        break;
+      }
+      case "subagent_end": {
+        const endedSub = ds.activeSubagents.get(event.toolCallId);
+        if (endedSub) {
+          endedSub.status = event.isError ? "error" : "completed";
+          endedSub.durationMs = event.durationMs;
+          endedSub.result = event.result;
+        }
+        break;
+      }
+      // ── Observational Memory ───────────────────────────────────────────
+      case "om_status": {
+        const w = event.windows;
+        ds.omProgress.pendingTokens = w.active.messages.tokens;
+        ds.omProgress.threshold = w.active.messages.threshold;
+        ds.omProgress.thresholdPercent = w.active.messages.threshold > 0 ? w.active.messages.tokens / w.active.messages.threshold * 100 : 0;
+        ds.omProgress.observationTokens = w.active.observations.tokens;
+        ds.omProgress.reflectionThreshold = w.active.observations.threshold;
+        ds.omProgress.reflectionThresholdPercent = w.active.observations.threshold > 0 ? w.active.observations.tokens / w.active.observations.threshold * 100 : 0;
+        ds.omProgress.buffered = {
+          observations: { ...w.buffered.observations },
+          reflection: { ...w.buffered.reflection }
+        };
+        ds.omProgress.generationCount = event.generationCount;
+        ds.omProgress.stepNumber = event.stepNumber;
+        ds.bufferingMessages = w.buffered.observations.status === "running";
+        ds.bufferingObservations = w.buffered.reflection.status === "running";
+        break;
+      }
+      case "om_observation_start":
+        ds.omProgress.status = "observing";
+        ds.omProgress.cycleId = event.cycleId;
+        ds.omProgress.startTime = Date.now();
+        break;
+      case "om_observation_end":
+        ds.omProgress.status = "idle";
+        ds.omProgress.cycleId = void 0;
+        ds.omProgress.startTime = void 0;
+        ds.omProgress.observationTokens = event.observationTokens;
+        ds.omProgress.pendingTokens = 0;
+        ds.omProgress.thresholdPercent = 0;
+        break;
+      case "om_observation_failed":
+        ds.omProgress.status = "idle";
+        ds.omProgress.cycleId = void 0;
+        ds.omProgress.startTime = void 0;
+        break;
+      case "om_reflection_start":
+        ds.omProgress.status = "reflecting";
+        ds.omProgress.cycleId = event.cycleId;
+        ds.omProgress.startTime = Date.now();
+        ds.omProgress.preReflectionTokens = ds.omProgress.observationTokens;
+        ds.omProgress.observationTokens = event.tokensToReflect;
+        ds.omProgress.reflectionThresholdPercent = ds.omProgress.reflectionThreshold > 0 ? event.tokensToReflect / ds.omProgress.reflectionThreshold * 100 : 0;
+        break;
+      case "om_reflection_end":
+        ds.omProgress.status = "idle";
+        ds.omProgress.cycleId = void 0;
+        ds.omProgress.startTime = void 0;
+        ds.omProgress.observationTokens = event.compressedTokens;
+        ds.omProgress.reflectionThresholdPercent = ds.omProgress.reflectionThreshold > 0 ? event.compressedTokens / ds.omProgress.reflectionThreshold * 100 : 0;
+        break;
+      case "om_reflection_failed":
+        ds.omProgress.status = "idle";
+        ds.omProgress.cycleId = void 0;
+        ds.omProgress.startTime = void 0;
+        break;
+      case "om_buffering_start":
+        if (event.operationType === "observation") {
+          ds.bufferingMessages = true;
+        } else {
+          ds.bufferingObservations = true;
+        }
+        break;
+      case "om_buffering_end":
+        if (event.operationType === "observation") {
+          ds.bufferingMessages = false;
+        } else {
+          ds.bufferingObservations = false;
+        }
+        break;
+      case "om_buffering_failed":
+        if (event.operationType === "observation") {
+          ds.bufferingMessages = false;
+        } else {
+          ds.bufferingObservations = false;
+        }
+        break;
+      case "om_activation":
+        if (event.operationType === "observation") {
+          ds.bufferingMessages = false;
+        } else {
+          ds.bufferingObservations = false;
+        }
+        break;
+      // ── Token usage ────────────────────────────────────────────────────
+      case "usage_update":
+        ds.tokenUsage = {
+          promptTokens: this.tokenUsage.promptTokens,
+          completionTokens: this.tokenUsage.completionTokens,
+          totalTokens: this.tokenUsage.totalTokens
+        };
+        break;
+      // ── Tasks ──────────────────────────────────────────────────────────
+      case "task_updated":
+        ds.previousTasks = [...ds.tasks];
+        ds.tasks = event.tasks;
+        break;
+      // ── Thread lifecycle ───────────────────────────────────────────────
+      case "thread_changed":
+        this.resetThreadDisplayState();
+        ds.tokenUsage = { ...this.tokenUsage };
+        break;
+      case "thread_created":
+        this.resetThreadDisplayState();
+        ds.tokenUsage = { promptTokens: 0, completionTokens: 0, totalTokens: 0 };
+        break;
+      // ── State changes (for OM threshold overrides) ──────────────────────
+      case "state_changed": {
+        const keys = event.changedKeys;
+        if (keys.includes("observationThreshold")) {
+          const value = event.state.observationThreshold;
+          if (typeof value === "number") {
+            ds.omProgress.threshold = value;
+            ds.omProgress.thresholdPercent = value > 0 ? ds.omProgress.pendingTokens / value * 100 : 0;
+          }
+        }
+        if (keys.includes("reflectionThreshold")) {
+          const value = event.state.reflectionThreshold;
+          if (typeof value === "number") {
+            ds.omProgress.reflectionThreshold = value;
+            ds.omProgress.reflectionThresholdPercent = value > 0 ? ds.omProgress.observationTokens / value * 100 : 0;
+          }
+        }
+        break;
+      }
+    }
+  }
   // ===========================================================================
   // Runtime Context
   // ===========================================================================
@@ -2124,6 +2539,6 @@ var Harness = class {
   }
 };
 
-export { Harness, askUserTool, parseSubagentMeta, submitPlanTool, taskCheckTool, taskWriteTool };
+export { Harness, askUserTool, defaultDisplayState, defaultOMProgressState, parseSubagentMeta, submitPlanTool, taskCheckTool, taskWriteTool };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map