nvidia-nat-opentelemetry 1.4.0a20251029__py3-none-any.whl → 1.4.0a20251030__py3-none-any.whl

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")
 
@@ -169,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)
@@ -179,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,
     }
@@ -199,7 +202,7 @@ class DBNLTelemetryExporter(BatchConfigMixin, TelemetryExporterBaseConfig, name=
     """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)
+    api_token: OptionalSecretStr = Field(description="The DBNL API token.", default=None)
     project_id: str | None = Field(description="The DBNL project id.", default=None)
 
 
@@ -209,7 +212,7 @@ async def dbnl_telemetry_exporter(config: DBNLTelemetryExporter, builder: Builde
 
     from nat.plugins.opentelemetry import OTLPSpanAdapterExporter
 
-    api_token = config.api_token or os.environ.get("DBNL_API_TOKEN")
+    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")

{nvidia_nat_opentelemetry-1.4.0a20251029.dist-info → nvidia_nat_opentelemetry-1.4.0a20251030.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-opentelemetry
-Version: 1.4.0a20251029
+Version: 1.4.0a20251030
 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.0a20251029
+Requires-Dist: nvidia-nat==v1.4.0a20251030
 Requires-Dist: opentelemetry-api~=1.2
 Requires-Dist: opentelemetry-exporter-otlp~=1.3
 Requires-Dist: opentelemetry-sdk~=1.3

{nvidia_nat_opentelemetry-1.4.0a20251029.dist-info → nvidia_nat_opentelemetry-1.4.0a20251030.dist-info}/RECORD

@@ -4,14 +4,14 @@ nat/plugins/opentelemetry/otel_span.py,sha256=MC_ROZ8gSTu0gxRaaz77UDbn1ouZTZP3N_
 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/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.0a20251029.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/METADATA,sha256=3Vu8S1cXBK7cCMtq0eI5Pncl2ecJso--3dUIoUwT16s,2039
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_opentelemetry-1.4.0a20251029.dist-info/RECORD,,
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/METADATA,sha256=QxzBBBGg1D24iGwPwhonozJX8nvxRpWgQwvXwUD7jMk,2039
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_opentelemetry-1.4.0a20251030.dist-info/RECORD,,