revenium-python-sdk 0.1.0__tar.gz

revenium_python_sdk-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+The MIT License
+
+Copyright (c) 2025 Revenium, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.

revenium_python_sdk-0.1.0/PKG-INFO

@@ -0,0 +1,252 @@
+Metadata-Version: 2.4
+Name: revenium-python-sdk
+Version: 0.1.0
+Summary: The official Revenium Python SDK — unified AI metering middleware for OpenAI, Anthropic, Google, Ollama, LiteLLM, and Perplexity.
+Author-email: Revenium <support@revenium.io>
+License: MIT
+Project-URL: Homepage, https://github.com/revenium/revenium-python-sdk
+Project-URL: Bug Tracker, https://github.com/revenium/revenium-python-sdk/issues
+Project-URL: Documentation, https://docs.revenium.io
+Keywords: sdk,ai,llm,middleware,metering,revenium,openai,anthropic,google,ollama,litellm,perplexity
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: revenium_metering>=6.8.2
+Provides-Extra: openai
+Requires-Dist: wrapt; extra == "openai"
+Requires-Dist: openai>=1.0.0; extra == "openai"
+Requires-Dist: python-dotenv>=0.19.0; extra == "openai"
+Provides-Extra: langchain
+Requires-Dist: langchain>=0.1.16; extra == "langchain"
+Requires-Dist: langchain-openai>=0.1.0; extra == "langchain"
+Requires-Dist: langchain-core>=0.1.0; extra == "langchain"
+Provides-Extra: perplexity
+Requires-Dist: wrapt; extra == "perplexity"
+Requires-Dist: python-dotenv>=0.19.0; extra == "perplexity"
+Provides-Extra: perplexity-openai
+Requires-Dist: wrapt; extra == "perplexity-openai"
+Requires-Dist: openai>=1.0.0; extra == "perplexity-openai"
+Requires-Dist: python-dotenv>=0.19.0; extra == "perplexity-openai"
+Provides-Extra: perplexity-native
+Requires-Dist: wrapt; extra == "perplexity-native"
+Requires-Dist: perplexityai>=0.1.0; extra == "perplexity-native"
+Requires-Dist: python-dotenv>=0.19.0; extra == "perplexity-native"
+Provides-Extra: google
+Requires-Dist: wrapt; extra == "google"
+Provides-Extra: google-genai
+Requires-Dist: wrapt; extra == "google-genai"
+Requires-Dist: google-genai>=0.1.0; extra == "google-genai"
+Requires-Dist: python-dotenv; extra == "google-genai"
+Provides-Extra: google-vertex
+Requires-Dist: wrapt; extra == "google-vertex"
+Requires-Dist: vertexai>=1.0.0; extra == "google-vertex"
+Requires-Dist: python-dotenv; extra == "google-vertex"
+Provides-Extra: anthropic
+Requires-Dist: wrapt; extra == "anthropic"
+Requires-Dist: anthropic; extra == "anthropic"
+Requires-Dist: python-dotenv>=0.19.0; extra == "anthropic"
+Provides-Extra: ollama
+Requires-Dist: wrapt; extra == "ollama"
+Requires-Dist: ollama; extra == "ollama"
+Provides-Extra: litellm
+Requires-Dist: wrapt; extra == "litellm"
+Requires-Dist: litellm; extra == "litellm"
+Provides-Extra: litellm-proxy
+Requires-Dist: wrapt; extra == "litellm-proxy"
+Requires-Dist: litellm[proxy]; extra == "litellm-proxy"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
+Requires-Dist: pytest-cov; extra == "dev"
+Requires-Dist: flake8; extra == "dev"
+Requires-Dist: black; extra == "dev"
+Requires-Dist: mypy; extra == "dev"
+Requires-Dist: freezegun; extra == "dev"
+Requires-Dist: openai-responses>=0.12.0; extra == "dev"
+Requires-Dist: requests; extra == "dev"
+Requires-Dist: wrapt; extra == "dev"
+Requires-Dist: python-dotenv>=0.19.0; extra == "dev"
+Requires-Dist: anthropic; extra == "dev"
+Requires-Dist: boto3; extra == "dev"
+Requires-Dist: ollama; extra == "dev"
+Dynamic: license-file
+
+# Revenium Python SDK
+
+[![PyPI version](https://img.shields.io/pypi/v/revenium-python-sdk.svg)](https://pypi.org/project/revenium-python-sdk/)
+[![Python Versions](https://img.shields.io/pypi/pyversions/revenium-python-sdk.svg)](https://pypi.org/project/revenium-python-sdk/)
+[![Documentation](https://img.shields.io/badge/docs-revenium.io-blue)](https://docs.revenium.io)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+The official Revenium Python SDK — unified AI metering middleware for deeply attributed AI usage metrics. Supports OpenAI, Anthropic, Google (Gemini/Vertex AI), Ollama, LiteLLM, and Perplexity.
+
+## Features
+
+- **Unified SDK**: Single package with middleware for all major AI providers
+- **Asynchronous Processing**: Background thread management for non-blocking metering operations
+- **Graceful Shutdown**: Ensures all metering data is properly sent even during application shutdown
+- **Decorator Support**: `@revenium_meter` and `@revenium_metadata` for easy integration
+- **Tool Metering**: Meter arbitrary tool/function calls alongside LLM API metering
+
+## Supported Providers
+
+| Provider | Extra | Install Command |
+|----------|-------|----------------|
+| OpenAI | `openai` | `pip install revenium-python-sdk[openai]` |
+| Anthropic | `anthropic` | `pip install revenium-python-sdk[anthropic]` |
+| Google Gemini | `google-genai` | `pip install revenium-python-sdk[google-genai]` |
+| Google Vertex AI | `google-vertex` | `pip install revenium-python-sdk[google-vertex]` |
+| Ollama | `ollama` | `pip install revenium-python-sdk[ollama]` |
+| LiteLLM | `litellm` | `pip install revenium-python-sdk[litellm]` |
+| LiteLLM Proxy | `litellm-proxy` | `pip install revenium-python-sdk[litellm-proxy]` |
+| Perplexity (OpenAI) | `perplexity-openai` | `pip install revenium-python-sdk[perplexity-openai]` |
+| Perplexity (Native) | `perplexity-native` | `pip install revenium-python-sdk[perplexity-native]` |
+| LangChain | `langchain` | `pip install revenium-python-sdk[langchain]` |
+
+## Installation
+
+```bash
+# Core SDK
+pip install revenium-python-sdk
+
+# With a specific provider
+pip install revenium-python-sdk[openai]
+
+# Multiple providers
+pip install revenium-python-sdk[openai,anthropic,ollama]
+```
+
+## Quick Start
+
+```python
+from revenium_middleware import client, run_async_in_thread, shutdown_event
+
+# Record usage directly
+client.record_usage(
+    model="gpt-4o",
+    prompt_tokens=500,
+    completion_tokens=200,
+    user_id="user123",
+    session_id="session456"
+)
+
+# Run async metering tasks in background threads
+async def async_metering_task():
+    await client.async_record_usage(
+        model="gpt-3.5-turbo",
+        prompt_tokens=300,
+        completion_tokens=150,
+        user_id="user789"
+    )
+
+thread = run_async_in_thread(async_metering_task())
+
+# Application continues while metering happens in background
+```
+
+## Provider-Specific Usage
+
+Each provider has its own middleware module. See the `examples/` directory for detailed usage:
+
+- `examples/openai/` — OpenAI and Azure OpenAI examples
+- `examples/anthropic/` — Anthropic and Bedrock examples
+- `examples/google/` — Google AI and Vertex AI examples
+- `examples/ollama/` — Ollama examples
+- `examples/litellm/` — LiteLLM client and proxy examples
+- `examples/perplexity/` — Perplexity examples
+
+## Tool Metering
+
+The `meter_tool` decorator lets you meter arbitrary tool/function calls (web scrapers, image generators, database lookups, etc.) alongside your LLM API metering. This is available via `revenium_metering` v6.8.2+.
+
+```python
+from revenium_middleware import meter_tool, configure
+
+# Configure the metering client
+configure(
+    metering_url="https://api.revenium.io/meter",
+    api_key="your-api-key",
+)
+
+# Decorate any tool function to automatically meter it
+@meter_tool("my-web-scraper", operation="scrape")
+def scrape_website(url):
+    # Your scraping logic here
+    return {"pages": 5, "data_mb": 2.3}
+
+# The decorator captures timing, success/failure, and reports to Revenium
+result = scrape_website("https://example.com")
+```
+
+You can also report tool calls manually:
+
+```python
+from revenium_middleware import report_tool_call
+
+report_tool_call(
+    tool_id="my-tool",
+    operation="fetch",
+    duration_ms=1234,
+    success=True,
+    usage_metadata={"records": 42},
+)
+```
+
+## Compatibility
+
+- Python 3.8+
+- Compatible with all supported AI providers
+
+## Logging
+
+This module uses Python's standard logging system. You can control the log level by setting the `REVENIUM_LOG_LEVEL` environment variable:
+
+```bash
+# Enable debug logging
+export REVENIUM_LOG_LEVEL=DEBUG
+
+# Or when running your script
+REVENIUM_LOG_LEVEL=DEBUG python your_script.py
+```
+
+Available log levels:
+- `DEBUG`: Detailed debugging information
+- `INFO`: General information (default)
+- `WARNING`: Warning messages only
+- `ERROR`: Error messages only
+- `CRITICAL`: Critical error messages only
+
+## Documentation
+
+For detailed documentation, visit [docs.revenium.io](https://docs.revenium.io)
+
+## Contributing
+
+See [CONTRIBUTING.md](./CONTRIBUTING.md)
+
+## Code of Conduct
+
+See [CODE_OF_CONDUCT.md](./CODE_OF_CONDUCT.md)
+
+## Security
+
+See [SECURITY.md](./SECURITY.md)
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](./LICENSE) file for details.
+
+## Acknowledgments
+
+- Built by the Revenium team

revenium_python_sdk-0.1.0/README.md

@@ -0,0 +1,168 @@
+# Revenium Python SDK
+
+[![PyPI version](https://img.shields.io/pypi/v/revenium-python-sdk.svg)](https://pypi.org/project/revenium-python-sdk/)
+[![Python Versions](https://img.shields.io/pypi/pyversions/revenium-python-sdk.svg)](https://pypi.org/project/revenium-python-sdk/)
+[![Documentation](https://img.shields.io/badge/docs-revenium.io-blue)](https://docs.revenium.io)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+The official Revenium Python SDK — unified AI metering middleware for deeply attributed AI usage metrics. Supports OpenAI, Anthropic, Google (Gemini/Vertex AI), Ollama, LiteLLM, and Perplexity.
+
+## Features
+
+- **Unified SDK**: Single package with middleware for all major AI providers
+- **Asynchronous Processing**: Background thread management for non-blocking metering operations
+- **Graceful Shutdown**: Ensures all metering data is properly sent even during application shutdown
+- **Decorator Support**: `@revenium_meter` and `@revenium_metadata` for easy integration
+- **Tool Metering**: Meter arbitrary tool/function calls alongside LLM API metering
+
+## Supported Providers
+
+| Provider | Extra | Install Command |
+|----------|-------|----------------|
+| OpenAI | `openai` | `pip install revenium-python-sdk[openai]` |
+| Anthropic | `anthropic` | `pip install revenium-python-sdk[anthropic]` |
+| Google Gemini | `google-genai` | `pip install revenium-python-sdk[google-genai]` |
+| Google Vertex AI | `google-vertex` | `pip install revenium-python-sdk[google-vertex]` |
+| Ollama | `ollama` | `pip install revenium-python-sdk[ollama]` |
+| LiteLLM | `litellm` | `pip install revenium-python-sdk[litellm]` |
+| LiteLLM Proxy | `litellm-proxy` | `pip install revenium-python-sdk[litellm-proxy]` |
+| Perplexity (OpenAI) | `perplexity-openai` | `pip install revenium-python-sdk[perplexity-openai]` |
+| Perplexity (Native) | `perplexity-native` | `pip install revenium-python-sdk[perplexity-native]` |
+| LangChain | `langchain` | `pip install revenium-python-sdk[langchain]` |
+
+## Installation
+
+```bash
+# Core SDK
+pip install revenium-python-sdk
+
+# With a specific provider
+pip install revenium-python-sdk[openai]
+
+# Multiple providers
+pip install revenium-python-sdk[openai,anthropic,ollama]
+```
+
+## Quick Start
+
+```python
+from revenium_middleware import client, run_async_in_thread, shutdown_event
+
+# Record usage directly
+client.record_usage(
+    model="gpt-4o",
+    prompt_tokens=500,
+    completion_tokens=200,
+    user_id="user123",
+    session_id="session456"
+)
+
+# Run async metering tasks in background threads
+async def async_metering_task():
+    await client.async_record_usage(
+        model="gpt-3.5-turbo",
+        prompt_tokens=300,
+        completion_tokens=150,
+        user_id="user789"
+    )
+
+thread = run_async_in_thread(async_metering_task())
+
+# Application continues while metering happens in background
+```
+
+## Provider-Specific Usage
+
+Each provider has its own middleware module. See the `examples/` directory for detailed usage:
+
+- `examples/openai/` — OpenAI and Azure OpenAI examples
+- `examples/anthropic/` — Anthropic and Bedrock examples
+- `examples/google/` — Google AI and Vertex AI examples
+- `examples/ollama/` — Ollama examples
+- `examples/litellm/` — LiteLLM client and proxy examples
+- `examples/perplexity/` — Perplexity examples
+
+## Tool Metering
+
+The `meter_tool` decorator lets you meter arbitrary tool/function calls (web scrapers, image generators, database lookups, etc.) alongside your LLM API metering. This is available via `revenium_metering` v6.8.2+.
+
+```python
+from revenium_middleware import meter_tool, configure
+
+# Configure the metering client
+configure(
+    metering_url="https://api.revenium.io/meter",
+    api_key="your-api-key",
+)
+
+# Decorate any tool function to automatically meter it
+@meter_tool("my-web-scraper", operation="scrape")
+def scrape_website(url):
+    # Your scraping logic here
+    return {"pages": 5, "data_mb": 2.3}
+
+# The decorator captures timing, success/failure, and reports to Revenium
+result = scrape_website("https://example.com")
+```
+
+You can also report tool calls manually:
+
+```python
+from revenium_middleware import report_tool_call
+
+report_tool_call(
+    tool_id="my-tool",
+    operation="fetch",
+    duration_ms=1234,
+    success=True,
+    usage_metadata={"records": 42},
+)
+```
+
+## Compatibility
+
+- Python 3.8+
+- Compatible with all supported AI providers
+
+## Logging
+
+This module uses Python's standard logging system. You can control the log level by setting the `REVENIUM_LOG_LEVEL` environment variable:
+
+```bash
+# Enable debug logging
+export REVENIUM_LOG_LEVEL=DEBUG
+
+# Or when running your script
+REVENIUM_LOG_LEVEL=DEBUG python your_script.py
+```
+
+Available log levels:
+- `DEBUG`: Detailed debugging information
+- `INFO`: General information (default)
+- `WARNING`: Warning messages only
+- `ERROR`: Error messages only
+- `CRITICAL`: Critical error messages only
+
+## Documentation
+
+For detailed documentation, visit [docs.revenium.io](https://docs.revenium.io)
+
+## Contributing
+
+See [CONTRIBUTING.md](./CONTRIBUTING.md)
+
+## Code of Conduct
+
+See [CODE_OF_CONDUCT.md](./CODE_OF_CONDUCT.md)
+
+## Security
+
+See [SECURITY.md](./SECURITY.md)
+
+## License
+
+This project is licensed under the MIT License - see the [LICENSE](./LICENSE) file for details.
+
+## Acknowledgments
+
+- Built by the Revenium team

revenium_python_sdk-0.1.0/pyproject.toml

@@ -0,0 +1,120 @@
+[build-system]
+requires = ["setuptools>=42", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "revenium-python-sdk"
+version = "0.1.0"
+description = "The official Revenium Python SDK — unified AI metering middleware for OpenAI, Anthropic, Google, Ollama, LiteLLM, and Perplexity."
+readme = "README.md"
+requires-python = ">=3.8"
+license = { text = "MIT" }
+authors = [
+    { name = "Revenium", email = "support@revenium.io" }
+]
+dependencies = [
+    "revenium_metering>=6.8.2"
+]
+keywords = ["sdk", "ai", "llm", "middleware", "metering", "revenium", "openai", "anthropic", "google", "ollama", "litellm", "perplexity"]
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+    "Intended Audience :: Developers",
+    "Topic :: Software Development :: Libraries :: Python Modules"
+]
+
+[project.urls]
+"Homepage" = "https://github.com/revenium/revenium-python-sdk"
+"Bug Tracker" = "https://github.com/revenium/revenium-python-sdk/issues"
+"Documentation" = "https://docs.revenium.io"
+
+[project.optional-dependencies]
+openai = [
+    "wrapt",
+    "openai>=1.0.0",
+    "python-dotenv>=0.19.0",
+]
+langchain = [
+    "langchain>=0.1.16",
+    "langchain-openai>=0.1.0",
+    "langchain-core>=0.1.0",
+]
+perplexity = [
+    "wrapt",
+    "python-dotenv>=0.19.0",
+]
+perplexity-openai = [
+    "wrapt",
+    "openai>=1.0.0",
+    "python-dotenv>=0.19.0",
+]
+perplexity-native = [
+    "wrapt",
+    "perplexityai>=0.1.0",
+    "python-dotenv>=0.19.0",
+]
+google = [
+    "wrapt",
+]
+google-genai = [
+    "wrapt",
+    "google-genai>=0.1.0",
+    "python-dotenv",
+]
+google-vertex = [
+    "wrapt",
+    "vertexai>=1.0.0",
+    "python-dotenv",
+]
+anthropic = [
+    "wrapt",
+    "anthropic",
+    "python-dotenv>=0.19.0",
+]
+ollama = [
+    "wrapt",
+    "ollama",
+]
+litellm = [
+    "wrapt",
+    "litellm",
+]
+litellm-proxy = [
+    "wrapt",
+    "litellm[proxy]",
+]
+
+dev = [
+    "pytest>=7.0.0",
+    "pytest-asyncio",
+    "pytest-cov",
+    "flake8",
+    "black",
+    "mypy",
+    "freezegun",
+    "openai-responses>=0.12.0",
+    "requests",
+    "wrapt",
+    "python-dotenv>=0.19.0",
+    "anthropic",
+    "boto3",
+    "ollama",
+]
+
+[tool.pytest.ini_options]
+markers = [
+    "unit: Unit tests (fast, no external dependencies)",
+    "e2e: end-to-end tests requiring live services",
+    "integration: integration tests requiring live services",
+]
+addopts = "-m 'not e2e and not integration'"
+
+[tool.setuptools.packages.find]
+include = ["revenium_middleware*"]