nvidia-nat-opentelemetry 1.3.0a20250910__py3-none-any.whl → 1.3.0a20250917__py3-none-any.whl

nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py

@@ -15,10 +15,12 @@
 
 import logging
 from collections.abc import Callable
+from collections.abc import Mapping
+from enum import Enum
+from typing import Any
 
 from nat.builder.context import ContextState
-from nat.observability.mixin.tagging_config_mixin import PrivacyLevel
-from nat.observability.processor.header_redaction_processor import HeaderRedactionProcessor
+from nat.observability.processor.redaction import SpanHeaderRedactionProcessor
 from nat.observability.processor.span_tagging_processor import SpanTaggingProcessor
 from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
@@ -61,11 +63,10 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
             endpoint="https://api.service.com/v1/traces",
             headers={"Authorization": "Bearer your-token"},
             redaction_attributes=["user.email", "request.body"],
-            redaction_header="x-user-id",
+            redaction_headers=["x-user-id"],
             redaction_callback=should_redact,
             redaction_value="REDACTED",
-            privacy_tag_key="privacy.level",
-            privacy_tag_value=PrivacyLevel.HIGH,
+            tags={"privacy.level": PrivacyLevel.HIGH, "service.type": "sensitive"},
             batch_size=50,
             flush_interval=10.0
         )
@@ -84,13 +85,13 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
             resource_attributes: dict[str, str] | None = None,
             # Redaction args
             redaction_attributes: list[str] | None = None,
-            redaction_header: str | None = None,
-            redaction_callback: Callable[[str], bool] | None = None,
+            redaction_headers: list[str] | None = None,
+            redaction_callback: Callable[..., Any] | None = None,
             redaction_enabled: bool = False,
             force_redaction: bool = False,
             redaction_value: str = "[REDACTED]",
-            privacy_tag_key: str | None = None,
-            privacy_tag_value: PrivacyLevel | None = None,
+            redaction_tag: str | None = None,
+            tags: Mapping[str, Enum | str] | None = None,
             # OTLPSpanExporterMixin args
             endpoint: str,
             headers: dict[str, str] | None = None,
@@ -99,20 +100,20 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
 
         Args:
             context_state: The context state for the exporter.
-            batch_size: Number of spans to batch before exporting.
-            flush_interval: Time in seconds between automatic batch flushes.
-            max_queue_size: Maximum number of spans to queue.
-            drop_on_overflow: Whether to drop spans when queue is full.
-            shutdown_timeout: Maximum time to wait for export completion during shutdown.
+            batch_size: Number of spans to batch before exporting, default is 100.
+            flush_interval: Time in seconds between automatic batch flushes, default is 5.0.
+            max_queue_size: Maximum number of spans to queue, default is 1000.
+            drop_on_overflow: Whether to drop spans when queue is full, default is False.
+            shutdown_timeout: Maximum time to wait for export completion during shutdown, default is 10.0.
             resource_attributes: Additional resource attributes for spans.
             redaction_attributes: List of span attribute keys to redact when conditions are met.
-            redaction_header: Header key to check for authentication/user identification.
-            redaction_callback: Function to determine if spans should be redacted based on header value.
-            redaction_enabled: Whether the redaction processor is enabled.
-            force_redaction: If True, always redact regardless of header checks.
-            redaction_value: Value to replace redacted attributes with.
-            privacy_tag_key: Key name for the privacy level tag to add to spans.
-            privacy_tag_value: Privacy level value to assign to spans.
+            redaction_headers: List of header keys to check for authentication/user identification.
+            redaction_callback: Function that returns true to redact spans based on header value, false otherwise.
+            redaction_enabled: Whether the redaction processor is enabled, default is False.
+            force_redaction: If True, always redact regardless of header checks, default is False.
+            redaction_value: Value to replace redacted attributes with, default is "[REDACTED]".
+            tags: Mapping of tag keys to their values (enums or strings) to add to spans.
+            redaction_tag: Tag to add to spans when redaction occurs.
             endpoint: The endpoint for the OTLP service.
             headers: The headers for the OTLP service.
             **otlp_kwargs: Additional keyword arguments for the OTLP service.
@@ -129,16 +130,14 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
                          **otlp_kwargs)
 
         # Insert redaction and tagging processors to the front of the processing pipeline
-        self.add_processor(HeaderRedactionProcessor(attributes=redaction_attributes,
-                                                    header=redaction_header,
-                                                    callback=redaction_callback,
-                                                    enabled=redaction_enabled,
-                                                    force_redact=force_redaction,
-                                                    redaction_value=redaction_value),
+        self.add_processor(SpanHeaderRedactionProcessor(attributes=redaction_attributes or [],
+                                                        headers=redaction_headers or [],
+                                                        callback=redaction_callback or (lambda _: False),
+                                                        enabled=redaction_enabled,
+                                                        force_redact=force_redaction,
+                                                        redaction_value=redaction_value,
+                                                        redaction_tag=redaction_tag),
                            name="header_redaction",
                            position=0)
 
-        self.add_processor(SpanTaggingProcessor(tag_key=privacy_tag_key,
-                                                tag_value=privacy_tag_value.value if privacy_tag_value else None),
-                           name="span_privacy_tagging",
-                           position=1)
+        self.add_processor(SpanTaggingProcessor(tags=tags), name="span_sensitivity_tagging", position=1)

{nvidia_nat_opentelemetry-1.3.0a20250910.dist-info → nvidia_nat_opentelemetry-1.3.0a20250917.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-opentelemetry
-Version: 1.3.0a20250910
+Version: 1.3.0a20250917
 Summary: Subpackage for OpenTelemetry integration in NeMo Agent toolkit
 Keywords: ai,observability,opentelemetry
 Classifier: Programming Language :: Python
@@ -9,7 +9,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat==v1.3.0a20250910
+Requires-Dist: nvidia-nat==v1.3.0a20250917
 Requires-Dist: opentelemetry-api~=1.2
 Requires-Dist: opentelemetry-exporter-otlp~=1.3
 Requires-Dist: opentelemetry-sdk~=1.3

{nvidia_nat_opentelemetry-1.3.0a20250910.dist-info → nvidia_nat_opentelemetry-1.3.0a20250917.dist-info}/RECORD

@@ -3,13 +3,13 @@ nat/plugins/opentelemetry/__init__.py,sha256=j-YKuxwSIzGziyDyunw8s_W7WiFiqKLdFjw
 nat/plugins/opentelemetry/otel_span.py,sha256=v5Bzy4Cq51Sph5fKErZjt_UPfPXsVJx2qllj7c5afe4,16511
 nat/plugins/opentelemetry/otel_span_exporter.py,sha256=YO7JsQgi8Cf2OQBJ_s78HwJjrWx9SdqMvPen3Pa2_bI,6533
 nat/plugins/opentelemetry/otlp_span_adapter_exporter.py,sha256=C5MaZOB_ZijAbRZpRUtjpBsH0-FHeon81l5xAFlUKos,3945
-nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py,sha256=rxbTcPCYPoLinseVId4jllVP9E5NHD2vgcsy4-NsE9E,7086
+nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py,sha256=FPh_wGpI-oYIX5FCDgVQOu9FSyrJ4yUtY-SlaxnL2FE,7217
 nat/plugins/opentelemetry/register.py,sha256=IWPDQBIhHB7OcykOlCQ5RlMrVWjWQvoJwlgy6QCjir8,9070
 nat/plugins/opentelemetry/span_converter.py,sha256=Gz3KvRNQeEBBlpaPO8YRAJkw4fmzV7m9bT6dGX0IV2E,8846
 nat/plugins/opentelemetry/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/opentelemetry/mixin/otlp_span_exporter_mixin.py,sha256=stTer5KT6l2RA93_pexoU8WqhQLCnYbjyOhIMbDEwWk,2807
-nvidia_nat_opentelemetry-1.3.0a20250910.dist-info/METADATA,sha256=383wC6qSke_K2R5xeDlAX9Df44lpylTrYRyNheVNhwM,1733
-nvidia_nat_opentelemetry-1.3.0a20250910.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_opentelemetry-1.3.0a20250910.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
-nvidia_nat_opentelemetry-1.3.0a20250910.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_opentelemetry-1.3.0a20250910.dist-info/RECORD,,
+nvidia_nat_opentelemetry-1.3.0a20250917.dist-info/METADATA,sha256=uuwWEvEcpYnUsKRQq4Tzj877fWLGc54xkHeeaykjr54,1733
+nvidia_nat_opentelemetry-1.3.0a20250917.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_opentelemetry-1.3.0a20250917.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
+nvidia_nat_opentelemetry-1.3.0a20250917.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_opentelemetry-1.3.0a20250917.dist-info/RECORD,,