shotgun-sh 0.2.5.dev1__tar.gz → 0.2.6.dev1__tar.gz

{shotgun_sh-0.2.5.dev1 → shotgun_sh-0.2.6.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.2.5.dev1
+Version: 0.2.6.dev1
 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.2.5.dev1 → shotgun_sh-0.2.6.dev1}/pyproject.toml

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

{shotgun_sh-0.2.5.dev1 → shotgun_sh-0.2.6.dev1}/src/shotgun/agents/agent_manager.py

@@ -36,6 +36,7 @@ from textual.message import Message
 from textual.widget import Widget
 
 from shotgun.agents.common import add_system_prompt_message, add_system_status_message
+from shotgun.agents.config.models import KeyProvider
 from shotgun.agents.models import AgentType, FileOperation
 from shotgun.posthog_telemetry import track_event
 from shotgun.tui.screens.chat_screen.hint_message import HintMessage
@@ -359,10 +360,18 @@ class AgentManager(Widget):
         model_name = ""
         if hasattr(deps, "llm_model") and deps.llm_model is not None:
             model_name = deps.llm_model.name
-        is_gpt5 = (  # streaming is likely not supported for gpt5. It varies between keys.
-            "gpt-5" in model_name.lower()
+
+        # Check if it's a Shotgun account
+        is_shotgun_account = (
+            hasattr(deps, "llm_model")
+            and deps.llm_model is not None
+            and deps.llm_model.key_provider == KeyProvider.SHOTGUN
         )
 
+        # Only disable streaming for GPT-5 if NOT a Shotgun account
+        # Shotgun accounts support streaming for GPT-5
+        is_gpt5_byok = "gpt-5" in model_name.lower() and not is_shotgun_account
+
         # Track message send event
         event_name = f"message_send_{self._current_agent_type.value}"
         track_event(
@@ -383,7 +392,9 @@ class AgentManager(Widget):
                 usage_limits=usage_limits,
                 message_history=message_history,
                 deferred_tool_results=deferred_tool_results,
-                event_stream_handler=self._handle_event_stream if not is_gpt5 else None,
+                event_stream_handler=self._handle_event_stream
+                if not is_gpt5_byok
+                else None,
                 **kwargs,
             )
         finally:

{shotgun_sh-0.2.5.dev1 → shotgun_sh-0.2.6.dev1}/src/shotgun/prompts/agents/partials/common_agent_system_prompt.j2

@@ -20,6 +20,7 @@ Your extensive expertise spans, among other things:
 9. **Be Creative**: If the user seems not to know something, always be creative and come up with ideas that fit their thinking.
 10. Greet the user when you're just starting to work.
 11. DO NOT repeat yourself.
+12. If a user has agreed to a plan, you DO NOT NEED TO FOLLOW UP with them after every step to ask "is this search query ok?".
 
 
 ## Quality Standards

{shotgun_sh-0.2.5.dev1 → shotgun_sh-0.2.6.dev1}/src/shotgun/tui/screens/chat_screen/history.py

@@ -215,8 +215,18 @@ class AgentResponseWidget(Widget):
         acc = ""
         if self.item is None:
             return ""
+
+        # Check if there's an ask_user tool call
+        has_ask_user = any(
+            isinstance(part, ToolCallPart) and part.tool_name == "ask_user"
+            for part in self.item.parts
+        )
+
         for idx, part in enumerate(self.item.parts):
             if isinstance(part, TextPart):
+                # Skip ALL text parts if there's an ask_user tool call
+                if has_ask_user:
+                    continue
                 # Only show the circle prefix if there's actual content
                 if part.content and part.content.strip():
                     acc += f"**⏺** {part.content}\n\n"
@@ -224,7 +234,25 @@ class AgentResponseWidget(Widget):
                 parts_str = self._format_tool_call_part(part)
                 acc += parts_str + "\n\n"
             elif isinstance(part, BuiltinToolCallPart):
-                acc += f"{part.tool_name}({part.args})\n\n"
+                # Format builtin tool calls better
+                if part.tool_name and "search" in part.tool_name.lower():
+                    args = self._parse_args(part.args)
+                    if isinstance(args, dict) and "query" in args:
+                        query = self._truncate(str(args.get("query", "")))
+                        acc += f'Searching: "{query}"\n\n'
+                    else:
+                        acc += f"{part.tool_name}()\n\n"
+                else:
+                    # For other builtin tools, show name only or with truncated args
+                    if part.args:
+                        args_str = (
+                            str(part.args)[:50] + "..."
+                            if len(str(part.args)) > 50
+                            else str(part.args)
+                        )
+                        acc += f"{part.tool_name}({args_str})\n\n"
+                    else:
+                        acc += f"{part.tool_name}()\n\n"
             elif isinstance(part, BuiltinToolReturnPart):
                 acc += f"builtin tool ({part.tool_name}) return: {part.content}\n\n"
             elif isinstance(part, ThinkingPart):
@@ -305,12 +333,16 @@ class AgentResponseWidget(Widget):
                 return f'Reading file: "{args["filename"]}"'
             return "Reading file"
 
-        # Web search tools
-        if part.tool_name in [
-            "openai_web_search_tool",
-            "anthropic_web_search_tool",
-            "gemini_web_search_tool",
-        ]:
+        # Web search tools - handle variations
+        if (
+            part.tool_name
+            in [
+                "openai_web_search_tool",
+                "anthropic_web_search_tool",
+                "gemini_web_search_tool",
+            ]
+            or "search" in part.tool_name.lower()
+        ):  # Catch other search variations
             if "query" in args:
                 query = self._truncate(str(args["query"]))
                 return f'Searching web: "{query}"'
@@ -332,7 +364,24 @@ class AgentResponseWidget(Widget):
                 return f"{part.tool_name}({args['name']})"
             return f"▪ {part.tool_name}()"
 
-        return f"{part.tool_name}({part.args})"
+        # Default case for unrecognized tools - format args properly
+        args = self._parse_args(part.args)
+        if args and isinstance(args, dict):
+            # Try to extract common fields
+            if "query" in args:
+                return f'{part.tool_name}: "{self._truncate(str(args["query"]))}"'
+            elif "question" in args:
+                return f'{part.tool_name}: "{self._truncate(str(args["question"]))}"'
+            else:
+                # Show tool name with truncated args
+                args_str = (
+                    str(part.args)[:50] + "..."
+                    if len(str(part.args)) > 50
+                    else str(part.args)
+                )
+                return f"{part.tool_name}({args_str})"
+        else:
+            return f"{part.tool_name}()"
 
     def _format_ask_user_part(
         self,