kiwi-code 0.0.439__tar.gz → 0.0.440__tar.gz

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiwi-code
-Version: 0.0.439
+Version: 0.0.440
 Summary: A textual-based terminal user interface application
 Project-URL: Homepage, https://meetkiwi.ai
 Project-URL: Repository, https://github.com/jetoslabs/kiwi-code
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <4.0,>=3.11
-Requires-Dist: autobots-client==0.1.0
+Requires-Dist: autobots-client==0.1.1
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: psutil>=5.9.0

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "kiwi-code"
-version = "0.0.439"
+version = "0.0.440"
 description = "A textual-based terminal user interface application"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.11,<4.0"
 dependencies = [
-    "autobots-client==0.1.0",
+    "autobots-client==0.1.1",
     "loguru>=0.7.3",
     "pydantic>=2.12.5",
     "textual>=8.1.1",

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/src/kiwi_cli/__init__.py

@@ -1,3 +1,3 @@
 """Kiwi CLI - command-line interface and shared infrastructure modules."""
 
-__version__ = "0.0.439"
+__version__ = "0.0.440"

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/src/kiwi_runtime/__init__.py

@@ -1,3 +1,3 @@
 """Kiwi Runtime — terminal agent that connects to the server via WebSocket."""
 
-__version__ = "0.0.439"
+__version__ = "0.0.440"

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/src/kiwi_runtime/main.py

@@ -3182,6 +3182,9 @@ async def connect(
 
 
 def main():
+    import setproctitle
+    setproctitle.setproctitle("kiwi")
+
     parser = argparse.ArgumentParser(
         description="Kiwi AI CLI Agent — execute terminal commands for LLM agents"
     )

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/src/kiwi_tui/__init__.py

@@ -1,3 +1,3 @@
 """Autobots TUI - A textual-based terminal user interface."""
 
-__version__ = "0.0.439"
+__version__ = "0.0.440"

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/src/kiwi_tui/screens/term_dashboard.py

@@ -1184,6 +1184,9 @@ class TermDashboardScreen(Screen):
         self._cmd_result_visible: bool = False  # True while inline cmd output is shown
         self._cmd_running: bool = False
         self._is_streaming: bool = False
+        # True when the current run's latest prompt is still processing server-side.
+        # Used to prevent sending concurrent prompts (especially after reattaching with /continue).
+        self._run_inflight: bool = False
         self._pending_urls: list[str] = []
         self._detach_selected_urls: set[str] = set()
         self._metadata: dict = {}
@@ -1198,6 +1201,7 @@ class TermDashboardScreen(Screen):
         self._spinner_i: int = 0
 
         self._last_copy_notify_at: float = 0.0
+        self._last_submit_block_notify_at: float = 0.0
         self._clickable_list_seq: int = 0
 
         # Inline list mode for #term-runs-list (affects click behavior)
@@ -1737,7 +1741,6 @@ class TermDashboardScreen(Screen):
             elif event.action == "logs":
                 self.action_show_logs()
             elif event.action == "quit":
-                # Use the app-level quit flow so we still prompt for runtime cleanup if needed.
                 try:
                     self.app.action_quit()
                 except Exception:
@@ -1872,7 +1875,6 @@ class TermDashboardScreen(Screen):
             logger.debug(f"_show_slash_autocomplete: {exc}")
             return
 
-        # Tell the input widget to intercept navigation keys
         try:
             inp = self.query_one("#term-chat-input", TermSlashInput)
             inp.autocomplete_active = True
@@ -1929,25 +1931,75 @@ class TermDashboardScreen(Screen):
         if not command.needs_args:
             self.call_after_refresh(lambda v=val: self._do_send_value(v))
 
+    def _toast_submit_blocked(self, message: str) -> None:
+        """Show a throttled toast when the user tries to submit while busy."""
+        try:
+            now = time.monotonic()
+            if (now - self._last_submit_block_notify_at) < 1.0:
+                return
+            self._last_submit_block_notify_at = now
+            self.notify(
+                message,
+                title="Please wait",
+                severity="warning",
+                timeout=2.5,
+                markup=False,
+            )
+        except Exception:
+            pass
+
     def _do_send_value(self, value: str) -> None:
         """Programmatically send a value as if the user typed + pressed Enter."""
         self._hide_slash_autocomplete()
+        val = str(value or "")
+        msg = val.strip()
+        if not msg:
+            return
+
         if self._is_streaming or self._cmd_running:
+            try:
+                inp = self.query_one("#term-chat-input", TermSlashInput)
+                if not inp.value.strip():
+                    inp.value = msg
+            except Exception:
+                pass
+            self._toast_submit_blocked("Busy streaming — you can keep typing, but can't send yet.")
+            return
+
+        # Slash commands are allowed when we're not actively streaming.
+        if msg.startswith("/"):
+            try:
+                inp = self.query_one("#term-chat-input", TermSlashInput)
+                inp.record(msg)
+                inp.value = ""
+            except Exception:
+                pass
+            self._dispatch_slash_command(msg)
+            return
+
+        if self._run_inflight:
+            # Keep the value in the input as a draft (best-effort).
+            try:
+                inp = self.query_one("#term-chat-input", TermSlashInput)
+                if not inp.value.strip():
+                    inp.value = msg
+            except Exception:
+                pass
+            self._toast_submit_blocked(
+                f"Run {self.current_run_id or '(unknown)'} is still processing — draft your next message and send when done."
+            )
             return
+
         try:
             inp = self.query_one("#term-chat-input", TermSlashInput)
-            inp.record(value)
+            inp.record(msg)
             inp.value = ""
         except Exception:
             pass
-        if value.strip().startswith("/"):
-            self._dispatch_slash_command(value.strip())
-        else:
-            self._enter_conversation_mode()
-            self._hide_cmd_result()
-            self._add_message(value, "user")
-            self._process_message(value)
-
+        self._enter_conversation_mode()
+        self._hide_cmd_result()
+        self._add_message(msg, "user")
+        self._process_message(msg)
     #  Top bar / hints 
     def _refresh_top_bar(self) -> None:
         try:
@@ -1962,6 +2014,7 @@ class TermDashboardScreen(Screen):
             run = self.current_run_id or ""
             run_short = (run[:8] + "…") if len(run) > 8 else run
             streaming_flag = "  ● streaming" if self._is_streaming else ""
+            processing_flag = "  ● processing" if (self._run_inflight and not self._is_streaming) else ""
 
             parts: list[str] = [f"Kiwi Code v{KIWI_CODE_VERSION}", now, cwd_short]
             if action_short:
@@ -1977,7 +2030,9 @@ class TermDashboardScreen(Screen):
                 x0 = cell_len(prefix)
                 self._top_bar_run_bounds = (x0, x0 + cell_len(run_part))
 
-            self.query_one("#term-top-bar", Static).update("  " + " ── ".join(parts) + streaming_flag)
+            self.query_one("#term-top-bar", Static).update(
+                "  " + " ── ".join(parts) + streaming_flag + processing_flag
+            )
             self._refresh_welcome_meta()
         except Exception:
             pass
@@ -1991,8 +2046,6 @@ class TermDashboardScreen(Screen):
         except Exception:
             pass
 
-
-
     def _refresh_hint_model(self) -> None:
         try:
             action = self.current_action_name or ""
@@ -2151,10 +2204,8 @@ class TermDashboardScreen(Screen):
                 pass
             self._on_files_selected(file_paths)
             return
-        # If the input is focused, let TextArea's paste handling do its normal thing.
         if chat_input.has_focus:
             return
-        # Forward paste to the input when focus is elsewhere (prevents "paste does nothing").
         event.prevent_default()
         event.stop()
         try:
@@ -2171,24 +2222,37 @@ class TermDashboardScreen(Screen):
                 pass
     # Send 
     def _do_send(self) -> None:
-        # Always dismiss autocomplete before sending
         self._hide_slash_autocomplete()
+
         if self._is_streaming or self._cmd_running:
+            self._toast_submit_blocked("Busy streaming — you can keep typing, but can't send yet.")
             return
+
         chat_input = self.query_one("#term-chat-input", ChatInput)
-        message = chat_input.value.strip()
-        if message:
-            chat_input.record(message)
-        chat_input.value = ""
+        raw = chat_input.value
+        message = raw.strip()
+        if not message:
+            return
 
         if message.startswith("/"):
+            chat_input.record(message)
+            chat_input.value = ""
             self._dispatch_slash_command(message)
-        else:
-            # First real prompt — switch to conversation mode
-            self._enter_conversation_mode()
-            self._hide_cmd_result()
-            self._add_message(message if message else " ", "user")
-            self._process_message(message)
+            return
+
+        if self._run_inflight:
+            self._toast_submit_blocked(
+                f"Run {self.current_run_id or '(unknown)'} is still processing — draft your next message and send when done."
+            )
+            return
+
+        chat_input.record(message)
+        chat_input.value = ""
+
+        self._enter_conversation_mode()
+        self._hide_cmd_result()
+        self._add_message(message, "user")
+        self._process_message(message)
 
     #  Messages
     @staticmethod
@@ -2298,6 +2362,7 @@ class TermDashboardScreen(Screen):
                     if not w.is_finished:
                         w.cancel()
                 self._set_streaming(False)
+                self._set_run_inflight(False)
                 self.current_action_id = self.DEFAULT_ACTION_ID
                 self.current_action_name = None
                 self.current_run_id = None
@@ -2345,6 +2410,7 @@ class TermDashboardScreen(Screen):
                 self.current_action_name = None
                 self.current_run_id = None
                 self.current_run_kind = None
+                self._set_run_inflight(False)
                 self._clear_messages()
                 self._refresh_top_bar()
                 self.run_worker(self._fetch_action_name_async(), group="status",
@@ -2862,6 +2928,7 @@ class TermDashboardScreen(Screen):
         self.current_action_name = None
         self.current_run_id = None
         self.current_run_kind = None
+        self._set_run_inflight(False)
         self._refresh_top_bar()
         self._clear_messages()
         self.run_worker(self._fetch_action_name_async(), group="status", exclusive=False, exit_on_error=False)
@@ -2972,12 +3039,10 @@ class TermDashboardScreen(Screen):
                 scope=effective_args.scope,
                 allow_dirs=effective_args.allow_dirs,
             )
-            # Ensure runtime won't attribute our restore edits to an active prompt.
             try:
                 set_active_entry(run_dir, None)
             except Exception:
                 pass
-            # If latest doesn't exist yet, create it from current state (no-op restore).
             try:
                 ensure_latest_snapshot_exists(run_dir=run_dir, meta=meta)
             except Exception:
@@ -3077,8 +3142,16 @@ class TermDashboardScreen(Screen):
         )
         if not success or not result:
             return
+        if self.current_run_id and run_id != self.current_run_id:
+            return
         self._render_history(result)
 
+        status = str(result.get("status", "") or "").lower()
+        inflight = status in ("processing", "running", "pending")
+        self._set_run_inflight(inflight)
+        if inflight:
+            self._start_reattach_stream(run_id)
+
     def _render_history(self, result: dict) -> None:
         if not isinstance(result, dict):
             return
@@ -3103,8 +3176,62 @@ class TermDashboardScreen(Screen):
                 if text_content:
                     self._add_message(text_content, "assistant")
 
+
+    def _start_reattach_stream(self, run_id: str) -> None:
+        """Resume SSE streaming for an *existing* run that is still processing."""
+        if self._is_streaming:
+            return
+        if not run_id:
+            return
+        if self.current_run_id and run_id != self.current_run_id:
+            return
+        if not hasattr(self.app, "autobots_client"):
+            return
+        # Mount a streaming placeholder if one isn't already present.
+        try:
+            existing = self._active_streaming_row()
+            if existing:
+                self._streaming_widget_ref = existing
+            else:
+                messages = self.query_one("#term-messages", Vertical)
+                placeholder = AssistantMessageRow(
+                    "",
+                    verb="Reattaching",
+                    streaming=True,
+                    classes="term-message term-assistant-message",
+                )
+                messages.mount(placeholder)
+                self._scroll_end_root(after_refresh=True)
+                self._streaming_widget_ref = placeholder
+        except Exception:
+            self._streaming_widget_ref = None
+        # Mark in-flight + disable input while actively streaming.
+        self._set_run_inflight(True)
+        self._set_streaming(True)
+        self.run_worker(
+            self._reattach_worker(run_id),
+            exclusive=True,
+            group="stream",
+            exit_on_error=False,
+        )
+
+    async def _reattach_worker(self, run_id: str) -> None:
+        """Worker that attaches to SSE/polling for an already-started run."""
+        try:
+            self.app._sync_autobots_client_from_disk()
+        except Exception:
+            pass
+        if not hasattr(self.app, "autobots_client"):
+            return
+        client = self.app.autobots_client
+        # Best-effort: ensure CLI runtime is connected for this run while we stream.
+        try:
+            self.app.ensure_runtime_for_run_id(run_id)
+        except Exception:
+            pass
+        await self._consume_sse(run_id, client)
+
     #  Checkpoints / rewind
-    # --------------------
     # kiwi-runtime has built-in checkpointing hooks for filesystem mutations, but only when
     # an "active entry" is set for the current run. The classic dashboard wires this up;
     # TermDashboard must do the same so /rewind has checkpoints to restore.
@@ -3188,6 +3315,7 @@ class TermDashboardScreen(Screen):
         except Exception:
             self._streaming_widget_ref = None
 
+        self._set_run_inflight(True)
         self._set_streaming(True)
         self.run_worker(self._sse_worker(user_input), exclusive=True, group="stream")
 
@@ -3210,7 +3338,6 @@ class TermDashboardScreen(Screen):
 
         def _cleanup_checkpoint() -> None:
             if checkpoint_run_dir is not None and checkpoint_entry_id is not None:
-                # Update latest snapshot so "Latest Code" stays accurate across prompts.
                 try:
                     from kiwi_cli.checkpoints import ensure_meta, update_latest_from_entry, workspace_root_from_cwd
                     workspace_root = workspace_root_from_cwd()
@@ -3230,15 +3357,11 @@ class TermDashboardScreen(Screen):
                     pass
                 self._checkpoint_clear_active(run_dir=checkpoint_run_dir)
 
-        # Auto-connect local CLI runtime for every prompt.
-        # For existing runs, ensure a pinned runtime is available before running the action.
         if self.current_run_id:
             try:
                 self.app.ensure_runtime_for_run_id(self.current_run_id)
             except Exception:
                 pass
-            # If this is a continuation, start the checkpoint entry before sending the request
-            # to reduce races with early file edits.
             checkpoint_run_dir, checkpoint_entry_id = self._checkpoint_start_for_prompt(
                 run_id=self.current_run_id,
                 user_prompt=user_input,
@@ -3260,6 +3383,7 @@ class TermDashboardScreen(Screen):
             self._add_message(f"Error starting action: {e}", "error")
             self._drop_placeholder()
             self._set_streaming(False)
+            self._set_run_inflight(False)
             _cleanup_checkpoint()
             return
 
@@ -3285,6 +3409,7 @@ class TermDashboardScreen(Screen):
                 self._add_message(f"Error starting action: {message}", "error")
                 self._drop_placeholder()
                 self._set_streaming(False)
+                self._set_run_inflight(False)
                 _cleanup_checkpoint()
                 return
 
@@ -3413,6 +3538,7 @@ class TermDashboardScreen(Screen):
                                             f"(HTTP {c2}). Please log in again."
                                         )
                                         got_final = True
+                                        self._set_run_inflight(False)
                                         self._set_streaming(False)
                                         return True
                     if not ok or not res:
@@ -3431,6 +3557,7 @@ class TermDashboardScreen(Screen):
                         else:
                             self._add_message(f"Error: {err_msg}", "error")
                         got_final = True
+                        self._set_run_inflight(False)
                         self._set_streaming(False)
                         return True
                     if status not in ("completed", "success", "finished"):
@@ -3447,6 +3574,7 @@ class TermDashboardScreen(Screen):
                             sw_ref[0] = None
                             self._streaming_widget_ref = None
                             got_final = True
+                            self._set_run_inflight(False)
                             self._set_streaming(False)
                             return True
                     return False
@@ -3716,6 +3844,13 @@ class TermDashboardScreen(Screen):
         except Exception:
             pass
 
+    def _set_run_inflight(self, inflight: bool) -> None:
+        """Track whether the backend is still processing the latest prompt for this run."""
+        self._run_inflight = bool(inflight)
+        self._apply_blocking_state()
+        self._refresh_top_bar()
+
+
     def _set_streaming(self, streaming: bool) -> None:
         self._is_streaming = streaming
         try:
@@ -3754,19 +3889,21 @@ class TermDashboardScreen(Screen):
         self._refresh_top_bar()
 
     def _apply_blocking_state(self) -> None:
-        disabled = self._is_streaming or self._cmd_running
+        blocked = self._is_streaming or self._cmd_running or self._run_inflight
         try:
             ci = self.query_one("#term-chat-input", ChatInput)
-            ci.disabled = disabled
+            ci.disabled = False
             if self._is_streaming:
-                ci.placeholder = "\u25cf Streaming\u2026"
+                ci.placeholder = "\u25cf Streaming\u2026 (draft now; send when finished)"
             elif self._cmd_running:
-                ci.placeholder = "Running command\u2026"
+                ci.placeholder = "Busy\u2026 (draft now; send when finished)"
+            elif self._run_inflight:
+                ci.placeholder = "\u25cf Processing\u2026 (draft now; send when finished)"
             else:
                 ci.placeholder = (
                     "Message\u2026" if self._has_conversation else "What would you like to do?"
                 )
-            if not disabled:
+            if not self._cmd_running:
                 ci.focus()
         except Exception:
             pass
@@ -3939,6 +4076,7 @@ class TermDashboardScreen(Screen):
             pass
         try:
             self._set_streaming(False)
+            self._set_run_inflight(False)
         except Exception:
             pass
         self._drop_placeholder()

{kiwi_code-0.0.439 → kiwi_code-0.0.440}/uv.lock

@@ -181,16 +181,16 @@ wheels = [
 
 [[package]]
 name = "autobots-client"
-version = "0.1.0"
+version = "0.1.1"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "attrs" },
     { name = "httpx" },
     { name = "python-dateutil" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/ac/53/976cb848a3662e429fd215399c232cdab6b4f7d75489e2a634c0e3ca5e4c/autobots_client-0.1.0.tar.gz", hash = "sha256:2df4614ba19c05d214f291471a80a7788eadfd71d0fabbe582f4d3db688ce89d", size = 179502, upload-time = "2026-03-12T22:49:44.669Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/1d/ab/1e6a7c2a71d3080dc0b07abffe023acf97c1aa23437b9c1fc1bdb493c33e/autobots_client-0.1.1.tar.gz", hash = "sha256:6c8164713793e6424458260a7a67216a8ffb52fccf6c34d6c8c45684fe78b7a3", size = 152893, upload-time = "2026-06-11T21:36:36.548Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/c6/c1/367c0f57909453125ac115770f426970c88012a7a5cb4076be89b613ae7f/autobots_client-0.1.0-py3-none-any.whl", hash = "sha256:ef3ca67267ac61006204a8fb9a79c9f5b6d1023ae27266c85973799888cc0108", size = 706909, upload-time = "2026-03-12T22:49:46.313Z" },
+    { url = "https://files.pythonhosted.org/packages/87/57/b5c5ed1a6cba22b6f243e13143d8b31ebb5075fec4451c0a9ea0b1f3046b/autobots_client-0.1.1-py3-none-any.whl", hash = "sha256:a3c2aa84c6b53eb6818f218fc0eff369d16b46244a239132ead5d0b111757183", size = 622952, upload-time = "2026-06-11T21:36:37.776Z" },
 ]
 
 [[package]]
@@ -397,7 +397,7 @@ wheels = [
 
 [[package]]
 name = "kiwi-code"
-version = "0.0.439"
+version = "0.0.440"
 source = { editable = "." }
 dependencies = [
     { name = "autobots-client" },
@@ -421,7 +421,7 @@ test = [
 
 [package.metadata]
 requires-dist = [
-    { name = "autobots-client", specifier = "==0.1.0" },
+    { name = "autobots-client", specifier = "==0.1.1" },
     { name = "httpx", specifier = ">=0.25.0" },
     { name = "loguru", specifier = ">=0.7.3" },
     { name = "psutil", specifier = ">=5.9.0" },