revenium-python-sdk 0.4.0__tar.gz → 0.5.0__tar.gz

{revenium_python_sdk-0.4.0 → revenium_python_sdk-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: revenium-python-sdk
-Version: 0.4.0
+Version: 0.5.0
 Summary: The official Revenium Python SDK — unified AI metering middleware for OpenAI, Anthropic, Google, Ollama, LiteLLM, Perplexity, and fal.ai.
 Author-email: Revenium <support@revenium.io>
 License: MIT
@@ -36,6 +36,9 @@ 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: griptape
+Requires-Dist: griptape<2.0.0,>=1.0.0; python_version >= "3.10" and extra == "griptape"
+Requires-Dist: requests>=2.25.0; extra == "griptape"
 Provides-Extra: perplexity
 Requires-Dist: wrapt>=1.14.0; extra == "perplexity"
 Requires-Dist: python-dotenv>=0.19.0; extra == "perplexity"
@@ -89,6 +92,7 @@ Requires-Dist: python-dotenv>=0.19.0; extra == "dev"
 Requires-Dist: anthropic>=0.30.0; extra == "dev"
 Requires-Dist: boto3; extra == "dev"
 Requires-Dist: ollama>=0.3.0; extra == "dev"
+Requires-Dist: griptape<2.0.0,>=1.0.0; python_version >= "3.10" and extra == "dev"
 Dynamic: license-file
 
 # Revenium Python SDK
@@ -132,6 +136,7 @@ The official Revenium Python SDK — unified AI metering middleware for deeply a
 | Perplexity (Native SDK) | `perplexity-native` | `pip install revenium-python-sdk[perplexity-native]` |
 | fal.ai | `fal` | `pip install revenium-python-sdk[fal]` |
 | LangChain | `langchain` | `pip install revenium-python-sdk[langchain]` |
+| Griptape | `griptape` | `pip install revenium-python-sdk[griptape]` |
 
 ## Feature Matrix
 
@@ -147,6 +152,7 @@ The official Revenium Python SDK — unified AI metering middleware for deeply a
 | Terminal Summary | Yes | Yes | Yes | Yes | Yes | - | - |
 | Azure / Bedrock | Azure | Bedrock | Vertex AI | - | All | - | - |
 | LangChain Integration | Yes | - | - | - | - | - | - |
+| Griptape Integration | Yes | Yes | - | Yes | Yes | - | - |
 | CrewAI Integration | - | - | - | - | Yes | - | - |
 | Proxy Mode | - | - | - | - | Yes | - | - |
 
@@ -959,6 +965,49 @@ Credentials come from the standard environment variables (`REVENIUM_METERING_API
 
 ---
 
+### Griptape
+
+Metered prompt and embedding drivers for [Griptape](https://github.com/griptape-ai/griptape) applications. Requires Python 3.10+.
+
+```bash
+pip install "revenium-python-sdk[griptape,openai]"     # OpenAI
+pip install "revenium-python-sdk[griptape,anthropic]"  # Anthropic
+pip install "revenium-python-sdk[griptape,ollama]"     # Ollama
+pip install "revenium-python-sdk[griptape,litellm,litellm-proxy]"  # 100+ providers via LiteLLM
+```
+
+`ReveniumDriver` auto-detects the provider from the model name (`gpt-*` → OpenAI, `claude-*` → Anthropic, `llama`/`mistral`/... → Ollama, anything else → LiteLLM) and wraps the matching Griptape prompt driver with Revenium metering:
+
+```python
+import os
+from griptape.structures import Agent
+from revenium_middleware.griptape import ReveniumDriver
+
+os.environ["REVENIUM_METERING_API_KEY"] = "your_revenium_key"
+
+agent = Agent(prompt_driver=ReveniumDriver(
+    model="gpt-4o-mini",
+    usage_metadata={"task_type": "demo"},
+))
+agent.run("Hello!")
+```
+
+Force a provider with `force_provider="litellm"`, or wrap an existing driver with `ReveniumDriver(base_driver=...)`.
+
+**Embeddings:**
+
+```python
+from revenium_middleware.griptape import ReveniumEmbeddingDriver
+
+driver = ReveniumEmbeddingDriver(model="text-embedding-3-large")
+```
+
+**Provider-specific drivers:** for direct control, use `ReveniumOpenAiDriver`, `ReveniumAnthropicDriver`, `ReveniumOllamaDriver`, `ReveniumLiteLLMDriver` or `ReveniumOpenAiEmbeddingDriver` — each subclasses the corresponding Griptape driver and accepts a `usage_metadata` dict (see [Metadata Fields](#metadata-fields)).
+
+**Migrating from `revenium-griptape`:** the standalone package is deprecated — install the `griptape` extra and change `from revenium_griptape import ReveniumDriver` to `from revenium_middleware.griptape import ReveniumDriver`. All driver class names are unchanged. One behaviour difference: the old package called `load_dotenv()` automatically at import time; the SDK never mutates your environment on import, so if you keep credentials in a `.env` file, call `load_dotenv()` yourself before creating a driver.
+
+---
+
 ## Metadata Fields
 
 Add business context to any API call by passing a `usage_metadata` dictionary. All fields are optional.

{revenium_python_sdk-0.4.0 → revenium_python_sdk-0.5.0}/README.md

@@ -39,6 +39,7 @@ The official Revenium Python SDK — unified AI metering middleware for deeply a
 | Perplexity (Native SDK) | `perplexity-native` | `pip install revenium-python-sdk[perplexity-native]` |
 | fal.ai | `fal` | `pip install revenium-python-sdk[fal]` |
 | LangChain | `langchain` | `pip install revenium-python-sdk[langchain]` |
+| Griptape | `griptape` | `pip install revenium-python-sdk[griptape]` |
 
 ## Feature Matrix
 
@@ -54,6 +55,7 @@ The official Revenium Python SDK — unified AI metering middleware for deeply a
 | Terminal Summary | Yes | Yes | Yes | Yes | Yes | - | - |
 | Azure / Bedrock | Azure | Bedrock | Vertex AI | - | All | - | - |
 | LangChain Integration | Yes | - | - | - | - | - | - |
+| Griptape Integration | Yes | Yes | - | Yes | Yes | - | - |
 | CrewAI Integration | - | - | - | - | Yes | - | - |
 | Proxy Mode | - | - | - | - | Yes | - | - |
 
@@ -866,6 +868,49 @@ Credentials come from the standard environment variables (`REVENIUM_METERING_API
 
 ---
 
+### Griptape
+
+Metered prompt and embedding drivers for [Griptape](https://github.com/griptape-ai/griptape) applications. Requires Python 3.10+.
+
+```bash
+pip install "revenium-python-sdk[griptape,openai]"     # OpenAI
+pip install "revenium-python-sdk[griptape,anthropic]"  # Anthropic
+pip install "revenium-python-sdk[griptape,ollama]"     # Ollama
+pip install "revenium-python-sdk[griptape,litellm,litellm-proxy]"  # 100+ providers via LiteLLM
+```
+
+`ReveniumDriver` auto-detects the provider from the model name (`gpt-*` → OpenAI, `claude-*` → Anthropic, `llama`/`mistral`/... → Ollama, anything else → LiteLLM) and wraps the matching Griptape prompt driver with Revenium metering:
+
+```python
+import os
+from griptape.structures import Agent
+from revenium_middleware.griptape import ReveniumDriver
+
+os.environ["REVENIUM_METERING_API_KEY"] = "your_revenium_key"
+
+agent = Agent(prompt_driver=ReveniumDriver(
+    model="gpt-4o-mini",
+    usage_metadata={"task_type": "demo"},
+))
+agent.run("Hello!")
+```
+
+Force a provider with `force_provider="litellm"`, or wrap an existing driver with `ReveniumDriver(base_driver=...)`.
+
+**Embeddings:**
+
+```python
+from revenium_middleware.griptape import ReveniumEmbeddingDriver
+
+driver = ReveniumEmbeddingDriver(model="text-embedding-3-large")
+```
+
+**Provider-specific drivers:** for direct control, use `ReveniumOpenAiDriver`, `ReveniumAnthropicDriver`, `ReveniumOllamaDriver`, `ReveniumLiteLLMDriver` or `ReveniumOpenAiEmbeddingDriver` — each subclasses the corresponding Griptape driver and accepts a `usage_metadata` dict (see [Metadata Fields](#metadata-fields)).
+
+**Migrating from `revenium-griptape`:** the standalone package is deprecated — install the `griptape` extra and change `from revenium_griptape import ReveniumDriver` to `from revenium_middleware.griptape import ReveniumDriver`. All driver class names are unchanged. One behaviour difference: the old package called `load_dotenv()` automatically at import time; the SDK never mutates your environment on import, so if you keep credentials in a `.env` file, call `load_dotenv()` yourself before creating a driver.
+
+---
+
 ## Metadata Fields
 
 Add business context to any API call by passing a `usage_metadata` dictionary. All fields are optional.

{revenium_python_sdk-0.4.0 → revenium_python_sdk-0.5.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "revenium-python-sdk"
-version = "0.4.0"
+version = "0.5.0"
 description = "The official Revenium Python SDK — unified AI metering middleware for OpenAI, Anthropic, Google, Ollama, LiteLLM, Perplexity, and fal.ai."
 readme = "README.md"
 requires-python = ">=3.8"
@@ -51,6 +51,10 @@ langchain = [
     "langchain-openai>=0.1.0",
     "langchain-core>=0.1.0",
 ]
+griptape = [
+    "griptape>=1.0.0,<2.0.0; python_version >= '3.10'",
+    "requests>=2.25.0",
+]
 perplexity = [
     "wrapt>=1.14.0",
     "python-dotenv>=0.19.0",
@@ -116,6 +120,7 @@ dev = [
     "anthropic>=0.30.0",
     "boto3",
     "ollama>=0.3.0",
+    "griptape>=1.0.0,<2.0.0; python_version >= '3.10'",
 ]
 
 [tool.pytest.ini_options]

revenium_python_sdk-0.5.0/revenium_middleware/griptape/__init__.py

@@ -0,0 +1,35 @@
+"""
+Revenium metering drivers for the Griptape framework.
+
+Universal support for LLM providers through the unified Revenium SDK.
+Install with: pip install "revenium-python-sdk[griptape]"
+
+Supported providers:
+- Tier 1 (direct): OpenAI, Anthropic, Ollama
+- Tier 2 (via LiteLLM): Google/Gemini, Cohere, Azure OpenAI, Bedrock, and 100+ others
+"""
+try:
+    import griptape  # noqa: F401
+except ImportError as exc:  # pragma: no cover
+    raise ImportError(
+        "The griptape package is required for revenium_middleware.griptape. "
+        "Install it with: pip install 'revenium-python-sdk[griptape]' "
+        "(requires Python 3.10+)."
+    ) from exc
+
+from .universal_driver import ReveniumDriver, ReveniumEmbeddingDriver
+from .openai_driver import ReveniumOpenAiDriver
+from .openai_embedding_driver import ReveniumOpenAiEmbeddingDriver
+from .anthropic_driver import ReveniumAnthropicDriver
+from .ollama_driver import ReveniumOllamaDriver
+from .litellm_driver import ReveniumLiteLLMDriver
+
+__all__ = [
+    "ReveniumDriver",
+    "ReveniumEmbeddingDriver",
+    "ReveniumOpenAiDriver",
+    "ReveniumOpenAiEmbeddingDriver",
+    "ReveniumAnthropicDriver",
+    "ReveniumOllamaDriver",
+    "ReveniumLiteLLMDriver",
+]

revenium_python_sdk-0.5.0/revenium_middleware/griptape/_metadata.py

@@ -0,0 +1,10 @@
+"""Shared usage_metadata helpers for the Griptape drivers."""
+
+# Keys that would corrupt the middleware's authentication state if they
+# reached a provider API call.
+_AUTH_KEYS = frozenset({"revenium_api_key", "revenium_api_base_url"})
+
+
+def strip_revenium_auth_keys(metadata):
+    """Return a copy of metadata without Revenium auth keys."""
+    return {k: v for k, v in metadata.items() if k not in _AUTH_KEYS}

revenium_python_sdk-0.5.0/revenium_middleware/griptape/anthropic_driver.py

@@ -0,0 +1,87 @@
+"""
+Revenium driver for Anthropic/Claude.
+
+This driver extends the Griptape AnthropicPromptDriver to automatically inject
+Revenium usage metadata for tracking and cost monitoring.
+"""
+import logging
+from typing import Dict, Any, Optional
+
+from griptape.drivers.prompt.anthropic import AnthropicPromptDriver
+from griptape.common.prompt_stack.prompt_stack import PromptStack
+
+from ._metadata import strip_revenium_auth_keys
+
+# Import the Revenium middleware for Anthropic - this auto-patches the anthropic client
+try:
+    import revenium_middleware.anthropic  # noqa: F401
+    MIDDLEWARE_AVAILABLE = True
+except ImportError:
+    MIDDLEWARE_AVAILABLE = False
+    logging.warning(
+        "Anthropic middleware could not be loaded. "
+        "Install it with: pip install 'revenium-python-sdk[anthropic]'"
+    )
+
+logger = logging.getLogger(__name__)
+
+
+class ReveniumAnthropicDriver(AnthropicPromptDriver):
+    """
+    Anthropic prompt driver with Revenium usage metering.
+    
+    This driver automatically injects usage metadata into Anthropic API calls
+    for tracking and cost monitoring through Revenium.
+    
+    Usage metadata is passed through to the LLM API after stripping the
+    Revenium auth keys (revenium_api_key, revenium_api_base_url). Do not
+    include other secrets in usage_metadata.
+    """
+
+    def __init__(self, usage_metadata: Optional[Dict[str, Any]] = None, **kwargs):
+        """
+        Initialize the Revenium Anthropic driver.
+        
+        Args:
+            usage_metadata: Optional metadata for tracking usage (user_id, task_type, etc.)
+            **kwargs: All other arguments passed to AnthropicPromptDriver
+        """
+        # Store usage metadata as instance variable
+        self.usage_metadata = usage_metadata or {}
+        
+        # Initialize parent without usage_metadata
+        super().__init__(**kwargs)
+        
+        if not MIDDLEWARE_AVAILABLE:
+            logger.warning(
+                "Running without the Anthropic middleware. "
+                "Usage metering will not be available."
+            )
+
+    def _base_params(self, prompt_stack: PromptStack) -> Dict[str, Any]:
+        """
+        Build the base parameters for the Anthropic API call.
+        
+        This method extends the base implementation to inject Revenium
+        usage metadata while preserving all other functionality.
+        
+        Args:
+            prompt_stack: The prompt stack for the request
+            
+        Returns:
+            Parameters dictionary for the Anthropic API call
+        """
+        # Get the base parameters from the parent class
+        params = super()._base_params(prompt_stack)
+        
+        # Only inject metadata if middleware is available and we have metadata
+        if MIDDLEWARE_AVAILABLE and self.usage_metadata:
+            # Copy only business metadata; auth credentials are excluded to
+            # prevent corruption of the middleware's authentication state.
+            clean_metadata = strip_revenium_auth_keys(self.usage_metadata)
+            if clean_metadata:
+                # Inject the clean metadata directly into the API call parameters
+                params['usage_metadata'] = clean_metadata
+                logger.debug(f"Injected usage metadata: {list(clean_metadata.keys())}")
+
+        return params