kiwi-code 0.0.38__tar.gz → 0.0.39__tar.gz

{kiwi_code-0.0.38 → kiwi_code-0.0.39}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kiwi-code
-Version: 0.0.38
+Version: 0.0.39
 Summary: A textual-based terminal user interface application
 Project-URL: Homepage, https://meetkiwi.ai
 Project-URL: Repository, https://github.com/jetoslabs/kiwi-code

{kiwi_code-0.0.38 → kiwi_code-0.0.39}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "kiwi-code"
-version = "0.0.38"
+version = "0.0.39"
 description = "A textual-based terminal user interface application"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.11,<4.0"

{kiwi_code-0.0.38 → kiwi_code-0.0.39}/src/kiwi_tui/screens/dashboard.py

@@ -2354,6 +2354,42 @@ class DashboardScreen(Screen):
             return None
         return None
 
+    _MESSAGES_AUTOFOLLOW_THRESHOLD: int = 2
+
+    def _messages_is_near_bottom(self, *, threshold: int | None = None) -> bool:
+        """Return True when the chat viewport is already at or near the bottom.
+
+        Streaming updates should only auto-follow when the user hasn't intentionally
+        scrolled upward to inspect earlier messages.
+        """
+        try:
+            messages = self.query_one("#messages", VerticalScroll)
+            margin = (
+                self._MESSAGES_AUTOFOLLOW_THRESHOLD
+                if threshold is None
+                else max(0, int(threshold))
+            )
+            current_position = max(
+                int(messages.scroll_offset.y),
+                int(messages.scroll_target_y),
+            )
+            remaining = max(0, int(messages.max_scroll_y) - current_position)
+            return remaining <= margin
+        except Exception:
+            # Fail open so unrelated rendering issues don't block visible output.
+            return True
+
+    def _scroll_messages_to_end(self, *, after_refresh: bool = False) -> None:
+        """Scroll the messages viewport to the bottom without animation."""
+        try:
+            messages = self.query_one("#messages", VerticalScroll)
+            if after_refresh:
+                self.call_after_refresh(messages.scroll_end, animate=False, immediate=True)
+            else:
+                messages.scroll_end(animate=False)
+        except Exception:
+            pass
+
     def _blink_streaming_dot_tick(self) -> None:
         """Timer tick: blink the dot for the active streaming assistant row."""
         w = getattr(self, "_streaming_widget_ref", None)
@@ -3113,6 +3149,7 @@ class DashboardScreen(Screen):
             return widget_ref
 
         messages = self.query_one("#messages", VerticalScroll)
+        should_autofollow = self._messages_is_near_bottom()
 
         markdown_text = self._prepare_markdown(text_content)
         if widget_ref is None:
@@ -3163,7 +3200,8 @@ class DashboardScreen(Screen):
                 except Exception:
                     pass
 
-        messages.scroll_end(animate=False)
+        if should_autofollow:
+            self._scroll_messages_to_end(after_refresh=True)
         return widget_ref
 
     def extract_text_from_output(self, output: any) -> str:

{kiwi_code-0.0.38 → kiwi_code-0.0.39}/tests/test_tui_headless.py

@@ -805,6 +805,114 @@ async def test_tui_allows_empty_prompt(isolated_home: Path, monkeypatch: pytest.
 
     assert called == [""]
 
+
+@pytest.mark.asyncio
+async def test_tui_streaming_requests_autofollow_when_viewport_is_near_bottom(
+    isolated_home: Path, monkeypatch: pytest.MonkeyPatch
+ ) -> None:
+    """Streaming updates should request auto-follow only when the viewport is near bottom."""
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test(size=(80, 24)) as pilot:
+        await pilot.pause()
+        assert type(app.screen).__name__ == "DashboardScreen"
+        screen = app.screen
+
+        calls: list[dict] = []
+        monkeypatch.setattr(screen, "_messages_is_near_bottom", lambda *args, **kwargs: True)
+        monkeypatch.setattr(
+            screen,
+            "_scroll_messages_to_end",
+            lambda *args, **kwargs: calls.append(dict(kwargs)),
+        )
+
+        widget = screen.update_streaming_message({"blocks": [{"text": "stream start"}]})
+        await pilot.pause()
+        assert widget is not None
+        assert calls == [{"after_refresh": True}]
+
+        calls.clear()
+        screen.update_streaming_message(
+            {"blocks": [{"text": "\n".join(f"stream line {i}" for i in range(10))}]},
+            widget,
+        )
+        await pilot.pause()
+        assert calls == [{"after_refresh": True}]
+
+
+@pytest.mark.asyncio
+async def test_tui_streaming_does_not_force_scroll_when_user_reads_older_messages(
+    isolated_home: Path,
+ ) -> None:
+    """Streaming updates should not yank the viewport back to the bottom."""
+    tokens_path = isolated_home / ".kiwi" / "tokens.json"
+    tokens_path.parent.mkdir(parents=True, exist_ok=True)
+    tokens_path.write_text(
+        json.dumps(
+            {
+                "access_token": "test-access-token",
+                "refresh_token": "test-refresh-token",
+                "token_type": "Bearer",
+                "expires_at": None,
+            }
+        ),
+        encoding="utf-8",
+    )
+
+    from kiwi_tui.main import AutobotsTUI
+
+    app = AutobotsTUI()
+    async with app.run_test(size=(80, 24)) as pilot:
+        await pilot.pause()
+        assert type(app.screen).__name__ == "DashboardScreen"
+        screen = app.screen
+        messages = screen.query_one("#messages")
+
+        for i in range(30):
+            screen.add_message(
+                "\n".join(f"history {i} line {j}" for j in range(4)),
+                "assistant",
+            )
+        await pilot.pause()
+        assert messages.max_scroll_y > 0
+
+        widget = screen.update_streaming_message({"blocks": [{"text": "stream start"}]})
+        await pilot.pause()
+        assert widget is not None
+
+        target_y = max(0, int(messages.max_scroll_y) - 5)
+        messages.scroll_to(y=target_y, animate=False, immediate=True)
+        await pilot.pause()
+        y_before = int(messages.scroll_offset.y)
+        target_before = int(messages.scroll_target_y)
+        assert y_before == target_before
+        assert y_before < int(messages.max_scroll_y)
+
+        screen.update_streaming_message(
+            {"blocks": [{"text": "\n".join(f"stream line {i}" for i in range(60))}]},
+            widget,
+        )
+        await pilot.pause()
+
+        assert int(messages.scroll_offset.y) == y_before
+        assert int(messages.scroll_target_y) == target_before
+        assert int(messages.scroll_offset.y) < int(messages.max_scroll_y)
+
 @pytest.mark.asyncio
 async def test_tui_quits_cleanly(isolated_home: Path) -> None:
     from kiwi_tui.main import AutobotsTUI

{kiwi_code-0.0.38 → kiwi_code-0.0.39}/uv.lock

@@ -397,7 +397,7 @@ wheels = [
 
 [[package]]
 name = "kiwi-code"
-version = "0.0.38"
+version = "0.0.39"
 source = { editable = "." }
 dependencies = [
     { name = "autobots-client" },