nvidia-nat-opentelemetry 1.4.0a20251010__py3-none-any.whl → 1.4.0a20251206__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

@@ -20,6 +20,9 @@ from pydantic import Field
 
 from nat.builder.builder import Builder
 from nat.cli.register_workflow import register_telemetry_exporter
+from nat.data_models.common import OptionalSecretStr
+from nat.data_models.common import SerializableSecretStr
+from nat.data_models.common import get_secret_value
 from nat.data_models.telemetry_exporter import TelemetryExporterBaseConfig
 from nat.observability.mixin.batch_config_mixin import BatchConfigMixin
 from nat.observability.mixin.collector_config_mixin import CollectorConfigMixin
@@ -31,8 +34,8 @@ class LangfuseTelemetryExporter(BatchConfigMixin, TelemetryExporterBaseConfig, n
     """A telemetry exporter to transmit traces to externally hosted langfuse service."""
 
     endpoint: str = Field(description="The langfuse OTEL endpoint (/api/public/otel/v1/traces)")
-    public_key: str = Field(description="The Langfuse public key", default="")
-    secret_key: str = Field(description="The Langfuse secret key", default="")
+    public_key: SerializableSecretStr = Field(description="The Langfuse public key", default="")
+    secret_key: SerializableSecretStr = Field(description="The Langfuse secret key", default="")
     resource_attributes: dict[str, str] = Field(default_factory=dict,
                                                 description="The resource attributes to add to the span")
 
@@ -44,8 +47,8 @@ async def langfuse_telemetry_exporter(config: LangfuseTelemetryExporter, builder
 
     from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
 
-    secret_key = config.secret_key or os.environ.get("LANGFUSE_SECRET_KEY")
-    public_key = config.public_key or os.environ.get("LANGFUSE_PUBLIC_KEY")
+    secret_key = get_secret_value(config.secret_key) if config.secret_key else os.environ.get("LANGFUSE_SECRET_KEY")
+    public_key = get_secret_value(config.public_key) if config.public_key else os.environ.get("LANGFUSE_PUBLIC_KEY")
     if not secret_key or not public_key:
         raise ValueError("secret and public keys are required for langfuse")
 
@@ -69,7 +72,7 @@ class LangsmithTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, Telemet
         description="The langsmith OTEL endpoint",
         default="https://api.smith.langchain.com/otel/v1/traces",
     )
-    api_key: str = Field(description="The Langsmith API key", default="")
+    api_key: SerializableSecretStr = Field(description="The Langsmith API key", default="")
     resource_attributes: dict[str, str] = Field(default_factory=dict,
                                                 description="The resource attributes to add to the span")
 
@@ -80,7 +83,7 @@ async def langsmith_telemetry_exporter(config: LangsmithTelemetryExporter, build
 
     from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
 
-    api_key = config.api_key or os.environ.get("LANGSMITH_API_KEY")
+    api_key = get_secret_value(config.api_key) if config.api_key else os.environ.get("LANGSMITH_API_KEY")
     if not api_key:
         raise ValueError("API key is required for langsmith")
 
@@ -134,7 +137,7 @@ async def otel_telemetry_exporter(config: OtelCollectorTelemetryExporter, builde
 class PatronusTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, TelemetryExporterBaseConfig, name="patronus"):
     """A telemetry exporter to transmit traces to Patronus service."""
 
-    api_key: str = Field(description="The Patronus API key", default="")
+    api_key: SerializableSecretStr = Field(description="The Patronus API key", default="")
     resource_attributes: dict[str, str] = Field(default_factory=dict,
                                                 description="The resource attributes to add to the span")
 
@@ -145,7 +148,7 @@ async def patronus_telemetry_exporter(config: PatronusTelemetryExporter, builder
 
     from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
 
-    api_key = config.api_key or os.environ.get("PATRONUS_API_KEY")
+    api_key = get_secret_value(config.api_key) if config.api_key else os.environ.get("PATRONUS_API_KEY")
     if not api_key:
         raise ValueError("API key is required for Patronus")
 
@@ -159,7 +162,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"):
@@ -168,7 +172,7 @@ class GalileoTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, Telemetry
     endpoint: str = Field(description="The galileo endpoint to export telemetry traces.",
                           default="https://app.galileo.ai/api/galileo/otel/traces")
     logstream: str = Field(description="The logstream name to group the telemetry traces.")
-    api_key: str = Field(description="The api key to authenticate with the galileo service.")
+    api_key: SerializableSecretStr = Field(description="The api key to authenticate with the galileo service.")
 
 
 @register_telemetry_exporter(config_type=GalileoTelemetryExporter)
@@ -178,7 +182,7 @@ async def galileo_telemetry_exporter(config: GalileoTelemetryExporter, builder:
     from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
 
     headers = {
-        "Galileo-API-Key": config.api_key,
+        "Galileo-API-Key": get_secret_value(config.api_key),
         "logstream": config.logstream,
         "project": config.project,
     }
@@ -192,3 +196,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: OptionalSecretStr = 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 = get_secret_value(config.api_token) if config.api_token else 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.0a20251010.dist-info → nvidia_nat_opentelemetry-1.4.0a20251206.dist-info}/METADATA

@@ -1,10 +1,10 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-opentelemetry
-Version: 1.4.0a20251010
+Version: 1.4.0a20251206
 Summary: Subpackage for OpenTelemetry integration in NeMo Agent toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation
-License-Expression: Apache-2.0
+License: Apache-2.0
 Project-URL: documentation, https://docs.nvidia.com/nemo/agent-toolkit/latest/
 Project-URL: source, https://github.com/NVIDIA/NeMo-Agent-Toolkit
 Keywords: ai,observability,opentelemetry
@@ -14,9 +14,9 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <3.14,>=3.11
 Description-Content-Type: text/markdown
-License-File: LICENSE.md
 License-File: LICENSE-3rd-party.txt
-Requires-Dist: nvidia-nat==v1.4.0a20251010
+License-File: LICENSE.md
+Requires-Dist: nvidia-nat==v1.4.0a20251206
 Requires-Dist: opentelemetry-api~=1.2
 Requires-Dist: opentelemetry-exporter-otlp~=1.3
 Requires-Dist: opentelemetry-sdk~=1.3

nvidia_nat_opentelemetry-1.4.0a20251206.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=EZpGOYQ5FVXfwDUea8TfFqWgZnhGP9NFFb1oGFajk5I,11186
+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.0a20251206.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/METADATA,sha256=dUafjNuyZdjnYKLdFsGNFHA1BBI1vyCL-FtAyFnhkVo,2039
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_opentelemetry-1.4.0a20251206.dist-info/RECORD,,

nvidia_nat_opentelemetry-1.4.0a20251010.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.0a20251010.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/METADATA,sha256=Vtm8fsJe3kfgnFZsVRii0RB6GQZYkMSHrznui2xw4Z4,2050
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_opentelemetry-1.4.0a20251010.dist-info/RECORD,,