paid-python 0.3.0__py3-none-any.whl → 0.3.2__py3-none-any.whl

paid/tracing/autoinstrumentation.py

@@ -21,13 +21,6 @@ try:
 except ImportError:
     ANTHROPIC_AVAILABLE = False
 
-try:
-    from opentelemetry.instrumentation.google_generativeai import GoogleGenerativeAiInstrumentor
-
-    GEMINI_AVAILABLE = True
-except ImportError:
-    GEMINI_AVAILABLE = False
-
 try:
     from opentelemetry.instrumentation.openai import OpenAIInstrumentor
 
@@ -79,7 +72,6 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
     Args:
         libraries: List of library names to instrument. Currently supported:
                   - "anthropic": Anthropic library
-                  - "gemini": Gemini library
                   - "openai": OpenAI library
                   - "openai-agents": OpenAI Agents SDK
                   - "bedrock": AWS Bedrock
@@ -114,7 +106,7 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
 
     # Default to all supported libraries if none specified
     if libraries is None:
-        libraries = ["anthropic", "gemini", "openai", "openai-agents", "bedrock", "langchain", "google-genai"]
+        libraries = ["anthropic", "openai", "openai-agents", "bedrock", "langchain", "google-genai"]
 
     for library in libraries:
         if library in _initialized_instrumentors:
@@ -123,8 +115,6 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
 
         if library == "anthropic":
             _instrument_anthropic()
-        elif library == "gemini":
-            _instrument_gemini()
         elif library == "openai":
             _instrument_openai()
         elif library == "openai-agents":
@@ -158,21 +148,6 @@ def _instrument_anthropic() -> None:
     logger.info("Anthropic auto-instrumentation enabled")
 
 
-def _instrument_gemini() -> None:
-    """
-    Instrument the Google Gemini library using opentelemetry-instrumentation-google-generativeai.
-    """
-    if not GEMINI_AVAILABLE:
-        logger.warning("Gemini library not available, skipping instrumentation")
-        return
-
-    # Instrument Gemini with Paid's tracer provider
-    GoogleGenerativeAiInstrumentor().instrument(tracer_provider=tracing.paid_tracer_provider)
-
-    _initialized_instrumentors.append("gemini")
-    logger.info("Gemini auto-instrumentation enabled")
-
-
 def _instrument_openai() -> None:
     """
     Instrument the OpenAI library using opentelemetry-instrumentation-openai.
@@ -227,12 +202,14 @@ def _instrument_langchain() -> None:
         return
 
     # Instrument LangChain with Paid's tracer provider
-    LangchainInstrumentor(disable_trace_context_propagation=True).instrument(tracer_provider=tracing.paid_tracer_provider)
-
+    LangchainInstrumentor(disable_trace_context_propagation=True).instrument(
+        tracer_provider=tracing.paid_tracer_provider
+    )
 
     _initialized_instrumentors.append("langchain")
     logger.info("LangChain auto-instrumentation enabled")
 
+
 def _instrument_google_genai() -> None:
     """
     Instrument Google GenAI using openinference-instrumentation-google-genai.
@@ -241,7 +218,6 @@ def _instrument_google_genai() -> None:
         logger.warning("Google GenAI instrumentation library not available, skipping instrumentation")
         return
 
-    
     GoogleGenAIInstrumentor().instrument(tracer_provider=tracing.paid_tracer_provider)
     _initialized_instrumentors.append("google-genai")
     logger.info("Google GenAI auto-instrumentation enabled")

paid/tracing/context_data.py

@@ -44,6 +44,8 @@ class ContextData:
 
     @classmethod
     def set_context_key(cls, key: str, value: Any) -> None:
+        if value is None:
+            return
         if key not in cls._context:
             logger.warning(f"Invalid context key: {key}")
             return
@@ -51,6 +53,18 @@ class ContextData:
         reset_tokens = cls._get_or_create_reset_tokens()
         reset_tokens[key] = reset_token
 
+    @classmethod
+    def unset_context_key(cls, key: str) -> None:
+        """Unset a specific context key"""
+        if key not in cls._context:
+            logger.warning(f"Invalid context key: {key}")
+            return
+        reset_tokens = cls._reset_tokens.get()
+        if reset_tokens:
+            _ = cls._context[key].set(None)
+            if key in reset_tokens:
+                del reset_tokens[key]
+
     @classmethod
     def reset_context(cls) -> None:
         reset_tokens = cls._reset_tokens.get()
@@ -58,3 +72,14 @@ class ContextData:
             for key, reset_token in reset_tokens.items():
                 cls._context[key].reset(reset_token)
             reset_tokens.clear()
+
+    @classmethod
+    def reset_context_key(cls, key: str) -> None:
+        """Reset a specific context key to its previous value using the stored reset token."""
+        if key not in cls._context:
+            logger.warning(f"Invalid context key: {key}")
+            return
+        reset_tokens = cls._reset_tokens.get()
+        if reset_tokens and key in reset_tokens:
+            cls._context[key].reset(reset_tokens[key])
+            del reset_tokens[key]

paid/tracing/distributed_tracing.py

@@ -110,4 +110,4 @@ def unset_tracing_token():
         DeprecationWarning,
         stacklevel=2,
     )
-    ContextData.set_context_key("trace_id", None)
+    ContextData.unset_context_key("trace_id")

paid/tracing/tracing.py

@@ -73,6 +73,7 @@ class PaidSpanProcessor(SpanProcessor):
         "gen_ai.response.messages",
         "llm.output_message",
         "llm.input_message",
+        "llm.invocation_parameters",
         "output.value",
         "input.value",
     }
@@ -137,12 +138,11 @@ class PaidSpanProcessor(SpanProcessor):
         """Called to force flush. Always returns True since there's nothing to flush."""
         return True
 
+
 def setup_graceful_termination():
     def flush_traces():
         try:
-            if not isinstance(paid_tracer_provider, NoOpTracerProvider) and not paid_tracer_provider.force_flush(
-                10000
-            ):
+            if not isinstance(paid_tracer_provider, NoOpTracerProvider) and not paid_tracer_provider.force_flush(10000):
                 logger.error("OTEL force flush : timeout reached")
         except Exception as e:
             logger.error(f"Error flushing traces: {e}")
@@ -169,8 +169,10 @@ def setup_graceful_termination():
         for sig in (signal.SIGINT, signal.SIGTERM):
             signal.signal(sig, create_chained_signal_handler(sig))
     except Exception as e:
-        logger.warning(f"Could not set up termination handlers: {e}"
-                       "\nConsider calling initialize_tracing() from the main thread during app initialization if you don't already")
+        logger.warning(
+            f"Could not set up termination handlers: {e}"
+            "\nConsider calling initialize_tracing() from the main thread during app initialization if you don't already"
+        )
 
 
 def initialize_tracing(api_key: Optional[str] = None, collector_endpoint: Optional[str] = DEFAULT_COLLECTOR_ENDPOINT):
@@ -220,7 +222,7 @@ def initialize_tracing(api_key: Optional[str] = None, collector_endpoint: Option
         span_processor = SimpleSpanProcessor(otlp_exporter)
         paid_tracer_provider.add_span_processor(span_processor)
 
-        setup_graceful_termination() # doesn't throw
+        setup_graceful_termination()  # doesn't throw
 
         logger.info("Paid tracing initialized successfully - collector at %s", collector_endpoint)
     except Exception as e:

{paid_python-0.3.0.dist-info → paid_python-0.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.3.0
+Version: 0.3.2
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers
@@ -25,7 +25,6 @@ Requires-Dist: openinference-instrumentation-openai-agents (>=1.0.0)
 Requires-Dist: opentelemetry-api (>=1.23.0)
 Requires-Dist: opentelemetry-exporter-otlp-proto-http (>=1.23.0)
 Requires-Dist: opentelemetry-instrumentation-anthropic (>=0.47.0)
-Requires-Dist: opentelemetry-instrumentation-google-generativeai (>=0.47.0)
 Requires-Dist: opentelemetry-instrumentation-langchain (>=0.47.0)
 Requires-Dist: opentelemetry-instrumentation-openai (>=0.47.0)
 Requires-Dist: opentelemetry-sdk (>=1.23.0)

{paid_python-0.3.0.dist-info → paid_python-0.3.2.dist-info}/RECORD

@@ -37,12 +37,12 @@ paid/orders/lines/raw_client.py,sha256=KZN_yBokCOkf1lUb4ZJtX_NZbqmTqCdJNoaIOdWar
 paid/orders/raw_client.py,sha256=650e1Sj2vi9KVJc15M3ENXIKYoth0qMz66dzvXy1Sb4,16245
 paid/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 paid/tracing/__init__.py,sha256=Pe55koIwqJ6Vv5-9Wqi8xIdwCS2BbxZds-MK5fD-F5Y,506
-paid/tracing/autoinstrumentation.py,sha256=LNZJ8wXCu4g8sgd7DkWJCLBljsePr2D0E06scJfXG9g,8470
-paid/tracing/context_data.py,sha256=UFCZxX6zGa9w5lGLQK5tXdMM69aV52nReUkxwfN9S6A,2378
+paid/tracing/autoinstrumentation.py,sha256=J1v1YNR6fISd0B-P9MPwWQGVqJ309PuYfaUHsfqGUGk,7650
+paid/tracing/context_data.py,sha256=oiLocz-9qDqB5nQzJlrLsc2Mkr9MaNt_yF_hjppobKc,3298
 paid/tracing/context_manager.py,sha256=ZQtsJ9JPxTwn2t4AW26WpYboaOEZdI2T1Sw0Rwsbf-E,8470
-paid/tracing/distributed_tracing.py,sha256=XCUGFBB2lksrR19Mry10cTZRFSCSztUbUjE_sKCwjX8,3995
+paid/tracing/distributed_tracing.py,sha256=Vht3U8QJmT5jlRVnrybTn-cI1RPuVtyb3V4eTu6gA4g,3991
 paid/tracing/signal.py,sha256=PfYxF6EFQS8j7RY5_C5NXrCBVu9Hq2E2tyG4fdQScJk,3252
-paid/tracing/tracing.py,sha256=CxumDRwQHJ5S4FnzyTfUi_s557Ka7LDxVYSId0jStSc,14783
+paid/tracing/tracing.py,sha256=-gnN75ihJdgq6zTxXbpJSloHJqcRYBhb-RLyk1Bkxho,14803
 paid/tracing/wrappers/__init__.py,sha256=IIleLB_JUbzLw7FshrU2VHZAKF3dZHMGy1O5zCBwwqM,1588
 paid/tracing/wrappers/anthropic/__init__.py,sha256=_x1fjySAQxuT5cIGO_jU09LiGcZH-WQLqKg8mUFAu2w,115
 paid/tracing/wrappers/anthropic/anthropicWrapper.py,sha256=pGchbOb41CbTxc7H8xXoM-LjR085spqrzXqCVC_rrFk,4913
@@ -99,7 +99,7 @@ paid/usage/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 paid/usage/client.py,sha256=280WJuepoovk3BAVbAx2yN2Q_qBdvx3CcPkLu8lXslc,3030
 paid/usage/raw_client.py,sha256=2acg5C4lxuZodZjepU9QYF0fmBxgG-3ZgXs1zUJG-wM,3709
 paid/version.py,sha256=QIpDFnOrxMxrs86eL0iNH0mSZ1DO078wWHYY9TYAoew,78
-paid_python-0.3.0.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
-paid_python-0.3.0.dist-info/METADATA,sha256=uwDSnsVHDB_IrUSbds61pzA6udH1VxceYb445lSfoAM,22411
-paid_python-0.3.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-paid_python-0.3.0.dist-info/RECORD,,
+paid_python-0.3.2.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
+paid_python-0.3.2.dist-info/METADATA,sha256=6oJYiI-SNj7ghh91lD6Sw4Av9vcwjnfxPxGP5V3GYU4,22335
+paid_python-0.3.2.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+paid_python-0.3.2.dist-info/RECORD,,