shotgun-sh 0.1.13__tar.gz → 0.1.14.dev1__tar.gz

{shotgun_sh-0.1.13 → shotgun_sh-0.1.14.dev1}/PKG-INFO

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

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

{shotgun_sh-0.1.13 → shotgun_sh-0.1.14.dev1}/src/shotgun/agents/agent_manager.py

@@ -31,7 +31,6 @@ from pydantic_ai.messages import (
     SystemPromptPart,
     ToolCallPart,
     ToolCallPartDelta,
-    ToolReturnPart,
 )
 from textual.message import Message
 from textual.widget import Widget
@@ -273,8 +272,15 @@ class AgentManager(Widget):
         # Use merged deps (shared state + agent-specific system prompt) if not provided
         if deps is None:
             deps = self._create_merged_deps(self._current_agent_type)
-        if not deferred_tool_results:
-            self.ensure_agent_canecelled_safely()
+        ask_user_part = self.get_unanswered_ask_user_part()
+        if ask_user_part and prompt:
+            if not deferred_tool_results:
+                deferred_tool_results = DeferredToolResults()
+            deferred_tool_results.calls[ask_user_part.tool_call_id] = UserAnswer(
+                answer=prompt,
+                tool_call_id=ask_user_part.tool_call_id,
+            )
+            prompt = None
 
         # Ensure deps is not None
         if deps is None:
@@ -677,31 +683,21 @@ class AgentManager(Widget):
         self.ui_message_history.append(message)
         self._post_messages_updated()
 
-    def ensure_agent_canecelled_safely(self) -> None:
+    def get_unanswered_ask_user_part(self) -> ToolCallPart | None:
         if not self.message_history:
-            return
+            return None
         self.last_response = self.message_history[-1]
         ## we're searching for unanswered ask_user parts
-        found_tool = None
-        for part in self.message_history[-1].parts:
-            if isinstance(part, ToolCallPart) and part.tool_name == "ask_user":
-                found_tool = part
-                break
-        if not found_tool:
-            return
-        tool_result = ModelRequest(
-            parts=[
-                ToolReturnPart(
-                    tool_call_id=found_tool.tool_call_id,
-                    tool_name=found_tool.tool_name,
-                    content=UserAnswer(
-                        answer="⚠️ Operation cancelled by user",
-                        tool_call_id=found_tool.tool_call_id,
-                    ),
-                )
-            ]
+        found_tool = next(
+            (
+                part
+                for part in self.message_history[-1].parts
+                if isinstance(part, ToolCallPart) and part.tool_name == "ask_user"
+            ),
+            None,
         )
-        self.message_history.append(tool_result)
+
+        return found_tool
 
 
 # Re-export AgentType for backward compatibility

{shotgun_sh-0.1.13 → shotgun_sh-0.1.14.dev1}/src/shotgun/build_constants.py

@@ -12,8 +12,8 @@ POSTHOG_API_KEY = ''
 POSTHOG_PROJECT_ID = '191396'
 
 # Logfire configuration embedded at build time (only for dev builds)
-LOGFIRE_ENABLED = ''
-LOGFIRE_TOKEN = ''
+LOGFIRE_ENABLED = 'true'
+LOGFIRE_TOKEN = 'pylf_v1_us_KZ5NM1pP3NwgJkbBJt6Ftdzk8mMhmrXcGJHQQgDJ1LfK'
 
 # Build metadata
 BUILD_TIME_ENV = "production" if SENTRY_DSN else "development"

{shotgun_sh-0.1.13 → shotgun_sh-0.1.14.dev1}/src/shotgun/tui/screens/chat.py

@@ -4,7 +4,7 @@ from dataclasses import dataclass
 from pathlib import Path
 from typing import cast
 
-from pydantic_ai import DeferredToolResults, RunContext
+from pydantic_ai import RunContext
 from pydantic_ai.messages import (
     ModelMessage,
     ModelRequest,
@@ -37,7 +37,6 @@ from shotgun.agents.models import (
     AgentDeps,
     AgentType,
     FileOperationTracker,
-    UserAnswer,
     UserQuestion,
 )
 from shotgun.codebase.core.manager import CodebaseAlreadyIndexedError
@@ -712,7 +711,6 @@ class ChatScreen(Screen[None]):
 
     @work
     async def run_agent(self, message: str) -> None:
-        deferred_tool_results = None
         prompt = None
         self.working = True
 
@@ -721,32 +719,15 @@ class ChatScreen(Screen[None]):
 
         self._current_worker = get_current_worker()
 
-        if self.question:
-            # This is a response to a question from the agent
-            self.question.result.set_result(
-                UserAnswer(answer=message, tool_call_id=self.question.tool_call_id)
-            )
-
-            deferred_tool_results = DeferredToolResults()
-
-            deferred_tool_results.calls[self.question.tool_call_id] = UserAnswer(
-                answer=message, tool_call_id=self.question.tool_call_id
-            )
-
-            self.question = None
-        else:
-            # This is a new user prompt
-            prompt = message
+        prompt = message
 
         try:
             await self.agent_manager.run(
                 prompt=prompt,
-                deferred_tool_results=deferred_tool_results,
             )
         except asyncio.CancelledError:
             # Handle cancellation gracefully - DO NOT re-raise
             self.mount_hint("⚠️ Operation cancelled by user")
-            self.agent_manager.ensure_agent_canecelled_safely()
         finally:
             self.working = False
             self._current_worker = None