polygram 0.4.8 → 0.4.10

package/.claude-plugin/plugin.json

@@ -1,7 +1,7 @@
 {
   "$schema": "https://anthropic.com/claude-code/plugin.schema.json",
   "name": "polygram",
-  "version": "0.4.8",
+  "version": "0.4.10",
   "description": "Telegram integration for Claude Code that preserves the OpenClaw per-chat session model. Migration target for OpenClaw users. Multi-bot, multi-chat, per-topic isolation; SQLite transcripts; inline-keyboard approvals. Bundles /polygram:status|logs|pair-code|approvals admin commands and a history skill.",
   "keywords": [
     "telegram",

package/lib/process-manager.js

@@ -54,11 +54,12 @@ class ProcessManager {
     db = null,
     logger = console,
     killTimeoutMs = DEFAULT_KILL_TIMEOUT_MS,
-    onInit = null,       // (sessionKey, event, entry) → void
-    onResult = null,     // (sessionKey, event, entry, pending) → void
-    onClose = null,      // (sessionKey, code, entry) → void
-    onStreamChunk = null,// (sessionKey, partialText, entry) → void — routes to pendingQueue[0]
-    onToolUse = null,    // (sessionKey, toolName, entry) → void — routes to pendingQueue[0]
+    onInit = null,        // (sessionKey, event, entry) → void
+    onResult = null,      // (sessionKey, event, entry, pending) → void
+    onClose = null,       // (sessionKey, code, entry) → void
+    onStreamChunk = null, // (sessionKey, partialText, entry) → void — routes to pendingQueue[0]
+    onToolUse = null,     // (sessionKey, toolName, entry) → void — routes to pendingQueue[0]
+    onRespawn = null,     // (sessionKey, reason, entry) → void — fires after graceful drain-and-kill
   } = {}) {
     if (!spawnFn) throw new Error('spawnFn required');
     this.cap = cap;
@@ -71,6 +72,7 @@ class ProcessManager {
     this.onClose = onClose;
     this.onStreamChunk = onStreamChunk;
     this.onToolUse = onToolUse;
+    this.onRespawn = onRespawn;
     this.procs = new Map();
   }
 
@@ -137,13 +139,22 @@ class ProcessManager {
       queued: entry.pendingQueue.length,
     });
     if (entry.pendingQueue.length === 0) {
-      // Fire-and-forget — caller doesn't need to await the kill.
-      this.kill(sessionKey).catch(() => {});
+      // Queue empty — kill immediately, fire onRespawn after close.
+      this._killAndNotifyRespawn(sessionKey, reason).catch(() => {});
       return { killed: true, queued: 0 };
     }
     return { killed: false, queued: entry.pendingQueue.length };
   }
 
+  async _killAndNotifyRespawn(sessionKey, reason) {
+    const entry = this.procs.get(sessionKey);
+    await this.kill(sessionKey);
+    if (this.onRespawn && entry) {
+      try { this.onRespawn(sessionKey, reason, entry); }
+      catch (err) { this.logger.error(`[pm] onRespawn: ${err.message}`); }
+    }
+  }
+
   async kill(sessionKey) {
     const entry = this.procs.get(sessionKey);
     if (!entry) return;
@@ -257,7 +268,8 @@ class ProcessManager {
         } else {
           entry.inFlight = false;
           // Graceful drain-and-respawn: if caller asked for a respawn
-          // (e.g. /model change) and we just emptied the queue, kill now.
+          // (e.g. /model change) and we just emptied the queue, kill now
+          // and fire onRespawn so the caller can post confirmation.
           if (entry.needsRespawn) {
             const reason = entry.needsRespawn;
             entry.needsRespawn = null;
@@ -266,7 +278,7 @@ class ProcessManager {
               chat_id: entry.chatId,
               reason,
             });
-            this.kill(sessionKey).catch(() => {});
+            this._killAndNotifyRespawn(sessionKey, reason).catch(() => {});
           }
         }
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "polygram",
-  "version": "0.4.8",
+  "version": "0.4.10",
   "description": "Telegram daemon for Claude Code that preserves the OpenClaw per-chat session model. Migration path for OpenClaw users moving to Claude Code.",
   "main": "lib/ipc-client.js",
   "bin": {

package/polygram.js

@@ -1021,11 +1021,9 @@ async function handleMessage(sessionKey, chatId, msg, bot) {
         old_value: oldModel, new_value: newModel,
         user: cmdUser, user_id: cmdUserId, source: 'command',
       }), 'log model change');
-      const { queued, anyActive } = requestRespawnForChat('model-change');
+      const { anyActive } = requestRespawnForChat('model-change');
       const ver = MODEL_VERSIONS[newModel] || newModel;
-      const suffix = anyActive
-        ? ` (applies after ${queued} in-flight turn${queued === 1 ? '' : 's'} complete${queued === 1 ? 's' : ''})`
-        : '';
+      const suffix = anyActive ? ` — I'll switch when I finish` : '';
       await sendReply(`Model → ${newModel} (${ver})${suffix}`);
     } else {
       await sendReply(`Unknown model. Use: opus, sonnet, haiku`);
@@ -1043,10 +1041,8 @@ async function handleMessage(sessionKey, chatId, msg, bot) {
         old_value: oldEffort, new_value: newEffort,
         user: cmdUser, user_id: cmdUserId, source: 'command',
       }), 'log effort change');
-      const { queued, anyActive } = requestRespawnForChat('effort-change');
-      const suffix = anyActive
-        ? ` (applies after ${queued} in-flight turn${queued === 1 ? '' : 's'} complete${queued === 1 ? 's' : ''})`
-        : '';
+      const { anyActive } = requestRespawnForChat('effort-change');
+      const suffix = anyActive ? ` — I'll switch when I finish` : '';
       await sendReply(`Effort → ${newEffort}${suffix}`);
     } else {
       await sendReply(`Unknown effort. Use: low, medium, high, xhigh, max`);
@@ -1908,6 +1904,25 @@ async function main() {
       const r = head?.context?.reactor;
       if (r) r.setState(classifyToolName(toolName));
     },
+    // Fires after a graceful /model or /effort drain has actually
+    // swapped to the new settings. Post a confirmation back to the
+    // chat so the user knows the switch happened.
+    onRespawn: (sessionKey, reason, entry) => {
+      const chatId = entry.chatId;
+      if (!chatId) return;
+      const chatConfig = config.chats[chatId];
+      if (!chatConfig) return;
+      const text = reason === 'model-change'
+        ? `✓ Using ${chatConfig.model} now.`
+        : reason === 'effort-change'
+        ? `✓ Effort is ${chatConfig.effort} now.`
+        : `✓ Ready.`;
+      const threadId = entry.threadId || undefined;
+      tg(bot, 'sendMessage', {
+        chat_id: chatId, text,
+        ...(threadId && { message_thread_id: threadId }),
+      }, { source: 'respawn-confirm', botName: BOT_NAME }).catch(() => {});
+    },
   });
 
   console.log(`polygram (LRU cap=${cap}, SQLite source of truth)`);