shotgun-sh 0.1.0.dev17__py3-none-any.whl → 0.1.0.dev18__py3-none-any.whl

shotgun/agents/agent_manager.py

@@ -17,17 +17,23 @@ from pydantic_ai.agent import AgentRunResult
 from pydantic_ai.messages import (
     AgentStreamEvent,
     FinalResultEvent,
+    FunctionToolCallEvent,
+    FunctionToolResultEvent,
     ModelMessage,
     ModelRequest,
     ModelResponse,
     ModelResponsePart,
     PartDeltaEvent,
     PartStartEvent,
+    SystemPromptPart,
+    ToolCallPart,
     ToolCallPartDelta,
 )
 from textual.message import Message
 from textual.widget import Widget
 
+from shotgun.agents.common import add_system_prompt_message
+
 from .history.compaction import apply_persistent_compaction
 from .models import AgentDeps, AgentRuntimeOptions, FileOperation
 from .plan import create_plan_agent
@@ -264,11 +270,6 @@ class AgentManager(Widget):
             self.ui_message_history.append(ModelRequest.user_text_prompt(prompt))
         self._post_messages_updated()
 
-        # Ensure system prompt is added to message history before running agent
-        from pydantic_ai.messages import SystemPromptPart
-
-        from shotgun.agents.common import add_system_prompt_message
-
         # Start with persistent message history
         message_history = self.message_history
 
@@ -389,19 +390,33 @@ class AgentManager(Widget):
                         state.latest_partial = partial_message
                         self._post_partial_message(partial_message, False)
 
+                elif isinstance(event, FunctionToolCallEvent):
+                    existing_call_idx = next(
+                        (
+                            i
+                            for i, part in enumerate(partial_parts)
+                            if isinstance(part, ToolCallPart)
+                            and part.tool_call_id == event.part.tool_call_id
+                        ),
+                        None,
+                    )
+                    if existing_call_idx is not None:
+                        partial_parts[existing_call_idx] = event.part
+                    else:
+                        partial_parts.append(event.part)
+                    partial_message = self._build_partial_response(partial_parts)
+                    if partial_message is not None:
+                        state.latest_partial = partial_message
+                        self._post_partial_message(partial_message, False)
+                elif isinstance(event, FunctionToolResultEvent):
+                    self.ui_message_history.append(ModelRequest(parts=[event.result]))
+                    self._post_messages_updated()  ## this is what the user responded with
                 elif isinstance(event, FinalResultEvent):
                     final_message = (
                         state.latest_partial
                         or self._build_partial_response(partial_parts)
                     )
-                    self._post_partial_message(final_message, True)
-                    state.latest_partial = None
-                    state.final_sent = True
-                    partial_parts.clear()
-                    self._stream_state = None
-                    break
-
-                # Ignore other AgentStreamEvent variants (e.g. tool call notifications) for partial UI updates.
+                    self._post_partial_message(final_message, False)
 
             except Exception:  # pragma: no cover - defensive logging
                 logger.exception(

shotgun/agents/tools/user_interaction.py

@@ -13,8 +13,9 @@ logger = get_logger(__name__)
 async def ask_user(ctx: RunContext[AgentDeps], question: str) -> str:
     """Ask the human a question and return the answer.
 
+
     Args:
-        question: The question to ask the user
+        question: The question to ask the user with a clear CTA at the end. Needs to be is readable, clear, and easy to understand. Use Markdown formatting. Make key phrases and words stand out.
 
     Returns:
         The user's response as a string

shotgun/codebase/core/manager.py

@@ -27,6 +27,14 @@ from shotgun.logging_config import get_logger
 logger = get_logger(__name__)
 
 
+class CodebaseAlreadyIndexedError(Exception):
+    """Raised when a codebase is already indexed."""
+
+    def __init__(self, repo_path: str):
+        self.repo_path = repo_path
+        super().__init__(f"Codebase already indexed: {repo_path}")
+
+
 class CodebaseFileHandler(FileSystemEventHandler):
     """Handles file system events for code graph updates."""
 
@@ -339,9 +347,7 @@ class CodebaseGraphManager:
 
         # Check if graph already exists
         if graph_path.exists():
-            raise ValueError(
-                f"Graph already exists for {repo_path}. Use update_graph() to modify it."
-            )
+            raise CodebaseAlreadyIndexedError(repo_path)
 
         # Import the builder from local core module
         from shotgun.codebase.core import CodebaseIngestor

shotgun/prompts/agents/research.j2

@@ -23,7 +23,7 @@ Use meaningful artifact IDs like: "api-design-patterns", "microservices-study",
 ## RESEARCH PRINCIPLES
 
 {% if interactive_mode -%}
-- CRITICAL: BEFORE RUNNING ANY SEARCH TOOL, ASK THE USER FOR CONFIRMATION USING ask_user().
+- CRITICAL: BEFORE RUNNING ANY SEARCH TOOL, ASK THE USER FOR APPROVAL USING ask_user(). FINISH THE QUESTION WITH ASKING FOR A GO AHEAD.
 {% endif -%}
 - Build upon existing research rather than starting from scratch
 - Focus on practical, actionable information over theoretical concepts

shotgun/tui/screens/chat.py

@@ -33,6 +33,7 @@ from shotgun.agents.models import (
     UserAnswer,
     UserQuestion,
 )
+from shotgun.codebase.core.manager import CodebaseAlreadyIndexedError
 from shotgun.sdk.codebase import CodebaseSDK
 from shotgun.sdk.exceptions import CodebaseNotFoundError, InvalidPathError
 from shotgun.sdk.services import get_artifact_service, get_codebase_service
@@ -248,6 +249,12 @@ class CodebaseIndexScreen(ModalScreen[CodebaseIndexSelection | None]):
                     disabled=True,
                 )
 
+    def on_mount(self) -> None:
+        name_input = self.query_one("#index-codebase-name", Input)
+        if not name_input.value and self.selected_path:
+            name_input.value = self.selected_path.name
+        self._update_confirm()
+
     def _update_confirm(self) -> None:
         confirm = self.query_one("#index-confirm", Button)
         name_input = self.query_one("#index-codebase-name", Input)
@@ -475,9 +482,14 @@ class ChatScreen(Screen[None]):
         if event.is_last:
             partial_response_widget.partial_response = None
 
+    def _clear_partial_response(self) -> None:
+        partial_response_widget = self.query_one(ChatHistory)
+        partial_response_widget.partial_response = None
+
     @on(MessageHistoryUpdated)
     def handle_message_history_updated(self, event: MessageHistoryUpdated) -> None:
         """Handle message history updates from the agent manager."""
+        self._clear_partial_response()
         self.messages = event.messages
 
         # If there are file operations, add a message showing the modified files
@@ -611,6 +623,11 @@ class ChatScreen(Screen[None]):
                 severity="information",
                 timeout=8,
             )
+
+            self.mount_hint(codebase_indexed_hint(selection.name))
+        except CodebaseAlreadyIndexedError as exc:
+            self.notify(str(exc), severity="warning")
+            return
         except InvalidPathError as exc:
             self.notify(str(exc), severity="error")
 
@@ -619,7 +636,6 @@ class ChatScreen(Screen[None]):
         finally:
             label.update("")
             label.refresh()
-            self.mount_hint(codebase_indexed_hint(selection.name))
 
     @work
     async def run_agent(self, message: str) -> None:

shotgun/tui/screens/chat_screen/history.py

@@ -1,14 +1,17 @@
 import json
+from collections.abc import Sequence
 
 from pydantic_ai.messages import (
     BuiltinToolCallPart,
     BuiltinToolReturnPart,
     ModelMessage,
     ModelRequest,
+    ModelRequestPart,
     ModelResponse,
     TextPart,
     ThinkingPart,
     ToolCallPart,
+    ToolReturnPart,
 )
 from textual.app import ComposeResult
 from textual.reactive import reactive
@@ -78,27 +81,32 @@ class ChatHistory(Widget):
 
     def compose(self) -> ComposeResult:
         self.vertical_tail = VerticalTail()
-        yield self.vertical_tail
-        yield PartialResponseWidget(self.partial_response).data_bind(
-            item=ChatHistory.partial_response
-        )
+        with self.vertical_tail:
+            for item in self.items:
+                if isinstance(item, ModelRequest):
+                    yield UserQuestionWidget(item)
+                elif isinstance(item, ModelResponse):
+                    yield AgentResponseWidget(item)
+            yield PartialResponseWidget(self.partial_response).data_bind(
+                item=ChatHistory.partial_response
+            )
+
+    def watch_partial_response(self, _partial_response: ModelMessage | None) -> None:
+        self.call_after_refresh(self.autoscroll)
 
     def update_messages(self, messages: list[ModelMessage]) -> None:
         """Update the displayed messages without recomposing."""
         if not self.vertical_tail:
             return
 
-        # Clear existing widgets
-        self.vertical_tail.remove_children()
+        self.items = messages
+        self.refresh(recompose=True)
 
-        # Add new message widgets
-        for item in messages:
-            if isinstance(item, ModelRequest):
-                self.vertical_tail.mount(UserQuestionWidget(item))
-            elif isinstance(item, ModelResponse):
-                self.vertical_tail.mount(AgentResponseWidget(item))
+        self.autoscroll()
 
-        self.items = messages
+    def autoscroll(self) -> None:
+        if self.vertical_tail:
+            self.vertical_tail.scroll_end(animate=False)
 
 
 class UserQuestionWidget(Widget):
@@ -116,6 +124,20 @@ class UserQuestionWidget(Widget):
             )
             yield Markdown(markdown=f"**>** {prompt}")
 
+    def format_prompt_parts(self, parts: Sequence[ModelRequestPart]) -> str:
+        acc = ""
+        for part in parts:
+            if isinstance(part, TextPart):
+                acc += (
+                    f"**>** {part.content if isinstance(part.content, str) else ''}\n\n"
+                )
+            elif isinstance(part, ToolCallPart):
+                if part.tool_name == "ask_user" and isinstance(part.content, dict):
+                    acc += f"**>** {part.content['answer']}\n\n"
+                else:
+                    acc += "∟ finished\n\n"  # let's not show anything yet
+        return acc
+
 
 class AgentResponseWidget(Widget):
     def __init__(self, item: ModelResponse | None) -> None:
@@ -133,34 +155,38 @@ class AgentResponseWidget(Widget):
         acc = ""
         if self.item is None:
             return ""
-        for part in self.item.parts:  # TextPart | ToolCallPart | BuiltinToolCallPart | BuiltinToolReturnPart | ThinkingPart
+        for idx, part in enumerate(self.item.parts):
             if isinstance(part, TextPart):
                 acc += part.content + "\n\n"
             elif isinstance(part, ToolCallPart):
                 parts_str = self._format_tool_call_part(part)
                 acc += parts_str + "\n\n"
+            elif isinstance(part, ToolReturnPart):
+                acc += (
+                    f"tool ({part.tool_name}) return: "
+                    + self._format_tool_return_call_part(part)
+                    + "\n\n"
+                )
             elif isinstance(part, BuiltinToolCallPart):
-                acc += f"{part.tool_name}({part.args})\n\n"
+                acc += f"builtin tool ({part.tool_name}): {part.args}\n\n"
             elif isinstance(part, BuiltinToolReturnPart):
-                acc += f"{part.tool_name}()\n\n"
+                acc += f"builtin tool ({part.tool_name}) return: {part.content}\n\n"
             elif isinstance(part, ThinkingPart):
-                acc += f"{part.content}\n\n"
+                if (
+                    idx == len(self.item.parts) - 1
+                ):  # show the thinking part only if it's the last part
+                    acc += (
+                        f"thinking: {part.content}\n\n"
+                        if part.content
+                        else "Thinking..."
+                    )
+                else:
+                    continue
         return acc.strip()
 
     def _format_tool_call_part(self, part: ToolCallPart) -> str:
         if part.tool_name == "ask_user":
-            if isinstance(part.args, str):
-                try:
-                    _args = json.loads(part.args) if part.args.strip() else {}
-                except json.JSONDecodeError:
-                    _args = {}
-            else:
-                _args = part.args
-
-            if isinstance(_args, dict) and "question" in _args:
-                return f"{_args['question']}"
-            else:
-                return "❓ "
+            return self._format_ask_user_part(part)
         if part.tool_name == "write_artifact_section":
             if isinstance(part.args, dict) and "section_title" in part.args:
                 return f"{part.tool_name}({part.args['section_title']})"
@@ -170,6 +196,31 @@ class AgentResponseWidget(Widget):
             if isinstance(part.args, dict) and "name" in part.args:
                 return f"{part.tool_name}({part.args['name']})"
             else:
-                return f"{part.tool_name}()"
+                return f"▪ {part.tool_name}()"
 
         return f"{part.tool_name}({part.args})"
+
+    def _format_ask_user_part(
+        self,
+        part: ToolCallPart,
+    ) -> str:
+        return "*Answer to continue*"
+        if isinstance(part.args, str):
+            try:
+                _args = json.loads(part.args) if part.args.strip() else {}
+            except json.JSONDecodeError:
+                _args = {}
+        else:
+            _args = part.args
+
+        if isinstance(_args, dict) and "question" in _args:
+            return f"{_args['question']}"
+        else:
+            return "❓ "
+
+    def _format_tool_return_call_part(self, part: ToolReturnPart) -> str:
+        content = part.content
+        if part.tool_name == "ask_user":
+            response = content.get("answer", "") if isinstance(content, dict) else ""
+            return f"**⏺** {response}"
+        return f"∟ {content}"

{shotgun_sh-0.1.0.dev17.dist-info → shotgun_sh-0.1.0.dev18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.1.0.dev17
+Version: 0.1.0.dev18
 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.1.0.dev17.dist-info → shotgun_sh-0.1.0.dev18.dist-info}/RECORD

@@ -7,7 +7,7 @@ shotgun/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 shotgun/sentry_telemetry.py,sha256=3r9on0GQposn9aX6Dkb9mrfaVQl_dIZzhu9BjE838AU,2854
 shotgun/telemetry.py,sha256=aBwCRFU97oiIK5K13OhT7yYCQUAVQyrvnoG-aX3k2ZE,3109
 shotgun/agents/__init__.py,sha256=8Jzv1YsDuLyNPFJyckSr_qI4ehTVeDyIMDW4omsfPGc,25
-shotgun/agents/agent_manager.py,sha256=VRMPfwPtgTBtzyp-JYyyitGk_ZQ8oh8hTmB2cEylyZ0,16024
+shotgun/agents/agent_manager.py,sha256=N9BUoGCydfiHVxLImG4JraSO4niQJqJPcj82M0ATT6g,16828
 shotgun/agents/artifact_state.py,sha256=WkspYQe-9CvBS90PapBsX997yvJ5KWH-qtSXIWmfvp0,2121
 shotgun/agents/common.py,sha256=my-Y7VNNpDDYxDJ4hlRmTs7lIS8tBKFvR9ew7uKxcPE,11369
 shotgun/agents/models.py,sha256=6edILmN7caDAmf8x-qgl7lA42tl2b-ZsiC3da8tS3Xw,7127
@@ -32,7 +32,7 @@ shotgun/agents/history/token_estimation.py,sha256=iNqhDSqFzG0YYxGijMRzj54GALFglO
 shotgun/agents/tools/__init__.py,sha256=QaN80IqWvB5qEcjHqri1-PYvYlO74vdhcwLugoEdblo,772
 shotgun/agents/tools/artifact_management.py,sha256=f8WvCCcXb_sMK0U4Y8D1RLUhcj2rhT6CrxB3bURYxcs,17205
 shotgun/agents/tools/file_management.py,sha256=6ru6DXAl-S6DiCt2HLGTDrK2rJBJpn-t6RkSHzYbxc4,4571
-shotgun/agents/tools/user_interaction.py,sha256=7l0OY8EdgO-9gkKy-yOv0V0P_Uzzfk0jMU39d4XN1xM,1087
+shotgun/agents/tools/user_interaction.py,sha256=b3ncEpvoD06Cz4hwsS-ppVbQajQj640iWnVfA5WBjAA,1236
 shotgun/agents/tools/codebase/__init__.py,sha256=ceAGkK006NeOYaIJBLQsw7Q46sAyCRK9PYDs8feMQVw,661
 shotgun/agents/tools/codebase/codebase_shell.py,sha256=2zEq8YXzdcYttYAfKso_JGRqXHyy3xgAuWlf0unopFg,8635
 shotgun/agents/tools/codebase/directory_lister.py,sha256=MCLGDEc0F-4J8-UrquxdJrIQYs5xzYgws65mjf7Q7X4,4724
@@ -79,14 +79,14 @@ shotgun/codebase/core/change_detector.py,sha256=kWCYLWzRzb3IGGOj71KBn7UOCOKMpINJ
 shotgun/codebase/core/code_retrieval.py,sha256=_JVyyQKHDFm3dxOOua1mw9eIIOHIVz3-I8aZtEsEj1E,7927
 shotgun/codebase/core/ingestor.py,sha256=zMjadeqDOEr2v3vhTS25Jvx0WsLPXpgwquZfbdiz57o,59810
 shotgun/codebase/core/language_config.py,sha256=vsqHyuFnumRPRBV1lMOxWKNOIiClO6FyfKQR0fGrtl4,8934
-shotgun/codebase/core/manager.py,sha256=5GlJKykDGvnb6nTr9w3kyCPTL4OQgmBoesnWr28wvTg,55419
+shotgun/codebase/core/manager.py,sha256=GpLeyxC25HJKxh3wQxTiTkXv9NUDQdH6Mi9IdaMUmVQ,55586
 shotgun/codebase/core/nl_query.py,sha256=iV6NbsyDd1SQpT9U9BtgxcZPsNoEW3OHrkk475r_jAY,11410
 shotgun/codebase/core/parser_loader.py,sha256=LZRrDS8Sp518jIu3tQW-BxdwJ86lnsTteI478ER9Td8,4278
 shotgun/prompts/__init__.py,sha256=RswUm0HMdfm2m2YKUwUsEdRIwoczdbI7zlucoEvHYRo,132
 shotgun/prompts/loader.py,sha256=jy24-E02pCSmz2651aCT2NgHfRrHAGMYvKrD6gs0Er8,4424
 shotgun/prompts/agents/__init__.py,sha256=YRIJMbzpArojNX1BP5gfxxois334z_GQga8T-xyWMbY,39
 shotgun/prompts/agents/plan.j2,sha256=mogmrjqLx5xSXYqWdvLqcObZSfLVLQI39JpVs3Z-V84,2494
-shotgun/prompts/agents/research.j2,sha256=LxlASYYt45sDmZW8YLJYH8ftGWa6Jvx5aD0MGNXXVjs,3025
+shotgun/prompts/agents/research.j2,sha256=cLwPdIvv1MvVaHZYLdBhON-8qk-c3kjrRTF1-7Bc-vI,3069
 shotgun/prompts/agents/specify.j2,sha256=1CC2SHsxA1Yma0gSFsq-k3VpwtEohN9nh2qeRMMRPRA,1809
 shotgun/prompts/agents/tasks.j2,sha256=OYW1zsYRJxoQF4yVMqJNgi4SNz3YmcP4sljHmmqAu4Q,3613
 shotgun/prompts/agents/partials/artifact_system.j2,sha256=kaqkMU-t2x3M7z-4HU4KffSFug1WM5VDsin6tCkxjHg,1528
@@ -123,20 +123,20 @@ shotgun/tui/components/prompt_input.py,sha256=Ss-htqraHZAPaehGE4x86ij0veMjc4Ugad
 shotgun/tui/components/spinner.py,sha256=ovTDeaJ6FD6chZx_Aepia6R3UkPOVJ77EKHfRmn39MY,2427
 shotgun/tui/components/splash.py,sha256=vppy9vEIEvywuUKRXn2y11HwXSRkQZHLYoVjhDVdJeU,1267
 shotgun/tui/components/vertical_tail.py,sha256=kkCH0WjAh54jDvRzIaOffRZXUKn_zHFZ_ichfUpgzaE,1071
-shotgun/tui/screens/chat.py,sha256=y63oP_doU2_83UPByB8IebP3SUDqDBz96XhiL8j-snU,23922
+shotgun/tui/screens/chat.py,sha256=2jtjRBU6r-V5frEud-NqIYLHTi0C5HX6uLAXOwYuUo8,24564
 shotgun/tui/screens/chat.tcss,sha256=2Yq3E23jxsySYsgZf4G1AYrYVcpX0UDW6kNNI0tDmtM,437
 shotgun/tui/screens/directory_setup.py,sha256=lIZ1J4A6g5Q2ZBX8epW7BhR96Dmdcg22CyiM5S-I5WU,3237
 shotgun/tui/screens/provider_config.py,sha256=A_tvDHF5KLP5PV60LjMJ_aoOdT3TjI6_g04UIUqGPqM,7126
 shotgun/tui/screens/splash.py,sha256=E2MsJihi3c9NY1L28o_MstDxGwrCnnV7zdq00MrGAsw,706
 shotgun/tui/screens/chat_screen/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 shotgun/tui/screens/chat_screen/command_providers.py,sha256=4Q8jR7wN4e2q0SWHSsISZP-POyu1B9lyXgoOTKB4ZLc,7198
-shotgun/tui/screens/chat_screen/history.py,sha256=QAVMmtPKXtonrAuyWV6aTteqErH6U9H9V8hB1EPbIkM,5546
+shotgun/tui/screens/chat_screen/history.py,sha256=0rSb512ZhkF8zzqkWfwHp-Cmu5VIqL-LzPkohwlHe4Q,7447
 shotgun/utils/__init__.py,sha256=WinIEp9oL2iMrWaDkXz2QX4nYVPAm8C9aBSKTeEwLtE,198
 shotgun/utils/env_utils.py,sha256=8QK5aw_f_V2AVTleQQlcL0RnD4sPJWXlDG46fsHu0d8,1057
 shotgun/utils/file_system_utils.py,sha256=l-0p1bEHF34OU19MahnRFdClHufThfGAjQ431teAIp0,1004
 shotgun/utils/update_checker.py,sha256=Xf-7w3Pos3etzCoT771gJe2HLkA8_V2GrqWy7ni9UqA,11373
-shotgun_sh-0.1.0.dev17.dist-info/METADATA,sha256=o4PPBFmx5v9oWsXqVW-4kNm-3LHh67PpS7J4yJvIORg,11271
-shotgun_sh-0.1.0.dev17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-shotgun_sh-0.1.0.dev17.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
-shotgun_sh-0.1.0.dev17.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
-shotgun_sh-0.1.0.dev17.dist-info/RECORD,,
+shotgun_sh-0.1.0.dev18.dist-info/METADATA,sha256=woXTk-ZMmEwPuNJXm-LS5ZG0DPFM8ZH39VLVWEhdvZU,11271
+shotgun_sh-0.1.0.dev18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+shotgun_sh-0.1.0.dev18.dist-info/entry_points.txt,sha256=asZxLU4QILneq0MWW10saVCZc4VWhZfb0wFZvERnzfA,45
+shotgun_sh-0.1.0.dev18.dist-info/licenses/LICENSE,sha256=YebsZl590zCHrF_acCU5pmNt0pnAfD2DmAnevJPB1tY,1065
+shotgun_sh-0.1.0.dev18.dist-info/RECORD,,