Skip to main content

Async LLM router across AWS Bedrock and Google Gemini providers

Project description

llm-verse-avneesh

Async LLM router across AWS Bedrock (Claude, Amazon Nova) and Google Gemini.

A single Router.get_response(...) call validates its input, dispatches to the requested model by name, and always returns a plain dict shaped like LLMResponse — regardless of which underlying provider handled the call.

Install

pip install "llm-verse-avneesh[aws]"      # Bedrock (Claude, Nova) only
pip install "llm-verse-avneesh[google]"   # Gemini only
pip install "llm-verse-avneesh[all]"      # both

Usage

import asyncio
from llm_verse_avneesh import Router

async def main():
    router = Router()
    result = await router.get_response(
        llm_name="nova-lite",
        system_prompt="You are a helpful assistant.",
        user_prompt="Say hello in one sentence.",
        context=None,
        temperature=0.5,
        pydantic_model=None,
        max_tokens=200,
        repo_name="my-repo",
        llm_identifier="greeting-1",
        region_name="us-east-1",
        aws_access_key_id="...",
        aws_secret_access_key="...",
    )
    print(result["response"])

asyncio.run(main())

For Gemini, pass google_api_key instead of the AWS credential fields.

Registered model names

Registry keys match the name each provider documents publicly for the model — not an internal AWS/Anthropic model ID:

llm_name Provider Notes
claude-haiku-4-5 Bedrock plain text / structured output
claude-haiku-4-5-agentic Bedrock tool-calling loop, pass tools=[...]
nova-lite Bedrock
nova-2-lite Bedrock
nova-2-lite-grounding Bedrock web grounding; US regions only
nova-pro Bedrock
gemini-3.1-flash-lite Google

Validation

  • LLMRequest (in models.py) does basic sanity checks: non-empty strings, non-negative temperature, positive max_tokens, and that credentials for the selected provider are present.
  • Router.get_response then enforces the actual per-model limits from limits.py — e.g. Gemini allows temperature up to 2.0 while the Bedrock models here cap at 1.0. Exceeding a model's limit raises RouterValidationError naming the offending model and its limit. Update limits.py if a provider changes its published limits.

Credentials

aws_secret_access_key and google_api_key are stored as pydantic.SecretStr so they don't leak into repr(), str(), logs, or model_dump() output by accident. Call .get_secret_value() only at the point where a client needs the raw value.

Exceptions

All exceptions inherit from LLMRouterError:

  • RouterValidationError — bad input or a model-specific limit was exceeded.
  • ProviderNotFoundErrorllm_name isn't registered.
  • LLMCallError — the underlying provider call failed or raised.

Development

pip install -e ".[dev]"
pytest

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_verse_avneesh-0.1.1.tar.gz (19.9 kB view details)

Uploaded Source

Built Distribution

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

llm_verse_avneesh-0.1.1-py3-none-any.whl (24.7 kB view details)

Uploaded Python 3

File details

Details for the file llm_verse_avneesh-0.1.1.tar.gz.

File metadata

  • Download URL: llm_verse_avneesh-0.1.1.tar.gz
  • Upload date:
  • Size: 19.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for llm_verse_avneesh-0.1.1.tar.gz
Algorithm Hash digest
SHA256 b82361b02129d42a15a6ec3b7afde8d9fd85104ad7540bf588e96125249bce13
MD5 0a4382925c94bc998000425f76856da0
BLAKE2b-256 d62fd0071a994533fd744f5bdb4afac0323672836e70d72e8c7a74e1e0b174de

See more details on using hashes here.

Provenance

The following attestation bundles were made for llm_verse_avneesh-0.1.1.tar.gz:

Publisher: publish.yml on avneeshrai07/llm-verse-avneesh

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_verse_avneesh-0.1.1-py3-none-any.whl.

File metadata

File hashes

Hashes for llm_verse_avneesh-0.1.1-py3-none-any.whl
Algorithm Hash digest
SHA256 c108853c2bd4c8ddd7b1ed26461540b8b2d6c3ffcbe6f2f5249146611b68f705
MD5 23296050e6a198d32140e39cbffe2bfc
BLAKE2b-256 a0ea938df40a96f78f02464dccfe8bdf0ba8a257bd1e082d7d20d5d9e905843e

See more details on using hashes here.

Provenance

The following attestation bundles were made for llm_verse_avneesh-0.1.1-py3-none-any.whl:

Publisher: publish.yml on avneeshrai07/llm-verse-avneesh

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