nvidia-nat-ragaai 1.3.0a20250923__py3-none-any.whl → 1.4.0a20251124__py3-none-any.whl

nat/plugins/ragaai/mixin/ragaai_catalyst_mixin.py

@@ -56,7 +56,7 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                                                    self.user_gt,
                                                    self.external_id)
             except Exception as e:
-                print(f"Error in convert_json_format function: {trace_id}: {e}")
+                logger.exception("Error in convert_json_format function: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
@@ -64,7 +64,7 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                 if interactions and 'workflow' in interactions:
                     ragaai_trace["workflow"] = interactions['workflow']
             except Exception as e:
-                print(f"Error in format_interactions function: {trace_id}: {e}")
+                logger.exception("Error in format_interactions function: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
@@ -72,20 +72,23 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                 files_to_zip = self.files_to_zip or []
                 hash_id, zip_path = zip_list_of_unique_files(files_to_zip, output_dir=self.tmp_dir)
             except Exception as e:
-                print(f"Error in zip_list_of_unique_files function: {trace_id}: {e}")
+                logger.exception("Error in zip_list_of_unique_files function: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
                 ragaai_trace["metadata"]["system_info"] = asdict(self.system_monitor.get_system_info())
                 ragaai_trace["metadata"]["resources"] = asdict(self.system_monitor.get_resources())
             except Exception as e:
-                print(f"Error in get_system_info or get_resources function: {trace_id}: {e}")
+                logger.exception("Error in get_system_info or get_resources function: %s: %s",
+                                 trace_id,
+                                 e,
+                                 exc_info=True)
                 return None
 
             try:
                 ragaai_trace["metadata"]["system_info"]["source_code"] = hash_id
             except Exception as e:
-                print(f"Error in adding source code hash: {trace_id}: {e}")
+                logger.exception("Error in adding source code hash: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
@@ -95,14 +98,14 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                     if "end_time" in ragaai_trace:
                         ragaai_trace["data"][0]["end_time"] = ragaai_trace["end_time"]
             except Exception as e:
-                print(f"Error in adding start_time or end_time: {trace_id}: {e}")
+                logger.exception("Error in adding start_time or end_time: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
                 if hasattr(self, 'project_name'):
                     ragaai_trace["project_name"] = self.project_name
             except Exception as e:
-                print(f"Error in adding project name: {trace_id}: {e}")
+                logger.exception("Error in adding project name: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
@@ -110,7 +113,7 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                 if hasattr(self, 'tracer_type'):
                     ragaai_trace["tracer_type"] = self.tracer_type
             except Exception as e:
-                print(f"Error in adding tracer type: {trace_id}: {e}")
+                logger.exception("Error in adding tracer type: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             # Add user passed metadata to the trace
@@ -127,7 +130,7 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
 
                 logger.debug("Completed adding user passed metadata")
             except Exception as e:
-                print(f"Error in adding metadata: {trace_id}: {e}")
+                logger.exception("Error in adding metadata: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             try:
@@ -140,12 +143,12 @@ class RAGATraceExporterOptWrite(RAGATraceExporter):
                     with open(os.path.join(os.getcwd(), 'rag_agent_traces.json'), 'w', encoding="utf-8") as f:
                         json.dump(ragaai_trace, f, cls=TracerJSONEncoder, indent=2)
             except Exception as e:
-                print(f"Error in saving trace json: {trace_id}: {e}")
+                logger.exception("Error in saving trace json: %s: %s", trace_id, e, exc_info=True)
                 return None
 
             return {'trace_file_path': trace_file_path, 'code_zip_path': zip_path, 'hash_id': hash_id}
         except Exception as e:
-            print(f"Error converting trace {trace_id}: {str(e)}")
+            logger.exception("Error converting trace %s: %s", trace_id, str(e), exc_info=True)
             return None
 
 
@@ -185,7 +188,8 @@ class RagaAICatalystMixin:
 
     This mixin is designed to be used with OtelSpanExporter as a base class:
 
-    Example:
+    Example::
+
         class MyCatalystExporter(OtelSpanExporter, RagaAICatalystMixin):
             def __init__(self, base_url, access_key, secret_key, project, dataset, **kwargs):
                 super().__init__(base_url=base_url, access_key=access_key,
@@ -211,9 +215,9 @@ class RagaAICatalystMixin:
             project: RagaAI Catalyst project name.
             dataset: RagaAI Catalyst dataset name.
             tracer_type: RagaAI Catalyst tracer type.
-            debug_mode: When False (default), creates local rag_agent_traces.json file.
-                       When True, skips local file creation for cleaner operation.
-            **kwargs: Additional keyword arguments passed to parent classes.
+            debug_mode: When False (default), creates local rag_agent_traces.json file. When True, skips local file
+            creation for cleaner operation.
+            kwargs: Additional keyword arguments passed to parent classes.
         """
         logger.info("RagaAICatalystMixin initialized with debug_mode=%s", debug_mode)
 
@@ -241,4 +245,4 @@ class RagaAICatalystMixin:
             loop = asyncio.get_event_loop()
             await loop.run_in_executor(None, lambda: self._exporter.export(spans))  # type: ignore[arg-type]
         except Exception as e:
-            logger.error("Error exporting spans: %s", e, exc_info=True)
+            logger.exception("Error exporting spans: %s", e, exc_info=True)

nat/plugins/ragaai/ragaai_catalyst_exporter.py

@@ -42,8 +42,8 @@ class RagaAICatalystExporter(RagaAICatalystMixin, OtelSpanExporter):
         project: Project name for trace grouping
         dataset: Dataset name for trace organization
         tracer_type: RagaAI Catalyst tracer type.
-        debug_mode: When False (default), creates local rag_agent_traces.json file.
-                   When True, skips local file creation for cleaner operation.
+        debug_mode: When False (default), creates local rag_agent_traces.json file. When True, skips local file
+        creation for cleaner operation.
         batch_size: Batch size for exporting
         flush_interval: Flush interval for exporting
         max_queue_size: Maximum queue size for exporting

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.3.0a20250923.dist-info → nvidia_nat_ragaai-1.4.0a20251124.dist-info}/METADATA

@@ -1,7 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-ragaai
-Version: 1.3.0a20250923
+Version: 1.4.0a20251124
 Summary: Subpackage for RagaAI Catalyst integration in NeMo Agent toolkit
+Author: NVIDIA Corporation
+Maintainer: NVIDIA Corporation
+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,ragaai catalyst
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.11
@@ -9,8 +14,11 @@ 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[opentelemetry]==v1.3.0a20250923
+License-File: LICENSE-3rd-party.txt
+License-File: LICENSE.md
+Requires-Dist: nvidia-nat[litellm,opentelemetry]==v1.4.0a20251124
 Requires-Dist: ragaai-catalyst<2.3,>=2.2.0
+Dynamic: license-file
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

nvidia_nat_ragaai-1.4.0a20251124.dist-info/RECORD

@@ -0,0 +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=0-Lgxx6rQF3Bp0JCsj2R3nxf1qTr5caBD4PEC5GM-6Q,3988
+nat/plugins/ragaai/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
+nat/plugins/ragaai/mixin/ragaai_catalyst_mixin.py,sha256=4XXlfInqMon6RdqizXKtjGedjqimvpxmNaAXX6fKBoc,11331
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/METADATA,sha256=_RQUI5jtke2Sea799859Eu3RwwY0cSlYjU3efhu6tCs,1981
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/entry_points.txt,sha256=UhynvDN2vlLQB-hmjrvXnbTSyAe8sUhT2dFb-16Hum8,58
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_ragaai-1.4.0a20251124.dist-info/RECORD,,