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

nat/plugins/weave/register.py

@@ -14,6 +14,7 @@
 # limitations under the License.
 
 import logging
+import typing
 
 from pydantic import Field
 
@@ -41,6 +42,8 @@ class WeaveTelemetryExporter(TelemetryExporterBaseConfig, name="weave"):
         default=None,
         description="Additional keys to redact from traces beyond the default (api_key, auth_headers, authorization).")
     verbose: bool = Field(default=False, description="Whether to enable verbose logging.")
+    attributes: dict[str, typing.Any] | None = Field(default=None,
+                                                     description="Custom attributes to include in the traces.")
 
 
 @register_telemetry_exporter(config_type=WeaveTelemetryExporter)
@@ -71,4 +74,7 @@ async def weave_telemetry_exporter(config: WeaveTelemetryExporter, builder: Buil
         for key in config.redact_keys:
             sanitize.add_redact_key(key)
 
-    yield WeaveExporter(project=config.project, entity=config.entity, verbose=config.verbose)
+    yield WeaveExporter(project=config.project,
+                        entity=config.entity,
+                        verbose=config.verbose,
+                        attributes=config.attributes)

nat/plugins/weave/weave_exporter.py

@@ -51,10 +51,12 @@ class WeaveExporter(SpanExporter[Span, Span]):
                  context_state=None,
                  entity: str | None = None,
                  project: str | None = None,
-                 verbose: bool = False):
+                 verbose: bool = False,
+                 attributes: dict[str, Any] | None = None):
         super().__init__(context_state=context_state)
         self._entity = entity
         self._project = project
+        self._attributes = attributes or {}
         self._gc = weave_client_context.require_weave_client()
 
         # Optionally, set log filtering for presidio-analyzer to reduce verbosity
@@ -160,11 +162,14 @@ class WeaveExporter(SpanExporter[Span, Span]):
                 inputs["input"] = str(step.payload.data.input)
 
         # Create the Weave call
+        attributes = span.attributes.copy()
+        attributes.update(self._attributes)
+
         call = self._gc.create_call(
             op_name,
             inputs=inputs,
             parent=parent_call,
-            attributes=span.attributes,
+            attributes=attributes,
             display_name=op_name,
         )
 

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

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

nvidia_nat_weave-1.4.0a20251024.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=JGhp7u_LheQH-zBFw3EfJCeUo7TY5WC-rZQuvyEyifs,3547
+nat/plugins/weave/weave_exporter.py,sha256=FkY-xkOyOIRrr3ls7uEMxCgnr3e6pVbaqqrfejffWbA,12446
+nvidia_nat_weave-1.4.0a20251024.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_weave-1.4.0a20251024.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_weave-1.4.0a20251024.dist-info/METADATA,sha256=AlEf0IgvoR20l61SLGyf-WLvAYJxl0IRNcM9RUxTBJ8,2014
+nvidia_nat_weave-1.4.0a20251024.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_weave-1.4.0a20251024.dist-info/entry_points.txt,sha256=xg4vW3wKsOLfHa-QR6JqWnq3DmdfI_z9IZfg4I9thTY,56
+nvidia_nat_weave-1.4.0a20251024.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_weave-1.4.0a20251024.dist-info/RECORD,,

nvidia_nat_weave-1.4.0a20251022.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=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,,