tunacode-cli 0.0.17__tar.gz → 0.0.19__tar.gz

{tunacode_cli-0.0.17 → tunacode_cli-0.0.19}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.17
+Version: 0.0.19
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License-Expression: MIT
@@ -58,14 +58,20 @@ Dynamic: license-file
 
 ---
 
-### Recent Updates (v0.0.14)
-
-- **🔧 JSON Tool Parsing Fallback**: Automatic recovery when API providers fail with structured tool calling
-- **⚡ Enhanced Reliability**: Fixed parameter naming issues that caused tool schema errors
-- **🔄 Configuration Management**: New `/refresh` command to reload config without restart
-- **🧠 Improved ReAct Reasoning**: Enhanced iteration limits (now defaults to 20) and better thought processing
-- **🛠️ New Debug Commands**: `/parsetools` for manual JSON parsing, `/iterations` for controlling reasoning depth
-- **📊 Better Error Recovery**: Multiple fallback mechanisms for various failure scenarios
+### Recent Updates (v0.0.18)
+
+- **Advanced Agent Orchestration**: New orchestrator system for complex multi-step tasks with planning visibility
+- **Background Task Manager**: Asynchronous background processing for long-running operations
+- **Read-Only Agent**: Specialized agent for safe codebase exploration without modification risks
+- **Planning Transparency**: See the AI's planning process before execution with detailed task breakdowns
+- **Shell Command Support**: Execute shell commands directly with `!command` or open interactive shell with `!`
+- **Enhanced Bash Tool**: Advanced bash execution with timeouts, working directory, and environment variables
+- **JSON Tool Parsing Fallback**: Automatic recovery when API providers fail with structured tool calling
+- **Enhanced Reliability**: Fixed parameter naming issues that caused tool schema errors
+- **Configuration Management**: New `/refresh` command to reload config without restart
+- **Improved ReAct Reasoning**: Enhanced iteration limits (now defaults to 20) and better thought processing
+- **New Debug Commands**: `/parsetools` for manual JSON parsing, `/iterations` for controlling reasoning depth
+- **Better Error Recovery**: Multiple fallback mechanisms for various failure scenarios
 
 ### Core Features
 
@@ -84,6 +90,7 @@ Dynamic: license-file
 ### **Developer Tools**
 
 - 6 core tools: bash, grep, read_file, write_file, update_file, run_command
+- Direct shell command execution with `!` prefix
 - MCP (Model Context Protocol) support
 - File operation confirmations with diffs
 - Per-project context guides (TUNACODE.md)
@@ -284,6 +291,8 @@ Learn more at [modelcontextprotocol.io](https://modelcontextprotocol.io/)
 | `/model <provider:name> default` | Set default model                |
 | `/branch <name>`                 | Create and switch Git branch     |
 | `/dump`                          | Show message history (debug)     |
+| `!<command>`                     | Run shell command                |
+| `!`                              | Open interactive shell           |
 | `exit`                           | Exit application                 |
 
 ### Debug & Recovery Commands
@@ -298,6 +307,42 @@ Learn more at [modelcontextprotocol.io](https://modelcontextprotocol.io/)
 
 ---
 
+## Available Tools
+
+### Bash Tool
+The enhanced bash tool provides advanced shell command execution with safety features:
+
+- **Working Directory Support**: Execute commands in specific directories
+- **Environment Variables**: Set custom environment variables for commands
+- **Timeout Control**: Configurable timeouts (1-300 seconds) to prevent hanging
+- **Output Capture**: Full stdout/stderr capture with truncation for large outputs
+- **Safety Checks**: Warns about potentially destructive commands
+- **Error Guidance**: Helpful error messages for common issues (command not found, permission denied, etc.)
+
+**Example usage by the AI:**
+```python
+# Simple command
+await bash("ls -la")
+
+# With working directory
+await bash("npm install", cwd="/path/to/project")
+
+# With timeout for long operations
+await bash("npm run build", timeout=120)
+
+# With environment variables
+await bash("python script.py", env={"API_KEY": "secret"})
+```
+
+### Other Core Tools
+- **grep**: Fast parallel content search across files
+- **read_file**: Read file contents with line numbers
+- **write_file**: Create new files (fails if file exists)
+- **update_file**: Modify existing files with precise replacements
+- **run_command**: Basic command execution (simpler than bash)
+
+---
+
 ## Reliability Features
 
 ### JSON Tool Parsing Fallback
@@ -360,7 +405,14 @@ src/tunacode/
 │
 ├── core/                 # Core Application Logic
 │   ├── agents/           # AI Agent System
-│   │   └── main.py       # Primary agent implementation (pydantic-ai)
+│   │   ├── main.py       # Primary agent implementation (pydantic-ai)
+│   │   ├── orchestrator.py # Complex task orchestration and planning
+│   │   ├── planner_schema.py # Planning data models
+│   │   └── readonly.py   # Read-only agent for safe exploration
+│   ├── background/       # Background Task Management
+│   │   └── manager.py    # Async background task execution
+│   ├── llm/              # LLM Integration
+│   │   └── planner.py    # LLM-based task planning
 │   ├── setup/            # Application Setup & Initialization
 │   │   ├── agent_setup.py     # Agent configuration
 │   │   ├── base.py           # Setup step base class
@@ -409,7 +461,7 @@ src/tunacode/
 ├── exceptions.py      # Custom exceptions
 ├── types.py           # Type definitions
 └── prompts/
-    └── system.txt     # System prompts for AI agent
+    └── system.md      # System prompts for AI agent
 ```
 
 ### Key Components
@@ -418,6 +470,8 @@ src/tunacode/
 | -------------------- | ------------------------ | ------------------------------- |
 | **CLI Layer**        | Command parsing and REPL | `cli/main.py`, `cli/repl.py`    |
 | **Agent System**     | AI-powered assistance    | `core/agents/main.py`           |
+| **Orchestrator**     | Complex task planning    | `core/agents/orchestrator.py`   |
+| **Background Tasks** | Async task execution     | `core/background/manager.py`    |
 | **Tool System**      | File/command operations  | `tools/*.py`                    |
 | **State Management** | Session state tracking   | `core/state.py`                 |
 | **UI Framework**     | Rich terminal interface  | `ui/output.py`, `ui/console.py` |
@@ -482,6 +536,9 @@ While TunaCode builds on the foundation of sidekick-cli, we've made several arch
 
 - **JSON Tool Parsing Fallback**: Added fallback parsing for when API providers fail with structured tool calling
 - **Parallel Search Tools**: New `bash` and `grep` tools with parallel execution for codebase navigation
+- **Agent Orchestration**: Advanced orchestrator for complex multi-step tasks with planning transparency
+- **Background Processing**: Asynchronous task manager for long-running operations
+- **Read-Only Agent**: Safe exploration mode that prevents accidental modifications
 - **ReAct Reasoning**: Implemented ReAct (Reasoning + Acting) patterns with configurable iteration limits
 - **Dynamic Configuration**: Added `/refresh` command and modified configuration management
 - **Safety Changes**: Removed automatic git commits and `/undo` command - requires explicit git usage

{tunacode_cli-0.0.17 → tunacode_cli-0.0.19}/README.md

@@ -22,14 +22,20 @@
 
 ---
 
-### Recent Updates (v0.0.14)
-
-- **🔧 JSON Tool Parsing Fallback**: Automatic recovery when API providers fail with structured tool calling
-- **⚡ Enhanced Reliability**: Fixed parameter naming issues that caused tool schema errors
-- **🔄 Configuration Management**: New `/refresh` command to reload config without restart
-- **🧠 Improved ReAct Reasoning**: Enhanced iteration limits (now defaults to 20) and better thought processing
-- **🛠️ New Debug Commands**: `/parsetools` for manual JSON parsing, `/iterations` for controlling reasoning depth
-- **📊 Better Error Recovery**: Multiple fallback mechanisms for various failure scenarios
+### Recent Updates (v0.0.18)
+
+- **Advanced Agent Orchestration**: New orchestrator system for complex multi-step tasks with planning visibility
+- **Background Task Manager**: Asynchronous background processing for long-running operations
+- **Read-Only Agent**: Specialized agent for safe codebase exploration without modification risks
+- **Planning Transparency**: See the AI's planning process before execution with detailed task breakdowns
+- **Shell Command Support**: Execute shell commands directly with `!command` or open interactive shell with `!`
+- **Enhanced Bash Tool**: Advanced bash execution with timeouts, working directory, and environment variables
+- **JSON Tool Parsing Fallback**: Automatic recovery when API providers fail with structured tool calling
+- **Enhanced Reliability**: Fixed parameter naming issues that caused tool schema errors
+- **Configuration Management**: New `/refresh` command to reload config without restart
+- **Improved ReAct Reasoning**: Enhanced iteration limits (now defaults to 20) and better thought processing
+- **New Debug Commands**: `/parsetools` for manual JSON parsing, `/iterations` for controlling reasoning depth
+- **Better Error Recovery**: Multiple fallback mechanisms for various failure scenarios
 
 ### Core Features
 
@@ -48,6 +54,7 @@
 ### **Developer Tools**
 
 - 6 core tools: bash, grep, read_file, write_file, update_file, run_command
+- Direct shell command execution with `!` prefix
 - MCP (Model Context Protocol) support
 - File operation confirmations with diffs
 - Per-project context guides (TUNACODE.md)
@@ -248,6 +255,8 @@ Learn more at [modelcontextprotocol.io](https://modelcontextprotocol.io/)
 | `/model <provider:name> default` | Set default model                |
 | `/branch <name>`                 | Create and switch Git branch     |
 | `/dump`                          | Show message history (debug)     |
+| `!<command>`                     | Run shell command                |
+| `!`                              | Open interactive shell           |
 | `exit`                           | Exit application                 |
 
 ### Debug & Recovery Commands
@@ -262,6 +271,42 @@ Learn more at [modelcontextprotocol.io](https://modelcontextprotocol.io/)
 
 ---
 
+## Available Tools
+
+### Bash Tool
+The enhanced bash tool provides advanced shell command execution with safety features:
+
+- **Working Directory Support**: Execute commands in specific directories
+- **Environment Variables**: Set custom environment variables for commands
+- **Timeout Control**: Configurable timeouts (1-300 seconds) to prevent hanging
+- **Output Capture**: Full stdout/stderr capture with truncation for large outputs
+- **Safety Checks**: Warns about potentially destructive commands
+- **Error Guidance**: Helpful error messages for common issues (command not found, permission denied, etc.)
+
+**Example usage by the AI:**
+```python
+# Simple command
+await bash("ls -la")
+
+# With working directory
+await bash("npm install", cwd="/path/to/project")
+
+# With timeout for long operations
+await bash("npm run build", timeout=120)
+
+# With environment variables
+await bash("python script.py", env={"API_KEY": "secret"})
+```
+
+### Other Core Tools
+- **grep**: Fast parallel content search across files
+- **read_file**: Read file contents with line numbers
+- **write_file**: Create new files (fails if file exists)
+- **update_file**: Modify existing files with precise replacements
+- **run_command**: Basic command execution (simpler than bash)
+
+---
+
 ## Reliability Features
 
 ### JSON Tool Parsing Fallback
@@ -324,7 +369,14 @@ src/tunacode/
 │
 ├── core/                 # Core Application Logic
 │   ├── agents/           # AI Agent System
-│   │   └── main.py       # Primary agent implementation (pydantic-ai)
+│   │   ├── main.py       # Primary agent implementation (pydantic-ai)
+│   │   ├── orchestrator.py # Complex task orchestration and planning
+│   │   ├── planner_schema.py # Planning data models
+│   │   └── readonly.py   # Read-only agent for safe exploration
+│   ├── background/       # Background Task Management
+│   │   └── manager.py    # Async background task execution
+│   ├── llm/              # LLM Integration
+│   │   └── planner.py    # LLM-based task planning
 │   ├── setup/            # Application Setup & Initialization
 │   │   ├── agent_setup.py     # Agent configuration
 │   │   ├── base.py           # Setup step base class
@@ -373,7 +425,7 @@ src/tunacode/
 ├── exceptions.py      # Custom exceptions
 ├── types.py           # Type definitions
 └── prompts/
-    └── system.txt     # System prompts for AI agent
+    └── system.md      # System prompts for AI agent
 ```
 
 ### Key Components
@@ -382,6 +434,8 @@ src/tunacode/
 | -------------------- | ------------------------ | ------------------------------- |
 | **CLI Layer**        | Command parsing and REPL | `cli/main.py`, `cli/repl.py`    |
 | **Agent System**     | AI-powered assistance    | `core/agents/main.py`           |
+| **Orchestrator**     | Complex task planning    | `core/agents/orchestrator.py`   |
+| **Background Tasks** | Async task execution     | `core/background/manager.py`    |
 | **Tool System**      | File/command operations  | `tools/*.py`                    |
 | **State Management** | Session state tracking   | `core/state.py`                 |
 | **UI Framework**     | Rich terminal interface  | `ui/output.py`, `ui/console.py` |
@@ -446,6 +500,9 @@ While TunaCode builds on the foundation of sidekick-cli, we've made several arch
 
 - **JSON Tool Parsing Fallback**: Added fallback parsing for when API providers fail with structured tool calling
 - **Parallel Search Tools**: New `bash` and `grep` tools with parallel execution for codebase navigation
+- **Agent Orchestration**: Advanced orchestrator for complex multi-step tasks with planning transparency
+- **Background Processing**: Asynchronous task manager for long-running operations
+- **Read-Only Agent**: Safe exploration mode that prevents accidental modifications
 - **ReAct Reasoning**: Implemented ReAct (Reasoning + Acting) patterns with configurable iteration limits
 - **Dynamic Configuration**: Added `/refresh` command and modified configuration management
 - **Safety Changes**: Removed automatic git commits and `/undo` command - requires explicit git usage

{tunacode_cli-0.0.17 → tunacode_cli-0.0.19}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tunacode-cli"
-version = "0.0.17"
+version = "0.0.19"
 description = "Your agentic CLI developer."
 keywords = ["cli", "agent", "development", "automation"]
 readme = "README.md"

{tunacode_cli-0.0.17 → tunacode_cli-0.0.19}/src/tunacode/cli/commands.py

@@ -6,7 +6,6 @@ from enum import Enum
 from typing import Any, Dict, List, Optional, Type
 
 from .. import utils
-from ..configuration.models import ModelRegistry
 from ..exceptions import ValidationError
 from ..types import CommandArgs, CommandContext, CommandResult, ProcessRequestCallback
 from ..ui import console as ui
@@ -116,9 +115,9 @@ class YoloCommand(SimpleCommand):
         state = context.state_manager.session
         state.yolo = not state.yolo
         if state.yolo:
-            await ui.success("Ooh shit, its YOLO time!\n")
+            await ui.success("All tools are now active ⚡ Please proceed with caution.\n")
         else:
-            await ui.info("Pfft, boring...\n")
+            await ui.info("Tool confirmations re-enabled for safety.\n")
 
 
 class DumpCommand(SimpleCommand):
@@ -168,6 +167,39 @@ class ThoughtsCommand(SimpleCommand):
         await ui.success(f"Thought display {status}")
 
 
+class ArchitectCommand(SimpleCommand):
+    """Toggle architect mode for task planning and orchestration."""
+
+    def __init__(self):
+        super().__init__(
+            CommandSpec(
+                name="architect",
+                aliases=["/architect"],
+                description="Toggle architect mode (task planning & orchestration)",
+                category=CommandCategory.DEBUG,
+            )
+        )
+
+    async def execute(self, args: List[str], context: CommandContext) -> None:
+        state = context.state_manager.session
+        if args:
+            arg = args[0].lower()
+            if arg in {"on", "1", "true"}:
+                state.architect_mode = True
+            elif arg in {"off", "0", "false"}:
+                state.architect_mode = False
+            else:
+                await ui.error("Usage: /architect [on|off]")
+                return
+        else:
+            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")
+        else:
+            await ui.success(f"Architect mode {status} - Using direct execution")
+
+
 class IterationsCommand(SimpleCommand):
     """Configure maximum agent iterations for ReAct reasoning."""
 
@@ -189,12 +221,12 @@ class IterationsCommand(SimpleCommand):
                 if new_limit < 1 or new_limit > 50:
                     await ui.error("Iterations must be between 1 and 50")
                     return
-                
+
                 # Update the user config
                 if "settings" not in state.user_config:
                     state.user_config["settings"] = {}
                 state.user_config["settings"]["max_iterations"] = new_limit
-                
+
                 await ui.success(f"Maximum iterations set to {new_limit}")
                 await ui.muted("Higher values allow more complex reasoning but may be slower")
             except ValueError:
@@ -221,8 +253,9 @@ class ClearCommand(SimpleCommand):
     async def execute(self, args: List[str], context: CommandContext) -> None:
         # Patch any orphaned tool calls before clearing
         from tunacode.core.agents.main import patch_tool_messages
+
         patch_tool_messages("Conversation cleared", context.state_manager)
-        
+
         await ui.clear()
         context.state_manager.session.messages = []
         await ui.success("Message history cleared")
@@ -243,17 +276,17 @@ class FixCommand(SimpleCommand):
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
         from tunacode.core.agents.main import patch_tool_messages
-        
+
         # Count current messages
         before_count = len(context.state_manager.session.messages)
-        
+
         # Patch orphaned tool calls
         patch_tool_messages("Tool call resolved by /fix command", context.state_manager)
-        
+
         # Count after patching
         after_count = len(context.state_manager.session.messages)
         patched_count = after_count - before_count
-        
+
         if patched_count > 0:
             await ui.success(f"Fixed {patched_count} orphaned tool call(s)")
             await ui.muted("You can now continue the conversation normally")
@@ -269,36 +302,37 @@ class ParseToolsCommand(SimpleCommand):
             CommandSpec(
                 name="parsetools",
                 aliases=["/parsetools"],
-                description="Parse JSON tool calls from last response when structured calling fails",
+                description=(
+                    "Parse JSON tool calls from last response when structured calling fails"
+                ),
                 category=CommandCategory.DEBUG,
             )
         )
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
         from tunacode.core.agents.main import extract_and_execute_tool_calls
-        
+
         # Find the last model response in messages
         messages = context.state_manager.session.messages
         if not messages:
             await ui.error("No message history found")
             return
-        
+
         # Look for the most recent response with text content
         found_content = False
         for msg in reversed(messages):
-            if hasattr(msg, 'parts'):
+            if hasattr(msg, "parts"):
                 for part in msg.parts:
-                    if hasattr(part, 'content') and isinstance(part.content, str):
+                    if hasattr(part, "content") and isinstance(part.content, str):
                         # Create tool callback
                         from tunacode.cli.repl import _tool_handler
+
                         def tool_callback_with_state(part, node):
                             return _tool_handler(part, node, context.state_manager)
-                        
+
                         try:
                             await extract_and_execute_tool_calls(
-                                part.content, 
-                                tool_callback_with_state, 
-                                context.state_manager
+                                part.content, tool_callback_with_state, context.state_manager
                             )
                             await ui.success("JSON tool parsing completed")
                             found_content = True
@@ -306,7 +340,7 @@ class ParseToolsCommand(SimpleCommand):
                         except Exception as e:
                             await ui.error(f"Failed to parse tools: {str(e)}")
                             return
-        
+
         if not found_content:
             await ui.error("No parseable content found in recent messages")
 
@@ -326,7 +360,7 @@ class RefreshConfigCommand(SimpleCommand):
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
         from tunacode.configuration.defaults import DEFAULT_USER_CONFIG
-        
+
         # Update current session config with latest defaults
         for key, value in DEFAULT_USER_CONFIG.items():
             if key not in context.state_manager.session.user_config:
@@ -336,9 +370,11 @@ class RefreshConfigCommand(SimpleCommand):
                 for subkey, subvalue in value.items():
                     if subkey not in context.state_manager.session.user_config[key]:
                         context.state_manager.session.user_config[key][subkey] = subvalue
-        
+
         # Show updated max_iterations
-        max_iterations = context.state_manager.session.user_config.get("settings", {}).get("max_iterations", 20)
+        max_iterations = context.state_manager.session.user_config.get("settings", {}).get(
+            "max_iterations", 20
+        )
         await ui.success(f"Configuration refreshed - max iterations: {max_iterations}")
 
 
@@ -507,23 +543,20 @@ class UpdateCommand(SimpleCommand):
         )
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
+        import shutil
         import subprocess
         import sys
-        import shutil
 
         await ui.info("Checking for TunaCode updates...")
 
         # Detect installation method
         installation_method = None
-        
+
         # Check if installed via pipx
         if shutil.which("pipx"):
             try:
                 result = subprocess.run(
-                    ["pipx", "list"], 
-                    capture_output=True, 
-                    text=True, 
-                    timeout=10
+                    ["pipx", "list"], capture_output=True, text=True, timeout=10
                 )
                 if "tunacode" in result.stdout.lower():
                     installation_method = "pipx"
@@ -534,10 +567,10 @@ class UpdateCommand(SimpleCommand):
         if not installation_method:
             try:
                 result = subprocess.run(
-                    [sys.executable, "-m", "pip", "show", "tunacode-cli"], 
-                    capture_output=True, 
-                    text=True, 
-                    timeout=10
+                    [sys.executable, "-m", "pip", "show", "tunacode-cli"],
+                    capture_output=True,
+                    text=True,
+                    timeout=10,
                 )
                 if result.returncode == 0:
                     installation_method = "pip"
@@ -556,10 +589,7 @@ class UpdateCommand(SimpleCommand):
             if installation_method == "pipx":
                 await ui.info("Updating via pipx...")
                 result = subprocess.run(
-                    ["pipx", "upgrade", "tunacode"],
-                    capture_output=True,
-                    text=True,
-                    timeout=60
+                    ["pipx", "upgrade", "tunacode"], capture_output=True, text=True, timeout=60
                 )
             else:  # pip
                 await ui.info("Updating via pip...")
@@ -567,16 +597,16 @@ class UpdateCommand(SimpleCommand):
                     [sys.executable, "-m", "pip", "install", "--upgrade", "tunacode-cli"],
                     capture_output=True,
                     text=True,
-                    timeout=60
+                    timeout=60,
                 )
 
             if result.returncode == 0:
                 await ui.success("TunaCode updated successfully!")
                 await ui.muted("Restart TunaCode to use the new version")
-                
+
                 # Show update output if available
                 if result.stdout.strip():
-                    output_lines = result.stdout.strip().split('\n')
+                    output_lines = result.stdout.strip().split("\n")
                     for line in output_lines[-5:]:  # Show last 5 lines
                         if line.strip():
                             await ui.muted(f"  {line}")
@@ -723,6 +753,7 @@ class CommandRegistry:
             YoloCommand,
             DumpCommand,
             ThoughtsCommand,
+            ArchitectCommand,
             IterationsCommand,
             ClearCommand,
             FixCommand,
@@ -798,8 +829,9 @@ class CommandRegistry:
             return await command.execute(args, context)
         else:
             # Ambiguous - show possibilities
+            matches_str = ", ".join(sorted(set(matches)))
             raise ValidationError(
-                f"Ambiguous command '{command_name}'. Did you mean: {', '.join(sorted(set(matches)))}?"
+                f"Ambiguous command '{command_name}'. Did you mean: {matches_str}?"
             )
 
     def find_matching_commands(self, partial_command: str) -> List[str]:

{tunacode_cli-0.0.17 → tunacode_cli-0.0.19}/src/tunacode/cli/main.py

@@ -24,36 +24,39 @@ state_manager = StateManager()
 def main(
     version: bool = typer.Option(False, "--version", "-v", help="Show version and exit."),
     run_setup: bool = typer.Option(False, "--setup", help="Run setup process."),
-    baseurl: str = typer.Option(None, "--baseurl", help="API base URL (e.g., https://openrouter.ai/api/v1)"),
+    baseurl: str = typer.Option(
+        None, "--baseurl", help="API base URL (e.g., https://openrouter.ai/api/v1)"
+    ),
     model: str = typer.Option(None, "--model", help="Default model to use (e.g., openai/gpt-4)"),
     key: str = typer.Option(None, "--key", help="API key for the provider"),
 ):
     """🚀 Start TunaCode - Your AI-powered development assistant"""
-    
-    if version:
-        asyncio.run(ui.version())
-        return
-
-    asyncio.run(ui.banner())
-
-    has_update, latest_version = check_for_updates()
-    if has_update:
-        asyncio.run(ui.show_update_message(latest_version))
-
-    # Pass CLI args to setup
-    cli_config = {}
-    if baseurl or model or key:
-        cli_config = {
-            "baseurl": baseurl,
-            "model": model,
-            "key": key
-        }
-
-    try:
-        asyncio.run(setup(run_setup, state_manager, cli_config))
-        asyncio.run(repl(state_manager))
-    except Exception as e:
-        asyncio.run(ui.error(str(e)))
+
+    async def async_main():
+        if version:
+            await ui.version()
+            return
+
+        await ui.banner()
+
+        # Start update check in background
+        update_task = asyncio.to_thread(check_for_updates)
+
+        cli_config = {}
+        if baseurl or model or key:
+            cli_config = {"baseurl": baseurl, "model": model, "key": key}
+
+        try:
+            await setup(run_setup, state_manager, cli_config)
+            await repl(state_manager)
+        except Exception as e:
+            await ui.error(str(e))
+
+        has_update, latest_version = await update_task
+        if has_update:
+            await ui.update_available(latest_version)
+
+    asyncio.run(async_main())
 
 
 if __name__ == "__main__":