shotgun-sh 0.4.0.dev2__tar.gz → 0.4.0.dev3__tar.gz

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.4.0.dev2
+Version: 0.4.0.dev3
 Summary: AI-powered research, planning, and task management CLI tool
 Project-URL: Homepage, https://shotgun.sh/
 Project-URL: Repository, https://github.com/shotgun-sh/shotgun

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "shotgun-sh"
-version = "0.4.0.dev2"
+version = "0.4.0.dev3"
 description = "AI-powered research, planning, and task management CLI tool"
 readme = "README.md"
 license = { text = "MIT" }

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/src/shotgun/agents/agent_manager.py

@@ -78,7 +78,7 @@ from .models import AgentDeps, AgentRuntimeOptions
 from .plan import create_plan_agent
 from .research import create_research_agent
 from .router import create_router_agent
-from .router.models import RouterDeps
+from .router.models import RouterDeps, RouterMode
 from .specify import create_specify_agent
 from .tasks import create_tasks_agent
 
@@ -1384,6 +1384,14 @@ class AgentManager(Widget):
             return
 
         plan_display = deps.current_plan.format_for_display()
+
+        # In drafting mode, if plan is not complete, prompt user to continue
+        if (
+            deps.router_mode == RouterMode.DRAFTING
+            and not deps.current_plan.is_complete()
+        ):
+            plan_display += "\n\n**Shall I continue?**"
+
         logger.debug("Adding plan hint to UI history")
         self.ui_message_history.append(
             HintMessage(message=f"**Current Plan**\n\n{plan_display}")

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/src/shotgun/agents/router/models.py

@@ -358,6 +358,10 @@ class RouterDeps(AgentDeps):
     # Set by create_plan tool when plan.needs_approval() returns True
     # Excluded from serialization as it's transient UI state
     pending_approval: PendingApproval | None = Field(default=None, exclude=True)
+    # Completion state for Drafting mode
+    # Set by mark_step_done when plan completes in drafting mode
+    # Excluded from serialization as it's transient UI state
+    pending_completion: bool = Field(default=False, exclude=True)
     # Event stream handler for forwarding sub-agent streaming events to UI
     # This is set by the AgentManager when running the router with streaming
     # Excluded from serialization as it's a callable

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/src/shotgun/agents/router/tools/plan_tools.py

@@ -167,6 +167,11 @@ async def mark_step_done(
             if plan.is_complete():
                 ctx.deps.is_executing = False
                 logger.debug("Plan complete, is_executing=False")
+                # Set pending completion for Drafting mode
+                # The TUI will detect this and show the completion message
+                if ctx.deps.router_mode == RouterMode.DRAFTING:
+                    ctx.deps.pending_completion = True
+                    logger.debug("Set pending_completion=True for drafting mode")
             # Set pending checkpoint for Planning mode
             # The TUI will detect this and show the StepCheckpointWidget
             elif ctx.deps.router_mode == RouterMode.PLANNING:

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/src/shotgun/tui/screens/chat/chat_screen.py

@@ -988,6 +988,10 @@ class ChatScreen(Screen[None]):
         self, event: MessageHistoryUpdated
     ) -> None:
         """Handle message history updates from the agent manager."""
+        logger.debug(
+            "[MSG_HISTORY] MessageHistoryUpdated received - %d messages",
+            len(event.messages),
+        )
         self._clear_partial_response()
         self.messages = event.messages
 
@@ -1623,6 +1627,9 @@ class ChatScreen(Screen[None]):
         # Check for pending checkpoint (Planning mode step completion)
         self._check_pending_checkpoint()
 
+        # Check for plan completion (Drafting mode)
+        self._check_plan_completion()
+
         # Save conversation after each interaction
         self._save_conversation()
 
@@ -1826,6 +1833,31 @@ class ChatScreen(Screen[None]):
             )
         )
 
+    def _check_plan_completion(self) -> None:
+        """Check if a plan was completed in Drafting mode and show completion message.
+
+        This is called after each agent run to check if mark_step_done
+        set pending_completion in Drafting mode.
+        """
+        logger.debug("[PLAN] _check_plan_completion called")
+        if not isinstance(self.deps, RouterDeps):
+            logger.debug("[PLAN] Not RouterDeps, skipping plan completion check")
+            return
+
+        if not self.deps.pending_completion:
+            logger.debug("[PLAN] No pending completion")
+            return
+
+        # Clear the pending state
+        self.deps.pending_completion = False
+
+        # Show completion message
+        logger.debug("[PLAN] Showing plan completion message for drafting mode")
+        self.mount_hint("✅ All plan steps completed!")
+
+        # Hide the plan panel since the plan is done
+        self._hide_plan_panel()
+
     # =========================================================================
     # Sub-Agent Lifecycle Handlers (Stage 8)
     # =========================================================================

{shotgun_sh-0.4.0.dev2 → shotgun_sh-0.4.0.dev3}/src/shotgun/tui/screens/chat_screen/history/chat_history.py

@@ -1,5 +1,6 @@
 """Chat history widget - main container for message display."""
 
+import logging
 from collections.abc import Generator, Sequence
 
 from pydantic_ai.messages import (
@@ -21,6 +22,8 @@ from .agent_response import AgentResponseWidget
 from .partial_response import PartialResponseWidget
 from .user_question import UserQuestionWidget
 
+logger = logging.getLogger(__name__)
+
 
 class ChatHistory(Widget):
     """Main widget for displaying chat message history."""
@@ -89,23 +92,44 @@ class ChatHistory(Widget):
     def update_messages(self, messages: list[ModelMessage | HintMessage]) -> None:
         """Update the displayed messages using incremental mounting."""
         if not self.vertical_tail:
+            logger.debug(
+                "[CHAT_HISTORY] update_messages called but vertical_tail is None"
+            )
             return
 
         self.items = messages
         filtered = list(self.filtered_items())
 
+        logger.debug(
+            "[CHAT_HISTORY] update_messages - total=%d, filtered=%d, rendered=%d",
+            len(messages),
+            len(filtered),
+            self._rendered_count,
+        )
+
         # Only mount new messages that haven't been rendered yet
         if len(filtered) > self._rendered_count:
             new_messages = filtered[self._rendered_count :]
+            logger.debug(
+                "[CHAT_HISTORY] Mounting %d new messages",
+                len(new_messages),
+            )
             for item in new_messages:
                 widget: Widget
                 if isinstance(item, ModelRequest):
                     widget = UserQuestionWidget(item)
+                    logger.debug("[CHAT_HISTORY] Mounting UserQuestionWidget")
                 elif isinstance(item, HintMessage):
                     widget = HintMessageWidget(item)
+                    logger.debug("[CHAT_HISTORY] Mounting HintMessageWidget")
                 elif isinstance(item, ModelResponse):
                     widget = AgentResponseWidget(item)
+                    logger.debug("[CHAT_HISTORY] Mounting AgentResponseWidget")
                 else:
+                    logger.debug(
+                        "[CHAT_HISTORY] Skipping unknown message type: %s",
+                        type(item).__name__,
+                    )
                     continue
 
                 # Mount before the PartialResponseWidget
@@ -115,6 +139,12 @@ class ChatHistory(Widget):
 
             # Scroll to bottom to show newly added messages
             self.vertical_tail.scroll_end(animate=False)
+        else:
+            logger.debug(
+                "[CHAT_HISTORY] No new messages to mount (filtered=%d, rendered=%d)",
+                len(filtered),
+                self._rendered_count,
+            )
 
     def on_click(self, event: events.Click) -> None:
         """Focus the prompt input when clicking on the history area."""