alloy-runtime-cli 0.1.0__py3-none-any.whl

cli/tui/chat/renderers/plain.py

@@ -0,0 +1,83 @@
+"""Plain text renderer for chat messages.
+
+Renders messages without any Rich markup formatting.
+"""
+
+from datetime import datetime
+
+from alloy_runtime_types.dtos.sessions import MessageResponse
+
+from cli.tui.chat.renderers.base import get_message_body, get_message_body_for_display
+from cli.tui.chat.types import format_time
+
+
+class PlainTextRenderer:
+    """Renders messages as plain text without formatting.
+
+    Useful for:
+        - Exporting conversations
+        - Copying to clipboard
+        - Accessibility contexts
+        - Testing and debugging
+
+    Note: Long content is still truncated to prevent TUI issues.
+    The truncation indicator appears as plain text.
+    """
+
+    def render(
+        self,
+        messages: tuple[MessageResponse, ...],
+        streaming_content: str = "",
+        streaming_thinking_content: str = "",
+        pending_user_message: str | None = None,
+        assistant_name: str = "Assistant",
+        is_streaming: bool = False,
+        thinking_collapsed: bool = False,
+    ) -> str:
+        """Render messages to a plain text string.
+
+        Args:
+            messages: Tuple of message responses to render.
+            streaming_content: Partial content being streamed (appended to end).
+            streaming_thinking_content: Partial thinking/reasoning content (ignored in plain).
+            pending_user_message: User message shown optimistically before server confirms.
+            assistant_name: Name to display for assistant (agent name or model).
+            is_streaming: Whether currently streaming a response.
+            thinking_collapsed: Whether thinking blocks are collapsed (ignored in plain).
+
+        Returns:
+            Plain text string without Rich markup.
+        """
+        lines: list[str] = []
+
+        for msg in messages:
+            time_str = format_time(msg.created_at)
+
+            if msg.role == "user":
+                # Only truncate user messages (template injections are user-initiated)
+                body, _was_truncated = get_message_body_for_display(msg)
+                lines.append(f"You [{time_str}]:")
+            else:
+                # Never truncate assistant messages
+                body = get_message_body(msg)
+                lines.append(f"{assistant_name} [{time_str}]:")
+
+            lines.append(body)
+            lines.append("")
+
+        # Show pending user message
+        if pending_user_message:
+            now_str = datetime.now().strftime("%H:%M")
+            lines.append(f"You [{now_str}]:")
+            lines.append(pending_user_message)
+            lines.append("")
+
+        # Show streaming response
+        if streaming_content or is_streaming:
+            lines.append(f"{assistant_name}:")
+            if streaming_content:
+                lines.append(streaming_content)
+            else:
+                lines.append("...")
+
+        return "\n".join(lines)