tunacode-cli 0.0.76.1__py3-none-any.whl → 0.0.76.2__py3-none-any.whl

tunacode/cli/commands/implementations/debug.py

@@ -122,7 +122,7 @@ class FixCommand(SimpleCommand):
     )
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
-        from tunacode.core.agents.main import patch_tool_messages
+        from tunacode.core.agents import patch_tool_messages
 
         # Count current messages
         before_count = len(context.state_manager.session.messages)
@@ -152,7 +152,7 @@ class ParseToolsCommand(SimpleCommand):
     )
 
     async def execute(self, args: List[str], context: CommandContext) -> None:
-        from tunacode.core.agents.main import extract_and_execute_tool_calls
+        from tunacode.core.agents import extract_and_execute_tool_calls
 
         # Find the last model response in messages
         messages = context.state_manager.session.messages

tunacode/cli/commands/implementations/system.py

@@ -40,7 +40,7 @@ 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
+        from tunacode.core.agents import patch_tool_messages
 
         patch_tool_messages("Conversation cleared", context.state_manager)
 

tunacode/cli/repl.py

@@ -14,8 +14,8 @@ from pydantic_ai.exceptions import UnexpectedModelBehavior
 
 from tunacode.configuration.models import ModelRegistry
 from tunacode.constants import DEFAULT_CONTEXT_WINDOW
-from tunacode.core.agents import main as agent
-from tunacode.core.agents.main import patch_tool_messages
+from tunacode.core import agents as agent
+from tunacode.core.agents import patch_tool_messages
 from tunacode.core.token_usage.api_response_parser import ApiResponseParser
 from tunacode.core.token_usage.cost_calculator import CostCalculator
 from tunacode.core.token_usage.usage_tracker import UsageTracker
@@ -238,7 +238,7 @@ async def _handle_plan_approval(state_manager, original_request=None):
                 action()
                 if key == "a" and original_request:
                     await ui.info("🚀 Executing implementation...")
-                    await process_request(
+                    await execute_repl_request(
                         _transform_to_implementation_request(original_request),
                         state_manager,
                         output=True,
@@ -259,16 +259,16 @@ _command_registry.register_all_default_commands()
 
 async def _handle_command(command: str, state_manager: StateManager) -> CommandResult:
     """Handles a command string using the command registry."""
-    context = CommandContext(state_manager=state_manager, process_request=process_request)
+    context = CommandContext(state_manager=state_manager, process_request=execute_repl_request)
     try:
-        _command_registry.set_process_request_callback(process_request)
+        _command_registry.set_process_request_callback(execute_repl_request)
         return await _command_registry.execute(command, context)
     except ValidationError as e:
         await ui.error(str(e))
         return None
 
 
-async def process_request(text: str, state_manager: StateManager, output: bool = True):
+async def execute_repl_request(text: str, state_manager: StateManager, output: bool = True):
     """Process input using the agent, handling cancellation safely."""
     import uuid
 
@@ -412,6 +412,10 @@ async def process_request(text: str, state_manager: StateManager, output: bool =
             )
 
 
+# Backwards compatibility: exported name expected by external integrations/tests
+process_request = execute_repl_request
+
+
 async def warm_code_index():
     """Pre-warm the code index in background for faster directory operations."""
     try:
@@ -533,7 +537,7 @@ async def repl(state_manager: StateManager):
                 state_manager.session.operation_cancelled = False
 
             state_manager.session.current_task = get_app().create_background_task(
-                process_request(line, state_manager)
+                execute_repl_request(line, state_manager)
             )
             await state_manager.session.current_task
 

tunacode/cli/repl_components/error_recovery.py

@@ -6,6 +6,7 @@ Error recovery utilities for the REPL.
 
 import logging
 
+import tunacode.core.agents as agent_api
 from tunacode.types import StateManager
 from tunacode.ui import console as ui
 
@@ -130,13 +131,12 @@ async def attempt_tool_recovery(e: Exception, state_manager: StateManager) -> bo
         )
 
         try:
-            from tunacode.core.agents.main import extract_and_execute_tool_calls
 
             def tool_callback_with_state(tool_part, _node):
                 return tool_handler(tool_part, state_manager)
 
             # This function now returns the number of tools found
-            tools_found = await extract_and_execute_tool_calls(
+            tools_found = await agent_api.extract_and_execute_tool_calls(
                 content_to_parse, tool_callback_with_state, state_manager
             )
 

tunacode/cli/repl_components/tool_executor.py

@@ -9,7 +9,7 @@ from asyncio.exceptions import CancelledError
 
 from prompt_toolkit.application import run_in_terminal
 
-from tunacode.core.agents.main import patch_tool_messages
+from tunacode.core.agents import patch_tool_messages
 from tunacode.core.tool_handler import ToolHandler
 from tunacode.exceptions import UserAbortError
 from tunacode.types import StateManager

tunacode/constants.py

@@ -9,7 +9,7 @@ from enum import Enum
 
 # Application info
 APP_NAME = "TunaCode"
-APP_VERSION = "0.0.76.1"
+APP_VERSION = "0.0.76.2"
 
 
 # File patterns
@@ -44,6 +44,7 @@ class ToolName(str, Enum):
     LIST_DIR = "list_dir"
     GLOB = "glob"
     TODO = "todo"
+    REACT = "react"
     EXIT_PLAN_MODE = "exit_plan_mode"
 
 
@@ -57,6 +58,7 @@ TOOL_GREP = ToolName.GREP
 TOOL_LIST_DIR = ToolName.LIST_DIR
 TOOL_GLOB = ToolName.GLOB
 TOOL_TODO = ToolName.TODO
+TOOL_REACT = ToolName.REACT
 TOOL_EXIT_PLAN_MODE = ToolName.EXIT_PLAN_MODE
 
 # Tool categorization
@@ -65,6 +67,7 @@ READ_ONLY_TOOLS = [
     ToolName.GREP,
     ToolName.LIST_DIR,
     ToolName.GLOB,
+    ToolName.REACT,
     ToolName.EXIT_PLAN_MODE,
 ]
 WRITE_TOOLS = [ToolName.WRITE_FILE, ToolName.UPDATE_FILE]

tunacode/core/agents/__init__.py

@@ -1,8 +1,45 @@
-"""Agent helper modules."""
+"""Public entry points for TunaCode agent orchestration."""
 
-from .main import get_or_create_agent, process_request
+from . import main as main
+from .agent_components import (
+    AgentRunWithState,
+    AgentRunWrapper,
+    ResponseState,
+    SimpleResult,
+    ToolBuffer,
+    _process_node,
+    check_task_completion,
+    execute_tools_parallel,
+    extract_and_execute_tool_calls,
+    get_model_messages,
+    get_or_create_agent,
+    parse_json_tool_calls,
+    patch_tool_messages,
+)
+from .main import (
+    check_query_satisfaction,
+    get_agent_tool,
+    get_mcp_servers,
+    process_request,
+)
 
 __all__ = [
     "process_request",
     "get_or_create_agent",
+    "extract_and_execute_tool_calls",
+    "parse_json_tool_calls",
+    "get_model_messages",
+    "patch_tool_messages",
+    "_process_node",
+    "ResponseState",
+    "SimpleResult",
+    "AgentRunWrapper",
+    "AgentRunWithState",
+    "ToolBuffer",
+    "check_task_completion",
+    "execute_tools_parallel",
+    "get_mcp_servers",
+    "check_query_satisfaction",
+    "get_agent_tool",
+    "main",
 ]

tunacode/core/agents/agent_components/__init__.py

@@ -1,5 +1,7 @@
 """Agent components package for modular agent functionality."""
 
+from tunacode.ui.tool_descriptions import get_batch_description
+
 from .agent_config import get_or_create_agent
 from .agent_helpers import (
     create_empty_response_message,
@@ -17,6 +19,7 @@ from .message_handler import get_model_messages, patch_tool_messages
 from .node_processor import _process_node
 from .response_state import ResponseState
 from .result_wrapper import AgentRunWithState, AgentRunWrapper, SimpleResult
+from .streaming import stream_model_request_node
 from .task_completion import check_task_completion
 from .tool_buffer import ToolBuffer
 from .tool_executor import execute_tools_parallel
@@ -44,4 +47,6 @@ __all__ = [
     "get_tool_description",
     "get_tool_summary",
     "get_user_prompt_part_class",
+    "stream_model_request_node",
+    "get_batch_description",
 ]

tunacode/core/agents/agent_components/task_completion.py

@@ -3,6 +3,11 @@
 import re
 from typing import Tuple
 
+_COMPLETION_MARKERS = (
+    re.compile(r"^\s*TUNACODE\s+DONE:\s*", re.IGNORECASE),
+    re.compile(r"^\s*TUNACODE[_\s]+TASK_COMPLETE\s*:?[\s]*", re.IGNORECASE),
+)
+
 
 def check_task_completion(content: str) -> Tuple[bool, str]:
     """
@@ -21,12 +26,16 @@ def check_task_completion(content: str) -> Tuple[bool, str]:
 
     lines = content.split("\n")
 
-    # New marker: any line starting with "TUNACODE DONE:" (case-insensitive, allow leading whitespace)
-    done_pattern = re.compile(r"^\s*TUNACODE\s+DONE:\s*", re.IGNORECASE)
     for idx, line in enumerate(lines):
-        if done_pattern.match(line):
-            # Remove the marker line and return remaining content
-            cleaned = "\n".join(lines[:idx] + lines[idx + 1 :]).strip()
-            return True, cleaned
+        for pattern in _COMPLETION_MARKERS:
+            match = pattern.match(line)
+            if match:
+                remainder = line[match.end() :].strip()
+                cleaned_lines = lines[:idx]
+                if remainder:
+                    cleaned_lines.append(remainder)
+                cleaned_lines.extend(lines[idx + 1 :])
+                cleaned = "\n".join(cleaned_lines).strip()
+                return True, cleaned
 
     return False, content