Function svg_node

experiments/human_eval/draw_llm_token_cost_tree.py:169–177

(x: float, y: float, text: str, font_size: int, color: tuple[int, int, int])

def svg_node(x: float, y: float, text: str, font_size: int, color: tuple[int, int, int]) -> list[str]:
    # Fixed-size nodes are sufficient here because all labels are known.
    widths = {"TokenCost": 150, "Closed API": 160, "Open / open-access": 245}
    w = widths[text]
    h = 44
    return [
        f'<rect x="{x - w / 2}" y="{y - h / 2}" width="{w}" height="{h}" rx="8" fill="white" stroke="{rgb(color)}" stroke-width="2"/>',
        f'<text x="{x}" y="{y + 8}" font-family="Times New Roman" font-size="{font_size}" font-weight="700" text-anchor="middle" fill="{rgb(color)}">{text}</text>',
    ]


def write_svg() -> None: