shotgun-sh 0.1.0.dev20__tar.gz → 0.1.0.dev23__tar.gz

{shotgun_sh-0.1.0.dev20 → shotgun_sh-0.1.0.dev23}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: shotgun-sh
-Version: 0.1.0.dev20
+Version: 0.1.0.dev23
 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
@@ -16,12 +16,11 @@ Classifier: Intended Audience :: Developers
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Utilities
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Requires-Dist: anthropic>=0.39.0
 Requires-Dist: google-generativeai>=0.8.5
 Requires-Dist: httpx>=0.27.0
@@ -177,7 +176,7 @@ The update command automatically detects and uses the appropriate method:
 
 ### Requirements
 
-- **Python 3.10+** (3.13 recommended)
+- **Python 3.11+** (3.13 recommended)
 - **uv** - Fast Python package installer and resolver
 - **actionlint** (optional) - For GitHub Actions workflow validation
 
@@ -289,17 +288,17 @@ go install github.com/rhysd/actionlint/cmd/actionlint@latest
 
 ### Python Version Management
 
-The project supports **Python 3.10+**. The `.python-version` file specifies Python 3.10 to ensure development against the minimum supported version.
+The project supports **Python 3.11+**. The `.python-version` file specifies Python 3.11 to ensure development against the minimum supported version.
 
 If using **pyenv**:
 ```bash
-pyenv install 3.10.16  # or latest 3.10.x
+pyenv install 3.11
 ```
 
 If using **uv** (recommended):
 ```bash
-uv python install 3.10
-uv sync --python 3.10
+uv python install 3.11
+uv sync --python 3.11
 ```
 
 ### Commit Message Convention
@@ -350,7 +349,7 @@ uv run cz commit
 
 GitHub Actions automatically:
 - Runs on pull requests and pushes to main
-- Tests with Python 3.10
+- Tests with Python 3.11
 - Validates code with ruff, ruff-format, and mypy
 - Ensures all checks pass before merge
 

{shotgun_sh-0.1.0.dev20 → shotgun_sh-0.1.0.dev23}/README.md

@@ -123,7 +123,7 @@ The update command automatically detects and uses the appropriate method:
 
 ### Requirements
 
-- **Python 3.10+** (3.13 recommended)
+- **Python 3.11+** (3.13 recommended)
 - **uv** - Fast Python package installer and resolver
 - **actionlint** (optional) - For GitHub Actions workflow validation
 
@@ -235,17 +235,17 @@ go install github.com/rhysd/actionlint/cmd/actionlint@latest
 
 ### Python Version Management
 
-The project supports **Python 3.10+**. The `.python-version` file specifies Python 3.10 to ensure development against the minimum supported version.
+The project supports **Python 3.11+**. The `.python-version` file specifies Python 3.11 to ensure development against the minimum supported version.
 
 If using **pyenv**:
 ```bash
-pyenv install 3.10.16  # or latest 3.10.x
+pyenv install 3.11
 ```
 
 If using **uv** (recommended):
 ```bash
-uv python install 3.10
-uv sync --python 3.10
+uv python install 3.11
+uv sync --python 3.11
 ```
 
 ### Commit Message Convention
@@ -296,7 +296,7 @@ uv run cz commit
 
 GitHub Actions automatically:
 - Runs on pull requests and pushes to main
-- Tests with Python 3.10
+- Tests with Python 3.11
 - Validates code with ruff, ruff-format, and mypy
 - Ensures all checks pass before merge
 

{shotgun_sh-0.1.0.dev20 → shotgun_sh-0.1.0.dev23}/pyproject.toml

@@ -1,20 +1,19 @@
 [project]
 name = "shotgun-sh"
-version = "0.1.0.dev20"
+version = "0.1.0.dev23"
 description = "AI-powered research, planning, and task management CLI tool"
 readme = "README.md"
 license = { text = "MIT" }
 authors = [
     { name = "Proofs.io", email = "hello@proofs.io" }
 ]
-requires-python = ">=3.10"
+requires-python = ">=3.11"
 keywords = ["cli", "ai", "agent", "research", "planning", "task-management", "productivity", "llm", "pydantic-ai"]
 classifiers = [
     "Development Status :: 3 - Alpha",
     "Intended Audience :: Developers",
     "License :: OSI Approved :: MIT License",
     "Programming Language :: Python :: 3",
-    "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
     "Topic :: Software Development :: Libraries :: Python Modules",
@@ -101,7 +100,7 @@ ignore = ["E501", "T201"]
 "test/**/*.py" = ["S101", "B017"]  # Allow assert statements and broad exceptions in tests
 
 [tool.mypy]
-python_version = "3.10"
+python_version = "3.11"
 strict = true
 warn_return_any = true
 warn_unused_configs = true

{shotgun_sh-0.1.0.dev20 → shotgun_sh-0.1.0.dev23}/src/shotgun/agents/agent_manager.py

@@ -1,10 +1,12 @@
 """Agent manager for coordinating multiple AI agents with shared message history."""
 
 import logging
-from collections.abc import AsyncIterable
-from dataclasses import dataclass, field
-from enum import Enum
-from typing import Any, cast
+from collections.abc import AsyncIterable, Sequence
+from dataclasses import dataclass, field, is_dataclass, replace
+from typing import TYPE_CHECKING, Any, cast
+
+if TYPE_CHECKING:
+    from shotgun.agents.conversation_history import ConversationState
 
 from pydantic_ai import (
     Agent,
@@ -21,6 +23,7 @@ from pydantic_ai.messages import (
     FunctionToolResultEvent,
     ModelMessage,
     ModelRequest,
+    ModelRequestPart,
     ModelResponse,
     ModelResponsePart,
     PartDeltaEvent,
@@ -32,11 +35,12 @@ from pydantic_ai.messages import (
 from textual.message import Message
 from textual.widget import Widget
 
-from shotgun.agents.common import add_system_prompt_message
+from shotgun.agents.common import add_system_prompt_message, add_system_status_message
+from shotgun.agents.models import AgentType, FileOperation
 
 from .export import create_export_agent
 from .history.compaction import apply_persistent_compaction
-from .models import AgentDeps, AgentRuntimeOptions, FileOperation
+from .models import AgentDeps, AgentRuntimeOptions
 from .plan import create_plan_agent
 from .research import create_research_agent
 from .specify import create_specify_agent
@@ -45,16 +49,6 @@ from .tasks import create_tasks_agent
 logger = logging.getLogger(__name__)
 
 
-class AgentType(Enum):
-    """Enumeration for available agent types (for Python < 3.11)."""
-
-    RESEARCH = "research"
-    PLAN = "plan"
-    TASKS = "tasks"
-    SPECIFY = "specify"
-    EXPORT = "export"
-
-
 class MessageHistoryUpdated(Message):
     """Event posted when the message history is updated."""
 
@@ -122,6 +116,7 @@ class AgentManager(Widget):
         agent_runtime_options = AgentRuntimeOptions(
             interactive_mode=self.deps.interactive_mode,
             working_directory=self.deps.working_directory,
+            is_tui_context=self.deps.is_tui_context,
             max_iterations=self.deps.max_iterations,
             queue=self.deps.queue,
             tasks=self.deps.tasks,
@@ -280,6 +275,11 @@ class AgentManager(Widget):
         # Start with persistent message history
         message_history = self.message_history
 
+        deps.agent_mode = self._current_agent_type
+
+        # Add a system status message so the agent knows whats going on
+        message_history = await add_system_status_message(deps, message_history)
+
         # Check if the message history already has a system prompt
         has_system_prompt = any(
             hasattr(msg, "parts")
@@ -459,3 +459,87 @@ class AgentManager(Widget):
                 file_operations=file_operations,
             )
         )
+
+    def _filter_system_prompts(
+        self, messages: list[ModelMessage]
+    ) -> list[ModelMessage]:
+        """Filter out system prompts from messages for UI display.
+
+        Args:
+            messages: List of messages that may contain system prompts
+
+        Returns:
+            List of messages without system prompt parts
+        """
+        from pydantic_ai.messages import SystemPromptPart
+
+        filtered_messages: list[ModelMessage] = []
+        for msg in messages:
+            parts: Sequence[ModelRequestPart] | Sequence[ModelResponsePart] | None = (
+                msg.parts if hasattr(msg, "parts") else None
+            )
+            if not parts:
+                filtered_messages.append(msg)
+                continue
+
+            non_system_parts = [
+                part for part in parts if not isinstance(part, SystemPromptPart)
+            ]
+
+            if not non_system_parts:
+                # Skip messages made up entirely of system prompt parts (e.g. system message)
+                continue
+
+            if len(non_system_parts) == len(parts):
+                # Nothing was filtered – keep original message
+                filtered_messages.append(msg)
+                continue
+
+            if is_dataclass(msg):
+                filtered_messages.append(
+                    # ignore types because of the convoluted Request | Response types
+                    replace(msg, parts=cast(Any, non_system_parts))
+                )
+            else:
+                filtered_messages.append(msg)
+        return filtered_messages
+
+    def get_conversation_state(self) -> "ConversationState":
+        """Get the current conversation state.
+
+        Returns:
+            ConversationState object containing UI and agent messages and current type
+        """
+        from shotgun.agents.conversation_history import ConversationState
+
+        return ConversationState(
+            agent_messages=self.message_history.copy(),
+            agent_type=self._current_agent_type.value,
+        )
+
+    def restore_conversation_state(self, state: "ConversationState") -> None:
+        """Restore conversation state from a saved state.
+
+        Args:
+            state: ConversationState object to restore
+        """
+        # Restore message history for agents (includes system prompts)
+        self.message_history = state.agent_messages.copy()
+
+        # Filter out system prompts for UI display
+        self.ui_message_history = self._filter_system_prompts(state.agent_messages)
+
+        # Restore agent type
+        self._current_agent_type = AgentType(state.agent_type)
+
+        # Notify listeners about the restored messages
+        self._post_messages_updated()
+
+
+# Re-export AgentType for backward compatibility
+__all__ = [
+    "AgentManager",
+    "AgentType",
+    "MessageHistoryUpdated",
+    "PartialResponseMessage",
+]

{shotgun_sh-0.1.0.dev20 → shotgun_sh-0.1.0.dev23}/src/shotgun/agents/common.py

@@ -15,16 +15,16 @@ from pydantic_ai.agent import AgentRunResult
 from pydantic_ai.messages import (
     ModelMessage,
     ModelRequest,
-    ModelResponse,
     SystemPromptPart,
-    TextPart,
 )
 
 from shotgun.agents.config import ProviderType, get_config_manager, get_provider_model
+from shotgun.agents.models import AgentType
 from shotgun.logging_config import get_logger
 from shotgun.prompts import PromptLoader
-from shotgun.sdk.services import get_artifact_service, get_codebase_service
+from shotgun.sdk.services import get_codebase_service
 from shotgun.utils import ensure_shotgun_directory_exists
+from shotgun.utils.file_system_utils import get_shotgun_base_path
 
 from .history import token_limit_compactor
 from .history.compaction import apply_persistent_compaction
@@ -40,14 +40,7 @@ from .tools import (
     retrieve_code,
     write_file,
 )
-from .tools.artifact_management import (
-    create_artifact,
-    list_artifact_templates,
-    list_artifacts,
-    read_artifact,
-    read_artifact_section,
-    write_artifact_section,
-)
+from .tools.file_management import AGENT_DIRECTORIES
 
 logger = get_logger(__name__)
 
@@ -71,21 +64,24 @@ async def add_system_status_message(
     message_history = message_history or []
     codebase_understanding_graphs = await deps.codebase_service.list_graphs()
 
-    # Collect artifact state information
-    from .artifact_state import collect_artifact_state
+    # Get existing files for the agent
+    existing_files = get_agent_existing_files(deps.agent_mode)
 
-    artifact_state = collect_artifact_state()
+    # Extract table of contents from the agent's markdown file
+    markdown_toc = extract_markdown_toc(deps.agent_mode)
 
     system_state = prompt_loader.render(
         "agents/state/system_state.j2",
         codebase_understanding_graphs=codebase_understanding_graphs,
-        **artifact_state,
+        is_tui_context=deps.is_tui_context,
+        existing_files=existing_files,
+        markdown_toc=markdown_toc,
     )
 
     message_history.append(
-        ModelResponse(
+        ModelRequest(
             parts=[
-                TextPart(content=system_state),
+                SystemPromptPart(content=system_state),
             ]
         )
     )
@@ -98,14 +94,17 @@ def create_base_agent(
     load_codebase_understanding_tools: bool = True,
     additional_tools: list[Any] | None = None,
     provider: ProviderType | None = None,
+    agent_mode: AgentType | None = None,
 ) -> tuple[Agent[AgentDeps, str | DeferredToolRequests], AgentDeps]:
     """Create a base agent with common configuration.
 
     Args:
         system_prompt_fn: Function that will be decorated as system_prompt
         agent_runtime_options: Agent runtime options for the agent
+        load_codebase_understanding_tools: Whether to load codebase understanding tools
         additional_tools: Optional list of additional tools
         provider: Optional provider override. If None, uses configured default
+        agent_mode: The mode of the agent (research, plan, tasks, specify, export)
 
     Returns:
         Tuple of (Configured Pydantic AI agent, Agent dependencies)
@@ -127,13 +126,12 @@ def create_base_agent(
 
         # Create deps with model config and services
         codebase_service = get_codebase_service()
-        artifact_service = get_artifact_service()
         deps = AgentDeps(
             **agent_runtime_options.model_dump(),
             llm_model=model_config,
             codebase_service=codebase_service,
-            artifact_service=artifact_service,
             system_prompt_fn=system_prompt_fn,
+            agent_mode=agent_mode,
         )
 
     except Exception as e:
@@ -181,14 +179,6 @@ def create_base_agent(
     agent.tool(append_file)
     agent.tool(read_file)
 
-    # Register artifact management tools (always available)
-    agent.tool(create_artifact)
-    agent.tool(list_artifacts)
-    agent.tool(list_artifact_templates)
-    agent.tool(read_artifact)
-    agent.tool(read_artifact_section)
-    agent.tool(write_artifact_section)
-
     # Register codebase understanding tools (conditional)
     if load_codebase_understanding_tools:
         agent.tool(query_graph)
@@ -200,10 +190,134 @@ def create_base_agent(
     else:
         logger.debug("🚫🧠 Codebase understanding tools not registered")
 
-    logger.debug("✅ Agent creation complete with artifact and codebase tools")
+    logger.debug("✅ Agent creation complete with codebase tools")
     return agent, deps
 
 
+def extract_markdown_toc(agent_mode: AgentType | None) -> str | None:
+    """Extract table of contents from agent's markdown file.
+
+    Args:
+        agent_mode: The agent mode to extract TOC for
+
+    Returns:
+        Formatted TOC string (up to 2000 chars) or None if not applicable
+    """
+    # Skip for EXPORT mode or no mode
+    if (
+        not agent_mode
+        or agent_mode == AgentType.EXPORT
+        or agent_mode not in AGENT_DIRECTORIES
+    ):
+        return None
+
+    base_path = get_shotgun_base_path()
+    md_file = AGENT_DIRECTORIES[agent_mode]
+    md_path = base_path / md_file
+
+    # Check if the markdown file exists
+    if not md_path.exists():
+        return None
+
+    try:
+        content = md_path.read_text(encoding="utf-8")
+        lines = content.split("\n")
+
+        # Extract headings
+        toc_lines = []
+        for line in lines:
+            stripped = line.strip()
+            if stripped.startswith("#"):
+                # Count the heading level
+                level = 0
+                for char in stripped:
+                    if char == "#":
+                        level += 1
+                    else:
+                        break
+
+                # Get the heading text (remove the # symbols and clean up)
+                heading_text = stripped[level:].strip()
+                if heading_text:
+                    # Add indentation based on level
+                    indent = "  " * (level - 1)
+                    toc_lines.append(f"{indent}{'#' * level} {heading_text}")
+
+        if not toc_lines:
+            return None
+
+        # Join and truncate to 2000 characters
+        toc = "\n".join(toc_lines)
+        if len(toc) > 2000:
+            toc = toc[:1997] + "..."
+
+        return toc
+
+    except Exception as e:
+        logger.debug(f"Failed to extract TOC from {md_file}: {e}")
+        return None
+
+
+def get_agent_existing_files(agent_mode: AgentType | None = None) -> list[str]:
+    """Get list of existing files for the given agent mode.
+
+    Args:
+        agent_mode: The agent mode to check files for. If None, lists all files.
+
+    Returns:
+        List of existing file paths relative to .shotgun directory
+    """
+    base_path = get_shotgun_base_path()
+    existing_files = []
+
+    # If no agent mode, list all files in base path and first level subdirectories
+    if agent_mode is None:
+        # List files in the root .shotgun directory
+        for item in base_path.iterdir():
+            if item.is_file():
+                existing_files.append(item.name)
+            elif item.is_dir():
+                # List files in first-level subdirectories
+                for subitem in item.iterdir():
+                    if subitem.is_file():
+                        relative_path = subitem.relative_to(base_path)
+                        existing_files.append(str(relative_path))
+        return existing_files
+
+    # Handle specific agent modes
+    if agent_mode not in AGENT_DIRECTORIES:
+        return []
+
+    if agent_mode == AgentType.EXPORT:
+        # For export agent, list all files in exports directory
+        exports_dir = base_path / "exports"
+        if exports_dir.exists():
+            for file_path in exports_dir.rglob("*"):
+                if file_path.is_file():
+                    relative_path = file_path.relative_to(base_path)
+                    existing_files.append(str(relative_path))
+    else:
+        # For other agents, check both .md file and directory with same name
+        allowed_file = AGENT_DIRECTORIES[agent_mode]
+
+        # Check for the .md file
+        md_file_path = base_path / allowed_file
+        if md_file_path.exists():
+            existing_files.append(allowed_file)
+
+        # Check for directory with same base name (e.g., research/ for research.md)
+        base_name = allowed_file.replace(".md", "")
+        dir_path = base_path / base_name
+        if dir_path.exists() and dir_path.is_dir():
+            # List all files in the directory
+            for file_path in dir_path.rglob("*"):
+                if file_path.is_file():
+                    relative_path = file_path.relative_to(base_path)
+                    existing_files.append(str(relative_path))
+
+    return existing_files
+
+
 def build_agent_system_prompt(
     agent_type: str,
     ctx: RunContext[AgentDeps],

shotgun_sh-0.1.0.dev23/src/shotgun/agents/conversation_history.py

@@ -0,0 +1,56 @@
+"""Models and utilities for persisting TUI conversation history."""
+
+from datetime import datetime
+from typing import Any
+
+from pydantic import BaseModel, ConfigDict, Field
+from pydantic_ai.messages import (
+    ModelMessage,
+    ModelMessagesTypeAdapter,
+)
+from pydantic_core import to_jsonable_python
+
+
+class ConversationState(BaseModel):
+    """Represents the complete state of a conversation in memory."""
+
+    agent_messages: list[ModelMessage]
+    agent_type: str  # Will store AgentType.value
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+
+class ConversationHistory(BaseModel):
+    """Persistent conversation history for TUI sessions."""
+
+    version: int = 1
+    agent_history: list[dict[str, Any]] = Field(
+        default_factory=list
+    )  # Will store serialized ModelMessage objects
+    last_agent_model: str = "research"
+    updated_at: datetime = Field(default_factory=datetime.now)
+
+    model_config = ConfigDict(arbitrary_types_allowed=True)
+
+    def set_agent_messages(self, messages: list[ModelMessage]) -> None:
+        """Set agent_history from a list of ModelMessage objects.
+
+        Args:
+            messages: List of ModelMessage objects to serialize and store
+        """
+        # Serialize ModelMessage list to JSON-serializable format
+        self.agent_history = to_jsonable_python(
+            messages, fallback=lambda x: str(x), exclude_none=True
+        )
+
+    def get_agent_messages(self) -> list[ModelMessage]:
+        """Get agent_history as a list of ModelMessage objects.
+
+        Returns:
+            List of deserialized ModelMessage objects
+        """
+        if not self.agent_history:
+            return []
+
+        # Deserialize from JSON format back to ModelMessage objects
+        return ModelMessagesTypeAdapter.validate_python(self.agent_history)