Fix session lifecycle to prevent premature termination

messaging/handler.py

@@ -639,16 +639,8 @@ class ClaudeMessageHandler:
                 )
         finally:
             logger.info(f"HANDLER: _process_node completed for node {node_id}")
-            # Free the session-manager slot. Session IDs are persisted in the tree and
-            # can be resumed later by ID; we don't need to keep a CLISession instance
-            # around after this node completes.
-            try:
-                if captured_session_id:
-                    await self.cli_manager.remove_session(captured_session_id)
-                elif temp_session_id:
-                    await self.cli_manager.remove_session(temp_session_id)
-            except Exception as e:
-                logger.debug(f"Failed to remove session for node {node_id}: {e}")
+            # DO NOT remove sessions here. They should persist for the duration of the conversation.
+            # sessions will be managed/cleaned up by CLISessionManager if needed (e.g. on timeout).
 
     async def _propagate_error_to_children(
         self,

messaging/platforms/telegram.py

@@ -717,10 +717,6 @@ class TelegramPlatform(MessagingPlatform):
         chat_id = str(update.effective_chat.id)
         text = update.message.text or ""
 
-        # DEBUG ECHO
-        if user_id == self.admin_id:
-            await update.message.reply_text(f"DEBUG: Received '{text}'")
-
         # 1. Handle Admin Buttons (Highest Priority)
         if user_id == self.admin_id:
             if text == "🤖 الموديل الحالي": return await self._on_model_command(update, context)
@@ -987,6 +983,9 @@ class TelegramPlatform(MessagingPlatform):
         finally:
             with contextlib.suppress(OSError):
                 tmp_path.unlink(missing_ok=True)
+            logger.info(f"HANDLER: _process_node completed for node {node_id}")
+            # DO NOT remove sessions here. They should persist for the duration of the conversation.
+            # sessions will be managed/cleaned up by CLISessionManager if needed (e.g. on timeout).
 
     async def _on_telegram_error_wrapper(self, update: Update, context: ContextTypes.DEFAULT_TYPE, func) -> None:
         """Wrapper to catch and report errors to chat."""