Unified Python interface for OpenAI, Anthropic, Google, and Ollama LLMs
Project description
LLMRing
A comprehensive Python library for LLM integration with unified interface, advanced features, and MCP support. Supports OpenAI, Anthropic, Google Gemini, and Ollama with consistent APIs.
โจ Key Features
- ๐ Unified Interface: Single API for all major LLM providers
- โก Streaming Support: Real streaming for all providers (not simulated)
- ๐ ๏ธ Native Tool Calling: Provider-native function calling with consistent interface
- ๐ Unified Structured Output: JSON schema works across all providers with automatic adaptation
- ๐ง Conversational Configuration: MCP chat interface for natural language lockfile setup
- ๐ Smart Aliases: Always-current semantic aliases (
deep,fast,balanced) via intelligent recommendations - ๐ฐ Cost Tracking: Automatic cost calculation and receipt generation
- ๐ฏ Registry Integration: Centralized model capabilities and pricing
- ๐ Fallback Models: Automatic failover to alternative models for resilience
- ๐ง Advanced Features:
- OpenAI: JSON schema, o1 models, PDF processing
- Anthropic: Prompt caching (90% cost savings)
- Google: Native function calling, multimodal, 2M+ context
- Ollama: Local models, streaming, custom options
- ๐ Type Safety: Comprehensive typed exceptions and error handling
- ๐ MCP Integration: Model Context Protocol support for tool ecosystems
- ๐ฌ MCP Chat Client: Generic chat interface with persistent history for any MCP server
๐ Quick Start
Installation
# With uv (recommended)
uv add llmring
# With pip
pip install llmring
Including Lockfiles in Your Package:
To ship your llmring.lock with your package (like llmring does), add to your pyproject.toml:
[tool.hatch.build]
include = [
"src/yourpackage/**/*.py",
"src/yourpackage/**/*.lock", # Include lockfiles
]
Basic Usage
from llmring.service import LLMRing
from llmring.schemas import LLMRequest, Message
# Initialize service with context manager (auto-closes resources)
async with LLMRing() as service:
# Simple chat
request = LLMRequest(
model="fast",
messages=[
Message(role="system", content="You are a helpful assistant."),
Message(role="user", content="Hello!")
]
)
response = await service.chat(request)
print(response.content)
Streaming
async with LLMRing() as service:
# Real streaming for all providers
request = LLMRequest(
model="balanced",
messages=[Message(role="user", content="Count to 10")],
stream=True
)
accumulated_usage = None
async for chunk in await service.chat(request):
print(chunk.delta, end="", flush=True)
# Capture final usage stats
if chunk.usage:
accumulated_usage = chunk.usage
print() # Newline after streaming
if accumulated_usage:
print(f"Tokens used: {accumulated_usage.get('total_tokens', 0)}")
Tool Calling
async with LLMRing() as service:
tools = [{
"type": "function",
"function": {
"name": "get_weather",
"description": "Get weather for a location",
"parameters": {
"type": "object",
"properties": {
"location": {"type": "string"}
},
"required": ["location"]
}
}
}]
request = LLMRequest(
model="balanced",
messages=[Message(role="user", content="What's the weather in NYC?")],
tools=tools
)
response = await service.chat(request)
if response.tool_calls:
print("Function called:", response.tool_calls[0]["function"]["name"])
๐ Resource Management
Context Manager (Recommended)
from llmring import LLMRing, LLMRequest, Message
# Automatic resource cleanup with context manager
async with LLMRing() as service:
request = LLMRequest(
model="fast",
messages=[Message(role="user", content="Hello!")]
)
response = await service.chat(request)
# Resources are automatically cleaned up when exiting the context
Manual Cleanup
# Manual resource management
service = LLMRing()
try:
response = await service.chat(request)
finally:
await service.close() # Ensure resources are cleaned up
๐ง Advanced Features
๐ฏ Unified Structured Output (All Providers)
# Same JSON schema API works across ALL providers!
request = LLMRequest(
model="balanced", # Works with any provider
messages=[Message(role="user", content="Generate a person")],
response_format={
"type": "json_schema",
"json_schema": {
"name": "person",
"schema": {
"type": "object",
"properties": {
"name": {"type": "string"},
"age": {"type": "integer"},
"email": {"type": "string"}
},
"required": ["name", "age"]
}
},
"strict": True # Validates across all providers
}
)
response = await service.chat(request)
print("JSON:", response.content) # Valid JSON string
print("Data:", response.parsed) # Python dict ready to use
Provider-Specific Parameters
# Anthropic: Prompt caching for 90% cost savings
request = LLMRequest(
model="balanced",
messages=[
Message(
role="system",
content="Very long system prompt...", # 1024+ tokens
metadata={"cache_control": {"type": "ephemeral"}}
),
Message(role="user", content="Hello")
]
)
# Extra parameters for provider-specific features
request = LLMRequest(
model="fast",
messages=[Message(role="user", content="Hello")],
extra_params={
"logprobs": True,
"top_logprobs": 5,
"presence_penalty": 0.1,
"seed": 12345
}
)
๐ง Model Aliases and Lockfiles
LLMRing uses lockfiles to map semantic aliases to models, with support for fallback models and environment-specific profiles:
# Initialize lockfile (explicit creation at current directory)
llmring lock init
# Conversational configuration with AI advisor (recommended)
llmring lock chat # Natural language interface for lockfile management
# Analyze your configuration
llmring lock analyze
# View current aliases
llmring aliases
Lockfile Resolution Order:
- Explicit path via
lockfile_pathparameter (file must exist) LLMRING_LOCKFILE_PATHenvironment variable (file must exist)./llmring.lockin current directory (if exists)- Bundled lockfile at
src/llmring/llmring.lock(minimal fallback with advisor alias)
Packaging Your Own Lockfile: Libraries using LLMRing can ship with their own lockfiles. See Lockfile Documentation for details on:
- Including lockfiles in your package distribution
- Lockfile resolution order and precedence
- Creating lockfiles with fallback models
- Environment-specific profiles and configuration
Conversational Configuration via llmring lock chat:
- Describe your requirements in natural language
- Get AI-powered recommendations based on registry analysis
- Configure aliases with multiple fallback models
- Understand cost implications and tradeoffs
- Set up environment-specific profiles
# Use semantic aliases (always current, with fallbacks)
request = LLMRequest(
model="deep", # โ most capable reasoning model
messages=[Message(role="user", content="Hello")]
)
# Or use other aliases:
# model="fast" โ cost-effective quick responses
# model="balanced" โ optimal all-around model
# model="advisor" โ Claude Opus 4.1 - powers conversational config
Key Benefits:
- Always current: Registry-based recommendations, no hardcoded models
- Resilient: Fallback models provide automatic failover
- Cost-aware: Transparent cost analysis and recommendations
- Environment-specific: Different configurations for dev/staging/prod
๐ญ Profiles: Environment-Specific Configurations
LLMRing supports profiles to manage different model configurations for different environments (dev, staging, prod, etc.):
# Use different models based on environment
# Development: Use cheaper/faster models
# Production: Use higher-quality models
# Set profile via environment variable
export LLMRING_PROFILE=dev # or prod, staging, etc.
# Or specify profile in code
async with LLMRing() as service:
# Uses 'dev' profile bindings
response = await service.chat(request, profile="dev")
Profile Configuration in Lockfiles:
# llmring.lock - Different models per environment
[profiles.default]
[[profiles.default.bindings]]
alias = "assistant"
models = ["anthropic:claude-3-5-sonnet"] # Production quality
[profiles.dev]
[[profiles.dev.bindings]]
alias = "assistant"
models = ["openai:gpt-4o-mini"] # Cheaper for development
[profiles.test]
[[profiles.test.bindings]]
alias = "assistant"
models = ["ollama:llama3"] # Local model for testing
Using Profiles with CLI:
# Bind aliases to specific profiles
llmring bind assistant "openai:gpt-4o-mini" --profile dev
llmring bind assistant "anthropic:claude-3-5-sonnet" --profile prod
# List aliases in a profile
llmring aliases --profile dev
# Use profile for chat
llmring chat "Hello" --profile dev
# Set default profile via environment
export LLMRING_PROFILE=dev
llmring chat "Hello" # Now uses dev profile
Profile Selection Priority:
- Explicit parameter:
profile="dev"or--profile dev(highest priority) - Environment variable:
LLMRING_PROFILE=dev - Default:
defaultprofile (if not specified)
Common Use Cases:
- Development: Use cheaper models to reduce costs during development
- Testing: Use local models (Ollama) or mock responses
- Staging: Use production models but with different rate limits
- Production: Use highest quality models for best user experience
- A/B Testing: Test different models for the same alias
๐ฏ Fallback Models
Aliases can specify multiple models for automatic failover:
# In llmring.lock
[[bindings]]
alias = "assistant"
models = [
"anthropic:claude-3-5-sonnet", # Primary
"openai:gpt-4o", # First fallback
"google:gemini-1.5-pro" # Second fallback
]
If the primary model fails (rate limit, availability, etc.), LLMRing automatically tries the fallbacks.
๐ช Advanced: Direct Model References
While aliases are recommended, you can still use direct provider:model references when needed:
# Direct model reference (escape hatch)
request = LLMRequest(
model="anthropic:claude-3-5-sonnet", # Direct provider:model reference
messages=[Message(role="user", content="Hello")]
)
# Or specify exact model versions
request = LLMRequest(
model="openai:gpt-4o", # Specific model version when needed
messages=[Message(role="user", content="Hello")]
)
Terminology:
- Alias: Semantic name like
fast,balanced,deep(recommended) - Model Reference: Full
provider:modelformat likeopenai:gpt-4o(escape hatch) - Raw SDK Access: Bypassing LLMRing entirely using provider clients directly (see Provider Guide)
Recommendation: Use aliases for maintainability and cost optimization. Use direct model references only when you need a specific model version or provider-specific features.
๐ช Raw SDK Access (Escape Hatch)
When you need the full power of the underlying SDKs:
# Access any provider's raw client for maximum SDK features
openai_client = service.get_provider("openai").client # openai.AsyncOpenAI
anthropic_client = service.get_provider("anthropic").client # anthropic.AsyncAnthropic
google_client = service.get_provider("google").client # google.genai.Client
ollama_client = service.get_provider("ollama").client # ollama.AsyncClient
# Use any SDK feature not exposed by LLMRing
response = await openai_client.chat.completions.create(
model="fast", # Use alias or provider:model format when needed
messages=[{"role": "user", "content": "Hello"}],
logprobs=True,
top_logprobs=10,
parallel_tool_calls=False,
# Any OpenAI parameter
)
# Anthropic with all SDK features
response = await anthropic_client.messages.create(
model="balanced", # Use alias or provider:model format when needed
messages=[{"role": "user", "content": "Hello"}],
max_tokens=100,
top_p=0.9,
top_k=40,
system=[{
"type": "text",
"text": "You are helpful",
"cache_control": {"type": "ephemeral"}
}]
)
# Google with native SDK features
response = google_client.models.generate_content(
model="balanced", # Use alias or provider:model format when needed
contents="Hello",
generation_config={
"temperature": 0.7,
"top_p": 0.8,
"top_k": 40,
"candidate_count": 3
},
safety_settings=[{
"category": "HARM_CATEGORY_HARASSMENT",
"threshold": "BLOCK_MEDIUM_AND_ABOVE"
}]
)
When to use raw clients:
- Advanced SDK features not in LLMRing
- Provider-specific optimizations
- Complex configurations
- Performance-critical applications
๐ Provider Support
| Provider | Models | Streaming | Tools | Special Features |
|---|---|---|---|---|
| OpenAI | GPT-4o, GPT-4o-mini, o1 | โ Real | โ Native | JSON schema, PDF processing |
| Anthropic | Claude 3.5 Sonnet/Haiku | โ Real | โ Native | Prompt caching, large context |
| Gemini 1.5/2.0 Pro/Flash | โ Real | โ Native | Multimodal, 2M+ context | |
| Ollama | Llama, Mistral, etc. | โ Real | ๐ง Prompt | Local models, custom options |
๐ฆ Setup
Environment Variables
# Add to your .env file
OPENAI_API_KEY=sk-...
ANTHROPIC_API_KEY=sk-ant-...
GOOGLE_GEMINI_API_KEY=AIza...
# Optional
OLLAMA_BASE_URL=http://localhost:11434 # Default
Conversational Setup
# Create optimized configuration with AI advisor
llmring lock chat
# This opens an interactive chat where you can describe your needs
# and get personalized recommendations based on the registry
Dependencies
# Required for specific providers
pip install openai>=1.0 # OpenAI
pip install anthropic>=0.67 # Anthropic
pip install google-genai # Google Gemini
pip install ollama>=0.4 # Ollama
๐ MCP Integration
from llmring.mcp.client import create_enhanced_llm
# Create MCP-enabled LLM with tool ecosystem
llm = await create_enhanced_llm(
model="fast",
mcp_server_path="path/to/mcp/server"
)
# Now has access to MCP tools
response = await llm.chat([
Message(role="user", content="Use available tools to help me")
])
๐ Documentation
- Lockfile Documentation - Complete guide to lockfiles, aliases, and profiles
- MCP Integration - Model Context Protocol and chat client
- API Reference - Core API documentation
- Provider Guide - Provider-specific features
- Structured Output - Unified JSON schema support
- File Utilities - Vision and multimodal file handling
- CLI Reference - Command-line interface guide
- Receipts & Cost Tracking - Cost tracking and receipt system
- Examples - Working code examples:
- Quick Start - Basic usage patterns
- MCP Chat - MCP integration
- Streaming - Streaming with tools
- Conversational Lockfile - Lockfile setup guide
๐งช Development
# Install for development
uv sync --group dev
# Run tests
uv run pytest
# Lint and format
uv run ruff check src/
uv run ruff format src/
๐ ๏ธ Error Handling
LLMRing uses typed exceptions for better error handling:
from llmring.exceptions import (
ProviderAuthenticationError,
ModelNotFoundError,
ProviderRateLimitError,
ProviderTimeoutError
)
try:
response = await service.chat(request)
except ProviderAuthenticationError:
print("Invalid API key")
except ModelNotFoundError:
print("Model not supported")
except ProviderRateLimitError as e:
print(f"Rate limited, retry after {e.retry_after}s")
๐ฏ Key Benefits
- ๐ Unified Interface: Switch providers without code changes
- โก Performance: Real streaming, prompt caching, optimized requests
- ๐ก๏ธ Reliability: Circuit breakers, retries, typed error handling
- ๐ Observability: Cost tracking, usage analytics, receipt generation
- ๐ง Flexibility: Provider-specific features + raw SDK access
- ๐ Standards: Type-safe, well-tested, production-ready
๐ License
MIT License - see LICENSE file for details.
๐ค Contributing
- Fork the repository
- Create a feature branch
- Add tests for your changes
- Ensure all tests pass:
uv run pytest - Submit a pull request
๐ Examples
See the examples/ directory for complete working examples:
- Basic chat and streaming
- Tool calling and function execution
- Provider-specific features
- MCP integration
- Cost tracking and receipts
LLMRing: The comprehensive LLM library for Python developers ๐
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
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file llmring-1.1.0.tar.gz.
File metadata
- Download URL: llmring-1.1.0.tar.gz
- Upload date:
- Size: 190.7 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
4edb2ee5c82c850a4765c1d051fc25144c9e865ca9495548e526e39f1f094658
|
|
| MD5 |
11175486c9e8556b1c070d6a6cf83afd
|
|
| BLAKE2b-256 |
961002e0306d00f34853b0806ac32cc4bc065ae3f3dabfa8d5b7f00a9a43b38d
|
File details
Details for the file llmring-1.1.0-py3-none-any.whl.
File metadata
- Download URL: llmring-1.1.0-py3-none-any.whl
- Upload date:
- Size: 238.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.7.4
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
1d9b634e1a0db2a5e11986e912ed5363d2d26207fed12de1d149363eb5d9c1cf
|
|
| MD5 |
bb21d99defb2ec7f73a7bc8afc4baad1
|
|
| BLAKE2b-256 |
10a27020521bef39deba490fcf9449b55481184018907c972f59aadc70e28614
|