Skip to main content

Intelligent LLM routing with weighted least-outstanding selection, 429 cooldown, and request class isolation

Project description

llm-route

Intelligent LLM routing library for Python. Drop-in replacement for direct Azure OpenAI calls with automatic load balancing, 429 failover, and request class isolation.

Built as a minimal-dependency alternative to LiteLLM Router, focused on security and transparency.

Features

  • Weighted least-outstanding routing — routes to the deployment with the lowest load relative to its capacity
  • 429-aware cooldown — respects Retry-After headers, automatically fails over to the next backend
  • Request class isolation — separate concurrency budgets for light/medium/heavy requests prevent expensive operations from starving fast ones
  • Token-aware capacity tracking — tracks actual token usage per deployment per minute window
  • Request deadline — enforces a total timeout across all retry attempts
  • Health reporting — exposes deployment health, inflight counts, 429 rates, and TPM usage

Install

pip install llm-route

Or with uv:

uv add llm-route

Quick Start

import asyncio
from llm_route import SmartRouter, RequestClass, RouterConfig
from llm_route.config import DeploymentConfig

config = RouterConfig(
    deployments=[
        DeploymentConfig(
            name="eastus-1",
            endpoint="https://my-resource.openai.azure.com/",
            api_key="your-key",
            deployment_name="gpt-4o",
            tpm_quota=120_000,
        ),
    ],
)

router = SmartRouter(config=config)

async def main():
    response = await router.complete(
        messages=[{"role": "user", "content": "Hello"}],
        request_class=RequestClass.LIGHT,
    )
    print(response.choices[0].message.content)

asyncio.run(main())

Configuration

JSON config file

{
  "deployments": [
    {
      "name": "eastus-1",
      "endpoint": "https://my-eastus-1.openai.azure.com/",
      "api_key": "your-api-key",
      "deployment_name": "gpt-4o",
      "tpm_quota": 120000
    },
    {
      "name": "eastus-2",
      "endpoint": "https://my-eastus-2.openai.azure.com/",
      "api_key": "your-api-key",
      "deployment_name": "gpt-4o",
      "tpm_quota": 60000
    }
  ],
  "default_timeout": 60.0,
  "max_retries": 3,
  "cooldown_seconds": 10.0,
  "concurrency": {
    "light": 20,
    "medium": 10,
    "heavy": 3
  }
}

Load it:

config = RouterConfig.from_file("config.json")
router = SmartRouter(config=config)

Environment variables

All settings can be set via env vars with LLM_ROUTE_ prefix:

LLM_ROUTE_DEFAULT_TIMEOUT=60.0
LLM_ROUTE_MAX_RETRIES=3
LLM_ROUTE_COOLDOWN_SECONDS=10.0

Request Classes

Request classes provide concurrency isolation per deployment. Heavy requests (full document review) won't starve light ones (quick text search).

Class Default concurrency / deployment Use case
LIGHT 20 Short extraction, find-text, quick Q&A
MEDIUM 10 Clause analysis, section review
HEAVY 3 Full document review, redline, long synthesis
await router.complete(
    messages=[...],
    request_class=RequestClass.HEAVY,  # uses the heavy concurrency budget
)

How Routing Works

  1. Filter — remove disabled, cooled-down, and already-tried deployments
  2. Filter — remove deployments with no available concurrency for the request class
  3. Scoreinflight / weight where weight = tpm_quota / min_tpm. Lower is better.
  4. Select — pick lowest score; break ties by remaining TPM headroom
  5. Execute — acquire semaphore, call Azure OpenAI with remaining deadline
  6. Failover — on 429 or 5xx, mark cooldown, try next deployment
  7. Deadline — if total timeout expires, raise RouterExhaustedError

Health Monitoring

health = router.health()
for dep in health.deployments:
    print(f"{dep.name}: healthy={dep.healthy}, inflight={dep.inflight}, "
          f"tpm={dep.tpm_used}/{dep.tpm_quota}, 429s={dep.total_429s}")

Expose as a FastAPI endpoint:

@app.get("/health/llm")
async def llm_health():
    return router.health().model_dump()

Dependencies

Minimal by design:

  • openai — Azure OpenAI SDK
  • pydantic — data validation
  • pydantic-settings — configuration management

Optional:

  • redis — shared state for multi-replica deployments (pip install llm-route[redis])

Security

This library was built in response to the LiteLLM supply chain attack (March 2026). Design principles:

  • Minimal dependencies — 3 required packages, all well-maintained
  • No build-time code execution — pure Python, no compiled extensions
  • uv.lock committed — exact dependency tree is auditable
  • OIDC publishing — no stored PyPI tokens in CI

License

MIT

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

llm_route-0.6.1.tar.gz (41.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

llm_route-0.6.1-py3-none-any.whl (27.0 kB view details)

Uploaded Python 3

File details

Details for the file llm_route-0.6.1.tar.gz.

File metadata

  • Download URL: llm_route-0.6.1.tar.gz
  • Upload date:
  • Size: 41.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for llm_route-0.6.1.tar.gz
Algorithm Hash digest
SHA256 4b9032f621a6f1c22e4e7662e357a76adec43df6a1a0191c5a80ce2ebc447b96
MD5 fd8ea271f0f2fa8b821c05e868bb2821
BLAKE2b-256 48e8f44ef3df39c8438de0087c89831ac32838c0f1a3d9c2ef7db71999511e9c

See more details on using hashes here.

Provenance

The following attestation bundles were made for llm_route-0.6.1.tar.gz:

Publisher: publish.yml on aipro99/llm-route

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file llm_route-0.6.1-py3-none-any.whl.

File metadata

  • Download URL: llm_route-0.6.1-py3-none-any.whl
  • Upload date:
  • Size: 27.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for llm_route-0.6.1-py3-none-any.whl
Algorithm Hash digest
SHA256 3360a2405d0c961176a087e11a58c1d98be28d94744ecf0dfd7e59f88e233f4d
MD5 33018848899bfb1ff6db1c8e407e8dab
BLAKE2b-256 1406b2d377cd5e63e844e3c8dd82ae9f2ca6f83c661aa458da806da159d7e021

See more details on using hashes here.

Provenance

The following attestation bundles were made for llm_route-0.6.1-py3-none-any.whl:

Publisher: publish.yml on aipro99/llm-route

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page