nvidia-nat-weave 1.4.0a20251015__py3-none-any.whl → 1.4.0a20251022__py3-none-any.whl

nat/plugins/weave/register.py

@@ -27,7 +27,11 @@ logger = logging.getLogger(__name__)
 class WeaveTelemetryExporter(TelemetryExporterBaseConfig, name="weave"):
     """A telemetry exporter to transmit traces to Weights & Biases Weave using OpenTelemetry."""
     project: str = Field(description="The W&B project name.")
-    entity: str | None = Field(default=None, description="The W&B username or team name.")
+    entity: str | None = Field(default=None,
+                               description="The W&B username or team name.",
+                               deprecated=('This field is deprecated and will be removed in future versions. '
+                                           'This value is set automatically by the weave library, and setting it will '
+                                           'have no effect.'))
     redact_pii: bool = Field(default=False, description="Whether to redact PII from the traces.")
     redact_pii_fields: list[str] | None = Field(
         default=None,

nat/plugins/weave/weave_exporter.py

@@ -16,12 +16,14 @@
 import logging
 from collections.abc import Generator
 from contextlib import contextmanager
+from typing import Any
 
 from nat.data_models.intermediate_step import IntermediateStep
 from nat.data_models.span import Span
 from nat.observability.exporter.base_exporter import IsolatedAttribute
 from nat.observability.exporter.span_exporter import SpanExporter
 from nat.utils.log_utils import LogFilter
+from nat.utils.string_utils import truncate_string
 from nat.utils.type_utils import override
 from weave.trace.context import weave_client_context
 from weave.trace.context.call_context import get_current_call
@@ -142,9 +144,9 @@ class WeaveExporter(SpanExporter[Span, Span]):
         # Generate a meaningful operation name based on event type
         event_type = step.payload.event_type.split(".")[-1]
         if step.payload.name:
-            op_name = f"aiq.{event_type}.{step.payload.name}"
+            op_name = f"nat.{event_type}.{step.payload.name}"
         else:
-            op_name = f"aiq.{event_type}"
+            op_name = f"nat.{event_type}"
 
         # Create input dictionary
         inputs = {}
@@ -152,6 +154,7 @@ class WeaveExporter(SpanExporter[Span, Span]):
             try:
                 # Add the input to the Weave call
                 inputs["input"] = step.payload.data.input
+                self._extract_input_message(step.payload.data.input, inputs)
             except Exception:
                 # If serialization fails, use string representation
                 inputs["input"] = str(step.payload.data.input)
@@ -176,6 +179,76 @@ class WeaveExporter(SpanExporter[Span, Span]):
 
         return call
 
+    def _extract_input_message(self, input_data: Any, inputs: dict[str, Any]) -> None:
+        """
+        Extract message content from input data and add to inputs dictionary.
+        Also handles websocket mode where message is located at messages[0].content[0].text.
+
+        Args:
+            input_data: The raw input data from the request
+            inputs: Dictionary to populate with extracted message content
+        """
+        # Extract message content if input has messages attribute
+        messages = getattr(input_data, 'messages', [])
+        if messages:
+            content = messages[0].content
+            if isinstance(content, list) and content:
+                inputs["input_message"] = getattr(content[0], 'text', content[0])
+            else:
+                inputs["input_message"] = content
+
+    def _extract_output_message(self, output_data: Any, outputs: dict[str, Any]) -> None:
+        """
+        Extract message content from various response formats and add a preview to the outputs dictionary.
+        No data is added to the outputs dictionary if the output format is not supported.
+
+        Supported output formats for message content include:
+
+        - output.choices[0].message.content     /chat endpoint
+        - output.value                          /generate endpoint
+        - output[0].choices[0].message.content  chat WS schema
+        - output[0].choices[0].delta.content    chat_stream WS schema, /chat/stream endpoint
+        - output[0].value                       generate & generate_stream WS schema, /generate/stream endpoint
+
+        Args:
+            output_data: The raw output data from the response
+            outputs: Dictionary to populate with extracted message content.
+        """
+        # Handle choices-keyed output object for /chat completion endpoint
+        choices = getattr(output_data, 'choices', None)
+        if choices:
+            outputs["output_message"] = truncate_string(choices[0].message.content)
+            return
+
+        # Handle value-keyed output object for union types common for /generate completion endpoint
+        value = getattr(output_data, 'value', None)
+        if value:
+            outputs["output_message"] = truncate_string(value)
+            return
+
+        # Handle list-based outputs (streaming or websocket)
+        if not isinstance(output_data, list) or not output_data:
+            return
+
+        choices = getattr(output_data[0], 'choices', None)
+        if choices:
+            # chat websocket schema
+            message = getattr(choices[0], 'message', None)
+            if message:
+                outputs["output_message"] = truncate_string(getattr(message, 'content', None))
+                return
+
+            # chat_stream websocket schema and /chat/stream completion endpoint
+            delta = getattr(choices[0], 'delta', None)
+            if delta:
+                outputs["output_preview"] = truncate_string(getattr(delta, 'content', None))
+                return
+
+        # generate & generate_stream websocket schema, and /generate/stream completion endpoint
+        value = getattr(output_data[0], 'value', None)
+        if value:
+            outputs["output_preview"] = truncate_string(str(value))
+
     def _finish_weave_call(self, step: IntermediateStep) -> None:
         """
         Finish a previously created Weave call.
@@ -196,6 +269,7 @@ class WeaveExporter(SpanExporter[Span, Span]):
             try:
                 # Add the output to the Weave call
                 outputs["output"] = step.payload.data.output
+                self._extract_output_message(step.payload.data.output, outputs)
             except Exception:
                 # If serialization fails, use string representation
                 outputs["output"] = str(step.payload.data.output)

{nvidia_nat_weave-1.4.0a20251015.dist-info → nvidia_nat_weave-1.4.0a20251022.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-weave
-Version: 1.4.0a20251015
+Version: 1.4.0a20251022
 Summary: Subpackage for Weave 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.0a20251015
+Requires-Dist: nvidia-nat==v1.4.0a20251022
 Requires-Dist: presidio-analyzer~=2.2
 Requires-Dist: presidio-anonymizer~=2.2
 Requires-Dist: weave==0.52.6

nvidia_nat_weave-1.4.0a20251022.dist-info/RECORD

@@ -0,0 +1,11 @@
+nat/meta/pypi.md,sha256=FVQR5lfZjqZHm4VWMmQJYgZbwJJjrfQZgEqHscDuMR8,1121
+nat/plugins/weave/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+nat/plugins/weave/register.py,sha256=EWd3c2eLuNeXQk8CWM5sbP9SUtj59hvhLZaEAQiqbRg,3252
+nat/plugins/weave/weave_exporter.py,sha256=zgR4-_Mv8PKAH7WvWwbL51Q89deH_2oiqkSrXd82KHg,12259
+nvidia_nat_weave-1.4.0a20251022.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_weave-1.4.0a20251022.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_weave-1.4.0a20251022.dist-info/METADATA,sha256=z2z8mUvT_lFV4C06cYjpLMUxPbxSMQt518C0tUCLnz0,2014
+nvidia_nat_weave-1.4.0a20251022.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_weave-1.4.0a20251022.dist-info/entry_points.txt,sha256=xg4vW3wKsOLfHa-QR6JqWnq3DmdfI_z9IZfg4I9thTY,56
+nvidia_nat_weave-1.4.0a20251022.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_weave-1.4.0a20251022.dist-info/RECORD,,

nvidia_nat_weave-1.4.0a20251015.dist-info/RECORD

@@ -1,11 +0,0 @@
-nat/meta/pypi.md,sha256=FVQR5lfZjqZHm4VWMmQJYgZbwJJjrfQZgEqHscDuMR8,1121
-nat/plugins/weave/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-nat/plugins/weave/register.py,sha256=NvK8roew0XjgrhVVw7sDKqZcjpy-LvNIv3UYEFCElB0,2927
-nat/plugins/weave/weave_exporter.py,sha256=5JygAVkNKDl_uj-YRTIrN8w4Si_H1vOVBh7OCRuCzf4,8798
-nvidia_nat_weave-1.4.0a20251015.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
-nvidia_nat_weave-1.4.0a20251015.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-nvidia_nat_weave-1.4.0a20251015.dist-info/METADATA,sha256=w1mYyaZjE9R9Qowthv4-pqyGuq_2hDc4Sz9myfN_tXo,2014
-nvidia_nat_weave-1.4.0a20251015.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_weave-1.4.0a20251015.dist-info/entry_points.txt,sha256=xg4vW3wKsOLfHa-QR6JqWnq3DmdfI_z9IZfg4I9thTY,56
-nvidia_nat_weave-1.4.0a20251015.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
-nvidia_nat_weave-1.4.0a20251015.dist-info/RECORD,,