shotgun-sh 0.2.11.dev7__py3-none-any.whl → 0.2.19__py3-none-any.whl

shotgun/tui/screens/chat/chat_screen.py

@@ -2,6 +2,7 @@
 
 import asyncio
 import logging
+import time
 from datetime import datetime, timezone
 from pathlib import Path
 from typing import cast
@@ -11,6 +12,7 @@ from pydantic_ai.messages import (
     ModelRequest,
     ModelResponse,
     TextPart,
+    ToolCallPart,
     ToolReturnPart,
     UserPromptPart,
 )
@@ -102,7 +104,6 @@ class ChatScreen(Screen[None]):
     history: PromptHistory = PromptHistory()
     messages = reactive(list[ModelMessage | HintMessage]())
     indexing_job: reactive[CodebaseIndexSelection | None] = reactive(None)
-    partial_message: reactive[ModelMessage | None] = reactive(None)
 
     # Q&A mode state (for structured output clarifying questions)
     qa_mode = reactive(False)
@@ -113,6 +114,10 @@ class ChatScreen(Screen[None]):
     # Working state - keep reactive for Textual watchers
     working = reactive(False)
 
+    # Throttle context indicator updates (in seconds)
+    _last_context_update: float = 0.0
+    _context_update_throttle: float = 5.0  # 5 seconds
+
     def __init__(
         self,
         agent_manager: AgentManager,
@@ -279,10 +284,8 @@ class ChatScreen(Screen[None]):
     def action_toggle_mode(self) -> None:
         # Prevent mode switching during Q&A
         if self.qa_mode:
-            self.notify(
-                "Cannot switch modes while answering questions",
-                severity="warning",
-                timeout=3,
+            self.agent_manager.add_hint_message(
+                HintMessage(message="⚠️ Cannot switch modes while answering questions")
             )
             return
 
@@ -304,14 +307,18 @@ class ChatScreen(Screen[None]):
         if usage_hint:
             self.mount_hint(usage_hint)
         else:
-            self.notify("No usage hint available", severity="error")
+            self.agent_manager.add_hint_message(
+                HintMessage(message="⚠️ No usage hint available")
+            )
 
     async def action_show_context(self) -> None:
         context_hint = await self.agent_manager.get_context_hint()
         if context_hint:
             self.mount_hint(context_hint)
         else:
-            self.notify("No context analysis available", severity="error")
+            self.agent_manager.add_hint_message(
+                HintMessage(message="⚠️ No context analysis available")
+            )
 
     def action_view_onboarding(self) -> None:
         """Show the onboarding modal."""
@@ -436,7 +443,9 @@ class ChatScreen(Screen[None]):
 
         except Exception as e:
             logger.error(f"Failed to compact conversation: {e}", exc_info=True)
-            self.notify(f"Failed to compact: {e}", severity="error")
+            self.agent_manager.add_hint_message(
+                HintMessage(message=f"❌ Failed to compact: {e}")
+            )
         finally:
             # Hide spinner
             self.processing_state.stop_processing()
@@ -484,7 +493,9 @@ class ChatScreen(Screen[None]):
 
         except Exception as e:
             logger.error(f"Failed to clear conversation: {e}", exc_info=True)
-            self.notify(f"Failed to clear: {e}", severity="error")
+            self.agent_manager.add_hint_message(
+                HintMessage(message=f"❌ Failed to clear: {e}")
+            )
 
     @work(exclusive=False)
     async def update_context_indicator(self) -> None:
@@ -573,8 +584,6 @@ class ChatScreen(Screen[None]):
 
     @on(PartialResponseMessage)
     def handle_partial_response(self, event: PartialResponseMessage) -> None:
-        self.partial_message = event.message
-
         # Filter event.messages to exclude ModelRequest with only ToolReturnPart
         # These are intermediate tool results that would render as empty (UserQuestionWidget
         # filters out ToolReturnPart in format_prompt_parts), causing user messages to disappear
@@ -598,16 +607,33 @@ class ChatScreen(Screen[None]):
         )
 
         # Use widget coordinator to set partial response
-        self.widget_coordinator.set_partial_response(
-            self.partial_message, new_message_list
+        self.widget_coordinator.set_partial_response(event.message, new_message_list)
+
+        # Skip context updates for file write operations (they don't add to input context)
+        has_file_write = any(
+            isinstance(msg, ModelResponse)
+            and any(
+                isinstance(part, ToolCallPart)
+                and part.tool_name in ("write_file", "append_file")
+                for part in msg.parts
+            )
+            for msg in event.messages
         )
 
-        # Update context indicator with full message history including streaming messages
-        # Combine existing agent history with new streaming messages for accurate token count
-        combined_agent_history = self.agent_manager.message_history + event.messages
-        self.update_context_indicator_with_messages(
-            combined_agent_history, new_message_list
-        )
+        if has_file_write:
+            return  # Skip context update for file writes
+
+        # Throttle context indicator updates to improve performance during streaming
+        # Only update at most once per 5 seconds to avoid excessive token calculations
+        current_time = time.time()
+        if current_time - self._last_context_update >= self._context_update_throttle:
+            self._last_context_update = current_time
+            # Update context indicator with full message history including streaming messages
+            # Combine existing agent history with new streaming messages for accurate token count
+            combined_agent_history = self.agent_manager.message_history + event.messages
+            self.update_context_indicator_with_messages(
+                combined_agent_history, new_message_list
+            )
 
     def _clear_partial_response(self) -> None:
         # Use widget coordinator to clear partial response
@@ -919,11 +945,15 @@ class ChatScreen(Screen[None]):
     async def delete_codebase(self, graph_id: str) -> None:
         try:
             await self.codebase_sdk.delete_codebase(graph_id)
-            self.notify(f"Deleted codebase: {graph_id}", severity="information")
+            self.agent_manager.add_hint_message(
+                HintMessage(message=f"✓ Deleted codebase: {graph_id}")
+            )
         except CodebaseNotFoundError as exc:
-            self.notify(str(exc), severity="error")
+            self.agent_manager.add_hint_message(HintMessage(message=f"❌ {exc}"))
         except Exception as exc:  # pragma: no cover - defensive UI path
-            self.notify(f"Failed to delete codebase: {exc}", severity="error")
+            self.agent_manager.add_hint_message(
+                HintMessage(message=f"❌ Failed to delete codebase: {exc}")
+            )
 
     def _is_kuzu_corruption_error(self, exception: Exception) -> bool:
         """Check if error is related to kuzu database corruption.
@@ -1030,9 +1060,10 @@ class ChatScreen(Screen[None]):
                     )
                     cleaned = await manager.cleanup_corrupted_databases()
                     logger.info(f"Cleaned up {len(cleaned)} corrupted database(s)")
-                    self.notify(
-                        f"Retrying indexing after cleanup (attempt {attempt + 1}/{max_retries})...",
-                        severity="information",
+                    self.agent_manager.add_hint_message(
+                        HintMessage(
+                            message=f"🔄 Retrying indexing after cleanup (attempt {attempt + 1}/{max_retries})..."
+                        )
                     )
 
                 # Pass the current working directory as the indexed_from_cwd
@@ -1060,22 +1091,22 @@ class ChatScreen(Screen[None]):
                 logger.info(
                     f"Successfully indexed codebase '{result.name}' (ID: {result.graph_id})"
                 )
-                self.notify(
-                    f"Indexed codebase '{result.name}' (ID: {result.graph_id})",
-                    severity="information",
-                    timeout=8,
+                self.agent_manager.add_hint_message(
+                    HintMessage(
+                        message=f"✓ Indexed codebase '{result.name}' (ID: {result.graph_id})"
+                    )
                 )
                 break  # Success - exit retry loop
 
             except CodebaseAlreadyIndexedError as exc:
                 progress_timer.stop()
                 logger.warning(f"Codebase already indexed: {exc}")
-                self.notify(str(exc), severity="warning")
+                self.agent_manager.add_hint_message(HintMessage(message=f"⚠️ {exc}"))
                 return
             except InvalidPathError as exc:
                 progress_timer.stop()
                 logger.error(f"Invalid path error: {exc}")
-                self.notify(str(exc), severity="error")
+                self.agent_manager.add_hint_message(HintMessage(message=f"❌ {exc}"))
                 return
 
             except Exception as exc:  # pragma: no cover - defensive UI path
@@ -1094,10 +1125,10 @@ class ChatScreen(Screen[None]):
                     f"Failed to index codebase after {attempt + 1} attempts - "
                     f"repo_path: {selection.repo_path}, name: {selection.name}, error: {exc}"
                 )
-                self.notify(
-                    f"Failed to index codebase after {attempt + 1} attempts: {exc}",
-                    severity="error",
-                    timeout=30,  # Keep error visible for 30 seconds
+                self.agent_manager.add_hint_message(
+                    HintMessage(
+                        message=f"❌ Failed to index codebase after {attempt + 1} attempts: {exc}"
+                    )
                 )
                 break
 

shotgun/tui/screens/chat_screen/command_providers.py

@@ -5,6 +5,7 @@ from textual.command import DiscoveryHit, Hit, Provider
 
 from shotgun.agents.models import AgentType
 from shotgun.codebase.models import CodebaseGraph
+from shotgun.tui.screens.chat_screen.hint_message import HintMessage
 from shotgun.tui.screens.model_picker import ModelPickerScreen
 from shotgun.tui.screens.provider_config import ProviderConfigScreen
 
@@ -271,8 +272,8 @@ class DeleteCodebasePaletteProvider(Provider):
         try:
             result = await self.chat_screen.codebase_sdk.list_codebases()
         except Exception as exc:  # pragma: no cover - defensive UI path
-            self.chat_screen.notify(
-                f"Unable to load codebases: {exc}", severity="error"
+            self.chat_screen.agent_manager.add_hint_message(
+                HintMessage(message=f"❌ Unable to load codebases: {exc}")
             )
             return []
         return result.graphs

shotgun/tui/screens/chat_screen/history/chat_history.py

@@ -47,7 +47,6 @@ class ChatHistory(Widget):
         super().__init__()
         self.items: Sequence[ModelMessage | HintMessage] = []
         self.vertical_tail: VerticalTail | None = None
-        self.partial_response = None
         self._rendered_count = 0  # Track how many messages have been mounted
 
     def compose(self) -> ComposeResult:
@@ -63,7 +62,7 @@ class ChatHistory(Widget):
                     yield HintMessageWidget(item)
                 elif isinstance(item, ModelResponse):
                     yield AgentResponseWidget(item)
-            yield PartialResponseWidget(self.partial_response).data_bind(
+            yield PartialResponseWidget(None).data_bind(
                 item=ChatHistory.partial_response
             )
 

shotgun/tui/screens/directory_setup.py

@@ -8,7 +8,7 @@ from textual import on
 from textual.app import ComposeResult
 from textual.containers import Horizontal, Vertical
 from textual.screen import Screen
-from textual.widgets import Button, Static
+from textual.widgets import Button, Label, Static
 
 from shotgun.utils.file_system_utils import ensure_shotgun_directory_exists
 
@@ -56,6 +56,14 @@ class DirectorySetupScreen(Screen[None]):
         #directory-actions > * {
             margin-right: 2;
         }
+
+        #directory-status {
+            height: auto;
+            padding: 0 1;
+            min-height: 1;
+            color: $error;
+            text-align: center;
+        }
     """
 
     BINDINGS = [
@@ -69,6 +77,7 @@ class DirectorySetupScreen(Screen[None]):
             yield Static("Shotgun keeps workspace data in a .shotgun directory.\n")
             yield Static("Initialize it in the current directory?\n")
             yield Static(f"[$foreground-muted]({Path.cwd().resolve()})[/]")
+        yield Label("", id="directory-status")
         with Horizontal(id="directory-actions"):
             yield Button(
                 "Initialize and proceed \\[ENTER]", variant="primary", id="initialize"
@@ -93,17 +102,17 @@ class DirectorySetupScreen(Screen[None]):
         self._exit_application()
 
     def _initialize_directory(self) -> None:
+        status_label = self.query_one("#directory-status", Label)
         try:
             path = ensure_shotgun_directory_exists()
         except Exception as exc:  # pragma: no cover - defensive; textual path
-            self.notify(f"Failed to initialize directory: {exc}", severity="error")
+            status_label.update(f"❌ Failed to initialize directory: {exc}")
             return
 
         # Double-check a directory now exists; guard against unexpected filesystem state.
         if not path.is_dir():
-            self.notify(
-                "Unable to initialize .shotgun directory due to filesystem conflict.",
-                severity="error",
+            status_label.update(
+                "❌ Unable to initialize .shotgun directory due to filesystem conflict."
             )
             return
 

shotgun/tui/screens/feedback.py

@@ -76,6 +76,13 @@ class FeedbackScreen(Screen[Feedback | None]):
         #feedback-type-list {
             padding: 1;
         }
+
+        #feedback-status {
+            height: auto;
+            padding: 0 1;
+            min-height: 1;
+            color: $error;
+        }
     """
 
     BINDINGS = [
@@ -96,6 +103,7 @@ class FeedbackScreen(Screen[Feedback | None]):
             "",
             id="feedback-description",
         )
+        yield Label("", id="feedback-status")
         with Horizontal(id="feedback-actions"):
             yield Button("Submit", variant="primary", id="submit")
             yield Button("Cancel \\[ESC]", id="cancel")
@@ -176,9 +184,8 @@ class FeedbackScreen(Screen[Feedback | None]):
         description = text_area.text.strip()
 
         if not description:
-            self.notify(
-                "Please enter a description before submitting.", severity="error"
-            )
+            status_label = self.query_one("#feedback-status", Label)
+            status_label.update("❌ Please enter a description before submitting.")
             return
 
         app = cast("ShotgunApp", self.app)

shotgun/tui/screens/github_issue.py

@@ -6,7 +6,7 @@ from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Vertical
 from textual.screen import ModalScreen
-from textual.widgets import Button, Markdown, Static
+from textual.widgets import Button, Label, Markdown, Static
 
 
 class GitHubIssueScreen(ModalScreen[None]):
@@ -47,6 +47,13 @@ class GitHubIssueScreen(ModalScreen[None]):
             margin: 1 1;
             min-width: 20;
         }
+
+        #issue-status {
+            height: auto;
+            padding: 1;
+            min-height: 1;
+            text-align: center;
+        }
     """
 
     BINDINGS = [
@@ -85,6 +92,7 @@ We review all issues and will respond as soon as possible!
                     id="issue-markdown",
                 )
             with Vertical(id="issue-buttons"):
+                yield Label("", id="issue-status")
                 yield Button(
                     "🐙 Open GitHub Issues", id="github-button", variant="primary"
                 )
@@ -94,7 +102,8 @@ We review all issues and will respond as soon as possible!
     def handle_github(self) -> None:
         """Open GitHub issues page in browser."""
         webbrowser.open("https://github.com/shotgun-sh/shotgun/issues")
-        self.notify("Opening GitHub Issues in your browser...")
+        status_label = self.query_one("#issue-status", Label)
+        status_label.update("✓ Opening GitHub Issues in your browser...")
 
     @on(Button.Pressed, "#close-button")
     def handle_close(self) -> None:

shotgun/tui/screens/model_picker.py

@@ -72,6 +72,11 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
             padding: 1 0;
             }
         }
+        #model-picker-status {
+            height: auto;
+            padding: 0 1;
+            color: $error;
+        }
         #model-actions {
             padding: 1;
         }
@@ -94,6 +99,7 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
                 id="model-picker-summary",
             )
         yield ListView(id="model-list")
+        yield Label("", id="model-picker-status")
         with Horizontal(id="model-actions"):
             yield Button("Select \\[ENTER]", variant="primary", id="select")
             yield Button("Done \\[ESC]", id="done")
@@ -349,4 +355,5 @@ class ModelPickerScreen(Screen[ModelConfigUpdated | None]):
                 )
             )
         except Exception as exc:  # pragma: no cover - defensive; textual path
-            self.notify(f"Failed to select model: {exc}", severity="error")
+            status_label = self.query_one("#model-picker-status", Label)
+            status_label.update(f"❌ Failed to select model: {exc}")

shotgun/tui/screens/pipx_migration.py

@@ -8,7 +8,7 @@ from textual import on
 from textual.app import ComposeResult
 from textual.containers import Container, Horizontal, VerticalScroll
 from textual.screen import ModalScreen
-from textual.widgets import Button, Markdown
+from textual.widgets import Button, Label, Markdown
 
 if TYPE_CHECKING:
     pass
@@ -51,6 +51,13 @@ class PipxMigrationScreen(ModalScreen[None]):
             margin: 0 1;
             min-width: 20;
         }
+
+        #migration-status {
+            height: auto;
+            padding: 1;
+            min-height: 1;
+            text-align: center;
+        }
     """
 
     BINDINGS = [
@@ -106,6 +113,7 @@ Or install permanently: `uv tool install shotgun-sh`
                 )
 
                 with Container(id="buttons-container"):
+                    yield Label("", id="migration-status")
                     with Horizontal(id="action-buttons"):
                         yield Button(
                             "Copy Instructions to Clipboard",
@@ -136,16 +144,14 @@ curl -LsSf https://astral.sh/uv/install.sh | sh
 
 # Step 3: Run shotgun with uvx
 uvx shotgun-sh"""
+        status_label = self.query_one("#migration-status", Label)
         try:
             import pyperclip  # type: ignore[import-untyped]  # noqa: PGH003
 
             pyperclip.copy(instructions)
-            self.notify("Copied migration instructions to clipboard!")
+            status_label.update("✓ Copied migration instructions to clipboard!")
         except ImportError:
-            self.notify(
-                "Clipboard not available. See instructions above.",
-                severity="warning",
-            )
+            status_label.update("⚠️ Clipboard not available. See instructions above.")
 
     @on(Button.Pressed, "#continue")
     def _continue(self) -> None:

shotgun/tui/screens/provider_config.py

@@ -77,6 +77,14 @@ class ProviderConfigScreen(Screen[None]):
         #provider-list {
             padding: 1;
         }
+        #provider-status {
+            height: auto;
+            padding: 0 1;
+            min-height: 1;
+        }
+        #provider-status.error {
+            color: $error;
+        }
     """
 
     BINDINGS = [
@@ -103,6 +111,7 @@ class ProviderConfigScreen(Screen[None]):
             password=True,
             id="api-key",
         )
+        yield Label("", id="provider-status")
         with Horizontal(id="provider-actions"):
             yield Button("Save key \\[ENTER]", variant="primary", id="save")
             yield Button("Authenticate", variant="success", id="authenticate")
@@ -280,9 +289,11 @@ class ProviderConfigScreen(Screen[None]):
         """Async implementation of API key saving."""
         input_widget = self.query_one("#api-key", Input)
         api_key = input_widget.value.strip()
+        status_label = self.query_one("#provider-status", Label)
 
         if not api_key:
-            self.notify("Enter an API key before saving.", severity="error")
+            status_label.update("❌ Enter an API key before saving.")
+            status_label.add_class("error")
             return
 
         try:
@@ -291,25 +302,29 @@ class ProviderConfigScreen(Screen[None]):
                 api_key=api_key,
             )
         except Exception as exc:  # pragma: no cover - defensive; textual path
-            self.notify(f"Failed to save key: {exc}", severity="error")
+            status_label.update(f"❌ Failed to save key: {exc}")
+            status_label.add_class("error")
             return
 
         input_widget.value = ""
         await self.refresh_provider_status()
         await self._update_done_button_visibility()
-        self.notify(
-            f"Saved API key for {self._provider_display_name(self.selected_provider)}."
+        status_label.update(
+            f"✓ Saved API key for {self._provider_display_name(self.selected_provider)}."
         )
+        status_label.remove_class("error")
 
     def _clear_api_key(self) -> None:
         self.run_worker(self._do_clear_api_key(), exclusive=True)
 
     async def _do_clear_api_key(self) -> None:
         """Async implementation of API key clearing."""
+        status_label = self.query_one("#provider-status", Label)
         try:
             await self.config_manager.clear_provider_key(self.selected_provider)
         except Exception as exc:  # pragma: no cover - defensive; textual path
-            self.notify(f"Failed to clear key: {exc}", severity="error")
+            status_label.update(f"❌ Failed to clear key: {exc}")
+            status_label.add_class("error")
             return
 
         await self.refresh_provider_status()
@@ -321,9 +336,10 @@ class ProviderConfigScreen(Screen[None]):
             auth_button = self.query_one("#authenticate", Button)
             auth_button.display = True
 
-        self.notify(
-            f"Cleared API key for {self._provider_display_name(self.selected_provider)}."
+        status_label.update(
+            f"✓ Cleared API key for {self._provider_display_name(self.selected_provider)}."
         )
+        status_label.remove_class("error")
 
     async def _start_shotgun_auth(self) -> None:
         """Launch Shotgun Account authentication flow."""
@@ -335,4 +351,5 @@ class ProviderConfigScreen(Screen[None]):
         # Refresh provider status after auth completes
         if result:
             await self.refresh_provider_status()
-            # Notify handled by auth screen
+            # Auto-dismiss provider config screen after successful auth
+            self.dismiss()

shotgun/tui/screens/shotgun_auth.py

@@ -182,12 +182,10 @@ class ShotgunAuthScreen(Screen[bool]):
             self.query_one("#status", Label).update(
                 f"❌ Error: Failed to create authentication token\n{e}"
             )
-            self.notify("Failed to start authentication", severity="error")
 
         except Exception as e:
             logger.error("Unexpected error during auth flow: %s", e)
             self.query_one("#status", Label).update(f"❌ Unexpected error: {e}")
-            self.notify("Authentication failed", severity="error")
 
     async def _poll_token_status(self) -> None:
         """Poll token status until completed or expired."""
@@ -224,17 +222,12 @@ class ShotgunAuthScreen(Screen[bool]):
                             "✅ Authentication successful! Saving credentials..."
                         )
                         await asyncio.sleep(1)
-                        self.notify(
-                            "Shotgun Account configured successfully!",
-                            severity="information",
-                        )
                         self.dismiss(True)
                     else:
                         logger.error("Completed but missing keys")
                         self.query_one("#status", Label).update(
                             "❌ Error: Authentication completed but keys are missing"
                         )
-                        self.notify("Authentication failed", severity="error")
                         await asyncio.sleep(3)
                         self.dismiss(False)
                     return
@@ -250,7 +243,6 @@ class ShotgunAuthScreen(Screen[bool]):
                         "❌ Authentication token expired (30 minutes)\n"
                         "Please try again."
                     )
-                    self.notify("Authentication token expired", severity="error")
                     await asyncio.sleep(3)
                     self.dismiss(False)
                     return
@@ -269,7 +261,6 @@ class ShotgunAuthScreen(Screen[bool]):
                     self.query_one("#status", Label).update(
                         "❌ Authentication token expired"
                     )
-                    self.notify("Authentication token expired", severity="error")
                     await asyncio.sleep(3)
                     self.dismiss(False)
                     return
@@ -290,6 +281,5 @@ class ShotgunAuthScreen(Screen[bool]):
         self.query_one("#status", Label).update(
             "❌ Authentication timeout (30 minutes)\nPlease try again."
         )
-        self.notify("Authentication timeout", severity="error")
         await asyncio.sleep(3)
         self.dismiss(False)

shotgun/tui/screens/welcome.py

@@ -85,6 +85,21 @@ class WelcomeScreen(Screen[None]):
             margin: 1 0 0 0;
             width: 100%;
         }
+
+        #migration-warning {
+            width: 80%;
+            height: auto;
+            padding: 2;
+            margin: 1 0;
+            border: solid $warning;
+            background: $warning 20%;
+        }
+
+        #migration-warning-title {
+            text-style: bold;
+            color: $warning;
+            padding: 0 0 1 0;
+        }
     """
 
     BINDINGS = [
@@ -99,6 +114,23 @@ class WelcomeScreen(Screen[None]):
                 id="welcome-subtitle",
             )
 
+        # Show migration warning if migration failed
+        app = cast("ShotgunApp", self.app)
+        # Note: This is a synchronous call in compose, but config should already be loaded
+        if hasattr(app, "config_manager") and app.config_manager._config:
+            config = app.config_manager._config
+            if config.migration_failed:
+                with Vertical(id="migration-warning"):
+                    yield Static(
+                        "⚠️  Configuration Migration Failed",
+                        id="migration-warning-title",
+                    )
+                    backup_msg = "Your previous configuration couldn't be migrated automatically."
+                    if config.migration_backup_path:
+                        backup_msg += f"\n\nYour old configuration (including API keys) has been backed up to:\n{config.migration_backup_path}"
+                    backup_msg += "\n\nYou'll need to reconfigure Shotgun by choosing an option below."
+                    yield Markdown(backup_msg)
+
         with Container(id="options-container"):
             with Horizontal(id="options"):
                 # Left box - Shotgun Account

shotgun/tui/widgets/widget_coordinator.py

@@ -166,8 +166,9 @@ class WidgetCoordinator:
 
         try:
             chat_history = self.screen.query_one(ChatHistory)
-            if message:
-                chat_history.partial_response = message
+            # Set the reactive attribute to trigger the PartialResponseWidget update
+            chat_history.partial_response = message
+            # Also update the full message list
             chat_history.update_messages(messages)
         except Exception as e:
             logger.exception(f"Failed to set partial response: {e}")