code-puppy 0.0.287__py3-none-any.whl → 0.0.323__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

@@ -227,7 +227,7 @@ class MessageQueue:
 
             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)
@@ -322,7 +322,7 @@ 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"""
     # TUI mode has been removed, always emit dividers
     emit_message(MessageType.DIVIDER, content, **metadata)
@@ -335,7 +335,7 @@ def emit_prompt(prompt_text: str, timeout: float = None) -> str:
         # Emit the prompt as a message for display
         from code_puppy.messaging import emit_info
 
-        emit_info(f"[yellow]{prompt_text}[/yellow]")
+        emit_info(prompt_text)
 
         # Get input directly
         try: