wehub-resource-sync
57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
"""Shared span-type glyphs/tags/colors and pass/fail pill styles.
|
|
|
|
Both panes import from here so jumping between the tree and the details
|
|
view keeps the same visual identity for each span type.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Tuple
|
|
|
|
from rich.text import Text
|
|
|
|
from deepeval.inspect.types import Trace, TraceOrSpan
|
|
|
|
|
|
# `(glyph, tag, rich style)` per span type. Tags are full words rather
|
|
# than abbreviations because the tree pane is wide enough to spell them
|
|
# out, and "RETRIEVER" reads instantly while "RET" trips users into
|
|
# mentally expanding it.
|
|
#
|
|
# Explicit hex values rather than named ANSI colors: named colors get
|
|
# theme-remapped and `dim` collapses to invisible on some palettes;
|
|
# truecolor hex survives every theme and degrades to the nearest 256-
|
|
# color match on older terminals.
|
|
TYPE_STYLE: dict[str, Tuple[str, str, str]] = {
|
|
"trace": ("◆", "TRACE", "bold #8be9fd"), # cyan
|
|
"base": ("▪", "BASE", "#a8a8a8"), # mid-gray
|
|
"agent": ("◉", "AGENT", "bold #ff79c6"), # pink
|
|
"llm": ("✦", "LLM", "bold #f1fa8c"), # yellow
|
|
"retriever": ("⤓", "RETRIEVER", "bold #bd93f9"), # purple
|
|
"tool": ("⚒", "TOOL", "bold #50fa7b"), # green
|
|
}
|
|
|
|
|
|
def type_style(node: TraceOrSpan) -> Tuple[str, str, str]:
|
|
if isinstance(node, Trace):
|
|
return TYPE_STYLE["trace"]
|
|
return TYPE_STYLE.get(node.type, TYPE_STYLE["base"])
|
|
|
|
|
|
def type_prefix(node: TraceOrSpan) -> Text:
|
|
"""`◆ TRACE ` styled, ready to append into a Rich `Text` row."""
|
|
|
|
glyph, tag, style = type_style(node)
|
|
text = Text()
|
|
text.append(f"{glyph} ", style=style)
|
|
text.append(tag, style=style)
|
|
text.append(" ")
|
|
return text
|
|
|
|
|
|
# Pill foregrounds picked for luminance contrast: dark text on high-
|
|
# luminance backgrounds (green/yellow), light text on mid-luminance (red).
|
|
PILL_PASS = "bold #1a1a2e on #50fa7b"
|
|
PILL_FAIL = "bold #f8f8f2 on #ff5555"
|
|
PILL_WARN = "bold #1a1a2e on #f1fa8c"
|