payi 0.1.0a60__tar.gz → 0.1.0a62__tar.gz

payi-0.1.0a62/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.62"
+}

{payi-0.1.0a60 → payi-0.1.0a62}/CHANGELOG.md

@@ -1,5 +1,21 @@
 # Changelog
 
+## 0.1.0-alpha.62 (2025-04-07)
+
+Full Changelog: [v0.1.0-alpha.61...v0.1.0-alpha.62](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.61...v0.1.0-alpha.62)
+
+### Features
+
+* feat: auto generate use case id ([b50f807](https://github.com/Pay-i/pay-i-python/commit/b50f807b571c917d39f37a1879bf2ea7e7f5d29d))
+
+## 0.1.0-alpha.61 (2025-04-07)
+
+Full Changelog: [v0.1.0-alpha.60...v0.1.0-alpha.61](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.60...v0.1.0-alpha.61)
+
+### Features
+
+* feat(api): global instrumentation ([6977d0c](https://github.com/Pay-i/pay-i-python/commit/6977d0c78ffaa12bb43a439827b0fba08b258e55))
+
 ## 0.1.0-alpha.60 (2025-04-07)
 
 Full Changelog: [v0.1.0-alpha.59...v0.1.0-alpha.60](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.59...v0.1.0-alpha.60)

{payi-0.1.0a60 → payi-0.1.0a62}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a60
+Version: 0.1.0a62
 Summary: The official Python library for the payi API
 Project-URL: Homepage, https://github.com/Pay-i/pay-i-python
 Project-URL: Repository, https://github.com/Pay-i/pay-i-python

{payi-0.1.0a60 → payi-0.1.0a62}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "payi"
-version = "0.1.0-alpha.60"
+version = "0.1.0-alpha.62"
 description = "The official Python library for the payi API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "payi"
-__version__ = "0.1.0-alpha.60"  # x-release-please-version
+__version__ = "0.1.0-alpha.62"  # x-release-please-version

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/lib/AnthropicInstrumentor.py

@@ -7,12 +7,12 @@ from wrapt import wrap_function_wrapper  # type: ignore
 from payi.types import IngestUnitsParams
 from payi.types.ingest_units_params import Units
 
-from .instrument import IsStreaming, PayiInstrumentor
+from .instrument import _IsStreaming, _PayiInstrumentor
 
 
 class AnthropicIntrumentor:
     @staticmethod
-    def instrument(instrumentor: PayiInstrumentor) -> None:
+    def instrument(instrumentor: _PayiInstrumentor) -> None:
         try:
             import anthropic  # type: ignore #  noqa: F401  I001
 
@@ -45,9 +45,9 @@ class AnthropicIntrumentor:
             return
 
 
-@PayiInstrumentor.payi_wrapper
+@_PayiInstrumentor.payi_wrapper
 def chat_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -58,16 +58,16 @@ def chat_wrapper(
         process_chunk,
         process_request,
         process_synchronous_response,
-        IsStreaming.kwargs,
+        _IsStreaming.kwargs,
         wrapped,
         instance,
         args,
         kwargs,
     )
 
-@PayiInstrumentor.payi_awrapper
+@_PayiInstrumentor.payi_awrapper
 async def achat_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -78,7 +78,7 @@ async def achat_wrapper(
         process_chunk,
         process_request,
         process_synchronous_response,
-        IsStreaming.kwargs,
+        _IsStreaming.kwargs,
         wrapped,
         instance,
         args,
@@ -93,7 +93,7 @@ def process_chunk(chunk: Any, ingest: IngestUnitsParams) -> None:
         usage = chunk.message.usage
         units = ingest["units"]
 
-        input = PayiInstrumentor.update_for_vision(usage.input_tokens, units)
+        input = _PayiInstrumentor.update_for_vision(usage.input_tokens, units)
 
         units["text"] = Units(input=input, output=0)
 
@@ -124,7 +124,7 @@ def process_synchronous_response(response: Any, ingest: IngestUnitsParams, log_p
         text_cache_read = usage.cache_read_input_tokens
         units["text_cache_read"] = Units(input=text_cache_read, output=0)
 
-    input = PayiInstrumentor.update_for_vision(input, units)
+    input = _PayiInstrumentor.update_for_vision(input, units)
 
     units["text"] = Units(input=input, output=output)
 
@@ -165,4 +165,4 @@ def process_request(ingest: IngestUnitsParams, *args: Any, **kwargs: Any) -> Non
     if not has_image or estimated_token_count == 0:
         return
 
-    ingest["units"][PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)
+    ingest["units"][_PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/lib/BedrockInstrumentor.py

@@ -8,12 +8,12 @@ from wrapt import ObjectProxy, wrap_function_wrapper  # type: ignore
 from payi.types.ingest_units_params import Units, IngestUnitsParams
 from payi.types.pay_i_common_models_api_router_header_info_param import PayICommonModelsAPIRouterHeaderInfoParam
 
-from .instrument import IsStreaming, PayiInstrumentor
+from .instrument import _IsStreaming, _PayiInstrumentor
 
 
 class BedrockInstrumentor:
     @staticmethod
-    def instrument(instrumentor: PayiInstrumentor) -> None:
+    def instrument(instrumentor: _PayiInstrumentor) -> None:
         try:
             import boto3  # type: ignore #  noqa: F401  I001
 
@@ -33,8 +33,8 @@ class BedrockInstrumentor:
             logging.debug(f"Error instrumenting bedrock: {e}")
             return
 
-@PayiInstrumentor.payi_wrapper
-def create_client_wrapper(instrumentor: PayiInstrumentor, wrapped: Any, instance: Any, *args: Any, **kwargs: Any) -> Any: #  noqa: ARG001
+@_PayiInstrumentor.payi_wrapper
+def create_client_wrapper(instrumentor: _PayiInstrumentor, wrapped: Any, instance: Any, *args: Any, **kwargs: Any) -> Any: #  noqa: ARG001
     if kwargs.get("service_name") != "bedrock-runtime":
         return wrapped(*args, **kwargs)
 
@@ -55,7 +55,7 @@ class InvokeResponseWrapper(ObjectProxy): # type: ignore
     def __init__(
         self,
         response: Any,
-        instrumentor: PayiInstrumentor,
+        instrumentor: _PayiInstrumentor,
         ingest: IngestUnitsParams,
         log_prompt_and_response: bool
         ) -> None:
@@ -95,7 +95,7 @@ class InvokeResponseWrapper(ObjectProxy): # type: ignore
 
         return data
 
-def wrap_invoke(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
+def wrap_invoke(instrumentor: _PayiInstrumentor, wrapped: Any) -> Any:
     @wraps(wrapped)
     def invoke_wrapper(*args: Any, **kwargs: 'dict[str, Any]') -> Any:
         modelId:str = kwargs.get("modelId", "") # type: ignore
@@ -106,7 +106,7 @@ def wrap_invoke(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
                 None,
                 process_invoke_request,  
                 process_synchronous_invoke_response,  
-                IsStreaming.false,
+                _IsStreaming.false,
                 wrapped,
                 None,
                 args,
@@ -116,7 +116,7 @@ def wrap_invoke(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
     
     return invoke_wrapper
 
-def wrap_invoke_stream(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
+def wrap_invoke_stream(instrumentor: _PayiInstrumentor, wrapped: Any) -> Any:
     @wraps(wrapped)
     def invoke_wrapper(*args: Any, **kwargs: Any) -> Any:
         modelId:str = kwargs.get("modelId", "") # type: ignore
@@ -127,7 +127,7 @@ def wrap_invoke_stream(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
                 process_invoke_streaming_anthropic_chunk if modelId.startswith("anthropic.") else process_invoke_streaming_llama_chunk,
                 process_invoke_request, 
                 None,  
-                IsStreaming.true,
+                _IsStreaming.true,
                 wrapped,
                 None,
                 args,
@@ -137,7 +137,7 @@ def wrap_invoke_stream(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
 
     return invoke_wrapper
 
-def wrap_converse(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
+def wrap_converse(instrumentor: _PayiInstrumentor, wrapped: Any) -> Any:
     @wraps(wrapped)
     def invoke_wrapper(*args: Any, **kwargs: 'dict[str, Any]') -> Any:
         modelId:str = kwargs.get("modelId", "") # type: ignore
@@ -148,7 +148,7 @@ def wrap_converse(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
                 None,
                 process_converse_request,  
                 process_synchronous_converse_response,  
-                IsStreaming.false,
+                _IsStreaming.false,
                 wrapped,
                 None,
                 args,
@@ -158,7 +158,7 @@ def wrap_converse(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
     
     return invoke_wrapper
 
-def wrap_converse_stream(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
+def wrap_converse_stream(instrumentor: _PayiInstrumentor, wrapped: Any) -> Any:
     @wraps(wrapped)
     def invoke_wrapper(*args: Any, **kwargs: Any) -> Any:
         modelId:str = kwargs.get("modelId", "") # type: ignore
@@ -169,7 +169,7 @@ def wrap_converse_stream(instrumentor: PayiInstrumentor, wrapped: Any) -> Any:
                 process_converse_streaming_chunk,
                 process_converse_request, 
                 None,  
-                IsStreaming.true,
+                _IsStreaming.true,
                 wrapped,
                 None,
                 args,
@@ -187,7 +187,7 @@ def process_invoke_streaming_anthropic_chunk(chunk: str, ingest: IngestUnitsPara
         usage = chunk_dict['message']['usage']
         units = ingest["units"]
 
-        input = PayiInstrumentor.update_for_vision(usage['input_tokens'], units)
+        input = _PayiInstrumentor.update_for_vision(usage['input_tokens'], units)
 
         units["text"] = Units(input=input, output=0)
 
@@ -215,7 +215,7 @@ def process_synchronous_invoke_response(
         response: Any,
         ingest: IngestUnitsParams,
         log_prompt_and_response: bool,
-        instrumentor: PayiInstrumentor,
+        instrumentor: _PayiInstrumentor,
         **kargs: Any) -> Any: #  noqa: ARG001
 
     metadata = response.get("ResponseMetadata", {})

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/lib/OpenAIInstrumentor.py

@@ -9,7 +9,7 @@ from wrapt import wrap_function_wrapper  # type: ignore
 from payi.types import IngestUnitsParams
 from payi.types.ingest_units_params import Units
 
-from .instrument import IsStreaming, PayiInstrumentor
+from .instrument import _IsStreaming, _PayiInstrumentor
 
 
 class OpenAiInstrumentor:
@@ -20,7 +20,7 @@ class OpenAiInstrumentor:
         return isinstance(instance._client, (AsyncAzureOpenAI, AzureOpenAI))
 
     @staticmethod
-    def instrument(instrumentor: PayiInstrumentor) -> None:
+    def instrument(instrumentor: _PayiInstrumentor) -> None:
         try:
             from openai import OpenAI  # type: ignore #  noqa: F401  I001
             
@@ -53,9 +53,9 @@ class OpenAiInstrumentor:
             return
 
 
-@PayiInstrumentor.payi_wrapper
+@_PayiInstrumentor.payi_wrapper
 def embeddings_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -66,16 +66,16 @@ def embeddings_wrapper(
         None, # process_chat_chunk,
         None, # process_chat_request,
         process_ebmeddings_synchronous_response,
-        IsStreaming.false,
+        _IsStreaming.false,
         wrapped,
         instance,
         args,
         kwargs,
     )
 
-@PayiInstrumentor.payi_wrapper
+@_PayiInstrumentor.payi_wrapper
 async def aembeddings_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -86,16 +86,16 @@ async def aembeddings_wrapper(
         None, # process_chat_chunk,
         None, # process_chat_request,
         process_ebmeddings_synchronous_response,
-        IsStreaming.false,
+        _IsStreaming.false,
         wrapped,
         instance,
         args,
         kwargs,
     )
 
-@PayiInstrumentor.payi_wrapper
+@_PayiInstrumentor.payi_wrapper
 def chat_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -106,16 +106,16 @@ def chat_wrapper(
         process_chat_chunk,
         process_chat_request,
         process_chat_synchronous_response,
-        IsStreaming.kwargs,
+        _IsStreaming.kwargs,
         wrapped,
         instance,
         args,
         kwargs,
     )
 
-@PayiInstrumentor.payi_awrapper
+@_PayiInstrumentor.payi_awrapper
 async def achat_wrapper(
-    instrumentor: PayiInstrumentor,
+    instrumentor: _PayiInstrumentor,
     wrapped: Any,
     instance: Any,
     *args: Any,
@@ -126,7 +126,7 @@ async def achat_wrapper(
         process_chat_chunk,
         process_chat_request,
         process_chat_synchronous_response,
-        IsStreaming.kwargs,
+        _IsStreaming.kwargs,
         wrapped,
         instance,
         args,
@@ -184,7 +184,7 @@ def add_usage_units(usage: "dict[str, Any]", units: "dict[str, Units]") -> None:
         if input_cache != 0:
             units["text_cache_read"] = Units(input=input_cache, output=0)
 
-    input = PayiInstrumentor.update_for_vision(input - input_cache, units)
+    input = _PayiInstrumentor.update_for_vision(input - input_cache, units)
 
     units["text"] = Units(input=input, output=output)
 
@@ -221,4 +221,4 @@ def process_chat_request(ingest: IngestUnitsParams, *args: Any, **kwargs: Any) -
     if not has_image or estimated_token_count == 0:
         return
 
-    ingest["units"][PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)
+    ingest["units"][_PayiInstrumentor.estimated_prompt_tokens] = Units(input=estimated_token_count, output=0)

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/lib/helpers.py

@@ -1,3 +1,4 @@
+import os
 from typing import Dict, List, Union
 
 PAYI_BASE_URL = "https://api.pay-i.com"
@@ -69,22 +70,25 @@ def create_headers(
         headers.update({ PayiHeaderNames.use_case_version: str(use_case_version)})
     return headers
 
+def _resolve_payi_base_url(payi_base_url: Union[str, None]) -> str:
+    if payi_base_url:
+        return payi_base_url
+
+    payi_base_url = os.environ.get("PAYI_BASE_URL", None)
+
+    if payi_base_url:
+        return payi_base_url
+
+    return PAYI_BASE_URL
+
 def payi_anthropic_url(payi_base_url: Union[str, None] = None) -> str:
-    if payi_base_url is None:
-        payi_base_url = PAYI_BASE_URL
-    return payi_base_url + "/api/v1/proxy/anthropic"
+    return _resolve_payi_base_url(payi_base_url=payi_base_url) + "/api/v1/proxy/anthropic"
 
 def payi_openai_url(payi_base_url: Union[str, None] = None) -> str:
-    if payi_base_url is None:
-        payi_base_url = PAYI_BASE_URL
-    return payi_base_url +  "/api/v1/proxy/openai/v1"
+    return _resolve_payi_base_url(payi_base_url=payi_base_url) +  "/api/v1/proxy/openai/v1"
 
 def payi_azure_openai_url(payi_base_url: Union[str, None] = None) -> str:
-    if payi_base_url is None:
-        payi_base_url = PAYI_BASE_URL
-    return payi_base_url + "/api/v1/proxy/azure.openai"
+    return _resolve_payi_base_url(payi_base_url=payi_base_url) + "/api/v1/proxy/azure.openai"
 
 def payi_aws_bedrock_url(payi_base_url: Union[str, None] = None) -> str:
-    if payi_base_url is None:
-        payi_base_url = PAYI_BASE_URL
-    return payi_base_url + "/api/v1/proxy/aws.bedrock"
+    return _resolve_payi_base_url(payi_base_url=payi_base_url) + "/api/v1/proxy/aws.bedrock"

{payi-0.1.0a60 → payi-0.1.0a62}/src/payi/lib/instrument.py

@@ -21,7 +21,18 @@ from .helpers import PayiCategories
 from .Stopwatch import Stopwatch
 
 
-class Context(TypedDict, total=False):
+class PayiInstrumentConfig(TypedDict, total=False):
+    proxy: bool
+    limit_ids: Optional["list[str]"]
+    request_tags: Optional["list[str]"]
+    experience_name: Optional[str]
+    experience_id: Optional[str]
+    use_case_name: Optional[str]
+    use_case_id: Optional[str]
+    use_case_version: Optional[int]
+    user_id: Optional[str]
+
+class _Context(TypedDict, total=False):
     proxy: bool
     experience_name: Optional[str]
     experience_id: Optional[str]
@@ -32,7 +43,7 @@ class Context(TypedDict, total=False):
     request_tags: Optional['list[str]']
     user_id: Optional[str]
 
-class ParentState(TypedDict, total=False):
+class _ParentState(TypedDict, total=False):
     experience_name: Optional[str]
     experience_id: Optional[str]
     use_case_name: Optional[str]
@@ -41,12 +52,12 @@ class ParentState(TypedDict, total=False):
     limit_ids: Optional['list[str]']
     request_tags: Optional['list[str]']
 
-class IsStreaming(Enum):
+class _IsStreaming(Enum):
     false = 0
     true = 1 
     kwargs = 2
 
-class PayiInstrumentor:
+class _PayiInstrumentor:
     estimated_prompt_tokens: str = "estimated_prompt_tokens"
 
     def __init__(
@@ -61,7 +72,12 @@ class PayiInstrumentor:
     ):
         self._payi: Optional[Payi] = payi
         self._apayi: Optional[AsyncPayi] = apayi
-        self._context_stack: list[Context] = []  # Stack of context dictionaries
+
+        if not self._payi and not self._apayi:
+            self._payi = Payi()
+            self._apayi = AsyncPayi()
+
+        self._context_stack: list[_Context] = []  # Stack of context dictionaries
         self._log_prompt_and_response: bool = log_prompt_and_response
         self._prompt_and_response_logger: Optional[Callable[[str, dict[str, str]], None]] = prompt_and_response_logger
 
@@ -233,9 +249,9 @@ class PayiInstrumentor:
 
     def _setup_call_func(
         self
-        ) -> 'tuple[Context, ParentState]':
-        context: Context = {}
-        parentState: ParentState = {}
+        ) -> 'tuple[_Context, _ParentState]':
+        context: _Context = {}
+        parentState: _ParentState = {}
 
         if len(self._context_stack) > 0:
             # copy current context into the upcoming context
@@ -253,8 +269,8 @@ class PayiInstrumentor:
 
     def _init_context(
         self,
-        context: Context,
-        parentState: ParentState,
+        context: _Context,
+        parentState: _ParentState,
         proxy: bool,
         limit_ids: Optional["list[str]"],
         request_tags: Optional["list[str]"],
@@ -395,12 +411,12 @@ class PayiInstrumentor:
         if self._context_stack:
             self._context_stack.pop()
 
-    def set_context(self, context: Context) -> None:
+    def set_context(self, context: _Context) -> None:
         # Update the current top of the stack with the provided context
         if self._context_stack:
             self._context_stack[-1].update(context)
 
-    def get_context(self) -> Optional[Context]:
+    def get_context(self) -> Optional[_Context]:
         # Return the current top of the stack
         return self._context_stack[-1] if self._context_stack else None
 
@@ -461,7 +477,7 @@ class PayiInstrumentor:
         process_chunk: Optional[Callable[[Any, IngestUnitsParams], None]],
         process_request: Optional[Callable[[IngestUnitsParams, Any, Any], None]],
         process_synchronous_response: Any,
-        is_streaming: IsStreaming,
+        is_streaming: _IsStreaming,
         wrapped: Any,
         instance: Any,
         args: Any,
@@ -523,9 +539,9 @@ class PayiInstrumentor:
         sw = Stopwatch()
         stream: bool = False
         
-        if is_streaming == IsStreaming.kwargs:
+        if is_streaming == _IsStreaming.kwargs:
             stream = kwargs.get("stream", False)
-        elif is_streaming == IsStreaming.true:
+        elif is_streaming == _IsStreaming.true:
             stream = True
         else:
             stream = False
@@ -581,7 +597,7 @@ class PayiInstrumentor:
         process_chunk: Optional[Callable[[Any, IngestUnitsParams], None]],
         process_request: Optional[Callable[[IngestUnitsParams, Any, Any], None]],
         process_synchronous_response: Any,
-        is_streaming: IsStreaming,
+        is_streaming: _IsStreaming,
         wrapped: Any,
         instance: Any,
         args: Any,
@@ -652,9 +668,9 @@ class PayiInstrumentor:
         sw = Stopwatch()
         stream: bool = False
         
-        if is_streaming == IsStreaming.kwargs:
+        if is_streaming == _IsStreaming.kwargs:
             stream = kwargs.get("stream", False)
-        elif is_streaming == IsStreaming.true:
+        elif is_streaming == _IsStreaming.true:
             stream = True
         else:
             stream = False
@@ -713,7 +729,7 @@ class PayiInstrumentor:
 
     @staticmethod
     def _update_headers(
-        context: Context,
+        context: _Context,
         extra_headers: "dict[str, str]",
     ) -> None:
         limit_ids: Optional[list[str]] = context.get("limit_ids")
@@ -797,8 +813,8 @@ class PayiInstrumentor:
 
     @staticmethod
     def update_for_vision(input: int, units: 'dict[str, Units]') -> int:
-        if PayiInstrumentor.estimated_prompt_tokens in units:
-            prompt_token_estimate: int = units.pop(PayiInstrumentor.estimated_prompt_tokens)["input"] # type: ignore
+        if _PayiInstrumentor.estimated_prompt_tokens in units:
+            prompt_token_estimate: int = units.pop(_PayiInstrumentor.estimated_prompt_tokens)["input"] # type: ignore
             vision = input - prompt_token_estimate
             if (vision > 0):
                 units["vision"] = Units(input=vision, output=0)
@@ -843,7 +859,7 @@ class ChatStreamWrapper(ObjectProxy):  # type: ignore
         self,
         response: Any,
         instance: Any,
-        instrumentor: PayiInstrumentor,
+        instrumentor: _PayiInstrumentor,
         ingest: IngestUnitsParams,
         stopwatch: Stopwatch,
         process_chunk: Optional[Callable[[Any, IngestUnitsParams], None]] = None,
@@ -977,13 +993,14 @@ class ChatStreamWrapper(ObjectProxy):  # type: ignore
             return json.dumps(chunk)
 
 global _instrumentor
-_instrumentor: Optional[PayiInstrumentor] = None
+_instrumentor: Optional[_PayiInstrumentor] = None
 
 def payi_instrument(
     payi: Optional[Union[Payi, AsyncPayi, 'list[Union[Payi, AsyncPayi]]']] = None,
     instruments: Optional[Set[str]] = None,
     log_prompt_and_response: bool = True,
     prompt_and_response_logger: Optional[Callable[[str, "dict[str, str]"], None]] = None,
+    config: Optional[PayiInstrumentConfig] = None,
 ) -> None:
     global _instrumentor
     if _instrumentor:
@@ -1002,10 +1019,21 @@ def payi_instrument(
                 payi_param = p
             elif isinstance(p, AsyncPayi): # type: ignore
                 apayi_param = p
+    
+    global_context: _Context = {}
 
-    # allow for both payi and apayi to be None for the @proxy case
+    if config and len(config) > 0:
+        if "proxy" not in config:
+            config["proxy"] = False
+        global_context = {**config}
+
+    # create the clients on their behalf if not provided for global ingest instrumentation
+    if not payi_param and not apayi_param and len(global_context) > 0 and global_context.get("proxy", False):
+        payi_param = Payi()
+        apayi_param = AsyncPayi()
     
-    _instrumentor = PayiInstrumentor(
+    # allow for both payi and apayi to be None for the @proxy case
+    _instrumentor = _PayiInstrumentor(
         payi=payi_param,
         apayi=apayi_param,
         instruments=instruments,
@@ -1013,6 +1041,14 @@ def payi_instrument(
         prompt_and_response_logger=prompt_and_response_logger,
     )
 
+    if len(global_context) > 0:
+        if  "use_case_name" in global_context and not "use_case_id" in global_context:
+            global_context["use_case_id"] = str(uuid.uuid4())
+        if "experience_name" in global_context and not "experience_id" in global_context:
+            global_context["experience_id"] = str(uuid.uuid4())
+
+        _instrumentor._context_stack.append(global_context)
+
 def ingest(
     limit_ids: Optional["list[str]"] = None,
     request_tags: Optional["list[str]"] = None,

payi-0.1.0a60/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.1.0-alpha.60"
-}