shotgun-sh 0.1.9__py3-none-any.whl → 0.2.11__py3-none-any.whl

shotgun/tui/screens/chat_screen/history.py

@@ -1,222 +0,0 @@
-import json
-from collections.abc import Sequence
-
-from pydantic_ai.messages import (
-    BuiltinToolCallPart,
-    BuiltinToolReturnPart,
-    ModelMessage,
-    ModelRequest,
-    ModelRequestPart,
-    ModelResponse,
-    TextPart,
-    ThinkingPart,
-    ToolCallPart,
-    ToolReturnPart,
-    UserPromptPart,
-)
-from textual.app import ComposeResult
-from textual.reactive import reactive
-from textual.widget import Widget
-from textual.widgets import Markdown
-
-from shotgun.tui.components.vertical_tail import VerticalTail
-from shotgun.tui.screens.chat_screen.hint_message import HintMessage, HintMessageWidget
-
-
-class PartialResponseWidget(Widget):  # TODO: doesn't work lol
-    DEFAULT_CSS = """
-        PartialResponseWidget {
-            height: auto;
-        }
-        Markdown, AgentResponseWidget, UserQuestionWidget {
-            height: auto;
-        }
-    """
-
-    item: reactive[ModelMessage | None] = reactive(None, recompose=True)
-
-    def __init__(self, item: ModelMessage | None) -> None:
-        super().__init__()
-        self.item = item
-
-    def compose(self) -> ComposeResult:
-        yield Markdown(markdown="**partial response**")
-        if self.item is None:
-            pass
-        elif self.item.kind == "response":
-            yield AgentResponseWidget(self.item)
-        elif self.item.kind == "request":
-            yield UserQuestionWidget(self.item)
-
-    def watch_item(self, item: ModelMessage | None) -> None:
-        if item is None:
-            self.display = False
-        else:
-            self.display = True
-
-
-class ChatHistory(Widget):
-    DEFAULT_CSS = """
-        VerticalTail {
-            align: left bottom;
-
-        }
-        VerticalTail > * {
-            height: auto;
-        }
-
-        Horizontal {
-            height: auto;
-            background: $secondary-muted;
-        }
-
-        Markdown {
-            height: auto;
-        }
-    """
-    partial_response: reactive[ModelMessage | None] = reactive(None)
-
-    def __init__(self) -> None:
-        super().__init__()
-        self.items: Sequence[ModelMessage | HintMessage] = []
-        self.vertical_tail: VerticalTail | None = None
-        self.partial_response = None
-
-    def compose(self) -> ComposeResult:
-        self.vertical_tail = VerticalTail()
-
-        with self.vertical_tail:
-            for item in self.items:
-                if isinstance(item, ModelRequest):
-                    yield UserQuestionWidget(item)
-                elif isinstance(item, HintMessage):
-                    yield HintMessageWidget(item)
-                elif isinstance(item, ModelResponse):
-                    yield AgentResponseWidget(item)
-            yield PartialResponseWidget(self.partial_response).data_bind(
-                item=ChatHistory.partial_response
-            )
-        self.call_later(self.autoscroll)
-
-    def update_messages(self, messages: list[ModelMessage | HintMessage]) -> None:
-        """Update the displayed messages without recomposing."""
-        if not self.vertical_tail:
-            return
-
-        self.items = messages
-        self.refresh(recompose=True)
-        self.call_later(self.autoscroll)
-
-    def autoscroll(self) -> None:
-        if self.vertical_tail:
-            self.vertical_tail.scroll_end(animate=False, immediate=False, force=True)
-
-
-class UserQuestionWidget(Widget):
-    def __init__(self, item: ModelRequest | None) -> None:
-        super().__init__()
-        self.item = item
-
-    def compose(self) -> ComposeResult:
-        self.display = self.item is not None
-        if self.item is None:
-            yield Markdown(markdown="")
-        else:
-            prompt = self.format_prompt_parts(self.item.parts)
-            yield Markdown(markdown=prompt)
-
-    def format_prompt_parts(self, parts: Sequence[ModelRequestPart]) -> str:
-        acc = ""
-        for part in parts:
-            if isinstance(part, UserPromptPart):
-                acc += (
-                    f"**>** {part.content if isinstance(part.content, str) else ''}\n\n"
-                )
-            elif isinstance(part, ToolReturnPart):
-                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
-                    pass
-            elif isinstance(part, UserPromptPart):
-                acc += f"**>** {part.content}\n\n"
-        return acc
-
-
-class AgentResponseWidget(Widget):
-    def __init__(self, item: ModelResponse | None) -> None:
-        super().__init__()
-        self.item = item
-
-    def compose(self) -> ComposeResult:
-        self.display = self.item is not None
-        if self.item is None:
-            yield Markdown(markdown="")
-        else:
-            yield Markdown(markdown=self.compute_output())
-
-    def compute_output(self) -> str:
-        acc = ""
-        if self.item is None:
-            return ""
-        for idx, part in enumerate(self.item.parts):
-            if isinstance(part, TextPart):
-                acc += f"**⏺** {part.content}\n\n"
-            elif isinstance(part, ToolCallPart):
-                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"
-            elif isinstance(part, BuiltinToolReturnPart):
-                acc += f"builtin tool ({part.tool_name}) return: {part.content}\n\n"
-            elif isinstance(part, ThinkingPart):
-                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":
-            return self._format_ask_user_part(part)
-        # write_file
-        if part.tool_name == "write_file" or part.tool_name == "append_file":
-            if isinstance(part.args, dict) and "filename" in part.args:
-                return f"{part.tool_name}({part.args['filename']})"
-            else:
-                return f"{part.tool_name}()"
-        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']})"
-            else:
-                return f"{part.tool_name}()"
-        if part.tool_name == "create_artifact":
-            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}({part.args})"
-
-    def _format_ask_user_part(
-        self,
-        part: ToolCallPart,
-    ) -> str:
-        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 "❓ "

shotgun_sh-0.1.9.dist-info/METADATA

@@ -1,466 +0,0 @@
-Metadata-Version: 2.4
-Name: shotgun-sh
-Version: 0.1.9
-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
-Project-URL: Issues, https://github.com/shotgun-sh/shotgun-alpha/issues
-Project-URL: Discord, https://discord.gg/5RmY6J2N7s
-Author-email: "Proofs.io" <hello@proofs.io>
-License: MIT
-License-File: LICENSE
-Keywords: agent,ai,cli,llm,planning,productivity,pydantic-ai,research,task-management
-Classifier: Development Status :: 3 - Alpha
-Classifier: Environment :: Console
-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.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Classifier: Topic :: Utilities
-Requires-Python: >=3.11
-Requires-Dist: anthropic>=0.39.0
-Requires-Dist: google-generativeai>=0.8.5
-Requires-Dist: httpx>=0.27.0
-Requires-Dist: jinja2>=3.1.0
-Requires-Dist: kuzu>=0.7.0
-Requires-Dist: logfire[pydantic-ai]>=2.0.0
-Requires-Dist: openai>=1.0.0
-Requires-Dist: packaging>=23.0
-Requires-Dist: posthog>=3.0.0
-Requires-Dist: pydantic-ai>=0.0.14
-Requires-Dist: rich>=13.0.0
-Requires-Dist: sentry-sdk[pure-eval]>=2.0.0
-Requires-Dist: textual-dev>=1.7.0
-Requires-Dist: textual>=6.1.0
-Requires-Dist: tiktoken>=0.7.0
-Requires-Dist: tree-sitter-go>=0.23.0
-Requires-Dist: tree-sitter-javascript>=0.23.0
-Requires-Dist: tree-sitter-python>=0.23.0
-Requires-Dist: tree-sitter-rust>=0.23.0
-Requires-Dist: tree-sitter-typescript>=0.23.0
-Requires-Dist: tree-sitter>=0.21.0
-Requires-Dist: typer>=0.12.0
-Requires-Dist: watchdog>=4.0.0
-Provides-Extra: dev
-Requires-Dist: commitizen>=3.13.0; extra == 'dev'
-Requires-Dist: lefthook>=1.12.0; extra == 'dev'
-Requires-Dist: mypy>=1.11.0; extra == 'dev'
-Requires-Dist: ruff>=0.6.0; extra == 'dev'
-Description-Content-Type: text/markdown
-
-# Shotgun
-
-A Python CLI tool for research, planning, and task management powered by AI agents.
-
-## Features
-
-- **Research**: Perform research with agentic loops
-- **Planning**: Generate structured plans for achieving goals
-- **Tasks**: Generate prioritized task lists with agentic approaches
-
-## Installation
-
-### From PyPI (Recommended)
-
-```bash
-pip install shotgun-sh
-```
-
-### From Source
-
-```bash
-git clone https://github.com/shotgun-sh/shotgun.git
-cd shotgun
-uv sync --all-extras
-```
-
-After installation from source, you can use either method:
-
-**Method 1: Direct command (after uv sync)**
-```bash
-shotgun --help
-```
-
-**Method 2: Via uv run**
-```bash
-uv run shotgun --help
-```
-
-If installed from PyPI, simply use:
-```bash
-shotgun --help
-```
-
-### Virtual Environment Setup (Optional)
-
-If you prefer using a local virtual environment:
-
-```bash
-uv venv
-source .venv/bin/activate  # On Windows: .venv\Scripts\activate
-uv sync --all-extras
-shotgun --help
-```
-
-## Usage
-
-### Using Direct Commands (after uv sync)
-
-```bash
-# Research a topic
-shotgun research "What is quantum computing?"
-
-# Generate a plan
-shotgun plan "Build a web application"
-shotgun plan "build me a house"
-
-# Generate tasks for a project
-shotgun tasks "Create a machine learning model"
-```
-
-### Using uv run
-
-```bash
-# Research a topic
-uv run shotgun research "What is quantum computing?"
-
-# Generate a plan
-uv run shotgun plan "Build a web application"
-
-# Generate tasks for a project
-uv run shotgun tasks "Create a machine learning model"
-```
-
-## Auto-Updates
-
-Shotgun automatically checks for updates to keep you on the latest version.
-
-### How it works
-
-- Checks for updates on startup (runs in background, non-blocking)
-- Caches results for 24 hours to minimize API calls
-- Shows notification after command execution if an update is available
-- Never auto-updates development versions
-
-### Update Commands
-
-```bash
-# Check for available updates
-shotgun update --check
-
-# Install available updates
-shotgun update
-
-# Force update (even for dev versions with confirmation)
-shotgun update --force
-```
-
-### Disable Update Checks
-
-```bash
-# Disable for a single command
-shotgun --no-update-check research "topic"
-```
-
-### Installation Methods
-
-The update command automatically detects and uses the appropriate method:
-- **pipx**: `pipx upgrade shotgun-sh`
-- **pip**: `pip install --upgrade shotgun-sh`
-- **venv**: Updates within the virtual environment
-
-## Development Setup
-
-### Requirements
-
-- **Python 3.11+** (3.13 recommended)
-- **uv** - Fast Python package installer and resolver
-- **actionlint** (optional) - For GitHub Actions workflow validation
-
-### Quick Start
-
-1. **Clone and setup**:
-   ```bash
-   git clone https://github.com/shotgun-sh/shotgun.git
-   cd shotgun
-   ```
-
-2. **Install uv** (if not already installed):
-   ```bash
-   # macOS/Linux
-   curl -LsSf https://astral.sh/uv/install.sh | sh
-   
-   # Or via brew
-   brew install uv
-   ```
-
-3. **Install dependencies**:
-   ```bash
-   uv sync --all-extras
-   ```
-
-4. **Install git hooks**:
-   ```bash
-   uv run lefthook install
-   ```
-
-5. **Verify setup**:
-   ```bash
-   uv run shotgun --version
-   ```
-
-### Development Commands
-
-```bash
-# Run the CLI
-uv run shotgun --help
-
-# Run the TUI
-uv run tui
-
-# Run tests
-uv run pytest
-
-# Run tests with coverage
-uv run pytest --cov=src --cov-report=term-missing --cov-report=html
-
-# Run linting
-uv run ruff check .
-
-# Run formatting
-uv run ruff format .
-
-# Run type checking
-uv run mypy src/
-
-# Run all pre-commit hooks manually
-uv run lefthook run pre-commit
-```
-
-### Code Coverage
-
-To analyze test coverage and identify areas that need testing:
-
-```bash
-# Run tests with coverage analysis
-uv run pytest --cov=src --cov-report=term-missing --cov-report=html
-```
-
-This will:
-- Display coverage summary in the terminal
-- Generate a detailed HTML coverage report
-
-**Viewing the coverage report:**
-Open `htmlcov/index.html` in your browser to see:
-- Overall coverage percentage
-- File-by-file coverage breakdown  
-- Line-by-line coverage highlighting
-- Missing coverage areas
-
-The coverage configuration is in `pyproject.toml` and will automatically run when you use `uv run pytest`.
-
-### Git Hooks (Lefthook)
-
-This project uses [lefthook](https://github.com/evilmartians/lefthook) for git hooks. The hooks automatically run:
-
-- **ruff** - Python linting with auto-fix
-- **ruff-format** - Code formatting  
-- **mypy** - Type checking
-- **commitizen** - Commit message validation
-- **actionlint** - GitHub Actions workflow validation (if installed)
-
-#### Installing actionlint (recommended)
-
-```bash
-# macOS
-brew install actionlint
-
-# Linux/macOS (direct download)
-curl -sSfL https://raw.githubusercontent.com/rhysd/actionlint/main/scripts/download-actionlint.bash | bash
-
-# Go install
-go install github.com/rhysd/actionlint/cmd/actionlint@latest
-```
-
-
-### Python Version Management
-
-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.11
-```
-
-If using **uv** (recommended):
-```bash
-uv python install 3.11
-uv sync --python 3.11
-```
-
-### Commit Message Convention
-
-This project enforces **Conventional Commits** specification. All commit messages must follow this format:
-
-```
-<type>[optional scope]: <description>
-```
-
-**Required commit types:**
-- `feat`: New feature
-- `fix`: Bug fix
-- `docs`: Documentation changes
-- `style`: Code formatting changes
-- `refactor`: Code restructuring without feature changes
-- `perf`: Performance improvements
-- `test`: Adding or updating tests
-- `build`: Build system changes
-- `ci`: CI/CD changes
-- `chore`: Maintenance tasks
-- `revert`: Reverting previous commits
-
-**Examples:**
-```bash
-feat: add user authentication system
-fix: resolve memory leak in data processing
-docs: update API documentation
-refactor: simplify user validation logic
-```
-
-**For interactive commit creation:**
-```bash
-uv run cz commit
-```
-
-### Contributing
-
-1. Fork the repository
-2. Create a feature branch: `git checkout -b feat/feature-name`
-3. Make your changes
-4. Run the pre-commit hooks: `uv run lefthook run pre-commit`
-5. Commit with conventional format: `git commit -m "feat: add new feature"`
-6. Push to your fork: `git push origin feat/feature-name`
-7. Create a Pull Request with conventional title format
-
-### CI/CD
-
-GitHub Actions automatically:
-- Runs on pull requests and pushes to main
-- Tests with Python 3.11
-- Validates code with ruff, ruff-format, and mypy
-- Ensures all checks pass before merge
-
-## Observability & Telemetry
-
-Shotgun includes built-in observability with Sentry for error tracking and Logfire for logging and tracing. Both services track users anonymously using a UUID generated on first run.
-
-### Anonymous User Tracking
-
-Each user gets a unique anonymous ID stored in their config:
-```bash
-# Get your anonymous user ID
-shotgun config get-user-id
-```
-
-This ID is automatically included in:
-- **Sentry**: Error reports and exceptions
-- **Logfire**: All logs, traces, and spans
-
-### Logfire Queries
-
-Logfire uses SQL for querying logs. Here are helpful queries for debugging and analysis:
-
-#### Find all logs for a specific user
-```sql
-SELECT * FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-ORDER BY timestamp DESC;
-```
-
-#### Track user actions
-```sql
-SELECT
-  timestamp,
-  span_name,
-  message,
-  attributes
-FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-  AND span_name LIKE '%research%'
-ORDER BY timestamp DESC;
-```
-
-#### Find slow operations for a user
-```sql
-SELECT
-  span_name,
-  duration_ms,
-  attributes
-FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-  AND duration_ms > 1000
-ORDER BY duration_ms DESC;
-```
-
-#### Find errors for a user
-```sql
-SELECT * FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-  AND level = 'error'
-ORDER BY timestamp DESC;
-```
-
-#### Analyze user's AI provider usage
-```sql
-SELECT
-  attributes->>'provider' as provider,
-  COUNT(*) as usage_count,
-  AVG(duration_ms) as avg_duration
-FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-  AND attributes->>'provider' IS NOT NULL
-GROUP BY provider;
-```
-
-#### Track feature usage by user
-```sql
-SELECT
-  span_name,
-  COUNT(*) as usage_count
-FROM records
-WHERE attributes->>'user_id' = 'your-user-id-here'
-  AND span_name IN ('research', 'plan', 'tasks')
-GROUP BY span_name
-ORDER BY usage_count DESC;
-```
-
-### Setting Up Observability (Optional)
-
-For local development with Logfire:
-```bash
-# Set environment variables
-export LOGFIRE_ENABLED=true
-export LOGFIRE_TOKEN=your-logfire-token
-
-# Run shotgun - will now send logs to Logfire
-shotgun research "topic"
-```
-
-For Sentry (automatically configured in production builds):
-```bash
-# Set for local development
-export SENTRY_DSN=your-sentry-dsn
-```
-
-### Privacy
-
-- **No PII collected**: Only anonymous UUIDs are used for identification
-- **Opt-in for development**: Telemetry requires explicit environment variables
-- **Automatic in production**: Production builds include telemetry for error tracking
-
-## Support
-
-Join our discord https://discord.gg/5RmY6J2N7s