paid-python 0.3.5__py3-none-any.whl → 0.4.0__py3-none-any.whl

paid/tracing/__init__.py

@@ -7,13 +7,14 @@ from .distributed_tracing import (
     unset_tracing_token,
 )
 from .signal import signal
-from .tracing import initialize_tracing
+from .tracing import initialize_tracing, get_paid_tracer_provider
 
 __all__ = [
     "generate_tracing_token",
     "paid_autoinstrument",
     "paid_tracing",
     "initialize_tracing",
+    "get_paid_tracer_provider",
     "set_tracing_token",
     "unset_tracing_token",
     "signal",

paid/tracing/autoinstrumentation.py

@@ -44,7 +44,7 @@ except ImportError:
     BEDROCK_AVAILABLE = False
 
 try:
-    from opentelemetry.instrumentation.langchain import LangchainInstrumentor
+    from openinference.instrumentation.langchain import LangChainInstrumentor
 
     LANGCHAIN_AVAILABLE = True
 except ImportError:
@@ -196,16 +196,14 @@ def _instrument_bedrock() -> None:
 
 def _instrument_langchain() -> None:
     """
-    Instrument LangChain using opentelemetry-instrumentation-langchain.
+    Instrument LangChain using openinference-instrumentation-langchain.
     """
     if not LANGCHAIN_AVAILABLE:
         logger.warning("LangChain instrumentation library not available, skipping instrumentation")
         return
 
     # Instrument LangChain with Paid's tracer provider
-    LangchainInstrumentor(disable_trace_context_propagation=True).instrument(
-        tracer_provider=tracing.paid_tracer_provider
-    )
+    LangChainInstrumentor().instrument(tracer_provider=tracing.paid_tracer_provider)
 
     _initialized_instrumentors.append("langchain")
     logger.info("LangChain auto-instrumentation enabled")

paid/tracing/tracing.py

@@ -56,6 +56,22 @@ def set_token(token: str) -> None:
 # Initialized at module load with defaults, never None (uses no-op provider if not initialized or API key isn't available)
 paid_tracer_provider: Union[TracerProvider, NoOpTracerProvider] = NoOpTracerProvider()
 
+def get_paid_tracer_provider() -> Optional[TracerProvider]:
+    """Export the tracer provider to the user.
+    Initialize tracing if not already. Never return NoOpTracerProvider.
+
+    Returns:
+        The tracer provider instance.
+    """
+    global paid_tracer_provider
+
+    if get_token() is None:
+        initialize_tracing()
+
+    if not isinstance(paid_tracer_provider, TracerProvider):
+        return None
+
+    return paid_tracer_provider
 
 class PaidSpanProcessor(SpanProcessor):
     """
@@ -63,25 +79,36 @@ class PaidSpanProcessor(SpanProcessor):
     1. Prefixes all span names with 'paid.trace.'
     2. Automatically adds external_customer_id and external_agent_id attributes
        to all spans based on context variables set by the tracing decorator.
+    3. Filters out prompt/response data unless store_prompt=True.
+    4. Filters out duplicate LangChain spans that may duplicate information from other instrumentations.
     """
 
     SPAN_NAME_PREFIX = "paid.trace."
     PROMPT_ATTRIBUTES_SUBSTRINGS = {
-        "prompt",
-        # "gen_ai.prompt",
         "gen_ai.completion",
         "gen_ai.request.messages",
         "gen_ai.response.messages",
         "llm.output_message",
         "llm.input_message",
         "llm.invocation_parameters",
+        "gen_ai.prompt",
+        "langchain.prompt",
         "output.value",
         "input.value",
-        # "langchain.prompt",
     }
 
     def on_start(self, span: Span, parent_context: Optional[Context] = None) -> None:
         """Called when a span is started. Prefix the span name and add attributes."""
+
+        LANGCHAIN_SPAN_FILTERS = ["ChatOpenAI", "ChatAnthropic"]
+        if any(f in span.name for f in LANGCHAIN_SPAN_FILTERS):
+            # HACK TO FILTER DUPLICATE SPANS CREATED BY LANGCHAIN INSTRUMENTATION.
+            # Langchain instrumentation creates spans, that are created by other instrumentations (ex. OpenAI, Anthropic).
+            # Not all spans need filtering (ex. ChatGoogleGenerativeAI), so first test actual telemetry before adding filters.
+            # TODO: maybe consider a dropping sampler for such spans instead of raising an exception?
+            logger.debug(f"Dropping Langchain span: {span.name}")
+            raise Exception(f"Dropping Langchain span: {span.name}")
+
         # Prefix the span name
         if span.name and not span.name.startswith(self.SPAN_NAME_PREFIX):
             span.update_name(f"{self.SPAN_NAME_PREFIX}{span.name}")

{paid_python-0.3.5.dist-info → paid_python-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.3.5
+Version: 0.4.0
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers
@@ -21,11 +21,11 @@ Requires-Dist: httpx (>=0.21.0)
 Requires-Dist: mutagen (>=1.47.0)
 Requires-Dist: openinference-instrumentation-bedrock (>=0.1.0)
 Requires-Dist: openinference-instrumentation-google-genai (>=0.1.8)
+Requires-Dist: openinference-instrumentation-langchain (>=0.1.55)
 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-langchain (>=0.47.0)
 Requires-Dist: opentelemetry-sdk (>=1.23.0)
 Requires-Dist: pydantic (>=1.9.0)
 Requires-Dist: pydantic-core (>=2.18.0)
@@ -767,6 +767,14 @@ async def do_work():
 await do_work()
 ```
 
+### Paid OTEL Tracer Provider
+
+If you would like to use the Paid OTEL tracer provider:
+```python
+from paid.tracing import get_paid_tracer_provider
+paid_tracer_provider = get_paid_tracer_provider()
+```
+
 ## Contributing
 
 While we value open-source contributions to this SDK, this library is generated programmatically.

{paid_python-0.3.5.dist-info → paid_python-0.4.0.dist-info}/RECORD

@@ -56,13 +56,13 @@ paid/orders/lines/client.py,sha256=GqSwiXdlu49KLHt7uccS_H4nkVQosM1_PQOcPA9v82A,4
 paid/orders/lines/raw_client.py,sha256=KZN_yBokCOkf1lUb4ZJtX_NZbqmTqCdJNoaIOdWar8I,4590
 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=Vdd0PrjLSzyxXujY45S9a0tAKjtLs8-SHbg81v23WkY,7784
+paid/tracing/__init__.py,sha256=WScAVz67nxynERd9lZhmrobPMa0uScbB2nSly2vA3r4,564
+paid/tracing/autoinstrumentation.py,sha256=kH2d1wGPLE6Ore60nNAwFUKAWkH-3N_B_6urJYoobMg,7732
 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=Vht3U8QJmT5jlRVnrybTn-cI1RPuVtyb3V4eTu6gA4g,3991
 paid/tracing/signal.py,sha256=PfYxF6EFQS8j7RY5_C5NXrCBVu9Hq2E2tyG4fdQScJk,3252
-paid/tracing/tracing.py,sha256=0eMY357by892_32a2qCT6nRCUOHt03gxKCmUH2wDjVg,15184
+paid/tracing/tracing.py,sha256=MCIqzRELFB6VbvFrnpIp7bLX7HYfpHnmUapDLcAEioE,16478
 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
@@ -119,7 +119,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.5.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
-paid_python-0.3.5.dist-info/METADATA,sha256=V99x_8mHOv0uUo6pwD7rcf2AEOYhSouqgMlJZAawyDI,23632
-paid_python-0.3.5.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-paid_python-0.3.5.dist-info/RECORD,,
+paid_python-0.4.0.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
+paid_python-0.4.0.dist-info/METADATA,sha256=CmtqOI_FpMWGVBkeDa8OogiYVw6smGXeoUBnuzNP_yc,23834
+paid_python-0.4.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+paid_python-0.4.0.dist-info/RECORD,,