revenium-python-sdk 0.1.10__tar.gz → 0.3.0__tar.gz

{revenium_python_sdk-0.1.10 → revenium_python_sdk-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: revenium-python-sdk
-Version: 0.1.10
+Version: 0.3.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
@@ -195,7 +195,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import openai
-import revenium_middleware_openai  # Auto-initializes on import
+import revenium_middleware.openai  # Auto-initializes on import
 
 client = openai.OpenAI()
 response = client.chat.completions.create(
@@ -335,7 +335,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import openai
-import revenium_middleware_openai  # Auto-initializes
+import revenium_middleware.openai  # Auto-initializes
 
 client = openai.OpenAI()
 
@@ -374,7 +374,7 @@ The middleware automatically detects Azure OpenAI when using `AzureOpenAI()` and
 
 ```python
 from openai import AzureOpenAI
-import revenium_middleware_openai
+import revenium_middleware.openai
 
 client = AzureOpenAI(
     azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
@@ -407,7 +407,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import anthropic
-import revenium_middleware_anthropic  # Auto-initializes
+import revenium_middleware.anthropic  # Auto-initializes
 
 client = anthropic.Anthropic()
 
@@ -442,7 +442,7 @@ The middleware provides complete AWS Bedrock integration with automatic detectio
 
 ```python
 import anthropic
-import revenium_middleware_anthropic
+import revenium_middleware.anthropic
 
 # Bedrock is automatically detected when AWS credentials are available
 # and base_url contains 'amazonaws.com'
@@ -511,7 +511,7 @@ pip install "revenium-python-sdk[google-vertex]"
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_google
+import revenium_middleware.google
 from google import genai
 
 client = genai.Client()
@@ -532,7 +532,7 @@ print(response.text)
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_google
+import revenium_middleware.google
 import vertexai
 from vertexai.generative_models import GenerativeModel
 
@@ -573,7 +573,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import ollama
-import revenium_middleware_ollama  # Auto-initializes
+import revenium_middleware.ollama  # Auto-initializes
 
 # Chat completion
 response = ollama.chat(
@@ -608,7 +608,7 @@ response = ollama.embed(model='nomic-embed-text', input=['Text 1', 'Text 2', 'Te
 
 ```python
 import openai
-import revenium_middleware_openai
+import revenium_middleware.openai
 
 openai.api_key = 'ollama'
 openai.base_url = 'http://localhost:11434/v1/'
@@ -636,7 +636,7 @@ Supports all LLM providers available through LiteLLM with two integration patter
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_litellm_client.middleware  # Auto-initializes
+import revenium_middleware.litellm.client.middleware  # Auto-initializes
 import litellm
 import os
 
@@ -659,7 +659,7 @@ Add the callback to your LiteLLM `config.yaml` for server-side integration:
 
 ```yaml
 litellm_settings:
-  callbacks: ["revenium_middleware_litellm_proxy.middleware.proxy_handler_instance"]
+  callbacks: ["revenium_middleware.litellm.proxy.middleware.proxy_handler_instance"]
 ```
 
 When using the LiteLLM proxy, pass metadata via HTTP headers (`x-revenium-*`).
@@ -684,7 +684,7 @@ All decorators support static values, extraction from function arguments (`name_
 #### CrewAI Integration
 
 ```bash
-pip install "revenium-middleware-litellm[crewai]"
+pip install "revenium-python-sdk[litellm]" crewai
 ```
 
 Pre-built wrapper for tracking CrewAI agent executions. **Note:** CrewAI requires Python 3.12 or earlier.
@@ -708,7 +708,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 from openai import OpenAI
-import revenium_middleware_perplexity  # Auto-patches OpenAI
+import revenium_middleware.perplexity  # Auto-patches OpenAI
 
 client = OpenAI(
     api_key=os.getenv("PERPLEXITY_API_KEY"),
@@ -726,7 +726,7 @@ response = client.chat.completions.create(
 
 ```python
 from perplexity import Perplexity
-import revenium_middleware_perplexity  # Auto-patches Perplexity
+import revenium_middleware.perplexity  # Auto-patches Perplexity
 
 client = Perplexity(api_key=os.getenv("PERPLEXITY_API_KEY"))
 
@@ -761,7 +761,7 @@ for chunk in stream:
 Supports image, video, and audio generation through fal.ai with automatic media type detection.
 
 ```python
-import revenium_middleware_fal  # Auto-activates
+import revenium_middleware.fal  # Auto-activates
 import fal_client
 
 result = fal_client.subscribe(
@@ -793,16 +793,23 @@ for image in result.get("images", []):
 
 Callback handler that automatically tracks LLM calls, chains, tools, and agent actions.
 
+```bash
+pip install "revenium-python-sdk[langchain]"
+```
+
+Wrap any LangChain LLM (or embeddings model) with `wrap()` — the Revenium callback handler is attached for you:
+
 ```python
 from langchain_openai import ChatOpenAI
-from revenium_middleware_langchain import ReveniumCallbackHandler
+from revenium_middleware.openai.langchain import wrap
 
-handler = ReveniumCallbackHandler(
-    trace_id="session-123",
-    agent_name="support_agent"
+llm = wrap(
+    ChatOpenAI(model="gpt-4o-mini"),
+    usage_metadata={
+        "trace_id": "session-123",
+        "agent": "support_agent",
+    },
 )
-
-llm = ChatOpenAI(model="gpt-4o-mini", callbacks=[handler])
 response = llm.invoke("Hello!")
 ```
 
@@ -813,13 +820,14 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 
 prompt = ChatPromptTemplate.from_template("Tell me a joke about {topic}")
-chain = prompt | llm | output_parser
+chain = prompt | llm | StrOutputParser()
 result = chain.invoke({"topic": "programming"})
 ```
 
 **With agents:**
 
 ```python
+from langchain_core.messages import HumanMessage
 from langchain_core.tools import tool
 from langgraph.prebuilt import create_react_agent
 
@@ -829,40 +837,32 @@ def get_weather(city: str) -> str:
     return f"Sunny, 72F in {city}"
 
 agent = create_react_agent(llm, [get_weather])
-result = agent.invoke(
-    {"messages": [HumanMessage(content="Weather in NYC?")]},
-    config={"callbacks": [handler]}
-)
+result = agent.invoke({"messages": [HumanMessage(content="Weather in NYC?")]})
 ```
 
-**Async support:**
+**Async support:** the handler is async-native — wrap once and use `ainvoke`/`astream` directly:
 
 ```python
-from revenium_middleware_langchain import AsyncReveniumCallbackHandler
-
-handler = AsyncReveniumCallbackHandler(trace_id="async-session")
-llm = ChatOpenAI(model="gpt-4o-mini", callbacks=[handler])
+llm = wrap(ChatOpenAI(model="gpt-4o-mini"))
 response = await llm.ainvoke("Hello!")
 ```
 
 **Supported providers:** OpenAI, Anthropic, Google, AWS Bedrock, Azure OpenAI, Cohere, HuggingFace, Ollama. Provider is auto-detected from LangChain class name or model name prefix.
 
-**Programmatic configuration:**
+**Attaching to an existing LLM:** use `attach_to()` to add tracking in-place, with any of the standard metadata fields (see [Metadata Fields](#metadata-fields)):
 
 ```python
-from revenium_middleware_langchain import ReveniumCallbackHandler, ReveniumConfig, SubscriberConfig
-
-config = ReveniumConfig(
-    api_key="hak_your_api_key",
-    environment="production",
-    organization_name="my_org",
-    product_name="my_product",
-    subscriber=SubscriberConfig(id="user_123", email="user@example.com"),
-)
+from revenium_middleware.openai.langchain import attach_to
 
-handler = ReveniumCallbackHandler(config=config, trace_id="session-123")
+attach_to(llm, usage_metadata={
+    "organizationName": "my_org",
+    "productName": "my_product",
+    "subscriber": {"id": "user_123", "email": "user@example.com"},
+})
 ```
 
+Credentials come from the standard environment variables (`REVENIUM_METERING_API_KEY`, `REVENIUM_METERING_BASE_URL`) or `revenium_middleware.configure()`.
+
 ---
 
 ## Metadata Fields
@@ -1179,7 +1179,7 @@ Each field has a maximum length of **50,000 characters**. If exceeded, it's trun
 import os
 os.environ["REVENIUM_CAPTURE_PROMPTS"] = "true"
 
-import revenium_middleware_openai
+import revenium_middleware.openai
 from openai import OpenAI
 
 client = OpenAI()
@@ -1316,6 +1316,44 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 |----------|-------------|
 | `REVENIUM_METERING_API_KEY` | Your Revenium API key (starts with `hak_` or `rev_`) |
 
+### Configuring After Import
+
+The metering client is normally built from the environment when
+`revenium_middleware` is first imported. If your credentials only become
+available later (a secrets-vault bootstrap, framework settings hooks, import
+ordering), you don't need to restart: as soon as `REVENIUM_METERING_API_KEY`
+appears in the environment, the next metered call picks it up automatically.
+You can also configure programmatically at any time:
+
+```python
+import revenium_middleware
+
+revenium_middleware.initialize_metering(
+    api_key="hak_your_key",                    # defaults to REVENIUM_METERING_API_KEY
+    base_url="https://api.revenium.ai",        # defaults to REVENIUM_METERING_BASE_URL
+)
+```
+
+`initialize_metering()` returns `True` when metering is enabled after the
+call; invoke it with no arguments to re-read the environment.
+
+### Delivery Resilience (Store-and-Forward)
+
+Metering events that still fail after the client's own retries (network
+outages, 5xx, rate limiting) are not lost: they are held in a bounded
+in-memory buffer and replayed automatically in the background every 30
+seconds, reusing each event's original `Idempotency-Key` so replays can
+never double-bill. Permanent failures (401/403/404/422) are never buffered.
+The buffer holds up to 1000 events (oldest evicted first) for at most 24
+hours, and is drained on graceful shutdown. Inspect it programmatically:
+
+```python
+from revenium_middleware import get_buffer_stats
+
+print(get_buffer_stats())
+# {'size': 0, 'max_size': 1000, 'total_buffered': 3, 'total_replayed': 3, ...}
+```
+
 ### Optional Environment Variables
 
 | Variable | Default | Description |
@@ -1334,6 +1372,8 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 | `REVENIUM_TRANSACTION_NAME` | - | Human-friendly operation name |
 | `REVENIUM_RETRY_NUMBER` | - | Retry attempt number |
 | `REVENIUM_BEDROCK_DISABLE` | - | Set to `1` to disable Bedrock auto-detection |
+| `REVENIUM_BUFFER_MAX_SIZE` | `1000` | Store-and-forward buffer capacity (oldest events evicted when full) |
+| `REVENIUM_BUFFER_FLUSH_INTERVAL` | `30` | Seconds between automatic replay attempts for buffered events |
 
 ### Provider-Specific Environment Variables
 
@@ -1376,7 +1416,7 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 
 **Force direct Anthropic API:** Set `REVENIUM_BEDROCK_DISABLE=1` to disable Bedrock auto-detection.
 
-**Check initialization status:** Use `revenium_middleware_<provider>.is_initialized()` to verify setup.
+**Check initialization status (Anthropic):** Use `revenium_middleware.anthropic.is_initialized()` to verify setup.
 
 ---
 

{revenium_python_sdk-0.1.10 → revenium_python_sdk-0.3.0}/README.md

@@ -102,7 +102,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import openai
-import revenium_middleware_openai  # Auto-initializes on import
+import revenium_middleware.openai  # Auto-initializes on import
 
 client = openai.OpenAI()
 response = client.chat.completions.create(
@@ -242,7 +242,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import openai
-import revenium_middleware_openai  # Auto-initializes
+import revenium_middleware.openai  # Auto-initializes
 
 client = openai.OpenAI()
 
@@ -281,7 +281,7 @@ The middleware automatically detects Azure OpenAI when using `AzureOpenAI()` and
 
 ```python
 from openai import AzureOpenAI
-import revenium_middleware_openai
+import revenium_middleware.openai
 
 client = AzureOpenAI(
     azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT"),
@@ -314,7 +314,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import anthropic
-import revenium_middleware_anthropic  # Auto-initializes
+import revenium_middleware.anthropic  # Auto-initializes
 
 client = anthropic.Anthropic()
 
@@ -349,7 +349,7 @@ The middleware provides complete AWS Bedrock integration with automatic detectio
 
 ```python
 import anthropic
-import revenium_middleware_anthropic
+import revenium_middleware.anthropic
 
 # Bedrock is automatically detected when AWS credentials are available
 # and base_url contains 'amazonaws.com'
@@ -418,7 +418,7 @@ pip install "revenium-python-sdk[google-vertex]"
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_google
+import revenium_middleware.google
 from google import genai
 
 client = genai.Client()
@@ -439,7 +439,7 @@ print(response.text)
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_google
+import revenium_middleware.google
 import vertexai
 from vertexai.generative_models import GenerativeModel
 
@@ -480,7 +480,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 import ollama
-import revenium_middleware_ollama  # Auto-initializes
+import revenium_middleware.ollama  # Auto-initializes
 
 # Chat completion
 response = ollama.chat(
@@ -515,7 +515,7 @@ response = ollama.embed(model='nomic-embed-text', input=['Text 1', 'Text 2', 'Te
 
 ```python
 import openai
-import revenium_middleware_openai
+import revenium_middleware.openai
 
 openai.api_key = 'ollama'
 openai.base_url = 'http://localhost:11434/v1/'
@@ -543,7 +543,7 @@ Supports all LLM providers available through LiteLLM with two integration patter
 from dotenv import load_dotenv
 load_dotenv()
 
-import revenium_middleware_litellm_client.middleware  # Auto-initializes
+import revenium_middleware.litellm.client.middleware  # Auto-initializes
 import litellm
 import os
 
@@ -566,7 +566,7 @@ Add the callback to your LiteLLM `config.yaml` for server-side integration:
 
 ```yaml
 litellm_settings:
-  callbacks: ["revenium_middleware_litellm_proxy.middleware.proxy_handler_instance"]
+  callbacks: ["revenium_middleware.litellm.proxy.middleware.proxy_handler_instance"]
 ```
 
 When using the LiteLLM proxy, pass metadata via HTTP headers (`x-revenium-*`).
@@ -591,7 +591,7 @@ All decorators support static values, extraction from function arguments (`name_
 #### CrewAI Integration
 
 ```bash
-pip install "revenium-middleware-litellm[crewai]"
+pip install "revenium-python-sdk[litellm]" crewai
 ```
 
 Pre-built wrapper for tracking CrewAI agent executions. **Note:** CrewAI requires Python 3.12 or earlier.
@@ -615,7 +615,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 from openai import OpenAI
-import revenium_middleware_perplexity  # Auto-patches OpenAI
+import revenium_middleware.perplexity  # Auto-patches OpenAI
 
 client = OpenAI(
     api_key=os.getenv("PERPLEXITY_API_KEY"),
@@ -633,7 +633,7 @@ response = client.chat.completions.create(
 
 ```python
 from perplexity import Perplexity
-import revenium_middleware_perplexity  # Auto-patches Perplexity
+import revenium_middleware.perplexity  # Auto-patches Perplexity
 
 client = Perplexity(api_key=os.getenv("PERPLEXITY_API_KEY"))
 
@@ -668,7 +668,7 @@ for chunk in stream:
 Supports image, video, and audio generation through fal.ai with automatic media type detection.
 
 ```python
-import revenium_middleware_fal  # Auto-activates
+import revenium_middleware.fal  # Auto-activates
 import fal_client
 
 result = fal_client.subscribe(
@@ -700,16 +700,23 @@ for image in result.get("images", []):
 
 Callback handler that automatically tracks LLM calls, chains, tools, and agent actions.
 
+```bash
+pip install "revenium-python-sdk[langchain]"
+```
+
+Wrap any LangChain LLM (or embeddings model) with `wrap()` — the Revenium callback handler is attached for you:
+
 ```python
 from langchain_openai import ChatOpenAI
-from revenium_middleware_langchain import ReveniumCallbackHandler
+from revenium_middleware.openai.langchain import wrap
 
-handler = ReveniumCallbackHandler(
-    trace_id="session-123",
-    agent_name="support_agent"
+llm = wrap(
+    ChatOpenAI(model="gpt-4o-mini"),
+    usage_metadata={
+        "trace_id": "session-123",
+        "agent": "support_agent",
+    },
 )
-
-llm = ChatOpenAI(model="gpt-4o-mini", callbacks=[handler])
 response = llm.invoke("Hello!")
 ```
 
@@ -720,13 +727,14 @@ from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 
 prompt = ChatPromptTemplate.from_template("Tell me a joke about {topic}")
-chain = prompt | llm | output_parser
+chain = prompt | llm | StrOutputParser()
 result = chain.invoke({"topic": "programming"})
 ```
 
 **With agents:**
 
 ```python
+from langchain_core.messages import HumanMessage
 from langchain_core.tools import tool
 from langgraph.prebuilt import create_react_agent
 
@@ -736,40 +744,32 @@ def get_weather(city: str) -> str:
     return f"Sunny, 72F in {city}"
 
 agent = create_react_agent(llm, [get_weather])
-result = agent.invoke(
-    {"messages": [HumanMessage(content="Weather in NYC?")]},
-    config={"callbacks": [handler]}
-)
+result = agent.invoke({"messages": [HumanMessage(content="Weather in NYC?")]})
 ```
 
-**Async support:**
+**Async support:** the handler is async-native — wrap once and use `ainvoke`/`astream` directly:
 
 ```python
-from revenium_middleware_langchain import AsyncReveniumCallbackHandler
-
-handler = AsyncReveniumCallbackHandler(trace_id="async-session")
-llm = ChatOpenAI(model="gpt-4o-mini", callbacks=[handler])
+llm = wrap(ChatOpenAI(model="gpt-4o-mini"))
 response = await llm.ainvoke("Hello!")
 ```
 
 **Supported providers:** OpenAI, Anthropic, Google, AWS Bedrock, Azure OpenAI, Cohere, HuggingFace, Ollama. Provider is auto-detected from LangChain class name or model name prefix.
 
-**Programmatic configuration:**
+**Attaching to an existing LLM:** use `attach_to()` to add tracking in-place, with any of the standard metadata fields (see [Metadata Fields](#metadata-fields)):
 
 ```python
-from revenium_middleware_langchain import ReveniumCallbackHandler, ReveniumConfig, SubscriberConfig
-
-config = ReveniumConfig(
-    api_key="hak_your_api_key",
-    environment="production",
-    organization_name="my_org",
-    product_name="my_product",
-    subscriber=SubscriberConfig(id="user_123", email="user@example.com"),
-)
+from revenium_middleware.openai.langchain import attach_to
 
-handler = ReveniumCallbackHandler(config=config, trace_id="session-123")
+attach_to(llm, usage_metadata={
+    "organizationName": "my_org",
+    "productName": "my_product",
+    "subscriber": {"id": "user_123", "email": "user@example.com"},
+})
 ```
 
+Credentials come from the standard environment variables (`REVENIUM_METERING_API_KEY`, `REVENIUM_METERING_BASE_URL`) or `revenium_middleware.configure()`.
+
 ---
 
 ## Metadata Fields
@@ -1086,7 +1086,7 @@ Each field has a maximum length of **50,000 characters**. If exceeded, it's trun
 import os
 os.environ["REVENIUM_CAPTURE_PROMPTS"] = "true"
 
-import revenium_middleware_openai
+import revenium_middleware.openai
 from openai import OpenAI
 
 client = OpenAI()
@@ -1223,6 +1223,44 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 |----------|-------------|
 | `REVENIUM_METERING_API_KEY` | Your Revenium API key (starts with `hak_` or `rev_`) |
 
+### Configuring After Import
+
+The metering client is normally built from the environment when
+`revenium_middleware` is first imported. If your credentials only become
+available later (a secrets-vault bootstrap, framework settings hooks, import
+ordering), you don't need to restart: as soon as `REVENIUM_METERING_API_KEY`
+appears in the environment, the next metered call picks it up automatically.
+You can also configure programmatically at any time:
+
+```python
+import revenium_middleware
+
+revenium_middleware.initialize_metering(
+    api_key="hak_your_key",                    # defaults to REVENIUM_METERING_API_KEY
+    base_url="https://api.revenium.ai",        # defaults to REVENIUM_METERING_BASE_URL
+)
+```
+
+`initialize_metering()` returns `True` when metering is enabled after the
+call; invoke it with no arguments to re-read the environment.
+
+### Delivery Resilience (Store-and-Forward)
+
+Metering events that still fail after the client's own retries (network
+outages, 5xx, rate limiting) are not lost: they are held in a bounded
+in-memory buffer and replayed automatically in the background every 30
+seconds, reusing each event's original `Idempotency-Key` so replays can
+never double-bill. Permanent failures (401/403/404/422) are never buffered.
+The buffer holds up to 1000 events (oldest evicted first) for at most 24
+hours, and is drained on graceful shutdown. Inspect it programmatically:
+
+```python
+from revenium_middleware import get_buffer_stats
+
+print(get_buffer_stats())
+# {'size': 0, 'max_size': 1000, 'total_buffered': 3, 'total_replayed': 3, ...}
+```
+
 ### Optional Environment Variables
 
 | Variable | Default | Description |
@@ -1241,6 +1279,8 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 | `REVENIUM_TRANSACTION_NAME` | - | Human-friendly operation name |
 | `REVENIUM_RETRY_NUMBER` | - | Retry attempt number |
 | `REVENIUM_BEDROCK_DISABLE` | - | Set to `1` to disable Bedrock auto-detection |
+| `REVENIUM_BUFFER_MAX_SIZE` | `1000` | Store-and-forward buffer capacity (oldest events evicted when full) |
+| `REVENIUM_BUFFER_FLUSH_INTERVAL` | `30` | Seconds between automatic replay attempts for buffered events |
 
 ### Provider-Specific Environment Variables
 
@@ -1283,7 +1323,7 @@ See [`examples/openai/openai_blocking_demo.py`](examples/openai/openai_blocking_
 
 **Force direct Anthropic API:** Set `REVENIUM_BEDROCK_DISABLE=1` to disable Bedrock auto-detection.
 
-**Check initialization status:** Use `revenium_middleware_<provider>.is_initialized()` to verify setup.
+**Check initialization status (Anthropic):** Use `revenium_middleware.anthropic.is_initialized()` to verify setup.
 
 ---
 

{revenium_python_sdk-0.1.10 → revenium_python_sdk-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "revenium-python-sdk"
-version = "0.1.10"
+version = "0.3.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"

{revenium_python_sdk-0.1.10 → revenium_python_sdk-0.3.0}/revenium_middleware/__init__.py

@@ -138,6 +138,9 @@ logger.propagate = True
 # Existing imports like `from revenium_middleware import client` keep working.
 from ._core import (  # noqa: E402
     client,
+    get_buffer_stats,
+    get_client,
+    initialize_metering,
     run_async_in_thread,
     shutdown_event,
     revenium_meter,
@@ -166,6 +169,9 @@ from .agentic_outcomes import AgenticOutcomeClient, AgenticOutcomeSettings  # no
 __all__ = [
     # Metering exports
     "client",
+    "get_buffer_stats",
+    "get_client",
+    "initialize_metering",
     "run_async_in_thread",
     "shutdown_event",
     # Decorator exports

{revenium_python_sdk-0.1.10 → revenium_python_sdk-0.3.0}/revenium_middleware/_core/__init__.py

@@ -5,7 +5,8 @@ This subpackage contains the foundational components shared across all
 provider-specific middleware implementations.
 """
 
-from .metering import run_async_in_thread, shutdown_event, client
+from .metering import run_async_in_thread, shutdown_event, client, get_client, initialize_metering
+from .metering_buffer import get_buffer_stats
 from .metering_submission import submit_ai_event
 from .exceptions import BudgetExceededError
 from .enforcement import check_enforcement, is_circuit_breaker_enabled, stop_polling
@@ -50,6 +51,9 @@ from .trace_fields import (
 __all__ = [
     # Metering
     "client",
+    "get_client",
+    "initialize_metering",
+    "get_buffer_stats",
     "run_async_in_thread",
     "shutdown_event",
     "submit_ai_event",