Privacy-focused Python library for interacting with LLM providers (OpenAI, Gemini). Zero-retention middleware with mandatory encryption and billing verification.
Project description
msgmodel
A unified Python library and CLI for interacting with multiple Large Language Model (LLM) providers.
Overview
msgmodel provides both a Python library and a command-line interface to interact with three major LLM providers:
- OpenAI (GPT models)
- Google Gemini
- Anthropic Claude
Use it as a library in your Python projects or as a CLI tool for quick interactions.
Features
- Unified API: Single
query()andstream()functions work with all providers - Library & CLI: Use as a Python module or command-line tool
- Streaming support: Stream responses in real-time
- File attachments: Process images, PDFs, and text files with your prompts
- Flexible configuration: Dataclass-based configs with sensible defaults
- Multiple API key sources: Direct parameter, environment variable, or key file
- Exception-based error handling: Clean errors, no
sys.exit()in library code - Type-safe: Full type hints throughout
- Privacy-focused: Minimal data retention settings by default
Installation
From PyPI (Recommended)
# Basic installation
pip install msgmodel
# With Claude support
pip install msgmodel[claude]
# With all optional dependencies
pip install msgmodel[all]
From Source
# Clone the repository
git clone https://github.com/LeoooDias/msgmodel.git
cd msgmodel
# Install the package
pip install -e .
# Or with Claude support
pip install -e ".[claude]"
# Or with development dependencies
pip install -e ".[dev]"
Prerequisites
- Python 3.10 or higher
- API keys from the providers you wish to use
Quick Start
As a Library
from msgmodel import query, stream
# Simple query (uses OPENAI_API_KEY env var)
response = query("openai", "What is Python?")
print(response.text)
# With explicit API key
response = query("gemini", "Hello!", api_key="your-api-key")
# Streaming
for chunk in stream("claude", "Tell me a story"):
print(chunk, end="", flush=True)
# With file attachment
response = query("gemini", "Describe this image", file_path="photo.jpg")
# With custom configuration
from msgmodel import OpenAIConfig
config = OpenAIConfig(model="gpt-4o-mini", temperature=0.7, max_tokens=2000)
response = query("openai", "Write a poem", config=config)
As a CLI
# Basic usage
python -m msgmodel -p openai "What is Python?"
# Using shorthand provider codes
python -m msgmodel -p g "Hello, Gemini!" # g = gemini
python -m msgmodel -p c "Hello, Claude!" # c = claude
python -m msgmodel -p o "Hello, OpenAI!" # o = openai
# With streaming
python -m msgmodel -p openai "Tell me a story" --stream
# From a file
python -m msgmodel -p gemini -f prompt.txt
# With system instruction
python -m msgmodel -p claude "Analyze this" -i "You are a data analyst"
# With file attachment
python -m msgmodel -p gemini "Describe this" -b image.jpg
# Custom parameters
python -m msgmodel -p openai "Hello" -m gpt-4o-mini -t 500 --temperature 0.7
# Get full JSON response instead of just text
python -m msgmodel -p openai "Hello" --json
# Verbose output (shows model, provider, token usage)
python -m msgmodel -p openai "Hello" -v
API Key Configuration
API keys can be provided in three ways (in order of priority):
- Direct parameter:
query("openai", "Hello", api_key="sk-...") - Environment variable:
OPENAI_API_KEYfor OpenAIGEMINI_API_KEYfor GeminiANTHROPIC_API_KEYfor Claude
- Key file in current directory:
openai-api.keygemini-api.keyclaude-api.key
Configuration
Each provider has its own configuration dataclass with sensible defaults:
from msgmodel import OpenAIConfig, GeminiConfig, ClaudeConfig
# OpenAI configuration
openai_config = OpenAIConfig(
model="gpt-4o", # Model to use
temperature=1.0, # Sampling temperature
top_p=1.0, # Nucleus sampling
max_tokens=1000, # Max output tokens
store_data=False, # Don't store data for training
)
# Gemini configuration
gemini_config = GeminiConfig(
model="gemini-2.5-flash",
temperature=1.0,
top_p=0.95,
top_k=40,
safety_threshold="BLOCK_NONE",
)
# Claude configuration
claude_config = ClaudeConfig(
model="claude-sonnet-4-20250514",
temperature=1.0,
top_p=0.95,
top_k=40,
)
Data Retention & Privacy
msgmodel is designed with statelessness as a core principle. Here's what you need to know:
OpenAI (Default: Zero Data Retention)
When using OpenAI with store_data=False (the default):
- What's protected: Input prompts, system instructions, and model responses
- How: The
X-OpenAI-No-Storeheader is automatically added to all Chat Completions requests - Result: OpenAI does not use these interactions for service improvements or model training
- Persistence: Inputs/outputs are not stored beyond the immediate request-response cycle
- File retention: Files uploaded for processing are automatically deleted after the request completes (unless
delete_files_after_use=False)
Important limitations:
- OpenAI's API logs may retain minimal metadata (timestamps, API version, token counts) for ~30 days for debugging purposes, but not the actual content
- Billing records will still show API usage but not interaction content
- Enabling
store_data=Truedisables ZDR and allows OpenAI to use your data for service improvements
Example (ZDR enabled):
from msgmodel import query, OpenAIConfig
config = OpenAIConfig(
store_data=False, # Enables Zero Data Retention (default)
delete_files_after_use=True # Auto-delete uploaded files (default)
)
response = query("openai", "Sensitive prompt", config=config)
Example (Opt-out of ZDR):
from msgmodel import query, OpenAIConfig
config = OpenAIConfig(store_data=True) # Disables ZDR
response = query("openai", "Can use for training", config=config)
Google Gemini (Service-Tier Dependent)
Google Gemini's data retention policy depends on which service tier you use. No API parameter controls this; it's determined by your Google Cloud account configuration.
Unpaid Services (Default: Free Tier, Google AI Studio)
When this applies: You're using the free API quota without Cloud Billing enabled
- What's retained: Prompts, system instructions, and all model responses
- How long: Indefinitely (for model training and product improvement)
- Additional processing: Human reviewers may read and annotate your prompts
- Statelessness: ❌ NOT POSSIBLE — data is fundamentally retained for training
Configuration in msgmodel:
from msgmodel import query, GeminiConfig
# Default (unpaid): Data IS retained for training
config = GeminiConfig(use_paid_api=False) # Default
response = query("gemini", "Your prompt", config=config)
# WARNING: Library will emit warning: "Gemini is configured for UNPAID SERVICES..."
Paid Services (Google Cloud Billing + Paid Quota)
When this applies: Your Google Cloud project has Cloud Billing enabled AND you're using paid API quota
- What's protected: Data is NOT used for model training or product improvement
- What IS retained: Prompts and responses retained temporarily for abuse detection and legal compliance (typically 24-72 hours; exact duration unspecified by Google)
- Human review: ❌ NO (unless abuse is detected)
- Statelessness: ✅ ACHIEVABLE — within abuse monitoring requirements
- Backups: Encrypted backups retained up to 6 months per Google's standard deletion process
Configuration in msgmodel:
from msgmodel import query, GeminiConfig
# Paid services: Data protected from training, used only for abuse monitoring
config = GeminiConfig(use_paid_api=True)
response = query("gemini", "Sensitive prompt", config=config)
# No warning; library assumes you have paid quota active
Important: Setting use_paid_api=True assumes your Google Cloud project has:
- Cloud Billing account linked
- Paid API quota enabled (not on free quota tier)
If this is not the case, Google will apply unpaid service terms regardless of your code setting.
Learn more: Google Gemini API Terms — How Google Uses Your Data
File Handling in Gemini
- Inline files (msgmodel default): Base64-encoded files embedded in each request; no persistent storage; stateless by design ✅
- Google Files API (not used by msgmodel): Would upload to Google's servers; 48-hour auto-delete; encrypted backup up to 6 months
- Verdict: Current inline approach is more privacy-preserving for statelessness goals
Anthropic Claude
- Retention period: Content retained for up to 30 days for abuse prevention
- No configuration available: Google and Anthropic do not provide client-side controls for this
- Statelessness: ❌ NOT ACHIEVABLE — 30-day minimum retention is inherent to the service
See Anthropic Privacy for details.
Summary Comparison
| Provider | Statelessness Achievable | How | Caveat |
|---|---|---|---|
| OpenAI | ✅ YES | store_data=False (default) |
Zero-retention header; metadata ~30 days |
| Gemini (Paid) | ✅ MOSTLY | Cloud Billing + use_paid_api=True |
Abuse monitoring retention ~24-72 hours |
| Gemini (Unpaid) | ❌ NO | No configuration possible | Data retained for training indefinitely |
| Claude | ❌ NO | No configuration possible | 30-day minimum retention |
For maximum privacy across all providers, consider:
- OpenAI with ZDR: True zero-retention option with
store_data=False(default) - Gemini with Paid Services: Near-stateless with Cloud Billing and
use_paid_api=True(abuse monitoring only) - Running models locally (e.g., Ollama, LLaMA)
- Using on-premise deployments
For detailed privacy analysis of Gemini, see GEMINI_PRIVACY_ANALYSIS.md.
Error Handling
The library uses exceptions instead of sys.exit():
from msgmodel import query, MsgModelError, AuthenticationError, APIError
try:
response = query("openai", "Hello")
except AuthenticationError as e:
print(f"API key issue: {e}")
except APIError as e:
print(f"API call failed: {e}")
print(f"Status code: {e.status_code}")
except MsgModelError as e:
print(f"General error: {e}")
Response Object
The query() function returns an LLMResponse object:
response = query("openai", "Hello")
print(response.text) # The generated text
print(response.model) # Model used (e.g., "gpt-4o")
print(response.provider) # Provider name (e.g., "openai")
print(response.usage) # Token usage dict (if available)
print(response.raw_response) # Complete API response
Project Structure
msgModel/
├── msgmodel/ # Python package
│ ├── __init__.py # Public API exports
│ ├── __main__.py # CLI entry point
│ ├── core.py # Core query/stream functions
│ ├── config.py # Configuration dataclasses
│ ├── exceptions.py # Custom exceptions
│ ├── py.typed # PEP 561 marker for typed package
│ └── providers/ # Provider implementations
│ ├── __init__.py
│ ├── openai.py
│ ├── gemini.py
│ └── claude.py
├── tests/ # Test suite
│ ├── test_config.py
│ ├── test_core.py
│ └── test_exceptions.py
├── pyproject.toml # Package configuration
├── LICENSE # MIT License
├── MANIFEST.in # Distribution manifest
├── requirements.txt # Dependencies
└── README.md
CLI Usage
After installation, the msgmodel command is available:
# Basic usage
msgmodel -p openai "What is Python?"
# Or using python -m
python -m msgmodel -p openai "What is Python?"
# Provider shortcuts: o=openai, g=gemini, c=claude
msgmodel -p g "Hello, Gemini!"
msgmodel -p c "Hello, Claude!"
# With streaming
msgmodel -p openai "Tell me a story" --stream
# From a file
msgmodel -p gemini -f prompt.txt
Running Tests
# Install dev dependencies
pip install -e ".[dev]"
# Run tests
pytest
# Run with coverage
pytest --cov=msgmodel
Building & Publishing
# Install build tools
pip install build twine
# Build the package
python -m build
# Check the distribution
twine check dist/*
# Upload to PyPI (requires PyPI account)
twine upload dist/*
# Upload to TestPyPI first (recommended)
twine upload --repository testpypi dist/*
License
MIT License - see LICENSE for details.
Author
Leo Dias (but mostly AI)
Project details
Release history Release notifications | RSS feed
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 msgmodel-3.0.0.tar.gz.
File metadata
- Download URL: msgmodel-3.0.0.tar.gz
- Upload date:
- Size: 29.9 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.11
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
aaedb457961678117f9e8dcb75f3f1bd52b19a20a174462b0120692513c3727a
|
|
| MD5 |
ff1a753f2f6c9510da61801a9fd3667d
|
|
| BLAKE2b-256 |
f2aa4132a0a37678eab94dd4b5e53ae1898ae1b7a4ce929a6c476be71047a833
|
File details
Details for the file msgmodel-3.0.0-py3-none-any.whl.
File metadata
- Download URL: msgmodel-3.0.0-py3-none-any.whl
- Upload date:
- Size: 26.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.11
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
30289877ec2288bfac78aae2058a4dd9847b568a7a9834a1f77b1cda9720417d
|
|
| MD5 |
62c7a73cda0f9ea713e5787c881ca0a7
|
|
| BLAKE2b-256 |
3e768e6b693569c0962978c4576df973a29ff116ea6cd20ecbee1c361ca41f84
|