nvidia-nat-weave 1.2.1__py3-none-any.whl → 1.3.0__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,
@@ -40,9 +44,8 @@ class WeaveTelemetryExporter(TelemetryExporterBaseConfig, name="weave"):
 
 
 @register_telemetry_exporter(config_type=WeaveTelemetryExporter)
-async def weave_telemetry_exporter(config: WeaveTelemetryExporter, builder: Builder):  # pylint: disable=unused-argument
+async def weave_telemetry_exporter(config: WeaveTelemetryExporter, builder: Builder):
     import weave
-
     from nat.plugins.weave.weave_exporter import WeaveExporter
 
     weave_settings = {}
@@ -63,14 +66,9 @@ async def weave_telemetry_exporter(config: WeaveTelemetryExporter, builder: Buil
 
     # Handle custom redact keys if specified
     if config.redact_keys and config.redact_pii:
-        # Need to create a new list combining default keys and custom ones
-        from weave.trace import sanitize
-        default_keys = sanitize.REDACT_KEYS
-
-        # Create a new list with all keys
-        all_keys = list(default_keys) + config.redact_keys
+        from weave.utils import sanitize
 
-        # Replace the default REDACT_KEYS with our extended list
-        sanitize.REDACT_KEYS = tuple(all_keys)
+        for key in config.redact_keys:
+            sanitize.add_redact_key(key)
 
     yield WeaveExporter(project=config.project, entity=config.entity, verbose=config.verbose)

nat/plugins/weave/weave_exporter.py

@@ -17,17 +17,16 @@ import logging
 from collections.abc import Generator
 from contextlib import contextmanager
 
-from weave.trace.context import weave_client_context
-from weave.trace.context.call_context import get_current_call
-from weave.trace.context.call_context import set_call_stack
-from weave.trace.weave_client import Call
-
 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.type_utils import override
+from weave.trace.context import weave_client_context
+from weave.trace.context.call_context import get_current_call
+from weave.trace.context.call_context import set_call_stack
+from weave.trace.weave_client import Call
 
 logger = logging.getLogger(__name__)
 
@@ -125,7 +124,7 @@ class WeaveExporter(SpanExporter[Span, Span]):
         # Extract parent call if applicable
         parent_call = None
 
-        # If we have an existing Weave call from another framework (e.g., LangChain),
+        # If we have an existing Weave call from another framework (e.g., LangChain/LangGraph),
         # use it as the parent
         if existing_call is not None:
             parent_call = existing_call

{nvidia_nat_weave-1.2.1.dist-info → nvidia_nat_weave-1.3.0.dist-info}/METADATA

@@ -1,15 +1,26 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-weave
-Version: 1.2.1
+Version: 1.3.0
 Summary: Subpackage for Weave 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,wandb,pii
 Classifier: Programming Language :: Python
-Requires-Python: <3.13,>=3.11
+Classifier: Programming Language :: Python :: 3.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==v1.2.1
+License-File: LICENSE-3rd-party.txt
+License-File: LICENSE.md
+Requires-Dist: nvidia-nat==v1.3.0
 Requires-Dist: presidio-analyzer~=2.2
 Requires-Dist: presidio-anonymizer~=2.2
-Requires-Dist: weave~=0.51
+Requires-Dist: weave==0.52.6
+Dynamic: license-file
 
 <!--
 SPDX-FileCopyrightText: Copyright (c) 2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.

nvidia_nat_weave-1.3.0.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=5JygAVkNKDl_uj-YRTIrN8w4Si_H1vOVBh7OCRuCzf4,8798
+nvidia_nat_weave-1.3.0.dist-info/licenses/LICENSE-3rd-party.txt,sha256=fOk5jMmCX9YoKWyYzTtfgl-SUy477audFC5hNY4oP7Q,284609
+nvidia_nat_weave-1.3.0.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+nvidia_nat_weave-1.3.0.dist-info/METADATA,sha256=abd20Y15YdkkHLUv3lOP9UkTmOlCz6QpuH0jDFDRzks,1996
+nvidia_nat_weave-1.3.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_weave-1.3.0.dist-info/entry_points.txt,sha256=xg4vW3wKsOLfHa-QR6JqWnq3DmdfI_z9IZfg4I9thTY,56
+nvidia_nat_weave-1.3.0.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_weave-1.3.0.dist-info/RECORD,,