sdkrouter 0.1.0

Unified SDK for AI services with OpenAI compatibility

SDKRouter

Unified Python SDK for AI services with OpenAI compatibility. Access 300+ LLM models through a single interface, plus vision analysis, CDN, URL shortening, and HTML cleaning tools.

Installation

pip install sdkrouter

Quick Start

from sdkrouter import SDKRouter

client = SDKRouter(api_key="your-api-key")

# OpenAI-compatible chat completions
response = client.chat.completions.create(
    model="openai/gpt-4o",
    messages=[{"role": "user", "content": "Hello!"}]
)
print(response.choices[0].message.content)

Features

Chat Completions (OpenAI-Compatible)

# Non-streaming
response = client.chat.completions.create(
    model="anthropic/claude-3.5-sonnet",
    messages=[{"role": "user", "content": "Explain quantum computing"}],
    max_tokens=500,
)

# Streaming
for chunk in client.chat.completions.create(
    model="openai/gpt-4o",
    messages=[{"role": "user", "content": "Count to 5"}],
    stream=True,
):
    print(chunk.choices[0].delta.content, end="")

Vision Analysis

result = client.vision.analyze(
    image_url="https://example.com/image.jpg",
    prompt="Describe this image",
)
print(result.description)
print(f"Cost: ${result.cost_usd:.6f}")

CDN File Storage

# Upload file
file = client.cdn.upload(
    b"file content",
    filename="document.txt",
    is_public=True,
)
print(file.url)

# List files with pagination
files = client.cdn.list(page=1, page_size=20)
for f in files.results:
    print(f"{f.filename}: {f.url}")

URL Shortener

# Create short link
link = client.shortlinks.create(target_url="https://example.com/long-url")
print(link.short_url)

# Get statistics
stats = client.shortlinks.stats()
print(f"Total links: {stats.total_links}")

HTML Cleaner

result = client.cleaner.clean(
    html_content,
    output_format="markdown",
    remove_scripts=True,
    remove_styles=True,
)
print(result.cleaned_html)
print(f"Compression: {result.compression_ratio:.1f}x")

LLM Models API

# List available models
models = client.models.list(page=1, page_size=50)
for m in models.results:
    print(f"{m.model_id}: {m.name}")

# Get model details
model = client.models.get("openai/gpt-4o")
print(f"Context: {model.context_length} tokens")
print(f"Price: ${model.pricing.prompt}/M input tokens")

# Calculate cost
cost = client.models.calculate_cost(
    "openai/gpt-4o",
    input_tokens=1000,
    output_tokens=500,
)
print(f"Total: ${cost.total_cost_usd:.6f}")

Async Support

All features support async operations:

from sdkrouter import AsyncSDKRouter
import asyncio

async def main():
    client = AsyncSDKRouter(api_key="your-api-key")

    # Async chat completion
    response = await client.chat.completions.create(
        model="openai/gpt-4o",
        messages=[{"role": "user", "content": "Hello!"}]
    )

    # Parallel requests
    results = await asyncio.gather(
        client.vision.analyze(image_url="..."),
        client.models.list(),
        client.cdn.stats(),
    )

asyncio.run(main())

Type Safety

All responses are fully typed with Pydantic models:

from sdkrouter.tools import (
    VisionAnalyzeResponse,
    CDNFileDetail,
    PaginatedLLMModelListList,
    CleanResponse,
)

# IDE autocomplete and type checking
result: VisionAnalyzeResponse = client.vision.analyze(...)
print(result.description)  # str
print(result.cost_usd)     # float
print(result.usage.total_tokens)  # int

Configuration

# Environment variables
# SDKROUTER_API_KEY - API key
# SDKROUTER_BASE_URL - Custom base URL (default: https://ai.sdkrouter.com)

# Or pass directly
client = SDKRouter(
    api_key="your-key",
    base_url="https://your-server.com",
    timeout=60.0,
    max_retries=3,
)

# Use OpenRouter directly
client = SDKRouter(
    openrouter_api_key="your-openrouter-key",
    use_self_hosted=False,
)

Supported Models

Access 300+ models from providers:

• OpenAI: GPT-4o, GPT-4 Turbo, GPT-3.5 Turbo
• Anthropic: Claude 3.5 Sonnet, Claude 3 Opus, Claude 3 Haiku
• Google: Gemini Pro, Gemini Flash
• Meta: Llama 3.1, Llama 3.2
• Mistral: Mistral Large, Mixtral
• And many more via OpenRouter

License

MIT