tunacode-cli 0.0.51__py3-none-any.whl → 0.0.53__py3-none-any.whl

tunacode/tools/grep_components/search_result.py

@@ -0,0 +1,35 @@
+"""
+Search result and configuration data structures for the grep tool.
+"""
+
+from dataclasses import dataclass
+from typing import List, Optional
+
+
+@dataclass
+class SearchResult:
+    """Represents a single search match with context."""
+
+    file_path: str
+    line_number: int
+    line_content: str
+    match_start: int
+    match_end: int
+    context_before: List[str]
+    context_after: List[str]
+    relevance_score: float = 0.0
+
+
+@dataclass
+class SearchConfig:
+    """Configuration for search operations."""
+
+    case_sensitive: bool = False
+    use_regex: bool = False
+    max_results: int = 50
+    context_lines: int = 2
+    include_patterns: Optional[List[str]] = None
+    exclude_patterns: Optional[List[str]] = None
+    max_file_size: int = 1024 * 1024  # 1MB
+    timeout_seconds: int = 30
+    first_match_deadline: float = 3.0  # Timeout for finding first match

tunacode/tools/todo.py

@@ -14,8 +14,8 @@ from tunacode.constants import (
     MAX_TODO_CONTENT_LENGTH,
     MAX_TODOS_PER_SESSION,
     TODO_PRIORITIES,
-    TODO_PRIORITY_MEDIUM,
-    TODO_STATUS_PENDING,
+    TodoPriority,
+    TodoStatus,
 )
 from tunacode.types import TodoItem, ToolResult, UILogger
 
@@ -65,10 +65,14 @@ class TodoTool(BaseTool):
             ModelRetry: When invalid parameters are provided
         """
         if action == "add":
+            if isinstance(content, list):
+                raise ModelRetry("Use 'add_multiple' action for adding multiple todos")
             return await self._add_todo(content, priority)
         elif action == "add_multiple":
             return await self._add_multiple_todos(content, todos, priority)
         elif action == "update":
+            if isinstance(content, list):
+                raise ModelRetry("Cannot update with list content")
             return await self._update_todo(todo_id, status, priority, content)
         elif action == "complete":
             return await self._complete_todo(todo_id)
@@ -102,16 +106,16 @@ class TodoTool(BaseTool):
         new_id = f"todo_{uuid.uuid4().hex[:8]}"
 
         # Default priority if not specified
-        todo_priority = priority or TODO_PRIORITY_MEDIUM
-        if todo_priority not in TODO_PRIORITIES:
+        todo_priority = priority or TodoPriority.MEDIUM
+        if todo_priority not in [p.value for p in TodoPriority]:
             raise ModelRetry(
-                f"Invalid priority '{todo_priority}'. Must be one of: {', '.join(TODO_PRIORITIES)}"
+                f"Invalid priority '{todo_priority}'. Must be one of: {', '.join([p.value for p in TodoPriority])}"
             )
 
         new_todo = TodoItem(
             id=new_id,
             content=content,
-            status=TODO_STATUS_PENDING,
+            status=TodoStatus.PENDING,
             priority=todo_priority,
             created_at=datetime.now(),
         )
@@ -136,7 +140,7 @@ class TodoTool(BaseTool):
                 if not isinstance(todo_data, dict) or "content" not in todo_data:
                     raise ModelRetry("Each todo must be a dict with 'content' field")
                 todo_content = todo_data["content"]
-                todo_priority = todo_data.get("priority", priority or TODO_PRIORITY_MEDIUM)
+                todo_priority = todo_data.get("priority", priority or TodoPriority.MEDIUM)
                 if todo_priority not in TODO_PRIORITIES:
                     raise ModelRetry(
                         f"Invalid priority '{todo_priority}'. Must be one of: {', '.join(TODO_PRIORITIES)}"
@@ -144,7 +148,7 @@ class TodoTool(BaseTool):
                 todos_to_add.append((todo_content, todo_priority))
         elif isinstance(content, list):
             # List of strings format: ["task1", "task2", ...]
-            default_priority = priority or TODO_PRIORITY_MEDIUM
+            default_priority = priority or TodoPriority.MEDIUM
             if default_priority not in TODO_PRIORITIES:
                 raise ModelRetry(
                     f"Invalid priority '{default_priority}'. Must be one of: {', '.join(TODO_PRIORITIES)}"
@@ -184,7 +188,7 @@ class TodoTool(BaseTool):
             new_todo = TodoItem(
                 id=new_id,
                 content=task_content,
-                status=TODO_STATUS_PENDING,
+                status=TodoStatus.PENDING,
                 priority=task_priority,
                 created_at=datetime.now(),
             )
@@ -220,19 +224,21 @@ class TodoTool(BaseTool):
 
         # Update status if provided
         if status:
-            if status not in ["pending", "in_progress", "completed"]:
+            if status not in [s.value for s in TodoStatus]:
                 raise ModelRetry(
-                    f"Invalid status '{status}'. Must be pending, in_progress, or completed"
+                    f"Invalid status '{status}'. Must be one of: {', '.join([s.value for s in TodoStatus])}"
                 )
             todo.status = status
-            if status == "completed" and not todo.completed_at:
+            if status == TodoStatus.COMPLETED.value and not todo.completed_at:
                 todo.completed_at = datetime.now()
             changes.append(f"status to {status}")
 
         # Update priority if provided
         if priority:
-            if priority not in ["high", "medium", "low"]:
-                raise ModelRetry(f"Invalid priority '{priority}'. Must be high, medium, or low")
+            if priority not in [p.value for p in TodoPriority]:
+                raise ModelRetry(
+                    f"Invalid priority '{priority}'. Must be one of: {', '.join([p.value for p in TodoPriority])}"
+                )
             todo.priority = priority
             changes.append(f"priority to {priority}")
 
@@ -257,7 +263,7 @@ class TodoTool(BaseTool):
         # Find and update the todo
         for todo in self.state_manager.session.todos:
             if todo.id == todo_id:
-                todo.status = "completed"
+                todo.status = TodoStatus.COMPLETED.value
                 todo.completed_at = datetime.now()
                 return f"Marked todo {todo_id} as completed: {todo.content}"
 
@@ -270,9 +276,9 @@ class TodoTool(BaseTool):
             return "No todos found"
 
         # Group by status for better organization
-        pending = [t for t in todos if t.status == "pending"]
-        in_progress = [t for t in todos if t.status == "in_progress"]
-        completed = [t for t in todos if t.status == "completed"]
+        pending = [t for t in todos if t.status == TodoStatus.PENDING.value]
+        in_progress = [t for t in todos if t.status == TodoStatus.IN_PROGRESS.value]
+        completed = [t for t in todos if t.status == TodoStatus.COMPLETED.value]
 
         lines = []
 
@@ -319,9 +325,9 @@ class TodoTool(BaseTool):
             return "No todos found"
 
         # Group by status for better organization
-        pending = [t for t in todos if t.status == "pending"]
-        in_progress = [t for t in todos if t.status == "in_progress"]
-        completed = [t for t in todos if t.status == "completed"]
+        pending = [t for t in todos if t.status == TodoStatus.PENDING.value]
+        in_progress = [t for t in todos if t.status == TodoStatus.IN_PROGRESS.value]
+        completed = [t for t in todos if t.status == TodoStatus.COMPLETED.value]
 
         lines = []
 

tunacode/types.py

@@ -8,7 +8,18 @@ used throughout the TunaCode codebase.
 from dataclasses import dataclass, field
 from datetime import datetime
 from pathlib import Path
-from typing import Any, Awaitable, Callable, Dict, List, Literal, Optional, Protocol, Tuple, Union
+from typing import (
+    Any,
+    Awaitable,
+    Callable,
+    Dict,
+    List,
+    Literal,
+    Optional,
+    Protocol,
+    Tuple,
+    Union,
+)
 
 # Try to import pydantic-ai types if available
 try:
@@ -17,12 +28,14 @@ try:
 
     PydanticAgent = Agent
     MessagePart = Union[ToolReturnPart, Any]
+    ModelRequest = ModelRequest  # type: ignore[misc]
+    ModelResponse = Any
 except ImportError:
     # Fallback if pydantic-ai is not available
     PydanticAgent = Any
-    MessagePart = Any
+    MessagePart = Any  # type: ignore[misc]
     ModelRequest = Any
-    ModelResponse = Any
+    ModelResponse = Any  # type: ignore[misc]
 
 
 @dataclass
@@ -94,7 +107,7 @@ ToolCallId = str
 class ToolFunction(Protocol):
     """Protocol for tool functions."""
 
-    async def __call__(self, *args, **kwargs) -> str: ...
+    async def __call__(self, *_args, **kwargs) -> str: ...
 
 
 @dataclass
@@ -158,6 +171,8 @@ class ResponseState:
 
     has_user_response: bool = False
     has_final_synthesis: bool = False
+    task_completed: bool = False
+    awaiting_user_guidance: bool = False
 
 
 @dataclass

tunacode/ui/completers.py

@@ -3,7 +3,12 @@
 import os
 from typing import TYPE_CHECKING, Iterable, Optional
 
-from prompt_toolkit.completion import CompleteEvent, Completer, Completion, merge_completers
+from prompt_toolkit.completion import (
+    CompleteEvent,
+    Completer,
+    Completion,
+    merge_completers,
+)
 from prompt_toolkit.document import Document
 
 if TYPE_CHECKING:

tunacode/ui/decorators.py

@@ -53,7 +53,7 @@ def create_sync_wrapper(async_func: F) -> F:
         )
 
     # Attach the sync version as an attribute
-    async_func.sync = sync_wrapper
+    setattr(async_func, "sync", sync_wrapper)
 
     # Return the original async function
-    return async_func
+    return async_func  # type: ignore[return-value]

tunacode/ui/keybindings.py

@@ -44,7 +44,7 @@ def create_key_bindings(state_manager: StateManager = None) -> KeyBindings:
         if session.last_esc_time and (current_time - session.last_esc_time) > 3.0:
             session.esc_press_count = 0
 
-        session.esc_press_count += 1
+        session.esc_press_count = (session.esc_press_count or 0) + 1
         session.last_esc_time = current_time
 
         logger.debug(f"ESC key pressed: count={session.esc_press_count}, time={current_time}")

tunacode/ui/panels.py

@@ -47,7 +47,7 @@ colors = DotDict(UI_COLORS)
 @create_sync_wrapper
 async def panel(
     title: str,
-    text: Union[str, Markdown, Pretty],
+    text: Union[str, Markdown, Pretty, Table],
     top: int = DEFAULT_PANEL_PADDING["top"],
     right: int = DEFAULT_PANEL_PADDING["right"],
     bottom: int = DEFAULT_PANEL_PADDING["bottom"],
@@ -57,6 +57,7 @@ async def panel(
 ) -> None:
     """Display a rich panel with modern styling."""
     border_style = border_style or kwargs.get("style") or colors.border
+
     panel_obj = Panel(
         Padding(text, (0, 1, 0, 1)),
         title=f"[bold]{title}[/bold]",
@@ -65,7 +66,10 @@ async def panel(
         padding=(0, 1),
         box=ROUNDED,  # Use ROUNDED box style
     )
-    await print(Padding(panel_obj, (top, right, bottom, left)), **kwargs)
+
+    final_padding = Padding(panel_obj, (top, right, bottom, left))
+
+    await print(final_padding, **kwargs)
 
 
 async def agent(text: str, bottom: int = 1) -> None:
@@ -83,7 +87,7 @@ class StreamingAgentPanel:
         self.content = ""
         self.live = None
 
-    def _create_panel(self) -> Panel:
+    def _create_panel(self) -> Padding:
         """Create a Rich panel with current content."""
         # Use the UI_THINKING_MESSAGE constant instead of hardcoded text
         from rich.text import Text
@@ -92,7 +96,7 @@ class StreamingAgentPanel:
 
         # Handle the default thinking message with Rich markup
         if not self.content:
-            content_renderable = Text.from_markup(UI_THINKING_MESSAGE)
+            content_renderable: Union[Text, Markdown] = Text.from_markup(UI_THINKING_MESSAGE)
         else:
             content_renderable = Markdown(self.content)
         panel_obj = Panel(
@@ -122,7 +126,11 @@ class StreamingAgentPanel:
 
     async def update(self, content_chunk: str):
         """Update the streaming display with new content."""
-        self.content += content_chunk
+        # Defensive: some providers may yield None chunks intermittently
+        if content_chunk is None:
+            content_chunk = ""
+        # Ensure type safety for concatenation
+        self.content = (self.content or "") + str(content_chunk)
         if self.live:
             self.live.update(self._create_panel())
 

tunacode/ui/prompt_manager.py

@@ -40,7 +40,7 @@ class PromptManager:
             state_manager: Optional state manager for session persistence
         """
         self.state_manager = state_manager
-        self._temp_sessions = {}  # For when no state manager is available
+        self._temp_sessions: dict[str, PromptSession] = {}  # For when no state manager is available
         self._style = self._create_style()
 
     def _create_style(self) -> Style:

tunacode/ui/tool_ui.py

@@ -2,6 +2,8 @@
 Tool confirmation UI components, separated from business logic.
 """
 
+from typing import TYPE_CHECKING, Optional
+
 from rich.box import ROUNDED
 from rich.markdown import Markdown
 from rich.padding import Padding
@@ -16,6 +18,9 @@ from tunacode.utils.diff_utils import render_file_diff
 from tunacode.utils.file_utils import DotDict
 from tunacode.utils.text_utils import ext_to_lang, key_to_title
 
+if TYPE_CHECKING:
+    from tunacode.core.state import StateManager
+
 
 class ToolUI:
     """Handles tool confirmation UI presentation."""
@@ -71,7 +76,8 @@ class ToolUI:
 
         # Show file content on write_file
         elif tool_name == TOOL_WRITE_FILE:
-            return self._create_code_block(args["filepath"], args["content"])
+            markdown_obj = self._create_code_block(args["filepath"], args["content"])
+            return str(markdown_obj)
 
         # Default to showing key and value on new line
         content = ""
@@ -92,7 +98,7 @@ class ToolUI:
         return content.strip()
 
     async def show_confirmation(
-        self, request: ToolConfirmationRequest, state_manager=None
+        self, request: ToolConfirmationRequest, state_manager: Optional["StateManager"] = None
     ) -> ToolConfirmationResponse:
         """
         Show tool confirmation UI and get user response.

tunacode/utils/bm25.py

@@ -16,14 +16,14 @@ class BM25:
         self.k1 = k1
         self.b = b
         self.documents = [tokenize(doc) for doc in corpus]
-        self.doc_freqs = []
-        self.doc_lens = []
-        self.idf = {}
+        self.doc_freqs: list[dict] = []
+        self.doc_lens: list[int] = []
+        self.idf: dict[str, float] = {}
         self.avgdl = 0.0
         self._initialize()
 
     def _initialize(self) -> None:
-        df = Counter()
+        df: dict[str, int] = Counter()
         for doc in self.documents:
             freqs = Counter(doc)
             self.doc_freqs.append(freqs)

tunacode/utils/file_utils.py

@@ -14,8 +14,8 @@ class DotDict(dict):
     """dot.notation access to dictionary attributes"""
 
     __getattr__ = dict.get
-    __setattr__ = dict.__setitem__
-    __delattr__ = dict.__delitem__
+    __setattr__ = dict.__setitem__  # type: ignore[assignment]
+    __delattr__ = dict.__delitem__  # type: ignore[assignment]
 
 
 @contextmanager

tunacode/utils/message_utils.py

@@ -1,7 +1,9 @@
 """Utilities for processing message history."""
 
+from typing import Any
 
-def get_message_content(message: any) -> str:
+
+def get_message_content(message: Any) -> str:
     """Extracts the content from a message object of any type."""
     if isinstance(message, str):
         return message

tunacode/utils/system.py

@@ -85,9 +85,7 @@ def _load_gitignore_patterns(filepath=".gitignore"):
                 line = line.strip()
                 if line and not line.startswith("#"):
                     patterns.add(line)
-        # print(f"Loaded {len(patterns)} patterns from {filepath}") # Debug print (optional)
     except FileNotFoundError:
-        # print(f"{filepath} not found.") # Debug print (optional)
         return None
     except Exception as e:
         print(f"Error reading {filepath}: {e}")
@@ -322,8 +320,6 @@ def list_cwd(max_depth=3):
             dir_rel_path = os.path.join(rel_root, d) if rel_root else d
             if not _is_ignored(dir_rel_path, d, ignore_patterns):
                 dirs.append(d)
-            # else: # Optional debug print
-            #     print(f"Ignoring dir: {dir_rel_path}")
 
         # --- File Processing ---
         if current_depth <= max_depth:

tunacode/utils/text_utils.py

@@ -81,7 +81,7 @@ def expand_file_refs(text: str) -> Tuple[str, List[str]]:
 
     # Regex now includes trailing / and ** to capture directory intentions
     pattern = re.compile(r"@([\w./\-_*]+)")
-    expanded_files = []
+    expanded_files: list[str] = []
 
     def replacer(match: re.Match) -> str:
         path_spec = match.group(1)

tunacode/utils/token_counter.py

@@ -2,13 +2,13 @@
 
 import logging
 from functools import lru_cache
-from typing import Optional
+from typing import Any, Optional
 
 # Get logger for this module
 logger = logging.getLogger(__name__)
 
 # Cache for tokenizer encodings
-_encoding_cache = {}
+_encoding_cache: dict[str, Any] = {}
 
 
 @lru_cache(maxsize=8)

{tunacode_cli-0.0.51.dist-info → tunacode_cli-0.0.53.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.51
+Version: 0.0.53
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License: MIT
@@ -35,6 +35,10 @@ Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
 Requires-Dist: textual-dev; extra == "dev"
 Requires-Dist: pre-commit; extra == "dev"
+Requires-Dist: vulture>=2.7; extra == "dev"
+Requires-Dist: unimport>=1.0.0; extra == "dev"
+Requires-Dist: autoflake>=2.0.0; extra == "dev"
+Requires-Dist: dead>=1.5.0; extra == "dev"
 Dynamic: license-file
 
 # TunaCode CLI
@@ -175,3 +179,144 @@ _Note: While the tool is fully functional, we're focusing on stability and core
 ---
 
 MIT License - see [LICENSE](LICENSE) file
+
+hello from tuna world
+
+hello world
+
+## Getting Started
+
+### Prerequisites
+
+Before you begin, ensure you have the following installed:
+
+- Python 3.10 or higher
+- Git
+
+### Installation
+
+To install TunaCode, you can use one of the following methods:
+
+```bash
+# Option 1: One-line install (Linux/macOS)
+wget -qO- https://raw.githubusercontent.com/alchemiststudiosDOTai/tunacode/master/scripts/install_linux.sh | bash
+
+# Option 2: pip install
+pip install tunacode-cli
+```
+
+### Development Installation
+
+For developers who want to contribute to TunaCode:
+
+```bash
+# Clone the repository
+git clone https://github.com/alchemiststudiosDOTai/tunacode.git
+cd tunacode
+
+# Quick setup (recommended)
+./scripts/setup_dev_env.sh
+
+# Or manual setup
+python3 -m venv venv
+source venv/bin/activate  # On Windows: venv\Scripts\activate
+pip install -e ".[dev]"
+
+# Verify installation
+python -m tunacode --version
+```
+
+See [Development Guide](docs/DEVELOPMENT.md) for detailed instructions.
+
+### Configuration
+
+Choose your AI provider and set your API key:
+
+```bash
+# OpenAI
+tunacode --model "openai:gpt-4o" --key "sk-your-openai-key"
+
+# Anthropic Claude
+tunacode --model "anthropic:claude-3.5-sonnet" --key "sk-ant-your-anthropic-key"
+
+# OpenRouter (100+ models)
+tunacode --model "openrouter:openai/gpt-4o" --key "sk-or-your-openrouter-key"
+```
+
+Your config is saved to `~/.config/tunacode.json` (edit directly with `nvim ~/.config/tunacode.json`)
+
+### Recommended Models
+
+Based on extensive testing, these models provide the best performance:
+
+- `google/gemini-2.5-pro` - Excellent for complex reasoning
+- `openai/gpt-4.1` - Strong general-purpose model
+- `deepseek/deepseek-r1-0528` - Great for code generation
+- `openai/gpt-4.1-mini` - Fast and cost-effective
+- `anthropic/claude-4-sonnet-20250522` - Superior context handling
+
+_Note: Formal evaluations coming soon. Any model can work, but these have shown the best results in practice._
+
+## Usage
+
+### Starting TunaCode
+
+```bash
+tunacode
+```
+
+### Basic Commands
+
+| Command                  | Description            |
+| ------------------------ | ---------------------- |
+| `/help`                  | Show all commands      |
+| `/model <provider:name>` | Switch model           |
+| `/clear`                 | Clear message history  |
+| `/compact`               | Summarize conversation |
+| `/branch <name>`         | Create Git branch      |
+| `/yolo`                  | Skip confirmations     |
+| `!<command>`             | Run shell command      |
+| `exit`                   | Exit TunaCode          |
+
+## Performance
+
+TunaCode leverages parallel execution for read-only operations, achieving **3x faster** file operations:
+
+![Parallel Execution Performance](docs/assets/parrelel_work_3x.png)
+
+Multiple file reads, directory listings, and searches execute concurrently using async I/O, making code exploration significantly faster.
+
+## Features in Development
+
+- **Streaming UI**: Currently working on implementing streaming responses for better user experience
+- **Bug Fixes**: Actively addressing issues - please report any bugs you encounter!
+
+_Note: While the tool is fully functional, we're focusing on stability and core features before optimizing for speed._
+
+## Safety First
+
+⚠️ **Important**: TunaCode can modify your codebase. Always:
+
+- Use Git branches before making changes
+- Review file modifications before confirming
+- Keep backups of important work
+
+## Documentation
+
+- [**Features**](docs/FEATURES.md) - All features, tools, and commands
+- [**Advanced Configuration**](docs/ADVANCED-CONFIG.md) - Provider setup, MCP, customization
+- [**Architecture**](docs/ARCHITECTURE.md) - Source code organization and design
+- [**Development**](docs/DEVELOPMENT.md) - Contributing and development setup
+- [**Troubleshooting**](docs/TROUBLESHOOTING.md) - Common issues and solutions
+
+## Links
+
+- [PyPI Package](https://pypi.org/project/tunacode-cli/)
+- [GitHub Repository](https://github.com/alchemiststudiosDOTai/tunacode)
+- [Report Issues](https://github.com/alchemiststudiosDOTai/tunacode/issues)
+
+---
+
+MIT License - see [LICENSE](LICENSE) file
+# Test
+# Test