customize-response-style

Customize OhMyCode's Response Style

Change how OhMyCode renders output in the terminal — colors, formatting, tool call display, token usage display.

When to Use

• User says "change the colors", "make output cleaner", "customize the theme"
• User wants to modify how tool calls, errors, or streaming text are displayed
• User wants to add/remove elements from the output (e.g., hide token counts)

Where Rendering Happens

All rendering logic is in ohmycode/cli.py → render_stream() function. It consumes events from the conversation loop and prints them using rich.console.Console.

Current rendering:

Event	Current Style
TextChunk	Plain text, streamed character by character
ToolCallStart	▶ <tool_name> in cyan + params in dim
ToolCallResult (success)	✓ + green output (truncated to 500 chars)
ToolCallResult (error)	✗ + red error message
TurnComplete	Tokens: prompt=X completion=Y total=Z in dim

How to Customize

Option 1: Edit render_stream() Directly

Open ohmycode/cli.py and find async def render_stream(). Modify the rich markup:

# Change tool call header color from cyan to blue
console.print(f"\n[blue]▶ {event.tool_name}[/blue]", highlight=False)

# Add a box around tool results
from rich.panel import Panel
console.print(Panel(output, title="Tool Result", border_style="green"))

# Hide token usage
# Comment out or delete the TurnComplete rendering block

Option 2: Add Rich Themes

Create a custom theme at the top of cli.py:

from rich.theme import Theme

ohmycode_theme = Theme({
    "tool.name": "bold magenta",
    "tool.success": "green",
    "tool.error": "bold red",
    "tool.params": "dim italic",
    "status": "dim cyan",
    "prompt": "bold yellow",
})

console = Console(theme=ohmycode_theme)

Then use style names in print calls:

console.print(f"▶ {event.tool_name}", style="tool.name")

Option 3: Add Markdown Rendering for AI Responses