tunacode-cli 0.0.36__tar.gz → 0.0.37__tar.gz

{tunacode_cli-0.0.36/src/tunacode_cli.egg-info → tunacode_cli-0.0.37}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tunacode-cli
-Version: 0.0.36
+Version: 0.0.37
 Summary: Your agentic CLI developer.
 Author-email: larock22 <noreply@github.com>
 License-Expression: MIT
@@ -26,9 +26,7 @@ Requires-Dist: pygments==2.19.1
 Requires-Dist: rich==14.0.0
 Provides-Extra: dev
 Requires-Dist: build; extra == "dev"
-Requires-Dist: black; extra == "dev"
-Requires-Dist: flake8; extra == "dev"
-Requires-Dist: isort; extra == "dev"
+Requires-Dist: ruff; extra == "dev"
 Requires-Dist: pytest; extra == "dev"
 Requires-Dist: pytest-cov; extra == "dev"
 Requires-Dist: pytest-asyncio; extra == "dev"
@@ -70,7 +68,7 @@ Choose your AI provider and set your API key:
 # OpenAI
 tunacode --model "openai:gpt-4o" --key "sk-your-openai-key"
 
-# Anthropic Claude  
+# Anthropic Claude
 tunacode --model "anthropic:claude-3.5-sonnet" --key "sk-ant-your-anthropic-key"
 
 # OpenRouter (100+ models)
@@ -82,13 +80,14 @@ Your config is saved to `~/.config/tunacode.json` (edit directly with `nvim ~/.c
 ### 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.*
+_Note: Formal evaluations coming soon. Any model can work, but these have shown the best results in practice._
 
 ## Start Coding
 
@@ -98,16 +97,16 @@ 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 |
+| 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
 
@@ -122,11 +121,12 @@ Multiple file reads, directory listings, and searches execute concurrently using
 - **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.*
+_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

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/README.md

@@ -33,7 +33,7 @@ Choose your AI provider and set your API key:
 # OpenAI
 tunacode --model "openai:gpt-4o" --key "sk-your-openai-key"
 
-# Anthropic Claude  
+# Anthropic Claude
 tunacode --model "anthropic:claude-3.5-sonnet" --key "sk-ant-your-anthropic-key"
 
 # OpenRouter (100+ models)
@@ -45,13 +45,14 @@ Your config is saved to `~/.config/tunacode.json` (edit directly with `nvim ~/.c
 ### 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.*
+_Note: Formal evaluations coming soon. Any model can work, but these have shown the best results in practice._
 
 ## Start Coding
 
@@ -61,16 +62,16 @@ 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 |
+| 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
 
@@ -85,11 +86,12 @@ Multiple file reads, directory listings, and searches execute concurrently using
 - **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.*
+_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
@@ -109,4 +111,4 @@ Multiple file reads, directory listings, and searches execute concurrently using
 
 ---
 
-MIT License - see [LICENSE](LICENSE) file
+MIT License - see [LICENSE](LICENSE) file

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "tunacode-cli"
-version = "0.0.36"
+version = "0.0.37"
 description = "Your agentic CLI developer."
 keywords = ["cli", "agent", "development", "automation"]
 readme = "README.md"
@@ -38,9 +38,7 @@ tunacode = "tunacode.cli.main:app"
 [project.optional-dependencies]
 dev = [
     "build",
-    "black", 
-    "flake8",
-    "isort",
+    "ruff",
     "pytest",
     "pytest-cov",
     "pytest-asyncio",
@@ -51,9 +49,27 @@ dev = [
 Homepage = "https://github.com/larock22/tunacode"
 Repository = "https://github.com/larock22/tunacode"
 
-[tool.black]
+[tool.ruff]
 line-length = 100
 
-[tool.isort]
-line_length = 100
-profile = "black"
+[tool.ruff.lint]
+extend-select = ["I"]
+ignore = ["E203", "E501"]
+exclude = [
+    ".bzr",
+    ".direnv",
+    ".eggs",
+    ".git",
+    ".hg",
+    ".mypy_cache",
+    ".nox",
+    ".pants.d",
+    ".ruff_cache",
+    ".svn",
+    ".tox",
+    ".venv",
+    "__pypackages__",
+    "build",
+    "dist",
+    "venv",
+]

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/setup.py

@@ -1,4 +1,4 @@
-from setuptools import setup, find_namespace_packages
+from setuptools import find_namespace_packages, setup
 
 setup(
     package_dir={"": "src"},

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/src/tunacode/cli/commands/implementations/__init__.py

@@ -12,13 +12,14 @@ from .debug import (
 )
 from .development import BranchCommand, InitCommand
 from .model import ModelCommand
-from .system import ClearCommand, HelpCommand, RefreshConfigCommand, UpdateCommand
+from .system import ClearCommand, HelpCommand, RefreshConfigCommand, StreamingCommand, UpdateCommand
 
 __all__ = [
     # System commands
     "HelpCommand",
     "ClearCommand",
     "RefreshConfigCommand",
+    "StreamingCommand",
     "UpdateCommand",
     # Debug commands
     "YoloCommand",

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/src/tunacode/cli/commands/implementations/system.py

@@ -175,3 +175,42 @@ class UpdateCommand(SimpleCommand):
             await ui.error(f"Update failed: {e}")
         except FileNotFoundError:
             await ui.error(f"Could not find {installation_method} executable")
+
+
+class StreamingCommand(SimpleCommand):
+    """Toggle streaming display on/off."""
+
+    spec = CommandSpec(
+        name="streaming",
+        aliases=["/streaming"],
+        description="Toggle streaming display on/off",
+        category=CommandCategory.SYSTEM,
+    )
+
+    async def execute(self, args: List[str], context: CommandContext) -> None:
+        current_setting = context.state_manager.session.user_config.get("settings", {}).get(
+            "enable_streaming", True
+        )
+
+        if args and args[0].lower() in ["on", "true", "1", "enable", "enabled"]:
+            new_setting = True
+        elif args and args[0].lower() in ["off", "false", "0", "disable", "disabled"]:
+            new_setting = False
+        else:
+            # Toggle current setting
+            new_setting = not current_setting
+
+        # Update the configuration
+        if "settings" not in context.state_manager.session.user_config:
+            context.state_manager.session.user_config["settings"] = {}
+        context.state_manager.session.user_config["settings"]["enable_streaming"] = new_setting
+
+        status = "enabled" if new_setting else "disabled"
+        await ui.success(f"Streaming display {status}")
+
+        if new_setting:
+            await ui.muted(
+                "Responses will be displayed progressively as they are generated (default)"
+            )
+        else:
+            await ui.muted("Responses will be displayed all at once after completion")

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/src/tunacode/cli/commands/registry.py

@@ -19,7 +19,13 @@ from .implementations.debug import (
 )
 from .implementations.development import BranchCommand, InitCommand
 from .implementations.model import ModelCommand
-from .implementations.system import ClearCommand, HelpCommand, RefreshConfigCommand, UpdateCommand
+from .implementations.system import (
+    ClearCommand,
+    HelpCommand,
+    RefreshConfigCommand,
+    StreamingCommand,
+    UpdateCommand,
+)
 
 
 @dataclass
@@ -106,6 +112,7 @@ class CommandRegistry:
             FixCommand,
             ParseToolsCommand,
             RefreshConfigCommand,
+            StreamingCommand,
             UpdateCommand,
             HelpCommand,
             BranchCommand,

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/src/tunacode/cli/repl.py

@@ -70,8 +70,9 @@ async def _tool_confirm(tool_call, node, state_manager: StateManager):
             await _tool_ui.log_mcp(title, args)
         return
 
-    # Stop spinner during user interaction
-    state_manager.session.spinner.stop()
+    # Stop spinner during user interaction (only if not streaming)
+    if not state_manager.session.is_streaming_active and state_manager.session.spinner:
+        state_manager.session.spinner.stop()
 
     # Create confirmation request
     request = tool_handler.create_confirmation_request(tool_call.tool_name, args)
@@ -84,7 +85,10 @@ async def _tool_confirm(tool_call, node, state_manager: StateManager):
         raise UserAbortError("User aborted.")
 
     await ui.line()  # Add line after user input
-    state_manager.session.spinner.start()
+
+    # Restart spinner (only if not streaming)
+    if not state_manager.session.is_streaming_active and state_manager.session.spinner:
+        state_manager.session.spinner.start()
 
 
 async def _tool_handler(part, node, state_manager: StateManager):
@@ -96,7 +100,19 @@ async def _tool_handler(part, node, state_manager: StateManager):
     if tool_handler.should_confirm(part.tool_name):
         await ui.info(f"Tool({part.tool_name})")
 
-    state_manager.session.spinner.stop()
+    # Stop spinner only if not streaming
+    if not state_manager.session.is_streaming_active and state_manager.session.spinner:
+        state_manager.session.spinner.stop()
+
+    # Track if we need to stop/restart streaming panel
+    streaming_panel = None
+    if state_manager.session.is_streaming_active and hasattr(
+        state_manager.session, "streaming_panel"
+    ):
+        streaming_panel = state_manager.session.streaming_panel
+        # Stop the streaming panel to prevent UI interference during confirmation
+        if streaming_panel and tool_handler.should_confirm(part.tool_name):
+            await streaming_panel.stop()
 
     try:
         args = _parse_args(part.args)
@@ -128,7 +144,13 @@ async def _tool_handler(part, node, state_manager: StateManager):
         patch_tool_messages("Operation aborted by user.", state_manager)
         raise
     finally:
-        state_manager.session.spinner.start()
+        # Restart streaming panel if it was stopped
+        if streaming_panel and tool_handler.should_confirm(part.tool_name):
+            await streaming_panel.start()
+
+        # Restart spinner only if not streaming
+        if not state_manager.session.is_streaming_active and state_manager.session.spinner:
+            state_manager.session.spinner.start()
 
 
 # Initialize command registry
@@ -195,38 +217,77 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
             await ui.error(str(e))
             return
 
-        # Use normal agent processing
-        res = await agent.process_request(
-            state_manager.session.current_model,
-            text,
-            state_manager,
-            tool_callback=tool_callback_with_state,
+        # Check if streaming is enabled (default: True for better UX)
+        enable_streaming = state_manager.session.user_config.get("settings", {}).get(
+            "enable_streaming", True
         )
+
+        if enable_streaming:
+            # Stop spinner before starting streaming display (Rich.Live conflict)
+            await ui.spinner(False, state_manager.session.spinner, state_manager)
+
+            # Mark that streaming is active to prevent spinner conflicts
+            state_manager.session.is_streaming_active = True
+
+            # Use streaming agent processing
+            streaming_panel = ui.StreamingAgentPanel()
+            await streaming_panel.start()
+
+            # Store streaming panel reference in session for tool handler access
+            state_manager.session.streaming_panel = streaming_panel
+
+            try:
+
+                async def streaming_callback(content: str):
+                    await streaming_panel.update(content)
+
+                res = await agent.process_request(
+                    state_manager.session.current_model,
+                    text,
+                    state_manager,
+                    tool_callback=tool_callback_with_state,
+                    streaming_callback=streaming_callback,
+                )
+            finally:
+                await streaming_panel.stop()
+                # Clear streaming panel reference
+                state_manager.session.streaming_panel = None
+                # Mark streaming as inactive
+                state_manager.session.is_streaming_active = False
+                # Don't restart spinner - it will be stopped in the outer finally block anyway
+        else:
+            # Use normal agent processing
+            res = await agent.process_request(
+                state_manager.session.current_model,
+                text,
+                state_manager,
+                tool_callback=tool_callback_with_state,
+            )
         if output:
             if state_manager.session.show_thoughts:
                 new_msgs = state_manager.session.messages[start_idx:]
                 for msg in new_msgs:
                     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"):
-                await ui.agent(res.result.output)
-                # Always show files in context after agent response
-                if state_manager.session.files_in_context:
-                    # Extract just filenames from full paths for readability
-                    filenames = [
-                        Path(f).name for f in sorted(state_manager.session.files_in_context)
-                    ]
-                    await ui.muted(f"\nFiles in context: {', '.join(filenames)}")
-            else:
-                # Fallback: show that the request was processed
-                await ui.muted("Request completed")
-                # Show files in context even for empty responses
-                if state_manager.session.files_in_context:
-                    filenames = [
-                        Path(f).name for f in sorted(state_manager.session.files_in_context)
-                    ]
-                    await ui.muted(f"Files in context: {', '.join(filenames)}")
+
+            # Only display result if not streaming (streaming already showed content)
+            if not enable_streaming:
+                # Check if result exists and has 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
+                    await ui.muted("Request completed")
+
+            # Always show files in context after agent response
+            if state_manager.session.files_in_context:
+                # Extract just filenames from full paths for readability
+                filenames = [Path(f).name for f in sorted(state_manager.session.files_in_context)]
+                await ui.muted(f"\nFiles in context: {', '.join(filenames)}")
     except CancelledError:
         await ui.muted("Request cancelled")
     except UserAbortError:

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/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.36"
+APP_VERSION = "0.0.37"
 
 # File patterns
 GUIDE_FILE_PATTERN = "{name}.md"

{tunacode_cli-0.0.36 → tunacode_cli-0.0.37}/src/tunacode/core/agents/main.py

@@ -12,6 +12,22 @@ from datetime import datetime, timezone
 from pathlib import Path
 from typing import Any, Iterator, List, Optional, Tuple
 
+from pydantic_ai import Agent
+
+# Import streaming types with fallback for older versions
+try:
+    from pydantic_ai.messages import (
+        PartDeltaEvent,
+        TextPartDelta,
+    )
+
+    STREAMING_AVAILABLE = True
+except ImportError:
+    # Fallback for older pydantic-ai versions
+    PartDeltaEvent = None
+    TextPartDelta = None
+    STREAMING_AVAILABLE = False
+
 from tunacode.constants import READ_ONLY_TOOLS
 from tunacode.core.state import StateManager
 from tunacode.services.mcp import get_mcp_servers
@@ -197,6 +213,7 @@ async def _process_node(
     tool_callback: Optional[ToolCallback],
     state_manager: StateManager,
     tool_buffer: Optional[ToolBuffer] = None,
+    streaming_callback: Optional[callable] = None,
 ):
     from tunacode.ui import console as ui
     from tunacode.utils.token_counter import estimate_tokens
@@ -216,6 +233,16 @@ async def _process_node(
     if hasattr(node, "model_response"):
         state_manager.session.messages.append(node.model_response)
 
+        # Stream content to callback if provided
+        # Use this as fallback when true token streaming is not available
+        if streaming_callback and not STREAMING_AVAILABLE:
+            for part in node.model_response.parts:
+                if hasattr(part, "content") and isinstance(part.content, str):
+                    content = part.content.strip()
+                    if content and not content.startswith('{"thought"'):
+                        # Stream non-JSON content (actual response content)
+                        await streaming_callback(content)
+
         # Enhanced display when thoughts are enabled
         if state_manager.session.show_thoughts:
             # Show raw API response data
@@ -683,6 +710,7 @@ async def process_request(
     message: str,
     state_manager: StateManager,
     tool_callback: Optional[ToolCallback] = None,
+    streaming_callback: Optional[callable] = None,
 ) -> AgentRun:
     agent = get_or_create_agent(model, state_manager)
     mh = state_manager.session.messages.copy()
@@ -723,7 +751,19 @@ async def process_request(
         i = 0
         async for node in agent_run:
             state_manager.session.current_iteration = i + 1
-            await _process_node(node, tool_callback, state_manager, tool_buffer)
+
+            # Handle token-level streaming for model request nodes
+            if streaming_callback and STREAMING_AVAILABLE and Agent.is_model_request_node(node):
+                async with node.stream(agent_run.ctx) as request_stream:
+                    async for event in request_stream:
+                        if isinstance(event, PartDeltaEvent) and isinstance(
+                            event.delta, TextPartDelta
+                        ):
+                            # Stream individual token deltas
+                            if event.delta.content_delta:
+                                await streaming_callback(event.delta.content_delta)
+
+            await _process_node(node, tool_callback, state_manager, tool_buffer, streaming_callback)
             if hasattr(node, "result") and node.result and hasattr(node.result, "output"):
                 if node.result.output:
                     response_state.has_user_response = True