From 1a0c72075de09311fc2f8ee5a700652770e6a4cf Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Thu, 19 Dec 2024 15:59:15 -0500 Subject: [PATCH] Stop using rules for output formatting. --- ra_aid/console/formatting.py | 9 +++------ ra_aid/tools/memory.py | 4 +--- ra_aid/tools/research.py | 12 ++++-------- 3 files changed, 8 insertions(+), 17 deletions(-) diff --git a/ra_aid/console/formatting.py b/ra_aid/console/formatting.py index 274bb31..5164f91 100644 --- a/ra_aid/console/formatting.py +++ b/ra_aid/console/formatting.py @@ -1,8 +1,6 @@ from rich.console import Console -from rich.rule import Rule from rich.panel import Panel from rich.markdown import Markdown -from rich.padding import Padding console = Console() @@ -31,10 +29,9 @@ def print_stage_header(stage: str) -> None: # Get appropriate icon with fallback icon = icons.get(stage_key, '🚀') - # Create styled rule with icon - rule_content = f"{icon} {stage_title}" - padded_rule = Padding(Rule(rule_content, style="green bold"), (1, 0)) - console.print(padded_rule) + # Create styled panel with icon + panel_content = f"{icon} {stage_title}" + console.print(Panel(panel_content, style="green bold", padding=0)) def print_task_header(task: str) -> None: """Print a task header with yellow styling and wrench emoji. Content is rendered as Markdown. diff --git a/ra_aid/tools/memory.py b/ra_aid/tools/memory.py index 81fea2d..82a24d3 100644 --- a/ra_aid/tools/memory.py +++ b/ra_aid/tools/memory.py @@ -3,7 +3,6 @@ from rich.console import Console from rich.markdown import Markdown from rich.panel import Panel from rich.rule import Rule -from rich.padding import Padding from langchain_core.tools import tool class SnippetInfo(TypedDict): @@ -185,8 +184,7 @@ def request_implementation() -> str: Empty string """ _global_memory['implementation_requested'] = True - padded_rule = Padding(Rule("🚀 Implementation Requested", style="yellow"), (1, 0)) - console.print(padded_rule) + console.print(Panel("🚀 Implementation Requested", style="yellow", padding=0)) return "" diff --git a/ra_aid/tools/research.py b/ra_aid/tools/research.py index c754f95..00aeaa9 100644 --- a/ra_aid/tools/research.py +++ b/ra_aid/tools/research.py @@ -1,7 +1,6 @@ from langchain_core.tools import tool from rich.console import Console -from rich.rule import Rule -from rich.padding import Padding +from rich.panel import Panel console = Console() @@ -10,8 +9,7 @@ def existing_project_detected() -> dict: """ When to call: Once you have confirmed that the current working directory contains project files. """ - padded_rule = Padding(Rule("📁 Existing Project Detected", style="bright_blue"), (1, 0)) - console.print(padded_rule) + console.print(Panel("📁 Existing Project Detected", style="bright_blue", padding=0)) return { 'hint': ( "You are working within an existing codebase that already has established patterns and standards. " @@ -30,8 +28,7 @@ def monorepo_detected() -> dict: """ When to call: After identifying that multiple packages or modules exist within a single repository. """ - padded_rule = Padding(Rule("📦 Monorepo Detected", style="bright_blue"), (1, 0)) - console.print(padded_rule) + console.print(Panel("📦 Monorepo Detected", style="bright_blue", padding=0)) return { 'hint': ( "You are researching in a monorepo environment that manages multiple packages or services under one roof. " @@ -53,8 +50,7 @@ def ui_detected() -> dict: """ When to call: After detecting that the project contains a user interface layer or front-end component. """ - padded_rule = Padding(Rule("🎯 UI Detected", style="bright_blue"), (1, 0)) - console.print(padded_rule) + console.print(Panel("🎯 UI Detected", style="bright_blue", padding=0)) return { 'hint': ( "You are working with a user interface component where established UI conventions, styles, and frameworks are likely in place. "