payi 0.1.0a131__tar.gz → 0.1.0a132__tar.gz

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

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

{payi-0.1.0a131 → payi-0.1.0a132}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.1.0-alpha.132 (2025-10-22)
+
+Full Changelog: [v0.1.0-alpha.131...v0.1.0-alpha.132](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.131...v0.1.0-alpha.132)
+
+### Features
+
+* thread local storage ([#385](https://github.com/Pay-i/pay-i-python/issues/385)) ([057de83](https://github.com/Pay-i/pay-i-python/commit/057de83b15fc791e6eb813d8cb62a54a0de0eea9))
+
 ## 0.1.0-alpha.131 (2025-10-21)
 
 Full Changelog: [v0.1.0-alpha.130...v0.1.0-alpha.131](https://github.com/Pay-i/pay-i-python/compare/v0.1.0-alpha.130...v0.1.0-alpha.131)

{payi-0.1.0a131 → payi-0.1.0a132}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: payi
-Version: 0.1.0a131
+Version: 0.1.0a132
 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.0a131 → payi-0.1.0a132}/pyproject.toml

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

{payi-0.1.0a131 → payi-0.1.0a132}/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.131"  # x-release-please-version
+__version__ = "0.1.0-alpha.132"  # x-release-please-version

{payi-0.1.0a131 → payi-0.1.0a132}/src/payi/lib/instrument.py

@@ -6,6 +6,7 @@ import atexit
 import asyncio
 import inspect
 import logging
+import threading
 import traceback
 from abc import abstractmethod
 from enum import Enum
@@ -210,6 +211,15 @@ class _StreamingType(Enum):
     iterator = 1
     stream_manager = 2
 
+class _ThreadLocalContextStorage(threading.local):
+    """
+    Thread-local storage for context stacks. Each thread gets its own context stack.
+    
+    Note: We don't use __init__ because threading.local's __init__ semantics are tricky.
+    Instead, we lazily initialize the context_stack attribute in the property accessor.
+    """
+    context_stack: "list[_Context]"
+
 class _InternalTrackContext:
     def __init__(
         self,
@@ -256,7 +266,12 @@ class _PayiInstrumentor:
         if self._apayi:
             _g_logger.debug(f"Pay-i instrumentor initialized with AsyncPayi instance: {self._apayi}")            
 
-        self._context_stack: list[_Context] = []  # Stack of context dictionaries
+        # Thread-local storage for context stacks - each thread gets its own stack
+        self._thread_local_storage = _ThreadLocalContextStorage()
+        
+        # Global immutable initial context that all threads inherit on first access
+        self._global_initial_context: Optional[_Context] = None
+        
         self._log_prompt_and_response: bool = log_prompt_and_response
 
         self._blocked_limits: set[str] = set()
@@ -316,7 +331,7 @@ class _PayiInstrumentor:
 
             self.__enter__()
 
-            # _init_current_context will update the currrent context stack location
+            # _init_current_context will update the current context stack location
             context: _Context = {}
             # Copy allowed keys from global_config into context
             # Dynamically use keys from _Context TypedDict
@@ -325,7 +340,12 @@ class _PayiInstrumentor:
                 if key in global_config:
                     context[key] = global_config[key] # type: ignore
 
-            self._init_current_context(**context) 
+            self._init_current_context(**context)
+            
+            # Store the initialized context as the global initial context (immutable after this point)
+            # All threads will inherit a copy of this context on their first access
+            current_context = self.get_context()
+            self._global_initial_context = current_context.copy() if current_context else None 
 
     def _ensure_payi_clients(self) -> None:
         if self._offline_instrumentation is not None:
@@ -689,6 +709,25 @@ class _PayiInstrumentor:
     def _ingest_units(self, request: _ProviderRequest) -> Optional[Union[XproxyResult, XproxyError]]:
         return self.set_xproxy_result(self._ingest_units_worker(request))
 
+    @property
+    def _context_stack(self) -> "list[_Context]":
+        """
+        Get the thread-local context stack. On first access per thread, 
+        initializes with the global initial context if one was set.
+        """
+        # Lazy-initialize the context_stack for this thread if it doesn't exist
+        if not hasattr(self._thread_local_storage, 'context_stack'):
+            self._thread_local_storage.context_stack = []
+        
+        stack = self._thread_local_storage.context_stack
+
+        # If this is the first access in this thread and we have a global initial context,
+        # initialize this thread's stack with it
+        if len(stack) == 0 and self._global_initial_context is not None:
+            stack.append(self._global_initial_context.copy())
+        
+        return stack
+
     def _setup_call_func(
         self
         ) -> _Context:

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

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