model-router

A single interface for sending text, vision, image, audio, and video queries to many model providers (Anthropic, OpenAI, Gemini, Together, Runware, ElevenLabs, Tripo), plus local models (Qwen2.5-VL, SAM 3).

Install

uv add antron-model-router
# with local models
uv add "antron-model-router[local]"   # or [qwen] / [sam]

The import name is model_router.

Usage

from model_router.router import ModelRouter
from model_router.model_interface import Capability
from model_router.query import Query

router = ModelRouter()
response = router.get_response(
    query=Query(query_text="Hello!"),
    capability=Capability.TEXT,
    interface_type="anthropic_sonnet45",  # or None to let the router choose
)

See src/model_router/demo.py (python -m model_router.demo) for an interactive tour of every interface.

API keys

Keys are read from environment variables when an interface is first used:

Mesh generation

Capability.MESH_GEN routes to Tripo H3.1 (tripo_h31) by default. Pass a MeshGenQuery with text only (text-to-mesh), one image (image-to-mesh), or front/left/back/right views (multiview):

from model_router.model_interface import Capability, MeshGenQuery

response = router.get_response(
    query=MeshGenQuery(images={"front": "front.png", "right": "right.png"}, texture=True, pbr=True),
    capability=Capability.MESH_GEN,
    interface_type=None,
)
glb_url = response["output"]["model_url"]

Local models

SAM 3 looks for sam3.pt in $MODEL_ROUTER_PRETRAINED_DIR/sam3 (default ./pretrained_models/sam3), otherwise it downloads from HuggingFace (facebook/sam3, gated — run hf auth login).

Development

uv sync
uv run pytest

License

MIT