nvidia-nat-opentelemetry 1.4.0a20251023__py3-none-any.whl → 1.4.0a20251028__py3-none-any.whl

nat/plugins/opentelemetry/__init__.py

@@ -13,11 +13,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+from nat.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPProtocol
 from nat.plugins.opentelemetry.otel_span_exporter import OtelSpanExporter
 from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 from nat.plugins.opentelemetry.otlp_span_redaction_adapter_exporter import OTLPSpanHeaderRedactionAdapterExporter
 
 __all__ = [
+    "OTLPProtocol",
     "OTLPSpanHeaderRedactionAdapterExporter",
     "OTLPSpanAdapterExporter",
     "OtelSpanExporter",

nat/plugins/opentelemetry/mixin/otlp_span_exporter_mixin.py

@@ -14,21 +14,25 @@
 # limitations under the License.
 
 import logging
+from typing import Literal
 
 from nat.plugins.opentelemetry.otel_span import OtelSpan
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
+from opentelemetry.exporter.otlp.proto.grpc.trace_exporter import OTLPSpanExporter as OTLPSpanExporterGRPC
+from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter as OTLPSpanExporterHTTP
 
 logger = logging.getLogger(__name__)
 
+OTLPProtocol = Literal['http', 'grpc']
+
 
 class OTLPSpanExporterMixin:
     """Mixin for OTLP span exporters.
 
     This mixin provides OTLP-specific functionality for OpenTelemetry span exporters.
-    It handles OTLP protocol transmission using the standard OpenTelemetry OTLP HTTP exporter.
+    It handles OTLP protocol transmission using the standard OpenTelemetry OTLP exporters.
 
     Key Features:
-    - Standard OTLP HTTP protocol support for span export
+    - Standard OTLP HTTP and gRPC protocol support for span export
     - Configurable endpoint and headers for authentication/routing
     - Integration with OpenTelemetry's OTLPSpanExporter for reliable transmission
     - Works with any OTLP-compatible collector or service
@@ -42,16 +46,29 @@ class OTLPSpanExporterMixin:
                 super().__init__(endpoint=endpoint, headers=headers, **kwargs)
     """
 
-    def __init__(self, *args, endpoint: str, headers: dict[str, str] | None = None, **kwargs):
+    def __init__(self,
+                 *args,
+                 endpoint: str,
+                 headers: dict[str, str] | None = None,
+                 protocol: OTLPProtocol = 'http',
+                 **kwargs):
         """Initialize the OTLP span exporter.
 
         Args:
             endpoint: OTLP service endpoint URL.
             headers: HTTP headers for authentication and metadata.
+            protocol: Transport protocol to use ('http' or 'grpc'). Defaults to 'http'.
         """
         # Initialize exporter before super().__init__() to ensure it's available
         # if parent class initialization potentially calls export_otel_spans()
-        self._exporter = OTLPSpanExporter(endpoint=endpoint, headers=headers)
+
+        if protocol == 'http':
+            self._exporter = OTLPSpanExporterHTTP(endpoint=endpoint, headers=headers)
+        elif protocol == 'grpc':
+            self._exporter = OTLPSpanExporterGRPC(endpoint=endpoint, headers=headers)
+        else:
+            raise ValueError(f"Invalid protocol: {protocol}")
+
         super().__init__(*args, **kwargs)
 
     async def export_otel_spans(self, spans: list[OtelSpan]) -> None:

nat/plugins/opentelemetry/otlp_span_adapter_exporter.py

@@ -16,6 +16,7 @@
 import logging
 
 from nat.builder.context import ContextState
+from nat.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPProtocol
 from nat.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPSpanExporterMixin
 from nat.plugins.opentelemetry.otel_span_exporter import OtelSpanExporter
 
@@ -32,7 +33,7 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):
     Key Features:
     - Complete span processing pipeline (IntermediateStep → Span → OtelSpan → Export)
     - Batching support for efficient transmission
-    - OTLP HTTP protocol for maximum compatibility
+    - OTLP HTTP and gRPC protocol for maximum compatibility
     - Configurable authentication via headers
     - Resource attribute management
     - Error handling and retry logic
@@ -48,6 +49,7 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):
         exporter = OTLPSpanAdapterExporter(
             endpoint="https://api.service.com/v1/traces",
             headers={"Authorization": "Bearer your-token"},
+            protocol='http',
             batch_size=50,
             flush_interval=10.0
         )
@@ -67,6 +69,7 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):
             # OTLPSpanExporterMixin args
             endpoint: str,
             headers: dict[str, str] | None = None,
+            protocol: OTLPProtocol = 'http',
             **otlp_kwargs):
         """Initialize the OTLP span exporter.
 
@@ -80,6 +83,7 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):
             resource_attributes: Additional resource attributes for spans.
             endpoint: The endpoint for the OTLP service.
             headers: The headers for the OTLP service.
+            protocol: The protocol to use for the OTLP service, default is 'http'.
             otlp_kwargs: Additional keyword arguments for the OTLP service.
         """
         super().__init__(context_state=context_state,
@@ -91,4 +95,5 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):
                          resource_attributes=resource_attributes,
                          endpoint=endpoint,
                          headers=headers,
+                         protocol=protocol,
                          **otlp_kwargs)

nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py

@@ -22,6 +22,7 @@ from typing import Any
 from nat.builder.context import ContextState
 from nat.observability.processor.redaction import SpanHeaderRedactionProcessor
 from nat.observability.processor.span_tagging_processor import SpanTaggingProcessor
+from nat.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPProtocol
 from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
 logger = logging.getLogger(__name__)
@@ -39,7 +40,7 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
     - Privacy level tagging for compliance and governance
     - Complete span processing pipeline (IntermediateStep → Span → Redaction → Tagging → OtelSpan → Batching → Export)
     - Batching support for efficient transmission
-    - OTLP HTTP protocol for maximum compatibility
+    - OTLP HTTP and gRPC protocol for maximum compatibility
     - Configurable authentication via headers
     - Resource attribute management
     - Error handling and retry logic
@@ -63,6 +64,7 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
         exporter = OTLPSpanRedactionAdapterExporter(
             endpoint="https://api.service.com/v1/traces",
             headers={"Authorization": "Bearer your-token"},
+            protocol='http',
             redaction_attributes=["user.email", "request.body"],
             redaction_headers=["x-user-id"],
             redaction_callback=should_redact,
@@ -96,6 +98,7 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
             # OTLPSpanExporterMixin args
             endpoint: str,
             headers: dict[str, str] | None = None,
+            protocol: OTLPProtocol = 'http',
             **otlp_kwargs):
         """Initialize the OTLP span exporter with redaction and tagging capabilities.
 
@@ -117,6 +120,7 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
             redaction_tag: Tag to add to spans when redaction occurs.
             endpoint: The endpoint for the OTLP service.
             headers: The headers for the OTLP service.
+            protocol: The protocol to use for the OTLP service, default is 'http'.
             otlp_kwargs: Additional keyword arguments for the OTLP service.
         """
         super().__init__(context_state=context_state,
@@ -128,6 +132,7 @@ class OTLPSpanHeaderRedactionAdapterExporter(OTLPSpanAdapterExporter):
                          resource_attributes=resource_attributes,
                          endpoint=endpoint,
                          headers=headers,
+                         protocol=protocol,
                          **otlp_kwargs)
 
         # Insert redaction and tagging processors to the front of the processing pipeline

nat/plugins/opentelemetry/register.py

@@ -159,7 +159,8 @@ async def patronus_telemetry_exporter(config: PatronusTelemetryExporter, builder
                                   flush_interval=config.flush_interval,
                                   max_queue_size=config.max_queue_size,
                                   drop_on_overflow=config.drop_on_overflow,
-                                  shutdown_timeout=config.shutdown_timeout)
+                                  shutdown_timeout=config.shutdown_timeout,
+                                  protocol="grpc")
 
 
 class GalileoTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, TelemetryExporterBaseConfig, name="galileo"):
@@ -192,3 +193,45 @@ async def galileo_telemetry_exporter(config: GalileoTelemetryExporter, builder:
         drop_on_overflow=config.drop_on_overflow,
         shutdown_timeout=config.shutdown_timeout,
     )
+
+
+class DBNLTelemetryExporter(BatchConfigMixin, TelemetryExporterBaseConfig, name="dbnl"):
+    """A telemetry exporter to transmit traces to DBNL."""
+
+    api_url: str | None = Field(description="The DBNL API URL.", default=None)
+    api_token: str | None = Field(description="The DBNL API token.", default=None)
+    project_id: str | None = Field(description="The DBNL project id.", default=None)
+
+
+@register_telemetry_exporter(config_type=DBNLTelemetryExporter)
+async def dbnl_telemetry_exporter(config: DBNLTelemetryExporter, builder: Builder):
+    """Create a DBNL telemetry exporter."""
+
+    from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
+
+    api_token = config.api_token or os.environ.get("DBNL_API_TOKEN")
+    if not api_token:
+        raise ValueError("API token is required for DBNL")
+    project_id = config.project_id or os.environ.get("DBNL_PROJECT_ID")
+    if not project_id:
+        raise ValueError("Project id is required for DBNL")
+
+    headers = {
+        "Authorization": f"Bearer {api_token}",
+        "x-dbnl-project-id": project_id,
+    }
+
+    api_url = config.api_url or os.environ.get("DBNL_API_URL")
+    if not api_url:
+        raise ValueError("API url is required for DBNL")
+    endpoint = api_url.rstrip("/") + "/otel/v1/traces"
+
+    yield OTLPSpanAdapterExporter(
+        endpoint=endpoint,
+        headers=headers,
+        batch_size=config.batch_size,
+        flush_interval=config.flush_interval,
+        max_queue_size=config.max_queue_size,
+        drop_on_overflow=config.drop_on_overflow,
+        shutdown_timeout=config.shutdown_timeout,
+    )

{nvidia_nat_opentelemetry-1.4.0a20251023.dist-info → nvidia_nat_opentelemetry-1.4.0a20251028.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-opentelemetry
-Version: 1.4.0a20251023
+Version: 1.4.0a20251028
 Summary: Subpackage for OpenTelemetry integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
@@ -16,7 +16,7 @@ Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE-3rd-party.txt
 License-File: LICENSE.md
-Requires-Dist: nvidia-nat==v1.4.0a20251023
+Requires-Dist: nvidia-nat==v1.4.0a20251028
 Requires-Dist: opentelemetry-api~=1.2
 Requires-Dist: opentelemetry-exporter-otlp~=1.3
 Requires-Dist: opentelemetry-sdk~=1.3

nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/RECORD

@@ -0,0 +1,17 @@
+nat/meta/pypi.md,sha256=_o1o1BLPY1pvjCkklWxlm7LlIDMPCk-2Rho85NUuN8U,1109
+nat/plugins/opentelemetry/__init__.py,sha256=6RYrwlYGss4SzyAwjcbgFTLKaLk9jEIHTGEAF8XiC5E,1181
+nat/plugins/opentelemetry/otel_span.py,sha256=MC_ROZ8gSTu0gxRaaz77UDbn1ouZTZP3N_-0PcN140U,16564
+nat/plugins/opentelemetry/otel_span_exporter.py,sha256=YO7JsQgi8Cf2OQBJ_s78HwJjrWx9SdqMvPen3Pa2_bI,6533
+nat/plugins/opentelemetry/otlp_span_adapter_exporter.py,sha256=cWBfPKHET7U1oRMs1QMXvg-z1c1DaqYrLzHIyUUbfGM,4237
+nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py,sha256=YnoQlktT6EJRM7o8bhvYQcdGrAdOXNEk8wOimK_Wy88,7509
+nat/plugins/opentelemetry/register.py,sha256=u3Hq5mhnDfYn4Q-XkKCGr13xuM6SpS8d3OhroTUjiCc,10711
+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=BN8fRHHEHEWZyrYoudl9lSeH--I2RY2dJVPyWIgDB0I,3454
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/METADATA,sha256=joY4cSLPTHUKhnqu_R6sv3tItHbyM5VAd3rVBkTKikc,2039
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_opentelemetry-1.4.0a20251028.dist-info/RECORD,,

nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/RECORD

@@ -1,17 +0,0 @@
-nat/meta/pypi.md,sha256=_o1o1BLPY1pvjCkklWxlm7LlIDMPCk-2Rho85NUuN8U,1109
-nat/plugins/opentelemetry/__init__.py,sha256=j-YKuxwSIzGziyDyunw8s_W7WiFiqKLdFjw-utwHPFk,1079
-nat/plugins/opentelemetry/otel_span.py,sha256=MC_ROZ8gSTu0gxRaaz77UDbn1ouZTZP3N_-0PcN140U,16564
-nat/plugins/opentelemetry/otel_span_exporter.py,sha256=YO7JsQgi8Cf2OQBJ_s78HwJjrWx9SdqMvPen3Pa2_bI,6533
-nat/plugins/opentelemetry/otlp_span_adapter_exporter.py,sha256=6xQHkKDhQk3-kObqj6kRv8ZlJtIV2Qqox6YkY0PCOYs,3945
-nat/plugins/opentelemetry/otlp_span_redaction_adapter_exporter.py,sha256=qRrUIxYRlCDaAkEGlFTaEkCJCsb68GBo11EGEiE6S8E,7217
-nat/plugins/opentelemetry/register.py,sha256=KnhV-axY0kJzZ3RReG4e_mTFR1dMr7d3a6ysYiCLTUI,9063
-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=3vK6DkTJXp6ZFH3AgNYUuuMOzjyskh_nVUWK-qMYKzM,2809
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/METADATA,sha256=R7MWQaHgT0BaJAM_fGZ0JL3NP5sC4MUrPcOtEiwzvcQ,2039
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_opentelemetry-1.4.0a20251023.dist-info/RECORD,,