Function estimate_output_budget

uncommon_route/router/structural.py:338–356

Estimate output token budget from structural signals.

(prompt: str, tier: str)

def estimate_output_budget(prompt: str, tier: str) -> int:
    """Estimate output token budget from structural signals."""
    tokens = estimate_tokens(prompt)
    if tokens < 10:
        return OutputBudget.SHORT

    code = score_code_markers(prompt)
    enum = score_enumeration_density(prompt)
    if code.score > 0.3 and enum.score > 0.3:
        return OutputBudget.LONG
    if tokens > 200:
        return OutputBudget.LONG

    tier_defaults = {
        "SIMPLE": OutputBudget.SHORT,
        "MEDIUM": OutputBudget.MEDIUM,
        "COMPLEX": OutputBudget.LONG,
    }
    return tier_defaults.get(tier, OutputBudget.MEDIUM)