linkgravity 1.9.0 → 1.9.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "linkgravity",
-  "version": "1.9.0",
+  "version": "1.9.1",
   "description": "Discord/Telegram bot bridge for the Antigravity (agy) CLI, with voice interaction support",
   "scripts": {
     "start": "node npm-scripts/run-dev.js",

package/src/core/agy_runner.py

@@ -344,7 +344,7 @@ async def agy_new_conversation(
     args = ["--dangerously-skip-permissions", "--print", content, "--print-timeout", "24h"]
     if model:
         args.extend(["--model", clean_model_name(model)])
-    result_text = await run_agy(*args, stream_queue=stream_queue, thread_id=thread_id, cwd=cwd)
+    result_text = await run_agy(*args, timeout=86400, stream_queue=stream_queue, thread_id=thread_id, cwd=cwd)
     conv_id = await _get_latest_conversation_id()
     return result_text, conv_id
 
@@ -360,7 +360,7 @@ async def agy_send_message(
     args = ["--dangerously-skip-permissions", "--print", content, "--conversation", conv_id, "--print-timeout", "24h"]
     if model:
         args.extend(["--model", clean_model_name(model)])
-    return await run_agy(*args, stream_queue=stream_queue, thread_id=thread_id, cwd=cwd)
+    return await run_agy(*args, timeout=86400, stream_queue=stream_queue, thread_id=thread_id, cwd=cwd)
 
 
 def get_current_model() -> str:

package/src/core/session_manager.py

@@ -14,6 +14,7 @@ class SessionManager:
         self.sessions: dict[str, dict] = self._load_sessions()
         self.active_queues: dict[str, asyncio.Queue] = {}
         self.active_tts_tasks: dict[str, asyncio.Task] = {}
+        self.active_handler_tasks: dict[str, asyncio.Task] = {}
         self.pending_approvals: dict[str, asyncio.Future] = {}
         self.pending_approval_types: dict[str, str] = {}
         self.pending_approval_messages: dict[str, Any] = {}
@@ -121,6 +122,15 @@ class SessionManager:
     def remove_tts_task(self, thread_id: str):
         self.active_tts_tasks.pop(str(thread_id), None)
 
+    def get_handler_task(self, thread_id: str) -> asyncio.Task | None:
+        return self.active_handler_tasks.get(str(thread_id))
+
+    def set_handler_task(self, thread_id: str, task: asyncio.Task):
+        self.active_handler_tasks[str(thread_id)] = task
+
+    def remove_handler_task(self, thread_id: str):
+        self.active_handler_tasks.pop(str(thread_id), None)
+
     def set_pending_approval(
         self,
         approval_key: str,

package/src/handlers/thread_reply.py

@@ -133,6 +133,17 @@ async def handle_existing_session(
         cleanup_images(image_paths)
 
 
+async def _run_tracked(thread_id: str, coro) -> None:
+    # Registered so /stop can cancel the whole turn, not just the agy process.
+    task = asyncio.current_task()
+    session_manager.set_handler_task(thread_id, task)
+    try:
+        await coro
+    finally:
+        if session_manager.get_handler_task(thread_id) is task:
+            session_manager.remove_handler_task(thread_id)
+
+
 async def handle_thread_reply(bot, incoming: IncomingMessage):
     session = session_manager.get_session(incoming.conversation_id)
     if not session:
@@ -202,10 +213,16 @@ async def handle_thread_reply(bot, incoming: IncomingMessage):
 
     if not conv_id:
         if session.get("status") == "pending":
-            await handle_pending_session(bot, incoming, session, agy_content, content, image_paths)
+            await _run_tracked(
+                incoming.conversation_id,
+                handle_pending_session(bot, incoming, session, agy_content, content, image_paths),
+            )
             return
         else:
             await adapter.send_message(thread, "⚠️ Session ID not found. Start a new session with `/new`.")
             return
 
-    await handle_existing_session(bot, incoming, session, conv_id, agy_content, image_paths)
+    await _run_tracked(
+        incoming.conversation_id,
+        handle_existing_session(bot, incoming, session, conv_id, agy_content, image_paths),
+    )