Skip to main content

AI cost management SDK with intelligent model routing, prompt compression, and real-time token tracking

Project description

Compress Light Reach

AI cost management SDK with intelligent model routing, prompt compression, and real-time token tracking

PyPI version Python 3.8+ License: MIT

Compress Light Reach is a Python SDK that provides intelligent model routing and prompt compression for LLM applications, reducing token usage and costs while maintaining quality.

Features

  • Intelligent Model Routing: Automatically selects optimal model based on quality requirements (HLE) and available provider keys
  • Token-aware Compression: Replaces repeated substrings with shorter placeholders
  • Dual Algorithms:
    • Fast greedy (~99% optimal) for daily use
    • Optimal DP (O(n²)) for critical prompts
  • Lossless: Perfect decompression guaranteed
  • Output Compression: Optional model output compression support
  • Cloud API: Uses Light Reach's cloud service for compression and routing
  • Multi-provider Support: OpenAI, Anthropic, Google, DeepSeek, Moonshot
  • BYOK: Provider API keys managed securely in dashboard (never passed through SDK)

Installation

pip install compress-lightreach

Quick Start

The SDK uses intelligent model routing and targets POST /api/v2/complete.

  • Authenticate with your LightReach API key (env var PCOMPRESLR_API_KEY or LIGHTREACH_API_KEY)
  • Manage provider keys (OpenAI/Anthropic/Google/etc.) in the dashboard (BYOK)
  • System automatically selects optimal model based on your requirements
from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

result = client.complete(
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Explain quantum computing in simple terms."},
    ],
    desired_hle=30,  # Quality preference (0-40, where 40 is SOTA)
)

print(result["decompressed_response"])
print(f"Selected: {result['routing_info']['selected_model']}")
print(f"Token savings: {result['compression_stats']['token_savings']}")

With Output Compression

result = client.complete(
    messages=[{"role": "user", "content": "Generate a long report..."}],
    desired_hle=25,
    compress_output=True,
)

print(result["decompressed_response"])

Intelligent Model Routing

The system automatically selects the optimal model based on quality requirements and your available provider keys:

from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

# Cross-provider optimization: system picks cheapest model meeting your quality bar
result = client.complete(
    messages=[{"role": "user", "content": "Explain quantum computing"}],
    desired_hle=30,  # Quality preference (0-40, where 40 is SOTA)
)

# Check what was selected
print(result["routing_info"]["selected_model"])           # e.g., "gpt-4o-mini"
print(result["routing_info"]["selected_provider"])        # e.g., "openai"
print(result["routing_info"]["model_hle"])                # e.g., 32.5
print(result["routing_info"]["model_price_per_million"])  # e.g., 0.15

Provider-Constrained Routing

Optionally constrain to a specific provider:

# Only use OpenAI models, but pick the cheapest one meeting HLE 35
result = client.complete(
    messages=[{"role": "user", "content": "Write a poem"}],
    llm_provider="openai",  # Optional: constrain to one provider
    desired_hle=35,
)

HLE Cascading with Admin Controls

Admins can set quality ceilings via the dashboard (global or per-tag) to control costs. Your desired_hle is a preference, but requests will error if they exceed the admin-set ceiling:

from pcompresslr import PcompresslrAPIClient, APIRequestError

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

# Admin set global HLE ceiling to 30%
# Requesting above the ceiling will error
try:
    result = client.complete(
        messages=[{"role": "user", "content": "Process payment"}],
        desired_hle=35,  # ERROR: exceeds ceiling of 30
        tags={"env": "production"},
    )
except APIRequestError as e:
    print(f"Error: {e}")  # "Requested HLE 35% exceeds workspace maximum of 30%"

# Correct usage: request within ceiling
result = client.complete(
    messages=[{"role": "user", "content": "Process payment"}],
    desired_hle=25,  # OK: below ceiling of 30
    tags={"env": "production"},
)

# Check if your HLE was lowered by admin ceiling
if result["routing_info"]["hle_clamped"]:
    print(f"HLE lowered from {result['routing_info']['requested_hle']} "
          f"to {result['routing_info']['effective_hle']} "
          f"by {result['routing_info']['hle_source']}-level ceiling")

HLE Ceiling Logic:

  • effective_hle = min(desired_hle, tag_hle, global_hle) - most restrictive ceiling wins
  • Lower ceiling = force cheaper models (better cost control)
  • Engineers get errors if requesting above ceiling
  • Tag-level ceilings can override global ceiling (lowest wins)

Using Message and CompressionConfig Classes

For more structured code, use the Message and CompressionConfig dataclasses:

from pcompresslr import PcompresslrAPIClient, Message, CompressionConfig

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

# Using Message dataclass
result = client.complete(
    messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": "Hello!"},
    ],
    desired_hle=30,
    compress=True,
    compress_output=False,
    compression_config={
        "compress_system": False,
        "compress_user": True,
        "compress_assistant": False,
        "compress_only_last_n_user": 1,
    },
    temperature=0.7,
    max_tokens=1000,
    tags={"env": "production"},
)

print(result["decompressed_response"])
print(f"Model used: {result['routing_info']['selected_model']}")

Note: The LightReach wrapper class is available for backwards compatibility but uses deprecated parameters (default_model, default_provider). Use PcompresslrAPIClient directly for v1.0.0 intelligent routing.

Compression Only (No LLM Call)

from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

# Compress text without making an LLM call
compressed = client.compress(
    prompt="Your text with repeated content here...",
    model="gpt-4",      # Model for tokenization
    algorithm="greedy", # 'greedy' or 'optimal'
    tags={"env": "dev"} # Optional tags
)

print(compressed["llm_format"])
print(f"Compression ratio: {compressed['compression_ratio']:.2%}")

# Decompress later
decompressed = client.decompress(compressed["llm_format"])
print(decompressed["decompressed"])

Command Line Interface

# Set your API key
export PCOMPRESLR_API_KEY=your-api-key

# Compress a prompt
pcompresslr "Your prompt with repeated text here..."

# Use optimal algorithm only
pcompresslr "Your prompt here" --optimal-only

# Use greedy algorithm only
pcompresslr "Your prompt here" --greedy-only

API Reference

PcompresslrAPIClient

Main API client for intelligent model routing and compression.

Constructor

PcompresslrAPIClient(
    api_key: str = None,  # Falls back to env vars
    api_url: str = None,  # Default: https://api.compress.lightreach.io
    timeout: int = 120    # Request timeout in seconds
)

Parameters:

  • api_key (str, optional): LightReach API key. Falls back to LIGHTREACH_API_KEY or PCOMPRESLR_API_KEY env vars.
  • api_url (str, optional): Override base API URL. Falls back to PCOMPRESLR_API_URL env var.
  • timeout (int, optional): Request timeout in seconds. Default: 120 (2 minutes for LLM calls).

Methods

complete(messages, ...)

Messages-first completion with intelligent routing (POST /api/v2/complete).

Parameters:

Parameter Type Default Description
messages list[dict] required Conversation history with role and content
llm_provider str None Provider constraint: "openai", "anthropic", "google", "deepseek", "moonshot". Omit for cross-provider optimization
desired_hle float None Quality preference (0-40, where 40 is SOTA). Must not exceed admin ceilings
compress bool True Whether to compress messages
compress_output bool False Whether to request compressed output from LLM
algorithm str "greedy" Compression algorithm: "greedy" or "optimal"
compression_config dict None Per-role compression settings (see below)
temperature float None LLM temperature parameter
max_tokens int None Maximum tokens to generate
tags dict[str, str] None Tags for cost attribution and tag-level HLE ceilings
max_history_messages int None Limit conversation history length

compression_config options:

{
    "compress_system": False,         # default
    "compress_user": True,            # default
    "compress_assistant": False,      # default
    "compress_only_last_n_user": 1,   # default (None = compress all)
}

Response (dict):

{
    "decompressed_response": str,     # Final decompressed LLM response
    "compression_stats": {
        "original_size_chars": int,
        "compressed_size_chars": int,
        "original_tokens": int,
        "compressed_tokens": int,
        "compression_ratio": float,
        "token_savings": int,
        "token_savings_percent": float,
        "processing_time_ms": float,
    },
    "llm_stats": {
        "prompt_tokens": int,
        "completion_tokens": int,
        "total_tokens": int,
    },
    "routing_info": {
        "selected_model": str,          # Model chosen by system
        "selected_provider": str,       # Provider chosen by system
        "model_hle": float,             # HLE score of selected model
        "model_price_per_million": float,
        "requested_hle": float | None,
        "effective_hle": float | None,  # Effective HLE after admin ceilings
        "hle_source": str,              # "request", "tag", "global", or "none"
        "hle_clamped": bool,            # True if admin ceiling lowered desired_hle
    },
    "warnings": list[str],
    "cost_estimate": float | None,
    "savings_estimate": float | None,
}

Deprecated parameters (ignored in v1.0.0):

  • model: System now selects models automatically
  • hle_target_percent: Use desired_hle instead
  • min_hle_score: Use desired_hle instead
  • auto_select_by_hle: Always auto-selects now
  • same_provider_only: Use llm_provider instead
compress(prompt, model, algorithm, tags)

Compression-only (POST /api/v1/compress).

Parameters:

  • prompt (str, required): Text to compress
  • model (str, optional): Model for tokenization. Default: "gpt-4"
  • algorithm (str, optional): "greedy" or "optimal". Default: "greedy"
  • tags (dict, optional): Tags for attribution

Response (dict):

{
    "compressed": str,
    "dictionary": dict[str, str],
    "llm_format": str,
    "compression_ratio": float,
    "original_size": int,
    "compressed_size": int,
    "processing_time_ms": float,
    "algorithm": str,
}
decompress(llm_format)

Decompress an LLM-formatted compressed prompt (POST /api/v1/decompress).

Parameters:

  • llm_format (str, required): The llm_format string from a compress response

Response (dict):

{
    "decompressed": str,
    "processing_time_ms": float,
}
health_check()

Check API health status (GET /health).

Response (dict):

{
    "status": str,
    "version": str,
}

LightReach Class (Legacy)

Deprecated: This wrapper uses default_model and default_provider parameters which are ignored in v1.0.0. Use PcompresslrAPIClient directly for intelligent routing.

Available for backwards compatibility. The Pcompresslr class is an alias for LightReach.

Data Classes

Message

from pcompresslr import Message

msg = Message(role="user", content="Hello!")
msg.to_dict()  # {"role": "user", "content": "Hello!"}

Roles: "system", "developer", "user", "assistant"

CompressionConfig

from pcompresslr import CompressionConfig

config = CompressionConfig(
    compress_system=False,        # default
    compress_user=True,           # default
    compress_assistant=False,     # default
    compress_only_last_n_user=1,  # default (None = compress all)
)
config.to_dict()

Environment Variables

Variable Description
PCOMPRESLR_API_KEY Your LightReach API key (primary)
LIGHTREACH_API_KEY Your LightReach API key (alternative)
PCOMPRESLR_API_URL Override the API base URL (advanced/testing)

Exceptions

Exception Description
PcompresslrAPIError Base exception class
APIKeyError Invalid or missing API key
RateLimitError Rate limit exceeded
APIRequestError General API errors (including routing failures, HLE ceiling exceeded)
from pcompresslr import (
    PcompresslrAPIClient,
    APIKeyError,
    RateLimitError,
    APIRequestError,
)

try:
    result = client.complete(messages=[...])
except APIKeyError as e:
    print("Invalid API key")
except RateLimitError as e:
    print("Rate limited, please retry later")
except APIRequestError as e:
    print(f"API error: {e}")

How It Works

Compress Light Reach uses intelligent algorithms to identify repeated substrings in your prompts and replace them with shorter placeholders.

The library:

  1. Identifies repeated substrings using efficient suffix array algorithms
  2. Calculates token savings for each potential replacement
  3. Selects optimal replacements that reduce total token count
  4. Intelligently routes to the best model based on your quality requirements
  5. Formats the result for easy LLM consumption
  6. Provides perfect decompression

Examples

Example 1: Complete with Compression

from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

prompt = """
Write a story about a cat. The cat is very friendly. 
Write a story about a dog. The dog is very friendly.
Write a story about a bird. The bird is very friendly.
"""

result = client.complete(
    messages=[{"role": "user", "content": prompt}],
    desired_hle=30,
)

print(result["decompressed_response"])
print(f"Model used: {result['routing_info']['selected_model']}")
print(f"Token savings: {result['compression_stats']['token_savings']} tokens")
print(f"Compression ratio: {result['compression_stats']['compression_ratio']:.2%}")

Example 2: Multi-turn Conversation

from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

result = client.complete(
    messages=[
        {"role": "system", "content": "You are a helpful coding assistant."},
        {"role": "user", "content": "How do I read a file in Python?"},
        {"role": "assistant", "content": "You can use open() with a context manager..."},
        {"role": "user", "content": "How about writing to a file?"},
    ],
    desired_hle=30,
    compression_config={
        "compress_system": False,
        "compress_user": True,
        "compress_assistant": False,
        "compress_only_last_n_user": 2,  # Only compress last 2 user messages
    },
)

Example 3: Provider-Constrained Request

from pcompresslr import PcompresslrAPIClient

client = PcompresslrAPIClient(api_key="your-lightreach-api-key")

# Constrain to Anthropic models only
result = client.complete(
    messages=[
        {"role": "system", "content": "You are a creative writing assistant."},
        {"role": "user", "content": "Write a haiku about coding."},
    ],
    llm_provider="anthropic",  # Only use Anthropic models
    desired_hle=30,
)

print(result["decompressed_response"])
print(f"Model: {result['routing_info']['selected_model']}")  # e.g., claude-3-haiku

Getting an API Key

To use Compress Light Reach, you need an API key from compress.lightreach.io.

  1. Visit compress.lightreach.io
  2. Sign up for an account
  3. Get your API key from the dashboard
  4. Set it as an environment variable: export PCOMPRESLR_API_KEY=your-key

Security & Privacy

BYOK model: Provider keys (OpenAI/Anthropic/Google/etc.) are managed in the dashboard and never passed through this SDK. The SDK only uses your LightReach API key for authentication with the service.

BYOK Provider Key Encryption (Required for Dashboard Settings → Provider Keys)

Provider keys are encrypted at rest using Fernet (symmetric authenticated encryption). The backend requires a Fernet key via:

  • API_KEY_ENCRYPTION_KEY

Generate a key:

python -c "from cryptography.fernet import Fernet; print(Fernet.generate_key().decode())"

Set it in your runtime environment (examples):

  • Docker Compose: set API_KEY_ENCRYPTION_KEY in your shell or .env before running docker compose up
  • GitHub Actions: store the value as a GitHub Secret, then map it to the environment variable API_KEY_ENCRYPTION_KEY in your deploy workflow

Requirements

  • Python 3.8+
  • tiktoken >= 0.5.0
  • requests >= 2.31.0
  • urllib3 >= 2.0.0
  • python-dotenv >= 1.0.0

License

MIT License - see LICENSE file for details.

Support

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

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

compress_lightreach-1.0.2.tar.gz (57.5 kB view details)

Uploaded Source

Built Distribution

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

compress_lightreach-1.0.2-py3-none-any.whl (18.5 kB view details)

Uploaded Python 3

File details

Details for the file compress_lightreach-1.0.2.tar.gz.

File metadata

  • Download URL: compress_lightreach-1.0.2.tar.gz
  • Upload date:
  • Size: 57.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.13.5

File hashes

Hashes for compress_lightreach-1.0.2.tar.gz
Algorithm Hash digest
SHA256 4b161d1519ce72bcc6d662020256d8fecdf744da9ab79eb7532276725a639bfc
MD5 a3ff39e4f5ce44aa52f7cb0ae30ba778
BLAKE2b-256 8a21d5f63fb745a2f41abca15bd439781545012efcfddfdfc791c24a2eb325b6

See more details on using hashes here.

File details

Details for the file compress_lightreach-1.0.2-py3-none-any.whl.

File metadata

File hashes

Hashes for compress_lightreach-1.0.2-py3-none-any.whl
Algorithm Hash digest
SHA256 bc0d2b674318fe3ba5d70e78cbf2d113acdb914a88664201597c9a205093ea95
MD5 fd70f0af1bce9eb03a2f028493d885e2
BLAKE2b-256 baccb9a76fd35f3327711d91dacadbbc8e4dec43e00a82519f322a6c03efe231

See more details on using hashes here.

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