tunacode-cli 0.0.22__tar.gz → 0.0.24__tar.gz

tunacode_cli-0.0.24/CLAUDE.md

@@ -0,0 +1,120 @@
+# CLAUDE.md
+
+This file provides guidance to Claude Code (claude.ai/code) when working with code in this repository.
+
+## Commands
+
+### Development Commands
+```bash
+# Install development environment
+pip install -e ".[dev]"
+
+# Run linting (black, isort, flake8)
+make lint
+
+# Run tests
+make test
+pytest tests/                 # Run all tests
+pytest tests/test_import.py  # Run single test file
+pytest -k "test_name"        # Run specific test
+
+# Run tests with coverage
+make coverage
+
+# Build distribution packages
+make build
+
+# Clean build artifacts
+make clean
+```
+
+### Version Management
+When updating versions, modify both:
+- `pyproject.toml`: version field
+- `src/tunacode/constants.py`: VERSION constant
+
+## Architecture
+
+TunaCode is a CLI tool that provides an AI-powered coding assistant using pydantic-ai. Key architectural decisions:
+
+### Agent System
+- Uses `pydantic-ai` for LLM agent implementation
+- Central agent in `src/tunacode/core/agents/main.py` with retryable tools
+- Supports multiple LLM providers (Anthropic, OpenAI, Google, OpenRouter) through unified interface
+- Model format: `provider:model-name` (e.g., `openai:gpt-4`, `anthropic:claude-3-opus`)
+
+### Tool System
+Four internal tools with confirmation UI:
+1. `read_file` - Read file contents
+2. `write_file` - Create new files (fails if exists)
+3. `update_file` - Update existing files with target/patch pattern
+4. `run_command` - Execute shell commands
+
+Tools extend `BaseTool` or `FileBasedTool` base classes. External tools supported via MCP (Model Context Protocol).
+
+### State Management
+- `StateManager` (core/state.py) maintains all session state
+- Includes user config, agent instances, message history, costs, permissions
+- Single source of truth passed throughout the application
+
+### Command System
+- Command registry pattern in `cli/commands.py`
+- Commands implement `BaseCommand` with `matches()` and `execute()` methods
+- Registered via `@CommandRegistry.register` decorator
+- Process flow: REPL → CommandRegistry → Command → Action
+
+### Setup Coordinator
+Modular setup with validation steps:
+1. Environment detection
+2. Model validation
+3. Configuration setup
+4. Git safety checks
+Each step implements `BaseSetupStep` interface.
+
+### UI Components
+- REPL uses `prompt_toolkit` for multiline input with syntax highlighting
+- Output formatting via `rich` library
+- Tool confirmations show diffs for file operations
+- Spinner during agent processing
+
+## Configuration
+
+### User Configuration
+Location: `~/.config/tunacode.json`
+```json
+{
+    "default_model": "provider:model-name",
+    "env": {
+        "ANTHROPIC_API_KEY": "...",
+        "OPENAI_API_KEY": "..."
+    }
+}
+```
+
+### Project Guide
+Location: `TUNACODE.md` in project root
+- Project-specific context for the AI assistant
+- Loaded automatically when present
+- Can include codebase conventions, architecture notes
+
+## Key Design Patterns
+
+### Error Handling
+- Custom exceptions in `exceptions.py`
+- `ModelRetry` from pydantic-ai for retryable errors
+- Graceful degradation for missing features
+
+### Permissions
+- File operation permissions tracked per session
+- "Yolo mode" to skip confirmations: `/yolo`
+- Permissions stored in StateManager
+
+### Async Architecture
+- All agent operations are async
+- Tool executions use async/await
+- REPL handles async with prompt_toolkit integration
+
+### Safety Notes
+- No automatic git commits (removed for safety)
+- File operations require explicit confirmation
+- Encourages git branches for experiments: `/branch <name>`

tunacode_cli-0.0.24/MANIFEST.in

@@ -0,0 +1,19 @@
+# Include all prompt files
+include src/tunacode/prompts/*.txt
+include src/tunacode/prompts/*.md
+
+# Include other important files
+include README.md
+include LICENSE
+include pyproject.toml
+include CLAUDE.md
+include TUNACODE.md
+
+# Include test files (optional, but good for development)
+recursive-include tests *.py
+
+# Exclude compiled files
+global-exclude *.pyc
+global-exclude __pycache__
+global-exclude *.so
+global-exclude *.egg-info

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.22
+Version: 0.0.24
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License-Expression: MIT
@@ -30,6 +30,7 @@ Requires-Dist: black; extra == "dev"
 Requires-Dist: flake8; extra == "dev"
 Requires-Dist: isort; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: textual-dev; extra == "dev"
 Dynamic: license-file
@@ -131,6 +132,24 @@ Dynamic: license-file
 pip install tunacode-cli
 ```
 
+#### Development Installation
+
+```bash
+# Clone the repository
+git clone https://github.com/larock22/tunacode.git
+cd tunacode
+
+# Run the setup script
+./scripts/setup_dev_env.sh
+
+# Or manually:
+python3 -m venv venv
+source venv/bin/activate
+pip install -e ".[dev]"
+```
+
+See [CONTRIBUTING.md](CONTRIBUTING.md) for detailed development setup.
+
 #### One-line Install (Linux/macOS)
 
 ```bash

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/README.md

@@ -95,6 +95,24 @@
 pip install tunacode-cli
 ```
 
+#### Development Installation
+
+```bash
+# Clone the repository
+git clone https://github.com/larock22/tunacode.git
+cd tunacode
+
+# Run the setup script
+./scripts/setup_dev_env.sh
+
+# Or manually:
+python3 -m venv venv
+source venv/bin/activate
+pip install -e ".[dev]"
+```
+
+See [CONTRIBUTING.md](CONTRIBUTING.md) for detailed development setup.
+
 #### One-line Install (Linux/macOS)
 
 ```bash

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tunacode-cli"
-version = "0.0.22"
+version = "0.0.24"
 description = "Your agentic CLI developer."
 keywords = ["cli", "agent", "development", "automation"]
 readme = "README.md"
@@ -42,6 +42,7 @@ dev = [
     "flake8",
     "isort",
     "pytest",
+    "pytest-asyncio",
     "pytest-cov",
     "textual-dev",
 ]

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/setup.py

@@ -5,6 +5,6 @@ setup(
     packages=find_namespace_packages(where="src"),
     include_package_data=True,
     package_data={
-        "tunacode": ["prompts/*.txt"],
+        "tunacode": ["prompts/*.txt", "prompts/*.md"],
     },
 )

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/cli/commands.py

@@ -192,7 +192,7 @@ class ArchitectCommand(SimpleCommand):
                 await ui.error("Usage: /architect [on|off]")
                 return
         else:
-            state.architect_mode = not getattr(state, 'architect_mode', False)
+            state.architect_mode = not getattr(state, "architect_mode", False)
         status = "ON" if state.architect_mode else "OFF"
         if state.architect_mode:
             await ui.success(f"Architect mode {status} - Requests will be planned before execution")

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/cli/main.py

@@ -14,6 +14,7 @@ from tunacode.core.state import StateManager
 from tunacode.setup import setup
 from tunacode.ui import console as ui
 from tunacode.utils.system import check_for_updates
+from tunacode.exceptions import UserAbortError
 
 app_settings = ApplicationSettings()
 app = typer.Typer(help="🐟 TunaCode - Your AI-powered development assistant")
@@ -49,8 +50,12 @@ def main(
         try:
             await setup(run_setup, state_manager, cli_config)
             await repl(state_manager)
+        except (KeyboardInterrupt, UserAbortError):
+            update_task.cancel()
+            return
         except Exception as e:
             from tunacode.exceptions import ConfigurationError
+
             if isinstance(e, ConfigurationError):
                 # ConfigurationError already printed helpful message, just exit cleanly
                 update_task.cancel()  # Cancel the update check

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/cli/repl.py

@@ -172,7 +172,7 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
             return _tool_handler(part, node, state_manager)
 
         # Check if architect mode is enabled
-        if getattr(state_manager.session, 'architect_mode', False):
+        if getattr(state_manager.session, "architect_mode", False):
             # Expand @file references before sending to the orchestrator
             try:
                 from tunacode.utils.text_utils import expand_file_refs
@@ -184,14 +184,18 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
             # Use orchestrator for planning and execution
             orchestrator = OrchestratorAgent(state_manager)
             results = await orchestrator.run(text, state_manager.session.current_model)
-            
+
             if output:
                 # Process results from all sub-agents
                 for res in results:
                     # Check if result exists and has output
-                    if hasattr(res, "result") and res.result is not None and hasattr(res.result, "output"):
+                    if (
+                        hasattr(res, "result")
+                        and res.result is not None
+                        and hasattr(res.result, "output")
+                    ):
                         await ui.agent(res.result.output)
-                
+
                 if not results:
                     # Fallback: show that the request was processed
                     await ui.muted("Request completed")
@@ -204,7 +208,7 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
             except ValueError as e:
                 await ui.error(str(e))
                 return
-            
+
             # Use normal agent processing
             res = await agent.process_request(
                 state_manager.session.current_model,
@@ -219,7 +223,11 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
                         if isinstance(msg, dict) and "thought" in msg:
                             await ui.muted(f"THOUGHT: {msg['thought']}")
                 # Check if result exists and has output
-                if hasattr(res, "result") and res.result is not None and hasattr(res.result, "output"):
+                if (
+                    hasattr(res, "result")
+                    and res.result is not None
+                    and hasattr(res.result, "output")
+                ):
                     await ui.agent(res.result.output)
                 else:
                     # Fallback: show that the request was processed
@@ -274,6 +282,7 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
 
 async def repl(state_manager: StateManager):
     action = None
+    ctrl_c_pressed = False
 
     # Professional startup information
     await ui.muted(f"• Model: {state_manager.session.current_model}")
@@ -286,12 +295,18 @@ async def repl(state_manager: StateManager):
         while True:
             try:
                 line = await ui.multiline_input(state_manager, _command_registry)
-            except (EOFError, KeyboardInterrupt):
-                break
+            except UserAbortError:
+                if ctrl_c_pressed:
+                    break
+                ctrl_c_pressed = True
+                await ui.warning("Hit Ctrl+C again to exit")
+                continue
 
             if not line:
                 continue
 
+            ctrl_c_pressed = False
+
             if line.lower() in ["exit", "quit"]:
                 break
 

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/cli/textual_app.py

@@ -11,18 +11,13 @@ Provides a rich, interactive terminal user interface with:
 import asyncio
 from typing import Optional
 
-from rich.markdown import Markdown
-from textual import on, work
+from textual import on
 from textual.app import App, ComposeResult
 from textual.binding import Binding
 from textual.containers import Container, Horizontal, Vertical, VerticalScroll
 from textual.message import Message
-from textual.reactive import reactive
-from textual.widgets import Button, Footer, Header, Input, Label
-from textual.widgets import Markdown as MarkdownWidget
-from textual.widgets import Static, TabbedContent, TabPane, TextArea
+from textual.widgets import Button, Footer, Header, Static, TextArea
 
-from tunacode.configuration.settings import ApplicationSettings
 from tunacode.core.state import StateManager
 from tunacode.setup import setup
 from tunacode.utils.system import check_for_updates
@@ -130,78 +125,78 @@ class TunaCodeApp(App):
         border: thick $primary;
         padding: 1;
     }
-    
+
     .sidebar-title {
         text-align: center;
         color: $primary;
         margin-bottom: 1;
     }
-    
+
     .section-title {
         color: $accent;
         margin: 1 0;
     }
-    
+
     .command-item {
         color: $text-muted;
         margin-left: 1;
     }
-    
+
     .status-ready {
         color: $success;
     }
-    
+
     .status-busy {
         color: $warning;
     }
-    
+
     .status-error {
         color: $error;
     }
-    
+
     ChatHistory {
         border: thick $primary;
         padding: 1;
         height: 1fr;
     }
-    
+
     .user-message {
         background: $surface;
         border-left: thick $primary;
         padding: 1;
         margin: 1 0;
     }
-    
+
     .agent-message {
         background: $surface;
         border-left: thick $success;
         padding: 1;
         margin: 1 0;
     }
-    
+
     .system-message {
         background: $surface;
         border-left: thick $warning;
         padding: 1;
         margin: 1 0;
     }
-    
+
     .tool-message {
         background: $surface;
         border-left: thick $accent;
         padding: 1;
         margin: 1 0;
     }
-    
+
     InputArea {
         height: 5;
         padding: 1;
     }
-    
+
     #message-input {
         height: 3;
     }
-    
+
     #send-button {
         width: 10;
         margin-left: 1;
@@ -250,7 +245,8 @@ class TunaCodeApp(App):
         )
         self.chat_history.add_message(
             "System",
-            "⚠️ IMPORTANT: Always use git branches before making major changes\nType '/help' for available commands",
+            "⚠️ IMPORTANT: Always use git branches before making major changes\n"
+            "Type '/help' for available commands",
             "system",
         )
 
@@ -280,7 +276,7 @@ class TunaCodeApp(App):
         """Handle slash commands."""
         if command == "/help":
             help_text = """Available Commands:
-            
+
 /help - Show this help message
 /clear - Clear chat history
 /model - Show current model info
@@ -371,8 +367,6 @@ def main():
 
     # Initialize state manager
     state_manager = StateManager()
-    app_settings = ApplicationSettings()
-
     # Show banner
     print("🐟 TunaCode - Modern AI Development Assistant")
     print("=" * 50)

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/cli/textual_bridge.py

@@ -5,9 +5,8 @@ This module adapts the existing REPL and agent processing logic to work
 with the new Textual-based interface while maintaining compatibility.
 """
 
-import asyncio
 from asyncio.exceptions import CancelledError
-from typing import Callable, Optional
+from typing import Callable
 
 from pydantic_ai.exceptions import UnexpectedModelBehavior
 
@@ -129,7 +128,7 @@ class TextualAgentBridge:
             # Check if confirmation is needed
             if tool_handler.should_confirm(part.tool_name):
                 # Create confirmation request
-                request = tool_handler.create_confirmation_request(part.tool_name, args)
+                tool_handler.create_confirmation_request(part.tool_name, args)
 
                 # For now, show a simple confirmation in the UI
                 # In a full implementation, this would show a proper modal dialog

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/constants.py

@@ -7,7 +7,7 @@ Centralizes all magic strings, UI text, error messages, and application constant
 
 # Application info
 APP_NAME = "TunaCode"
-APP_VERSION = "0.0.22"
+APP_VERSION = "0.0.24"
 
 # File patterns
 GUIDE_FILE_PATTERN = "{name}.md"

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/context.py

@@ -1,5 +1,3 @@
-import json
-import os
 import subprocess
 from pathlib import Path
 from typing import Dict, List

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/core/agents/main.py

@@ -7,11 +7,22 @@ Handles agent creation, configuration, and request processing.
 import json
 import re
 from datetime import datetime, timezone
+from pathlib import Path
 from typing import Optional
 
-# Lazy import for Agent and Tool
+from tunacode.core.state import StateManager
+from tunacode.services.mcp import get_mcp_servers
+from tunacode.tools.bash import bash
+from tunacode.tools.grep import grep
+from tunacode.tools.read_file import read_file
+from tunacode.tools.run_command import run_command
+from tunacode.tools.update_file import update_file
+from tunacode.tools.write_file import write_file
+from tunacode.types import (AgentRun, ErrorMessage, ModelName, PydanticAgent, ToolCallback,
+                            ToolCallId, ToolName)
 
 
+# Lazy import for Agent and Tool
 def get_agent_tool():
     import importlib
 
@@ -26,18 +37,6 @@ def get_model_messages():
     return messages.ModelRequest, messages.ToolReturnPart
 
 
-from tunacode.core.state import StateManager
-from tunacode.services.mcp import get_mcp_servers
-from tunacode.tools.bash import bash
-from tunacode.tools.grep import grep
-from tunacode.tools.read_file import read_file
-from tunacode.tools.run_command import run_command
-from tunacode.tools.update_file import update_file
-from tunacode.tools.write_file import write_file
-from tunacode.types import (AgentRun, ErrorMessage, ModelName, PydanticAgent, ToolCallback,
-                            ToolCallId, ToolName)
-
-
 async def _process_node(node, tool_callback: Optional[ToolCallback], state_manager: StateManager):
     if hasattr(node, "request"):
         state_manager.session.messages.append(node.request)
@@ -125,21 +124,25 @@ async def _process_node(node, tool_callback: Optional[ToolCallback], state_manag
 
 def get_or_create_agent(model: ModelName, state_manager: StateManager) -> PydanticAgent:
     if model not in state_manager.session.agents:
-        max_retries = state_manager.session.user_config["settings"]["max_retries"]
+        max_retries = state_manager.session.user_config.get("settings", {}).get("max_retries", 3)
 
         # Lazy import Agent and Tool
         Agent, Tool = get_agent_tool()
 
         # Load system prompt
-        import os
-        from pathlib import Path
-
         prompt_path = Path(__file__).parent.parent.parent / "prompts" / "system.md"
         try:
             with open(prompt_path, "r", encoding="utf-8") as f:
                 system_prompt = f.read().strip()
         except FileNotFoundError:
-            system_prompt = None
+            # Fallback to system.txt if system.md not found
+            prompt_path = Path(__file__).parent.parent.parent / "prompts" / "system.txt"
+            try:
+                with open(prompt_path, "r", encoding="utf-8") as f:
+                    system_prompt = f.read().strip()
+            except FileNotFoundError:
+                # Use a default system prompt if neither file exists
+                system_prompt = "You are a helpful AI assistant for software development tasks."
 
         state_manager.session.agents[model] = Agent(
             model=model,

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/core/llm/planner.py

@@ -1,4 +1,3 @@
-import json
 from typing import List
 
 from ...types import ModelName

{tunacode_cli-0.0.22 → tunacode_cli-0.0.24}/src/tunacode/core/setup/config_setup.py

@@ -99,11 +99,15 @@ class ConfigSetup(BaseSetup):
                     "  [green]tunacode --model 'anthropic:claude-3-opus' --key 'your-key'[/green]"
                 )
                 console.print(
-                    "  [green]tunacode --model 'openrouter:anthropic/claude-3.5-sonnet' --key 'your-key' --baseurl 'https://openrouter.ai/api/v1'[/green]"
+                    "  [green]tunacode --model 'openrouter:anthropic/claude-3.5-sonnet' "
+                    "--key 'your-key' --baseurl 'https://openrouter.ai/api/v1'[/green]"
                 )
                 console.print("\n[yellow]Run 'tunacode --help' for more options[/yellow]\n")
                 from tunacode.exceptions import ConfigurationError
-                raise ConfigurationError("No configuration found. Please use CLI flags to configure.")
+
+                raise ConfigurationError(
+                    "No configuration found. Please use CLI flags to configure."
+                )
 
         if not self.state_manager.session.user_config.get("default_model"):
             raise ConfigurationError(