Use Rich padding.

This commit is contained in:
AI Christianson 2024-12-19 15:05:00 -05:00
parent 22aeb2812d
commit 5029106bdd
3 changed files with 13 additions and 15 deletions

View File

@ -2,6 +2,7 @@ from rich.console import Console
from rich.rule import Rule from rich.rule import Rule
from rich.panel import Panel from rich.panel import Panel
from rich.markdown import Markdown from rich.markdown import Markdown
from rich.padding import Padding
console = Console() console = Console()
@ -32,9 +33,8 @@ def print_stage_header(stage: str) -> None:
# Create styled rule with icon # Create styled rule with icon
rule_content = f"{icon} {stage_title}" rule_content = f"{icon} {stage_title}"
console.print() padded_rule = Padding(Rule(rule_content, style="green bold"), (1, 0))
console.print(Rule(rule_content, style="green bold")) console.print(padded_rule)
console.print()
def print_task_header(task: str) -> None: def print_task_header(task: str) -> None:
"""Print a task header with yellow styling and wrench emoji. Content is rendered as Markdown. """Print a task header with yellow styling and wrench emoji. Content is rendered as Markdown.

View File

@ -3,6 +3,7 @@ from rich.console import Console
from rich.markdown import Markdown from rich.markdown import Markdown
from rich.panel import Panel from rich.panel import Panel
from rich.rule import Rule from rich.rule import Rule
from rich.padding import Padding
from langchain_core.tools import tool from langchain_core.tools import tool
class SnippetInfo(TypedDict): class SnippetInfo(TypedDict):
@ -184,9 +185,8 @@ def request_implementation() -> str:
Empty string Empty string
""" """
_global_memory['implementation_requested'] = True _global_memory['implementation_requested'] = True
console.print() padded_rule = Padding(Rule("🚀 Implementation Requested", style="yellow"), (1, 0))
console.print(Rule("🚀 Implementation Requested", style="yellow")) console.print(padded_rule)
console.print()
return "" return ""

View File

@ -1,6 +1,7 @@
from langchain_core.tools import tool from langchain_core.tools import tool
from rich.console import Console from rich.console import Console
from rich.rule import Rule from rich.rule import Rule
from rich.padding import Padding
console = Console() console = Console()
@ -9,9 +10,8 @@ def existing_project_detected() -> dict:
""" """
When to call: Once you have confirmed that the current working directory contains project files. When to call: Once you have confirmed that the current working directory contains project files.
""" """
print() padded_rule = Padding(Rule("📁 Existing Project Detected", style="bright_blue"), (1, 0))
console.print(Rule("📁 Existing Project Detected", style="bright_blue")) console.print(padded_rule)
print()
return { return {
'hint': ( 'hint': (
"You are working within an existing codebase that already has established patterns and standards. " "You are working within an existing codebase that already has established patterns and standards. "
@ -30,9 +30,8 @@ def monorepo_detected() -> dict:
""" """
When to call: After identifying that multiple packages or modules exist within a single repository. When to call: After identifying that multiple packages or modules exist within a single repository.
""" """
print() padded_rule = Padding(Rule("📦 Monorepo Detected", style="bright_blue"), (1, 0))
console.print(Rule("📦 Monorepo Detected", style="bright_blue")) console.print(padded_rule)
print()
return { return {
'hint': ( 'hint': (
"You are researching in a monorepo environment that manages multiple packages or services under one roof. " "You are researching in a monorepo environment that manages multiple packages or services under one roof. "
@ -54,9 +53,8 @@ def ui_detected() -> dict:
""" """
When to call: After detecting that the project contains a user interface layer or front-end component. When to call: After detecting that the project contains a user interface layer or front-end component.
""" """
print() padded_rule = Padding(Rule("🎯 UI Detected", style="bright_blue"), (1, 0))
console.print(Rule("🎯 UI Detected", style="bright_blue")) console.print(padded_rule)
print()
return { return {
'hint': ( 'hint': (
"You are working with a user interface component where established UI conventions, styles, and frameworks are likely in place. " "You are working with a user interface component where established UI conventions, styles, and frameworks are likely in place. "