nvidia-nat-ragaai 1.4.0a20251029__py3-none-any.whl → 1.4.0a20251031__py3-none-any.whl

nat/plugins/ragaai/register.py

@@ -19,6 +19,8 @@ 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 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
@@ -29,8 +31,8 @@ logger = logging.getLogger(__name__)
 class CatalystTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, TelemetryExporterBaseConfig, name="catalyst"):
     """A telemetry exporter to transmit traces to RagaAI catalyst."""
     endpoint: str = Field(description="The RagaAI Catalyst endpoint", default="https://catalyst.raga.ai/api")
-    access_key: str = Field(description="The RagaAI Catalyst API access key", default="")
-    secret_key: str = Field(description="The RagaAI Catalyst API secret key", default="")
+    access_key: SerializableSecretStr = Field(description="The RagaAI Catalyst API access key", default="")
+    secret_key: SerializableSecretStr = Field(description="The RagaAI Catalyst API secret key", default="")
     dataset: str | None = Field(description="The RagaAI Catalyst dataset name", default=None)
     tracer_type: str = Field(description="The RagaAI Catalyst tracer type", default="agentic/nemo-framework")
 
@@ -49,8 +51,8 @@ async def catalyst_telemetry_exporter(config: CatalystTelemetryExporter, builder
 
         from nat.plugins.ragaai.ragaai_catalyst_exporter import RagaAICatalystExporter
 
-        access_key = config.access_key or os.environ.get("CATALYST_ACCESS_KEY")
-        secret_key = config.secret_key or os.environ.get("CATALYST_SECRET_KEY")
+        access_key = get_secret_value(config.access_key) if config.access_key else os.environ.get("CATALYST_ACCESS_KEY")
+        secret_key = get_secret_value(config.secret_key) if config.secret_key else os.environ.get("CATALYST_SECRET_KEY")
         endpoint = config.endpoint or os.environ.get("CATALYST_ENDPOINT")
 
         assert endpoint is not None, "catalyst endpoint is not set"

{nvidia_nat_ragaai-1.4.0a20251029.dist-info → nvidia_nat_ragaai-1.4.0a20251031.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-ragaai
-Version: 1.4.0a20251029
+Version: 1.4.0a20251031
 Summary: Subpackage for RagaAI Catalyst 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[litellm,opentelemetry]==v1.4.0a20251029
+Requires-Dist: nvidia-nat[litellm,opentelemetry]==v1.4.0a20251031
 Requires-Dist: ragaai-catalyst<2.3,>=2.2.0
 Dynamic: license-file
 

{nvidia_nat_ragaai-1.4.0a20251029.dist-info → nvidia_nat_ragaai-1.4.0a20251031.dist-info}/RECORD

@@ -1,13 +1,13 @@
 nat/meta/pypi.md,sha256=0pjv0kFWJXwkDfjOfF0atoxFZLL9Anh_mnHVR8PUB0I,1112
 nat/plugins/ragaai/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/ragaai/ragaai_catalyst_exporter.py,sha256=6ycaprMCTbPgzzdEMYu3FsnvePj4BjwBN3Pu1mGbYF4,2958
-nat/plugins/ragaai/register.py,sha256=RQeKR_LAha0mokRqjoMOlzcgZ-J1BDK1aLYPi3s7B3o,3761
+nat/plugins/ragaai/register.py,sha256=0-Lgxx6rQF3Bp0JCsj2R3nxf1qTr5caBD4PEC5GM-6Q,3988
 nat/plugins/ragaai/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 nat/plugins/ragaai/mixin/ragaai_catalyst_mixin.py,sha256=-0It6fYTTqYbRpUjUXTJRt7Z5EZW7Xip3Fq2jQvJcY0,10909
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/METADATA,sha256=PhIOZ6COOKz5-qGjkL6yNsbC0oj73du_r8njWXt98pI,1981
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/entry_points.txt,sha256=UhynvDN2vlLQB-hmjrvXnbTSyAe8sUhT2dFb-16Hum8,58
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_ragaai-1.4.0a20251029.dist-info/RECORD,,
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/METADATA,sha256=TbUAFlVeoO6IGEq0zjfSRQwaytc6_jWhMTRCJ1HN4M8,1981
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/entry_points.txt,sha256=UhynvDN2vlLQB-hmjrvXnbTSyAe8sUhT2dFb-16Hum8,58
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_ragaai-1.4.0a20251031.dist-info/RECORD,,