chalk-remote-call-python 1.6.0__tar.gz → 1.6.2__tar.gz

{chalk_remote_call_python-1.6.0 → chalk_remote_call_python-1.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 1.6.0
+Version: 1.6.2
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -31,6 +31,10 @@ Requires-Dist: grpcio-health-checking>=1.60.0; extra == "dev"
 Requires-Dist: grpcio-reflection>=1.60.0; extra == "dev"
 Requires-Dist: protobuf>=4.25.0; extra == "dev"
 Requires-Dist: grpcio-tools>=1.60.0; extra == "dev"
+Provides-Extra: tracing
+Requires-Dist: opentelemetry-api>=1.20.0; extra == "tracing"
+Requires-Dist: opentelemetry-sdk>=1.20.0; extra == "tracing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.20.0; extra == "tracing"
 
 # chalk-remote-call-python
 

chalk_remote_call_python-1.6.2/chalk_remote_call/_version.py

@@ -0,0 +1 @@
+__version__ = "1.6.2"

{chalk_remote_call_python-1.6.0 → chalk_remote_call_python-1.6.2}/chalk_remote_call/tracing.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import atexit
 import contextlib
 import importlib
 import os
@@ -122,11 +123,20 @@ def _configure_runtime_tracing(sample_rate: float) -> None:
 
         trace_api.set_tracer_provider(provider)
         _runtime_tracer_provider = provider
+        atexit.register(provider.shutdown)
     except Exception:
         pass
     _runtime_tracing_configured = True
 
 
+def _sample_rate_for_inherited_parent(policy: str) -> float | None:
+    if policy == _TRACE_POLICY_PARENT_BASED_ALWAYS_OFF:
+        return 0.0
+    if policy == _TRACE_POLICY_PARENT_BASED_TRACE_ID_RATIO:
+        return _trace_sample_rate()
+    return None
+
+
 def _span_metadata(
     first_metadata: Mapping[str, Any] | None,
     traceparent_metadata: Mapping[str, Any] | None,
@@ -146,6 +156,11 @@ def _span_metadata(
         parent_span_context = trace.get_current_span(parent_context).get_span_context()
         if not parent_span_context.is_valid:
             return None, None, None
+
+        sample_rate = _sample_rate_for_inherited_parent(policy)
+        if sample_rate is None:
+            return None, None, None
+        _configure_runtime_tracing(sample_rate)
         return traceparent_metadata, parent_context, otel_modules
 
     if policy != _TRACE_POLICY_PARENT_BASED_TRACE_ID_RATIO:

{chalk_remote_call_python-1.6.0 → chalk_remote_call_python-1.6.2}/chalk_remote_call_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: chalk-remote-call-python
-Version: 1.6.0
+Version: 1.6.2
 Summary: Chalk remote call Python runtime interface client
 Author: Chalk AI, Inc.
 Project-URL: Homepage, https://chalk.ai
@@ -31,6 +31,10 @@ Requires-Dist: grpcio-health-checking>=1.60.0; extra == "dev"
 Requires-Dist: grpcio-reflection>=1.60.0; extra == "dev"
 Requires-Dist: protobuf>=4.25.0; extra == "dev"
 Requires-Dist: grpcio-tools>=1.60.0; extra == "dev"
+Provides-Extra: tracing
+Requires-Dist: opentelemetry-api>=1.20.0; extra == "tracing"
+Requires-Dist: opentelemetry-sdk>=1.20.0; extra == "tracing"
+Requires-Dist: opentelemetry-exporter-otlp-proto-grpc>=1.20.0; extra == "tracing"
 
 # chalk-remote-call-python
 

{chalk_remote_call_python-1.6.0 → chalk_remote_call_python-1.6.2}/chalk_remote_call_python.egg-info/requires.txt

@@ -9,3 +9,8 @@ grpcio-health-checking>=1.60.0
 grpcio-reflection>=1.60.0
 protobuf>=4.25.0
 grpcio-tools>=1.60.0
+
+[tracing]
+opentelemetry-api>=1.20.0
+opentelemetry-sdk>=1.20.0
+opentelemetry-exporter-otlp-proto-grpc>=1.20.0

{chalk_remote_call_python-1.6.0 → chalk_remote_call_python-1.6.2}/pyproject.toml

@@ -44,6 +44,11 @@ dev = [
     "protobuf>=4.25.0",
     "grpcio-tools>=1.60.0",
 ]
+tracing = [
+    "opentelemetry-api>=1.20.0",
+    "opentelemetry-sdk>=1.20.0",
+    "opentelemetry-exporter-otlp-proto-grpc>=1.20.0",
+]
 
 [project.urls]
 Homepage = "https://chalk.ai"

chalk_remote_call_python-1.6.0/chalk_remote_call/_version.py

@@ -1 +0,0 @@
-__version__ = "1.6.0"