code-puppy 0.0.169__py3-none-any.whl → 0.0.366__py3-none-any.whl

code_puppy/messaging/commands.py

@@ -0,0 +1,167 @@
+"""Command models for User → Agent communication in Code Puppy's messaging system.
+
+This module defines Pydantic models for commands that flow FROM the UI TO the Agent.
+This is the opposite direction of messages.py (which flows Agent → UI).
+
+Commands are used for:
+- Controlling agent execution (cancel, interrupt)
+- Responding to agent requests for user input
+- Providing confirmations and selections
+
+The UI layer creates these commands and sends them to the agent/runtime.
+The agent processes them and may emit messages in response.
+
+    ┌─────────┐   Commands    ┌─────────┐
+    │   UI    │ ────────────> │  Agent  │
+    │ (User)  │               │         │
+    │         │ <──────────── │         │
+    └─────────┘   Messages    └─────────┘
+
+NO Rich markup or formatting should be embedded in any string fields.
+"""
+
+from datetime import datetime, timezone
+from typing import Optional, Union
+from uuid import uuid4
+
+from pydantic import BaseModel, Field
+
+# =============================================================================
+# Base Command
+# =============================================================================
+
+
+class BaseCommand(BaseModel):
+    """Base class for all commands with auto-generated id and timestamp."""
+
+    id: str = Field(
+        default_factory=lambda: str(uuid4()),
+        description="Unique identifier for this command instance",
+    )
+    timestamp: datetime = Field(
+        default_factory=lambda: datetime.now(timezone.utc),
+        description="When this command was created (UTC)",
+    )
+
+    model_config = {"frozen": False, "extra": "forbid"}
+
+
+# =============================================================================
+# Agent Control Commands
+# =============================================================================
+
+
+class CancelAgentCommand(BaseCommand):
+    """Signals the agent to stop current execution gracefully.
+
+    The agent should finish any in-progress atomic operation, clean up,
+    and return control to the user. This is a soft cancellation.
+    """
+
+    reason: Optional[str] = Field(
+        default=None,
+        description="Optional reason for cancellation (for logging/debugging)",
+    )
+
+
+class InterruptShellCommand(BaseCommand):
+    """Signals to interrupt a currently running shell command.
+
+    This is equivalent to pressing Ctrl+C in a terminal. The shell process
+    should receive SIGINT and terminate. Use this when a command is taking
+    too long or producing unwanted output.
+    """
+
+    command_id: Optional[str] = Field(
+        default=None,
+        description="ID of the specific shell command to interrupt (None = current)",
+    )
+
+
+# =============================================================================
+# User Interaction Responses
+# =============================================================================
+
+
+class UserInputResponse(BaseCommand):
+    """Response to a UserInputRequest from the agent.
+
+    The prompt_id must match the prompt_id from the original UserInputRequest
+    so the agent can correlate the response with the request.
+    """
+
+    prompt_id: str = Field(
+        description="ID of the prompt this responds to (must match request)"
+    )
+    value: str = Field(description="The user's input value")
+
+
+class ConfirmationResponse(BaseCommand):
+    """Response to a ConfirmationRequest from the agent.
+
+    The user can confirm or deny, and optionally provide feedback text
+    if the original request had allow_feedback=True.
+    """
+
+    prompt_id: str = Field(
+        description="ID of the prompt this responds to (must match request)"
+    )
+    confirmed: bool = Field(
+        description="Whether the user confirmed (True) or denied (False)"
+    )
+    feedback: Optional[str] = Field(
+        default=None,
+        description="Optional feedback text from the user",
+    )
+
+
+class SelectionResponse(BaseCommand):
+    """Response to a SelectionRequest from the agent.
+
+    Contains both the index and the value for convenience and validation.
+    The agent can verify that selected_value matches options[selected_index].
+    """
+
+    prompt_id: str = Field(
+        description="ID of the prompt this responds to (must match request)"
+    )
+    selected_index: int = Field(
+        ge=0,
+        description="Zero-based index of the selected option",
+    )
+    selected_value: str = Field(description="The value of the selected option")
+
+
+# =============================================================================
+# Union Type for Type Checking
+# =============================================================================
+
+
+# All concrete command types (excludes BaseCommand itself)
+AnyCommand = Union[
+    CancelAgentCommand,
+    InterruptShellCommand,
+    UserInputResponse,
+    ConfirmationResponse,
+    SelectionResponse,
+]
+"""Union of all command types for type checking."""
+
+
+# =============================================================================
+# Export all public symbols
+# =============================================================================
+
+__all__ = [
+    # Base
+    "BaseCommand",
+    # Agent control
+    "CancelAgentCommand",
+    "InterruptShellCommand",
+    # User interaction responses
+    "UserInputResponse",
+    "ConfirmationResponse",
+    "SelectionResponse",
+    # Union type
+    "AnyCommand",
+]

code_puppy/messaging/markdown_patches.py

@@ -0,0 +1,57 @@
+"""Patches for Rich's Markdown rendering.
+
+This module provides customizations to Rich's default Markdown rendering,
+particularly for header justification which is hardcoded to center in Rich.
+"""
+
+from rich import box
+from rich.markdown import Heading, Markdown
+from rich.panel import Panel
+from rich.text import Text
+
+
+class LeftJustifiedHeading(Heading):
+    """A heading that left-justifies text instead of centering.
+
+    Rich's default Heading class hardcodes `text.justify = 'center'`,
+    which can look odd in a CLI context. This subclass overrides that
+    to use left justification instead.
+    """
+
+    def __rich_console__(self, console, options):
+        """Render the heading with left justification."""
+        text = self.text
+        text.justify = "left"  # Override Rich's default 'center'
+
+        if self.tag == "h1":
+            # Draw a border around h1s (same as Rich default)
+            yield Panel(
+                text,
+                box=box.HEAVY,
+                style="markdown.h1.border",
+            )
+        else:
+            # Styled text for h2 and beyond (same as Rich default)
+            if self.tag == "h2":
+                yield Text("")
+            yield text
+
+
+_patched = False
+
+
+def patch_markdown_headings():
+    """Patch Rich's Markdown to use left-justified headings.
+
+    This function is idempotent - calling it multiple times has no effect
+    after the first call.
+    """
+    global _patched
+    if _patched:
+        return
+
+    Markdown.elements["heading_open"] = LeftJustifiedHeading
+    _patched = True
+
+
+__all__ = ["patch_markdown_headings", "LeftJustifiedHeading"]

code_puppy/messaging/message_queue.py

@@ -1,8 +1,7 @@
 """
 Message queue system for decoupling Rich console output from renderers.
 
-This allows both TUI and interactive modes to consume the same messages
-but render them differently based on their capabilities.
+This allows interactive mode to consume messages and render them appropriately.
 """
 
 import asyncio
@@ -218,35 +217,23 @@ class MessageQueue:
 
         start_time = time.time()
 
-        # Check if we're in TUI mode - if so, try to yield control to the event loop
-        from code_puppy.state_management import is_tui_mode
-
-        sleep_interval = 0.05 if is_tui_mode() else 0.1
-
-        # Debug logging for TUI mode
-        if is_tui_mode():
-            print(f"[DEBUG] Waiting for prompt response: {prompt_id}")
+        # TUI mode has been removed, use standard sleep interval
+        sleep_interval = 0.1
 
         while True:
             if prompt_id in self._prompt_responses:
                 response = self._prompt_responses.pop(prompt_id)
-                if is_tui_mode():
-                    print(f"[DEBUG] Got response for {prompt_id}: {response[:20]}...")
                 return response
 
             if timeout and (time.time() - start_time) > timeout:
                 raise TimeoutError(
-                    f"No response received for prompt {prompt_id} within {timeout} seconds"
+                    f"No response for prompt {prompt_id} within {timeout}s"
                 )
 
             time.sleep(sleep_interval)
 
     def provide_prompt_response(self, prompt_id: str, response: str):
         """Provide a response to a human input request."""
-        from code_puppy.state_management import is_tui_mode
-
-        if is_tui_mode():
-            print(f"[DEBUG] Providing response for {prompt_id}: {response[:20]}...")
         self._prompt_responses[prompt_id] = response
 
 
@@ -335,44 +322,26 @@ def emit_system_message(content: Any, **metadata):
     emit_message(MessageType.SYSTEM, content, **metadata)
 
 
-def emit_divider(content: str = "[dim]" + "─" * 100 + "\n" + "[/dim]", **metadata):
+def emit_divider(content: str = "─" * 100 + "\n", **metadata):
     """Emit a divider line"""
-    from code_puppy.state_management import is_tui_mode
-
-    if not is_tui_mode():
-        emit_message(MessageType.DIVIDER, content, **metadata)
-    else:
-        pass
+    # TUI mode has been removed, always emit dividers
+    emit_message(MessageType.DIVIDER, content, **metadata)
 
 
 def emit_prompt(prompt_text: str, timeout: float = None) -> str:
-    """Emit a human input request and wait for response."""
-    from code_puppy.state_management import is_tui_mode
+    """Emit a human input request and wait for response.
 
-    # In interactive mode, use direct input instead of the queue system
-    if not is_tui_mode():
-        # Emit the prompt as a message for display
-        from code_puppy.messaging import emit_info
+    Uses safe_input for cross-platform compatibility, especially on Windows
+    where raw input() can fail after prompt_toolkit Applications.
+    """
+    from code_puppy.command_line.utils import safe_input
+    from code_puppy.messaging import emit_info
 
-        emit_info(f"[yellow]{prompt_text}[/yellow]")
+    emit_info(prompt_text)
 
-        # Get input directly
-        try:
-            # Try to use rich console for better formatting
-            from rich.console import Console
-
-            console = Console()
-            response = console.input("[cyan]>>> [/cyan]")
-            return response
-        except Exception:
-            # Fallback to basic input
-            response = input(">>> ")
-            return response
-
-    # In TUI mode, use the queue system
-    queue = get_global_queue()
-    prompt_id = queue.create_prompt_request(prompt_text)
-    return queue.wait_for_prompt_response(prompt_id, timeout)
+    # Use safe_input which resets Windows console state before reading
+    response = safe_input(">>> ")
+    return response
 
 
 def provide_prompt_response(prompt_id: str, response: str):