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

paid/tracing/tracing.py

@@ -66,8 +66,9 @@ class PaidSpanProcessor(SpanProcessor):
     """
 
     SPAN_NAME_PREFIX = "paid.trace."
-    PROMPT_ATTRIBUTES_PREFIXES = {
-        "gen_ai.prompt",
+    PROMPT_ATTRIBUTES_SUBSTRINGS = {
+        "prompt",
+        # "gen_ai.prompt",
         "gen_ai.completion",
         "gen_ai.request.messages",
         "gen_ai.response.messages",
@@ -76,6 +77,7 @@ class PaidSpanProcessor(SpanProcessor):
         "llm.invocation_parameters",
         "output.value",
         "input.value",
+        # "langchain.prompt",
     }
 
     def on_start(self, span: Span, parent_context: Optional[Context] = None) -> None:
@@ -125,7 +127,7 @@ class PaidSpanProcessor(SpanProcessor):
             filtered_attrs = {
                 k: v
                 for k, v in original_attributes.items()
-                if not any(k.startswith(prefix) for prefix in self.PROMPT_ATTRIBUTES_PREFIXES)
+                if not any(substr in k for substr in self.PROMPT_ATTRIBUTES_SUBSTRINGS)
             }
             # This works because the exporter reads attributes during serialization
             object.__setattr__(span, "_attributes", filtered_attrs)
@@ -139,7 +141,7 @@ class PaidSpanProcessor(SpanProcessor):
         return True
 
 
-def setup_graceful_termination():
+def setup_graceful_termination(paid_tracer_provider: TracerProvider):
     def flush_traces():
         try:
             if not isinstance(paid_tracer_provider, NoOpTracerProvider) and not paid_tracer_provider.force_flush(10000):
@@ -189,6 +191,12 @@ def initialize_tracing(api_key: Optional[str] = None, collector_endpoint: Option
         collector_endpoint = DEFAULT_COLLECTOR_ENDPOINT
 
     try:
+        # Check if tracing is disabled via environment variable
+        paid_enabled = os.environ.get("PAID_ENABLED", "true").lower()
+        if paid_enabled == "false":
+            logger.info("Paid tracing is disabled via PAID_ENABLED environment variable")
+            return
+
         if get_token() is not None:
             logger.warning("Tracing is already initialized - skipping re-initialization")
             return
@@ -222,7 +230,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(paid_tracer_provider)  # doesn't throw
 
         logger.info("Paid tracing initialized successfully - collector at %s", collector_endpoint)
     except Exception as e:

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: paid-python
-Version: 0.3.2
+Version: 0.3.4
 Summary: 
 Requires-Python: >=3.9,<3.14
 Classifier: Intended Audience :: Developers
@@ -123,6 +123,56 @@ export PAID_LOG_LEVEL=DEBUG
 
 Defaults to ERROR.
 
+## Environment Variables
+
+The Paid SDK supports the following environment variables for configuration:
+
+### `PAID_API_KEY`
+
+Your Paid API key for authentication. This is used as a fallback when you don't explicitly pass the `token` parameter to the `Paid()` client or `initialize_tracing()`.
+
+```bash
+export PAID_API_KEY="your_api_key_here"
+```
+
+You can then initialize the client without passing the token:
+
+```python
+from paid import Paid
+
+# API key is read from PAID_API_KEY environment variable
+client = Paid()
+```
+
+### `PAID_ENABLED`
+
+Controls whether Paid tracing is enabled. Set to `false` (case-insensitive) to disable all tracing functionality.
+
+```bash
+export PAID_ENABLED=false
+```
+
+This is useful for:
+- Development/testing environments where tracing isn't needed
+- Temporarily disabling tracing without modifying code
+- Feature flagging in different deployment environments
+
+Defaults to `true` if not set.
+
+### `PAID_LOG_LEVEL`
+
+Sets the logging level for Paid SDK operations. See the [Logging](#logging) section for details.
+
+### `PAID_OTEL_COLLECTOR_ENDPOINT`
+
+Overrides the default OpenTelemetry collector endpoint URL. Only needed if you want to route traces to a custom endpoint.
+
+```bash
+export PAID_OTEL_COLLECTOR_ENDPOINT="https://your-custom-endpoint.com:4318/v1/traces"
+```
+
+Defaults to `https://collector.agentpaid.io:4318/v1/traces`.
+
 ## Cost Tracking via OTEL tracing
 
 ### Simple Decorator and Context Manager Methods

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

@@ -42,7 +42,7 @@ paid/tracing/context_data.py,sha256=oiLocz-9qDqB5nQzJlrLsc2Mkr9MaNt_yF_hjppobKc,
 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=-gnN75ihJdgq6zTxXbpJSloHJqcRYBhb-RLyk1Bkxho,14803
+paid/tracing/tracing.py,sha256=0eMY357by892_32a2qCT6nRCUOHt03gxKCmUH2wDjVg,15184
 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.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,,
+paid_python-0.3.4.dist-info/LICENSE,sha256=Nz4baY1zvv0Qy7lqrQtbaiMhmEeGr2Q7A93aqzpml4c,1071
+paid_python-0.3.4.dist-info/METADATA,sha256=r3juPpujn-xEF7IRftwtqe-CdnV9aUffhrsa6ijd4qg,23695
+paid_python-0.3.4.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
+paid_python-0.3.4.dist-info/RECORD,,