@yeaft/webchat-agent 0.1.776 → 0.1.779

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yeaft/webchat-agent",
-  "version": "0.1.776",
+  "version": "0.1.779",
   "description": "Remote agent for Yeaft WebChat — connects worker machines to the central server",
   "main": "index.js",
   "type": "module",

package/unify/engine.js

@@ -326,10 +326,12 @@ export class Engine {
    *   config: object,
    *   conversationStore?: import('./conversation/persist.js').ConversationStore,
    *   memoryIndex?: import('./memory/index-db.js').SegmentIndex,
+   *   amsRegistry?: object,
    *   toolRegistry?: import('./tools/registry.js').ToolRegistry,
    *   skillManager?: import('./skills.js').SkillManager,
    *   mcpManager?: import('./mcp.js').MCPManager,
    *   yeaftDir?: string,
+   *   toolStats?: import('./stats/tool-usage.js').ToolUsageStats,
    * }} params
    */
   constructor({ adapter, trace, config, conversationStore, memoryIndex, amsRegistry, toolRegistry, skillManager, mcpManager, yeaftDir, toolStats = null }) {
@@ -772,6 +774,11 @@ export class Engine {
         parentVpPersona: vpCtx?.vpPersona || null,
         onEvent: this.#subAgentEventSink || null,
         language: this.#config?.language || 'en',
+        // Forward the session-shared ToolUsageStats so sub-agent
+        // engines record tool calls into the same on-disk snapshot
+        // (~/.yeaft/stats/tool-usage.json) the parent engine writes
+        // to. Null when the parent has no stats wired (e.g. tests).
+        toolStats: this.#toolStats || null,
       },
     };
   }

package/unify/eval/runner.js

@@ -90,7 +90,11 @@ export async function runSingleEval(evalCase, { adapter, model, config = {} }) {
   const trace = new NullTrace();
   const engineConfig = { model, maxOutputTokens: 4096, ...config };
 
-  // Build engine — optionally with ToolRegistry
+  // Build engine — optionally with ToolRegistry. NOTE: intentionally
+  // no `toolStats` here. The eval runner is an offline scoring harness
+  // and must NOT pollute the user-facing `~/.yeaft/stats/tool-usage.json`
+  // snapshot (see PR #782 for the session-shared ToolUsageStats
+  // wiring on the runtime paths).
   const engineOpts = { adapter, trace, config: engineConfig };
 
   if (evalCase.registryTools) {

package/unify/sub-agent/runner.js

@@ -86,6 +86,7 @@ export function isRestrictedToolName(name) {
  *   parentName?: string,
  *   parentVpId?: string,
  *   parentVpPersona?: object,
+ *   toolStats?: object,
  *   onEvent?: (agentId: string, evt: object) => void,
  *   language?: 'en'|'zh',
  * }} deps
@@ -112,6 +113,11 @@ export function startSubAgent(agent, deps = {}) {
     skillManager: deps.skillManager || null,
     mcpManager: deps.mcpManager || null,
     yeaftDir: deps.yeaftDir || null,
+    // Share the session-shared ToolUsageStats so sub-agent tool calls
+    // land in the same on-disk snapshot the parent records into. Sub-
+    // agents are often the heaviest tool users — leaving them out
+    // skewed `unify_fetch_tool_stats` output.
+    toolStats: deps.toolStats || null,
   });
 
   agent.subEngine = subEngine;

package/unify/web-bridge.js

@@ -499,6 +499,19 @@ export function __testGroupContextEntry(groupId) {
   return groupContexts.get(groupId);
 }
 
+/**
+ * Test-only: build (or return cached) per-VP Engine for a session that
+ * was wired via `__testSetSession`. Lets tests assert that the engine's
+ * dependencies (notably `toolStats`) come from the session reference —
+ * see `test/agent/web-bridge-vp-engine-tool-stats.test.js`.
+ *
+ * @param {string} groupId
+ * @param {string} vpId
+ */
+export function __testGetOrCreateVpEngine(groupId, vpId) {
+  return getOrCreateVpEngine(groupId, vpId);
+}
+
 /** Whether we've already sent a permission warning to the UI */
 let _permissionDiagnosticSent = false;
 
@@ -540,6 +553,12 @@ function getOrCreateVpEngine(groupId, vpId) {
     skillManager: session.skillManager,
     mcpManager: session.mcpManager,
     yeaftDir: session.yeaftDir,
+    // Share the session-shared ToolUsageStats so per-VP tool calls land
+    // in the same on-disk snapshot the `unify_fetch_tool_stats` handler
+    // reads. Without this, engine's record-on-tool-exec guard
+    // (`if (this.#toolStats && ...)`) is false and group VP tool calls
+    // are silently dropped.
+    toolStats: session.toolStats || null,
   });
   vpEngines.set(key, eng);
   return eng;