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

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

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

@@ -1,6 +1,6 @@
 [project]
 name = "shotgun-sh"
-version = "0.1.13"
+version = "0.1.14"
 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}/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}/src/shotgun/codebase/core/manager.py

@@ -1221,26 +1221,11 @@ class CodebaseGraphManager:
 
         removed_graphs = []
 
-        # Find all .kuzu files (can be files or directories)
+        # Find all .kuzu databases (files in v0.11.2, directories in newer versions)
         for path in self.storage_dir.glob("*.kuzu"):
             graph_id = path.stem
 
-            # If it's a plain file (not a directory), it's corrupted
-            # Valid Kuzu databases are always directories
-            if path.is_file():
-                logger.warning(
-                    f"Detected corrupted database file (should be directory) at {path}, removing it"
-                )
-                try:
-                    await anyio.to_thread.run_sync(path.unlink)
-                    removed_graphs.append(graph_id)
-                    logger.info(f"Removed corrupted database file: {graph_id}")
-                except Exception as e:
-                    logger.error(
-                        f"Failed to remove corrupted database file {graph_id}: {e}"
-                    )
-                continue
-
+            # Try to open and validate the database
             try:
                 # Try to open the database with a timeout to prevent hanging
                 async def try_open_database(
@@ -1346,7 +1331,7 @@ class CodebaseGraphManager:
         """
         graphs = []
 
-        # Find all .kuzu files
+        # Find all .kuzu database files (Kuzu v0.11.2 creates files, not directories)
         for path in self.storage_dir.glob("*.kuzu"):
             if path.is_file():
                 graph_id = path.stem
@@ -1439,6 +1424,8 @@ class CodebaseGraphManager:
         Args:
             graph_id: Graph to delete
         """
+        import shutil
+
         # Stop watcher if running
         if graph_id in self._watchers:
             await self.stop_watcher(graph_id)
@@ -1453,11 +1440,14 @@ class CodebaseGraphManager:
                 self._databases[graph_id].close()
                 del self._databases[graph_id]
 
-        # Delete files
+        # Delete database (files in v0.11.2, directories in newer versions)
         graph_path = self.storage_dir / f"{graph_id}.kuzu"
         if graph_path.exists():
-            # Delete the database file
-            await anyio.to_thread.run_sync(graph_path.unlink)
+            if graph_path.is_dir():
+                await anyio.to_thread.run_sync(shutil.rmtree, graph_path)
+            else:
+                # File-based database (Kuzu v0.11.2)
+                await anyio.to_thread.run_sync(graph_path.unlink)
 
         # Also delete the WAL file if it exists
         wal_path = self.storage_dir / f"{graph_id}.kuzu.wal"

{shotgun_sh-0.1.13 → shotgun_sh-0.1.14}/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