micro-models-agent 0.50.4 → 0.51.1

package/dist/main.js

@@ -12639,16 +12639,41 @@ ${t("tool.truncated", { tokens: Math.ceil(removedChars / 2) })}`;
     }
     contextManager.addMessage({ role: "user", content: input });
     slog.logUser(input);
-    if (config.moe?.enabled) {
-      return runWithMoE({
+    const startedAt = Date.now();
+    let toolCalls = 0;
+    const countTool = (ev) => {
+      if (ev.type === "end")
+        toolCalls++;
+      onTool?.(ev);
+    };
+    const emitTurnEnd = (result, err) => {
+      const interrupted = err instanceof Error && err.name === "AbortError" || this.shutdownRequested;
+      pluginManager.runOnTurnEnd?.({
+        logger,
+        sessionManager: sessionManager?.getActiveMeta(),
+        contextManager
+      }, {
+        success: !!result && result.success !== false && !err,
+        durationMs: Date.now() - startedAt,
+        textLength: (result?.text || "").length,
+        toolCalls,
+        interrupted: interrupted || undefined
+      });
+    };
+    try {
+      const result = config.moe?.enabled === true ? await runWithMoE({
         config,
         llmProvider,
         toolExecutor,
         logger,
         baseDir: this.deps.baseDir
-      }, input, () => this.executeSingleAgentLoop(input, onChunk, onMeta, onTool, onPhase), { onMeta, onTool, onPhase });
+      }, input, () => this.executeSingleAgentLoop(input, onChunk, onMeta, countTool, onPhase), { onMeta, onTool: countTool, onPhase }) : await this.executeSingleAgentLoop(input, onChunk, onMeta, countTool, onPhase);
+      emitTurnEnd(result);
+      return result;
+    } catch (err) {
+      emitTurnEnd(null, err);
+      throw err;
     }
-    return this.executeSingleAgentLoop(input, onChunk, onMeta, onTool, onPhase);
   }
   async executeSingleAgentLoop(input, onChunk, onMeta, onTool, onPhase) {
     const {
@@ -14303,6 +14328,18 @@ var init_checker = __esm(() => {
 // src/modules/plugins/manager.ts
 class PluginManager {
   plugins = [];
+  hostBridge = null;
+  setHostBridge(bridge) {
+    this.hostBridge = bridge;
+  }
+  withHost(ctx) {
+    if (!ctx || typeof ctx !== "object")
+      return ctx;
+    if (!this.hostBridge || ctx.hostBridge !== undefined) {
+      return ctx;
+    }
+    return { ...ctx, hostBridge: this.hostBridge };
+  }
   register(plugin, source) {
     if (source)
       plugin.source = source;
@@ -14341,7 +14378,8 @@ class PluginManager {
       }
     });
   }
-  async runOnBeforeTool(ctx, call) {
+  async runOnBeforeTool(rawCtx, call) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onBeforeTool) {
         try {
@@ -14353,7 +14391,8 @@ class PluginManager {
     }
     return true;
   }
-  async runOnAfterTool(ctx, call, result) {
+  async runOnAfterTool(rawCtx, call, result) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onAfterTool) {
         try {
@@ -14362,7 +14401,8 @@ class PluginManager {
       }
     }
   }
-  async runOnError(ctx, error) {
+  async runOnError(rawCtx, error) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onError) {
         try {
@@ -14371,7 +14411,8 @@ class PluginManager {
       }
     }
   }
-  runOnSessionStart(ctx) {
+  runOnSessionStart(rawCtx) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onSessionStart) {
         try {
@@ -14380,7 +14421,8 @@ class PluginManager {
       }
     }
   }
-  runOnBeforeThink(ctx) {
+  runOnBeforeThink(rawCtx) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onBeforeThink) {
         try {
@@ -14389,7 +14431,8 @@ class PluginManager {
       }
     }
   }
-  runOnAfterThink(ctx, response) {
+  runOnAfterThink(rawCtx, response) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onAfterThink) {
         try {
@@ -14398,7 +14441,8 @@ class PluginManager {
       }
     }
   }
-  runOnSessionEnd(ctx) {
+  runOnSessionEnd(rawCtx) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onSessionEnd) {
         try {
@@ -14407,7 +14451,8 @@ class PluginManager {
       }
     }
   }
-  runOnToolCall(ctx) {
+  runOnToolCall(rawCtx) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onToolCall) {
         try {
@@ -14416,7 +14461,8 @@ class PluginManager {
       }
     }
   }
-  runOnPhase(ctx, phase) {
+  runOnPhase(rawCtx, phase) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onPhase) {
         try {
@@ -14425,7 +14471,8 @@ class PluginManager {
       }
     }
   }
-  runOnToolStart(ctx, call) {
+  runOnToolStart(rawCtx, call) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onToolStart) {
         try {
@@ -14434,7 +14481,8 @@ class PluginManager {
       }
     }
   }
-  runOnToolEnd(ctx, call, result, durationMs) {
+  runOnToolEnd(rawCtx, call, result, durationMs) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onToolEnd) {
         try {
@@ -14443,7 +14491,8 @@ class PluginManager {
       }
     }
   }
-  runOnText(ctx, chunk) {
+  runOnText(rawCtx, chunk) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onText) {
         try {
@@ -14455,7 +14504,8 @@ class PluginManager {
     }
     return chunk;
   }
-  runOnMeta(ctx, chunk) {
+  runOnMeta(rawCtx, chunk) {
+    const ctx = this.withHost(rawCtx);
     for (const plugin of this.plugins) {
       if (plugin.onMeta) {
         try {
@@ -14467,6 +14517,16 @@ class PluginManager {
     }
     return chunk;
   }
+  runOnTurnEnd(rawCtx, summary) {
+    const ctx = this.withHost(rawCtx);
+    for (const plugin of this.plugins) {
+      if (plugin.onTurnEnd) {
+        try {
+          plugin.onTurnEnd(ctx, summary);
+        } catch {}
+      }
+    }
+  }
 }
 var init_manager3 = __esm(() => {
   init_checker();
@@ -35050,6 +35110,22 @@ function formatContextBar(used, limit, compactions, quality) {
   }
   return line;
 }
+function createHostBridge(opts) {
+  return {
+    isBusy: opts.isBusy,
+    async submit(text, o) {
+      if (opts.isBusy())
+        throw new Error("agent is busy");
+      for (const url of o?.images ?? [])
+        opts.addPendingImage(url);
+      return opts.runAgent(text);
+    },
+    interrupt() {
+      if (opts.isBusy())
+        opts.interruptRun();
+    }
+  };
+}
 
 class Repl {
   completer = new Completer;
@@ -35122,6 +35198,19 @@ class Repl {
     registerAllCommands(this);
     this.setupCompleter();
     this.setupListeners();
+    if (pluginManager) {
+      pluginManager.setHostBridge(createHostBridge({
+        isBusy: () => this.agentRunning,
+        addPendingImage: (url) => {
+          this.agent.contextManager?.addPendingImage?.({
+            type: "image_url",
+            image_url: { url }
+          });
+        },
+        runAgent: (text) => this.runAgent(text),
+        interruptRun: () => this.agent.shutdown()
+      }));
+    }
   }
   loadHistory() {
     if (existsSync52(this.historyPath)) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "micro-models-agent",
-  "version": "0.50.4",
+  "version": "0.51.1",
   "description": "Micro Models Agent (MMA) — LLM agent harness for small models (Qwen3.5-9B, 32K-64K context)",
   "type": "module",
   "bin": {