Function infer_capabilities

uncommon_route/model_map.py:238–283

Infer model capabilities from its name and pricing. Heuristic accuracy ~85%. The bandit/feedback system corrects errors over time via runtime observations.

(
    model_id: str,
    pricing: ModelPricing,
    *,
    has_explicit_pricing: bool = True,
)

def infer_capabilities(
    model_id: str,
    pricing: ModelPricing,
    *,
    has_explicit_pricing: bool = True,
) -> ModelCapabilities:
    """Infer model capabilities from its name and pricing.

    Heuristic accuracy ~85%.  The bandit/feedback system corrects errors
    over time via runtime observations.
    """
    name = model_id.lower()
    core = name.split("/", 1)[-1] if "/" in name else name
    provider = name.split("/", 1)[0] if "/" in name else ""

    reasoning = False
    if any(p in core for p in _REASONING_POSITIVE):
        if not any(p in core for p in _REASONING_NEGATIVE):
            reasoning = True
    if provider == "anthropic" and "opus" in core:
        reasoning = True

    image_generation = not is_routable_chat_model(model_id) and "image" in core

    vision = False
    if any(p in core for p in ("-vl", "vision", "image")):
        vision = True
    if provider in ("anthropic", "google"):
        vision = True
    if provider == "openai" and any(p in core for p in ("4o", "gpt-5", "gpt-4")):
        vision = True

    tool_calling = not image_generation

    free = (
        has_explicit_pricing
        and pricing.input_price <= 0.0
        and pricing.output_price <= 0.0
    )

    return ModelCapabilities(
        tool_calling=tool_calling,
        vision=vision,
        reasoning=reasoning,
        free=free,
    )


# ---------------------------------------------------------------------------