paid-python 0.1.1__py3-none-any.whl → 0.2.0__py3-none-any.whl

paid/client.py

@@ -17,7 +17,7 @@ from .tracing.distributed_tracing import (
 from .tracing.signal import signal
 from .tracing.tracing import (
     DEFAULT_COLLECTOR_ENDPOINT,
-    initialize_tracing_,
+    initialize_tracing,
     trace_async_,
     trace_sync_,
 )
@@ -114,7 +114,7 @@ class Paid:
             stacklevel=2,
         )
         token = self._client_wrapper._get_token()
-        initialize_tracing_(token, collector_endpoint=collector_endpoint)
+        initialize_tracing(token, collector_endpoint=collector_endpoint)
 
     def generate_tracing_token(self) -> int:
         """
@@ -394,7 +394,7 @@ class AsyncPaid:
             stacklevel=2,
         )
         token = self._client_wrapper._get_token()
-        initialize_tracing_(token, collector_endpoint=collector_endpoint)
+        initialize_tracing(token, collector_endpoint=collector_endpoint)
 
     def generate_tracing_token(self) -> int:
         """

paid/tracing/__init__.py

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

paid/tracing/autoinstrumentation.py

@@ -8,7 +8,7 @@ sending traces to the Paid collector endpoint.
 from typing import List, Optional
 
 from . import tracing
-from .tracing import initialize_tracing_
+from .tracing import initialize_tracing
 from opentelemetry.trace import NoOpTracerProvider
 
 from paid.logger import logger
@@ -49,6 +49,13 @@ try:
 except ImportError:
     BEDROCK_AVAILABLE = False
 
+try:
+    from opentelemetry.instrumentation.langchain import LangchainInstrumentor
+
+    LANGCHAIN_AVAILABLE = True
+except ImportError:
+    LANGCHAIN_AVAILABLE = False
+
 
 # Track which instrumentors have been initialized
 _initialized_instrumentors: List[str] = []
@@ -69,6 +76,7 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
                   - "openai": OpenAI library
                   - "openai-agents": OpenAI Agents SDK
                   - "bedrock": AWS Bedrock
+                  - "langchain": LangChain library
                   If None, all supported libraries that are installed will be instrumented.
 
     Note:
@@ -94,11 +102,11 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
     # Initialize tracing if not already initialized
     if isinstance(tracing.paid_tracer_provider, NoOpTracerProvider):
         logger.info("Tracing not initialized, initializing automatically")
-        initialize_tracing_()
+        initialize_tracing()
 
     # Default to all supported libraries if none specified
     if libraries is None:
-        libraries = ["anthropic", "gemini", "openai", "openai-agents", "bedrock"]
+        libraries = ["anthropic", "gemini", "openai", "openai-agents", "bedrock", "langchain"]
 
     for library in libraries:
         if library in _initialized_instrumentors:
@@ -115,9 +123,11 @@ def paid_autoinstrument(libraries: Optional[List[str]] = None) -> None:
             _instrument_openai_agents()
         elif library == "bedrock":
             _instrument_bedrock()
+        elif library == "langchain":
+            _instrument_langchain()
         else:
             logger.warning(
-                f"Unknown library '{library}' - supported libraries: anthropic, gemini, openai, openai-agents, bedrock"
+                f"Unknown library '{library}' - supported libraries: anthropic, gemini, openai, openai-agents, bedrock, langchain"
             )
 
     logger.info(f"Auto-instrumentation enabled for: {', '.join(_initialized_instrumentors)}")
@@ -196,3 +206,18 @@ def _instrument_bedrock() -> None:
 
     _initialized_instrumentors.append("bedrock")
     logger.info("Bedrock auto-instrumentation enabled")
+
+
+def _instrument_langchain() -> None:
+    """
+    Instrument LangChain using opentelemetry-instrumentation-langchain.
+    """
+    if not LANGCHAIN_AVAILABLE:
+        logger.warning("LangChain instrumentation library not available, skipping instrumentation")
+        return
+
+    # Instrument LangChain with Paid's tracer provider
+    LangchainInstrumentor().instrument(tracer_provider=tracing.paid_tracer_provider)
+
+    _initialized_instrumentors.append("langchain")
+    logger.info("LangChain auto-instrumentation enabled")

paid/tracing/context_manager.py

@@ -4,7 +4,7 @@ import functools
 from typing import Any, Callable, Dict, Optional, Tuple
 
 from . import distributed_tracing, tracing
-from .tracing import get_paid_tracer, get_token, initialize_tracing_, trace_async_, trace_sync_
+from .tracing import get_paid_tracer, get_token, initialize_tracing, trace_async_, trace_sync_
 from opentelemetry import trace
 from opentelemetry.context import Context
 from opentelemetry.trace import NonRecordingSpan, Span, SpanContext, Status, StatusCode, TraceFlags
@@ -88,7 +88,7 @@ class paid_tracing:
         ] = None
 
         if not get_token():
-            initialize_tracing_(None, self.collector_endpoint)
+            initialize_tracing(None, self.collector_endpoint)
 
     def _setup_context(self) -> Optional[Context]:
         """Set up context variables and return OTEL context if needed."""
@@ -190,7 +190,7 @@ class paid_tracing:
                 # Auto-initialize tracing if not done
                 if get_token() is None:
                     try:
-                        initialize_tracing_(None, self.collector_endpoint)
+                        initialize_tracing(None, self.collector_endpoint)
                     except Exception as e:
                         logger.error(f"Failed to auto-initialize tracing: {e}")
                         # Fall back to executing function without tracing
@@ -219,7 +219,7 @@ class paid_tracing:
                 # Auto-initialize tracing if not done
                 if get_token() is None:
                     try:
-                        initialize_tracing_(None, self.collector_endpoint)
+                        initialize_tracing(None, self.collector_endpoint)
                     except Exception as e:
                         logger.error(f"Failed to auto-initialize tracing: {e}")
                         # Fall back to executing function without tracing

paid/tracing/tracing.py

@@ -155,8 +155,43 @@ 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
+            ):
+                logger.error("OTEL force flush : timeout reached")
+        except Exception as e:
+            logger.error(f"Error flushing traces: {e}")
+
+    def create_chained_signal_handler(signum: int):
+        current_handler = signal.getsignal(signum)
+
+        def chained_handler(_signum, frame):
+            logger.warning(f"Received signal {_signum}, flushing traces")
+            flush_traces()
+            # Restore the original handler
+            signal.signal(_signum, current_handler)
+            # Re-raise the signal to let the original handler (or default) handle it
+            os.kill(os.getpid(), _signum)
+
+        return chained_handler
 
-def initialize_tracing_(api_key: Optional[str] = None, collector_endpoint: Optional[str] = DEFAULT_COLLECTOR_ENDPOINT):
+    try:
+        # This is already done by default OTEL shutdown,
+        # but user might turn that off - so register it explicitly
+        atexit.register(flush_traces)
+
+        # signal handlers
+        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")
+
+
+def initialize_tracing(api_key: Optional[str] = None, collector_endpoint: Optional[str] = DEFAULT_COLLECTOR_ENDPOINT):
     """
     Initialize OpenTelemetry with OTLP exporter for Paid backend.
 
@@ -203,36 +238,7 @@ def initialize_tracing_(api_key: Optional[str] = None, collector_endpoint: Optio
         span_processor = SimpleSpanProcessor(otlp_exporter)
         paid_tracer_provider.add_span_processor(span_processor)
 
-        # Terminate gracefully and don't lose traces
-        def flush_traces():
-            try:
-                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}")
-
-        def create_chained_signal_handler(signum: int):
-            current_handler = signal.getsignal(signum)
-
-            def chained_handler(_signum, frame):
-                logger.warning(f"Received signal {_signum}, flushing traces")
-                flush_traces()
-                # Restore the original handler
-                signal.signal(_signum, current_handler)
-                # Re-raise the signal to let the original handler (or default) handle it
-                os.kill(os.getpid(), _signum)
-
-            return chained_handler
-
-        # This is already done by default OTEL shutdown,
-        # but user might turn that off - so register it explicitly
-        atexit.register(flush_traces)
-
-        # Handle signals
-        for sig in (signal.SIGINT, signal.SIGTERM):
-            signal.signal(sig, create_chained_signal_handler(sig))
+        setup_graceful_termination() # doesn't throw
 
         logger.info("Paid tracing initialized successfully - collector at %s", collector_endpoint)
     except Exception as e:

{paid_python-0.1.1.dist-info → paid_python-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.1.1
+Version: 0.2.0
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers
@@ -25,6 +25,7 @@ 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)
 Requires-Dist: pydantic (>=1.9.0)
@@ -162,6 +163,13 @@ Both approaches:
 - Gracefully fall back to normal execution if tracing fails
 - Support the same parameters: `external_customer_id`, `external_agent_id`, `tracing_token`, `store_prompt`, `metadata`
 
+* Note - if it happens that you're calling `paid_tracing` from non-main thread, then it's advised to initialize from main thread:
+```python
+from paid.tracing import initialize_tracing
+initialize_tracing()
+```
+* `initialize_tracing` also accepts optional arguments like OTEL collector endpoint and api key if you want to reroute your tracing somewhere else :)
+
 ### Using the Paid wrappers
 
 You can track usage costs by using Paid wrappers around your AI provider's SDK.
@@ -182,9 +190,11 @@ Example usage:
 
 ```python
 from openai import OpenAI
-from paid.tracing import paid_tracing
+from paid.tracing import paid_tracing, initialize_tracing
 from paid.tracing.wrappers.openai import PaidOpenAI
 
+initialize_tracing()
+
 openAIClient = PaidOpenAI(OpenAI(
     # This is the default and can be omitted
     api_key="<OPENAI_API_KEY>",
@@ -212,10 +222,12 @@ You can attach custom metadata to your traces by passing a `metadata` dictionary
 <Tabs>
   <Tab title="Python - Decorator">
     ```python
-    from paid.tracing import paid_tracing, signal
+    from paid.tracing import paid_tracing, signal, initialize_tracing
     from paid.tracing.wrappers import PaidOpenAI
     from openai import OpenAI
 
+    initialize_tracing()
+
     openai_client = PaidOpenAI(OpenAI(api_key="<OPENAI_API_KEY>"))
 
     @paid_tracing(
@@ -243,10 +255,12 @@ You can attach custom metadata to your traces by passing a `metadata` dictionary
 
   <Tab title="Python - Context Manager">
     ```python
-    from paid.tracing import paid_tracing, signal
+    from paid.tracing import paid_tracing, signal, initialize_tracing
     from paid.tracing.wrappers import PaidOpenAI
     from openai import OpenAI
 
+    initialize_tracing()
+
     openai_client = PaidOpenAI(OpenAI(api_key="<OPENAI_API_KEY>"))
 
     def process_event(event):
@@ -305,14 +319,14 @@ For maximum convenience, you can use OpenTelemetry auto-instrumentation to autom
 
 ```python
 from paid import Paid
-from paid.tracing import paid_autoinstrument
+from paid.tracing import paid_autoinstrument, initialize_tracing
 from openai import OpenAI
 
 # Initialize Paid SDK
 client = Paid(token="PAID_API_KEY")
+initialize_tracing()
 
-# Enable auto-instrumentation for all supported libraries
-paid_autoinstrument()  # instruments all available: anthropic, gemini, openai, openai-agents, bedrock
+paid_autoinstrument()  # instruments all available: anthropic, gemini, openai, openai-agents, bedrock, langchain
 
 # Now all OpenAI calls will be automatically traced
 openai_client = OpenAI(api_key="<OPENAI_API_KEY>")
@@ -338,6 +352,7 @@ gemini             - Google Generative AI (google-generativeai)
 openai             - OpenAI Python SDK
 openai-agents      - OpenAI Agents SDK
 bedrock            - AWS Bedrock (boto3)
+langchain          - LangChain framework
 ```
 
 #### Selective Instrumentation
@@ -437,10 +452,12 @@ For such cases, you can pass a tracing token directly to `@paid_tracing()` or co
 The simplest way to implement distributed tracing is to pass the token directly to the decorator or context manager:
 
 ```python
-from paid.tracing import paid_tracing, signal, generate_tracing_token
+from paid.tracing import paid_tracing, signal, generate_tracing_token, initialize_tracing
 from paid.tracing.wrappers.openai import PaidOpenAI
 from openai import OpenAI
 
+initialize_tracing()
+
 openai_client = PaidOpenAI(OpenAI(api_key="<OPENAI_API_KEY>"))
 
 # Process 1: Generate token and do initial work
@@ -482,10 +499,12 @@ process_part_2()
 Using context manager instead of decorator:
 
 ```python
-from paid.tracing import paid_tracing, signal, generate_tracing_token
+from paid.tracing import paid_tracing, signal, generate_tracing_token, initialize_tracing
 from paid.tracing.wrappers.openai import PaidOpenAI
 from openai import OpenAI
 
+initialize_tracing()
+
 # Initialize
 openai_client = PaidOpenAI(OpenAI(api_key="<OPENAI_API_KEY>"))
 
@@ -643,9 +662,11 @@ The `@paid_tracing` decorator automatically handles both sync and async function
 
 ```python
 from openai import AsyncOpenAI
-from paid.tracing import paid_tracing
+from paid.tracing import paid_tracing, initialize_tracing
 from paid.tracing.wrappers.openai import PaidAsyncOpenAI
 
+initialize_tracing()
+
 # Wrap the async OpenAI client
 openai_client = PaidAsyncOpenAI(AsyncOpenAI(api_key="<OPENAI_API_KEY>"))
 
@@ -669,10 +690,12 @@ await generate_image()
 The `signal()` function works seamlessly in async contexts:
 
 ```python
-from paid.tracing import paid_tracing, signal
+from paid.tracing import paid_tracing, signal, initialize_tracing
 from paid.tracing.wrappers.openai import PaidAsyncOpenAI
 from openai import AsyncOpenAI
 
+initialize_tracing()
+
 openai_client = PaidAsyncOpenAI(AsyncOpenAI(api_key="<OPENAI_API_KEY>"))
 
 @paid_tracing("your_external_customer_id", "your_external_agent_id")

{paid_python-0.1.1.dist-info → paid_python-0.2.0.dist-info}/RECORD

@@ -2,7 +2,7 @@ paid/__init__.py,sha256=D1SeLoeTlySo_vZCZrxFX3y5KhKGrHflphLXoewImfk,1826
 paid/agents/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 paid/agents/client.py,sha256=ojc3H-nx4MqDrb74_i6JE_wjHSJaVAErsIunfNeffMo,23305
 paid/agents/raw_client.py,sha256=jN9LvPK2-bGeNQzcV3iRmprpegXKtO2JaOEXjnPfz9Y,26833
-paid/client.py,sha256=AWo35RlJ5bP2A5bGre3iY1GbBdYW7gu7HlioQIOoltk,23038
+paid/client.py,sha256=2GGQByab__kDKaWeNy4wK_T6RkS36TX_mA6fsO08Ww4,23035
 paid/contacts/__init__.py,sha256=_VhToAyIt_5axN6CLJwtxg3-CO7THa_23pbUzqhXJa4,85
 paid/contacts/client.py,sha256=sNm-yAg4dR9AyYWL7-RC_CuCCvOXX7YlDAUqn47yZhE,14058
 paid/contacts/raw_client.py,sha256=ZYNWuekHiL2sqK_gHR0IzcrLAopUKRXIqMUi-fuLGe4,19211
@@ -36,12 +36,12 @@ 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=IP6OkV885_xlK1H68RxGTX_IhpVAkY266zsWgnQKKTs,440
-paid/tracing/autoinstrumentation.py,sha256=p57bU87x1Xi-nb_2C4O56tcgfvN2UYZAd9pJ0Vyh9Nw,6765
-paid/tracing/context_manager.py,sha256=qXWYwUisUnAGUjsGNDRR4b0m8DMiu5IfNVnBhE4bwlE,9733
+paid/tracing/__init__.py,sha256=Pe55koIwqJ6Vv5-9Wqi8xIdwCS2BbxZds-MK5fD-F5Y,506
+paid/tracing/autoinstrumentation.py,sha256=qlLLiP9cWNIXX226dLDLvvlioLCPO-0F81mU_0fBb9s,7604
+paid/tracing/context_manager.py,sha256=Qtl59mjDsacX53LPFFAVXsfeGCEA6CIfxPsvjO-Kmx8,9729
 paid/tracing/distributed_tracing.py,sha256=CpUWpHai-4LxLLHbGxz41r9h5wLG-dC83YL5Vg29OBI,3967
 paid/tracing/signal.py,sha256=PfYxF6EFQS8j7RY5_C5NXrCBVu9Hq2E2tyG4fdQScJk,3252
-paid/tracing/tracing.py,sha256=Pqf7A9I0YU_sk8KEZlS1DF8-hdhNmqGtmfumoX_9jvQ,15878
+paid/tracing/tracing.py,sha256=w07I7KQ4O8yJdvLs9RDAQVVofVWeR4xNei3zC5-EE6c,16080
 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
@@ -98,7 +98,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.1.1.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
-paid_python-0.1.1.dist-info/METADATA,sha256=YMILAQJcALoQrcvdhyNUjn90d8nNGhJX9bJIVyf0tNw,21580
-paid_python-0.1.1.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-paid_python-0.1.1.dist-info/RECORD,,
+paid_python-0.2.0.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
+paid_python-0.2.0.dist-info/METADATA,sha256=tFvHkGpKqlXPnxZhDhxFm-vETz6RbVihbJZOEdjGPUw,22343
+paid_python-0.2.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+paid_python-0.2.0.dist-info/RECORD,,