nvidia-nat-opentelemetry 1.2.0rc5__py3-none-any.whl → 1.2.0rc6__py3-none-any.whl

{aiq → nat}/meta/pypi.md

@@ -15,7 +15,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image"
+![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image"
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for OpenTelemetry integration for observability.

{aiq → nat}/plugins/opentelemetry/mixin/otlp_span_exporter_mixin.py

@@ -17,7 +17,7 @@ import logging
 
 from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
 
-from aiq.plugins.opentelemetry.otel_span import OtelSpan
+from nat.plugins.opentelemetry.otel_span import OtelSpan
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/plugins/opentelemetry/otel_span.py

@@ -106,7 +106,7 @@ class OtelSpan(Span):  # pylint: disable=too-many-public-methods
         self._status = status or Status(StatusCode.UNSET)
         self._ended = False
         self._resource = resource or Resource.create()
-        self._instrumentation_scope = instrumentation_scope or InstrumentationScope("aiq", "1.0.0")
+        self._instrumentation_scope = instrumentation_scope or InstrumentationScope("nat", "1.0.0")
         self._dropped_attributes = 0
         self._dropped_events = 0
         self._dropped_links = 0

{aiq → nat}/plugins/opentelemetry/otel_span_exporter.py

@@ -20,13 +20,13 @@ from importlib.metadata import version
 
 from opentelemetry.sdk.resources import Resource
 
-from aiq.builder.context import AIQContextState
-from aiq.data_models.span import Span
-from aiq.observability.exporter.span_exporter import SpanExporter
-from aiq.observability.processor.batching_processor import BatchingProcessor
-from aiq.observability.processor.processor import Processor
-from aiq.plugins.opentelemetry.otel_span import OtelSpan
-from aiq.plugins.opentelemetry.span_converter import convert_span_to_otel
+from nat.builder.context import ContextState
+from nat.data_models.span import Span
+from nat.observability.exporter.span_exporter import SpanExporter
+from nat.observability.processor.batching_processor import BatchingProcessor
+from nat.observability.processor.processor import Processor
+from nat.plugins.opentelemetry.otel_span import OtelSpan
+from nat.plugins.opentelemetry.span_converter import convert_span_to_otel
 
 logger = logging.getLogger(__name__)
 
@@ -89,7 +89,7 @@ class OtelSpanExporter(SpanExporter[Span, OtelSpan]):  # pylint: disable=R0901
     """
 
     def __init__(self,
-                 context_state: AIQContextState | None = None,
+                 context_state: ContextState | None = None,
                  batch_size: int = 100,
                  flush_interval: float = 5.0,
                  max_queue_size: int = 1000,

{aiq → nat}/plugins/opentelemetry/otlp_span_adapter_exporter.py

@@ -15,9 +15,9 @@
 
 import logging
 
-from aiq.builder.context import AIQContextState
-from aiq.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPSpanExporterMixin
-from aiq.plugins.opentelemetry.otel_span_exporter import OtelSpanExporter
+from nat.builder.context import ContextState
+from nat.plugins.opentelemetry.mixin.otlp_span_exporter_mixin import OTLPSpanExporterMixin
+from nat.plugins.opentelemetry.otel_span_exporter import OtelSpanExporter
 
 logger = logging.getLogger(__name__)
 
@@ -56,7 +56,7 @@ class OTLPSpanAdapterExporter(OTLPSpanExporterMixin, OtelSpanExporter):  # pylin
             self,
             *,
             # OtelSpanExporter args
-            context_state: AIQContextState | None = None,
+            context_state: ContextState | None = None,
             batch_size: int = 100,
             flush_interval: float = 5.0,
             max_queue_size: int = 1000,

{aiq → nat}/plugins/opentelemetry/register.py

@@ -18,11 +18,11 @@ import os
 
 from pydantic import Field
 
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_telemetry_exporter
-from aiq.data_models.telemetry_exporter import TelemetryExporterBaseConfig
-from aiq.observability.mixin.batch_config_mixin import BatchConfigMixin
-from aiq.observability.mixin.collector_config_mixin import CollectorConfigMixin
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_telemetry_exporter
+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
 
 logger = logging.getLogger(__name__)
 
@@ -42,7 +42,7 @@ async def langfuse_telemetry_exporter(config: LangfuseTelemetryExporter, builder
 
     import base64
 
-    from aiq.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
+    from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
     secret_key = config.secret_key or os.environ.get("LANGFUSE_SECRET_KEY")
     public_key = config.public_key or os.environ.get("LANGFUSE_PUBLIC_KEY")
@@ -78,7 +78,7 @@ class LangsmithTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, Telemet
 async def langsmith_telemetry_exporter(config: LangsmithTelemetryExporter, builder: Builder):  # pylint: disable=W0613
     """Create a Langsmith telemetry exporter."""
 
-    from aiq.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
+    from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
     api_key = config.api_key or os.environ.get("LANGSMITH_API_KEY")
     if not api_key:
@@ -108,8 +108,8 @@ class OtelCollectorTelemetryExporter(BatchConfigMixin,
 async def otel_telemetry_exporter(config: OtelCollectorTelemetryExporter, builder: Builder):  # pylint: disable=W0613
     """Create an OpenTelemetry telemetry exporter."""
 
-    from aiq.plugins.opentelemetry.otel_span_exporter import get_opentelemetry_sdk_version
-    from aiq.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
+    from nat.plugins.opentelemetry.otel_span_exporter import get_opentelemetry_sdk_version
+    from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
     # Default resource attributes
     default_resource_attributes = {
@@ -143,7 +143,7 @@ class PatronusTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, Telemetr
 async def patronus_telemetry_exporter(config: PatronusTelemetryExporter, builder: Builder):  # pylint: disable=W0613
     """Create a Patronus telemetry exporter."""
 
-    from aiq.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
+    from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
     api_key = config.api_key or os.environ.get("PATRONUS_API_KEY")
     if not api_key:
@@ -176,7 +176,7 @@ class GalileoTelemetryExporter(BatchConfigMixin, CollectorConfigMixin, Telemetry
 async def galileo_telemetry_exporter(config: GalileoTelemetryExporter, builder: Builder):  # pylint: disable=W0613
     """Create a Galileo telemetry exporter."""
 
-    from aiq.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
+    from nat.plugins.opentelemetry.otlp_span_adapter_exporter import OTLPSpanAdapterExporter
 
     headers = {
         "Galileo-API-Key": config.api_key,

{aiq → nat}/plugins/opentelemetry/span_converter.py

@@ -24,9 +24,9 @@ from opentelemetry.trace import Status
 from opentelemetry.trace import StatusCode
 from opentelemetry.trace import TraceFlags
 
-from aiq.data_models.span import Span
-from aiq.data_models.span import SpanStatusCode
-from aiq.plugins.opentelemetry.otel_span import OtelSpan
+from nat.data_models.span import Span
+from nat.data_models.span import SpanStatusCode
+from nat.plugins.opentelemetry.otel_span import OtelSpan
 
 logger = logging.getLogger(__name__)
 
@@ -83,7 +83,7 @@ def _get_shared_instrumentation_scope():
     """Get shared instrumentation scope to avoid repeated creation."""
     if _shared.instrumentation_scope is None:
         from opentelemetry.sdk.trace import InstrumentationScope
-        _shared.instrumentation_scope = InstrumentationScope("aiq", "1.0.0")  # type: ignore
+        _shared.instrumentation_scope = InstrumentationScope("nat", "1.0.0")  # type: ignore
     return _shared.instrumentation_scope
 
 
@@ -99,11 +99,11 @@ def convert_event_type_to_span_kind(event_type: str) -> OpenInferenceSpanKindVal
     return SPAN_EVENT_TYPE_TO_SPAN_KIND_MAP.get(event_type, OpenInferenceSpanKindValues.UNKNOWN)
 
 
-def convert_span_status_code(aiq_status_code: SpanStatusCode) -> StatusCode:
-    """Convert AIQ SpanStatusCode to OpenTelemetry StatusCode.
+def convert_span_status_code(nat_status_code: SpanStatusCode) -> StatusCode:
+    """Convert NAT SpanStatusCode to OpenTelemetry StatusCode.
 
     Args:
-        aiq_status_code (SpanStatusCode): The AIQ span status code to convert
+        nat_status_code (SpanStatusCode): The NAT span status code to convert
 
     Returns:
         StatusCode: The corresponding OpenTelemetry StatusCode
@@ -113,31 +113,31 @@ def convert_span_status_code(aiq_status_code: SpanStatusCode) -> StatusCode:
         SpanStatusCode.ERROR: StatusCode.ERROR,
         SpanStatusCode.UNSET: StatusCode.UNSET,
     }
-    return status_map.get(aiq_status_code, StatusCode.UNSET)
+    return status_map.get(nat_status_code, StatusCode.UNSET)
 
 
-def convert_span_to_otel(aiq_span: Span) -> OtelSpan:
-    """Convert an AIQ Span to an OtelSpan using ultra-fast conversion.
+def convert_span_to_otel(nat_span: Span) -> OtelSpan:
+    """Convert a NAT Span to an OtelSpan using ultra-fast conversion.
 
     Args:
-        aiq_span (Span): The AIQ span to convert
+        nat_span (Span): The NAT span to convert
 
     Returns:
         OtelSpan: The converted OtelSpan with proper parent hierarchy.
     """
     # Fast path for spans without context
-    if not aiq_span.context:
+    if not nat_span.context:
         # Create minimal OtelSpan bypassing expensive constructor
         otel_span = object.__new__(OtelSpan)  # Bypass __init__
-        otel_span._name = aiq_span.name
+        otel_span._name = nat_span.name
         otel_span._context = None  # type: ignore
         otel_span._parent = None
-        otel_span._attributes = aiq_span.attributes.copy()
+        otel_span._attributes = nat_span.attributes.copy()
         otel_span._events = []
         otel_span._links = []
         otel_span._kind = SpanKind.INTERNAL
-        otel_span._start_time = aiq_span.start_time
-        otel_span._end_time = aiq_span.end_time
+        otel_span._start_time = nat_span.start_time
+        otel_span._end_time = nat_span.end_time
         otel_span._status = Status(StatusCode.UNSET)
         otel_span._ended = False
         otel_span._resource = _get_shared_resource()  # type: ignore
@@ -150,36 +150,36 @@ def convert_span_to_otel(aiq_span: Span) -> OtelSpan:
 
     # Process parent efficiently (if needed)
     parent_otel_span = None
-    trace_id = aiq_span.context.trace_id
+    trace_id = nat_span.context.trace_id
 
-    if aiq_span.parent:
-        parent_otel_span = convert_span_to_otel(aiq_span.parent)
+    if nat_span.parent:
+        parent_otel_span = convert_span_to_otel(nat_span.parent)
         parent_context = parent_otel_span.get_span_context()
         trace_id = parent_context.trace_id
 
     # Create SpanContext efficiently
     otel_span_context = SpanContext(
         trace_id=trace_id,
-        span_id=aiq_span.context.span_id,
+        span_id=nat_span.context.span_id,
         is_remote=False,
         trace_flags=_SAMPLED_TRACE_FLAGS,  # Reuse flags object
     )
 
     # Create OtelSpan bypassing expensive constructor
     otel_span = object.__new__(OtelSpan)  # Bypass __init__
-    otel_span._name = aiq_span.name
+    otel_span._name = nat_span.name
     otel_span._context = otel_span_context
     otel_span._parent = parent_otel_span
-    otel_span._attributes = aiq_span.attributes.copy()
+    otel_span._attributes = nat_span.attributes.copy()
     otel_span._events = []
     otel_span._links = []
     otel_span._kind = SpanKind.INTERNAL
-    otel_span._start_time = aiq_span.start_time
-    otel_span._end_time = aiq_span.end_time
+    otel_span._start_time = nat_span.start_time
+    otel_span._end_time = nat_span.end_time
 
     # Reuse status conversion
-    status_code = convert_span_status_code(aiq_span.status.code)
-    otel_span._status = Status(status_code, aiq_span.status.message)
+    status_code = convert_span_status_code(nat_span.status.code)
+    otel_span._status = Status(status_code, nat_span.status.message)
 
     otel_span._ended = False
     otel_span._resource = _get_shared_resource()  # type: ignore
@@ -190,37 +190,37 @@ def convert_span_to_otel(aiq_span: Span) -> OtelSpan:
     otel_span._status_description = None
 
     # Set span kind efficiently (direct attribute modification)
-    event_type = aiq_span.attributes.get("aiq.event_type", "UNKNOWN")
+    event_type = nat_span.attributes.get("nat.event_type", "UNKNOWN")
     span_kind = SPAN_EVENT_TYPE_TO_SPAN_KIND_MAP.get(event_type, OpenInferenceSpanKindValues.UNKNOWN)
     otel_span._attributes[SpanAttributes.OPENINFERENCE_SPAN_KIND] = span_kind.value
 
     # Process events (only if they exist)
-    if aiq_span.events:
-        for aiq_event in aiq_span.events:
+    if nat_span.events:
+        for nat_event in nat_span.events:
             # Optimize timestamp handling
-            if isinstance(aiq_event.timestamp, int):
-                event_timestamp_ns = aiq_event.timestamp
-            elif aiq_event.timestamp:
-                event_timestamp_ns = int(aiq_event.timestamp)
+            if isinstance(nat_event.timestamp, int):
+                event_timestamp_ns = nat_event.timestamp
+            elif nat_event.timestamp:
+                event_timestamp_ns = int(nat_event.timestamp)
             else:
                 event_timestamp_ns = int(time.time() * 1e9)
 
             # Add event directly to internal list (bypass add_event method)
             otel_span._events.append({
-                "name": aiq_event.name, "attributes": aiq_event.attributes, "timestamp": event_timestamp_ns
+                "name": nat_event.name, "attributes": nat_event.attributes, "timestamp": event_timestamp_ns
             })
 
     return otel_span
 
 
 def convert_spans_to_otel_batch(spans: list[Span]) -> list[OtelSpan]:
-    """Convert a list of AIQ spans to OtelSpans using stateless conversion.
+    """Convert a list of NAT spans to OtelSpans using stateless conversion.
 
     This is useful for batch processing or demos. Each span is converted
     independently using the stateless approach.
 
     Args:
-        spans (list[Span]): List of AIQ spans to convert
+        spans (list[Span]): List of NAT spans to convert
 
     Returns:
         list[OtelSpan]: List of converted OtelSpans with proper parent-child relationships

{nvidia_nat_opentelemetry-1.2.0rc5.dist-info → nvidia_nat_opentelemetry-1.2.0rc6.dist-info}/METADATA

@@ -1,12 +1,12 @@
 Metadata-Version: 2.4
 Name: nvidia-nat-opentelemetry
-Version: 1.2.0rc5
-Summary: Subpackage for OpenTelemetry integration in AIQtoolkit
+Version: 1.2.0rc6
+Summary: Subpackage for OpenTelemetry integration in NeMo Agent toolkit
 Keywords: ai,observability,opentelemetry
 Classifier: Programming Language :: Python
 Requires-Python: <3.13,>=3.11
 Description-Content-Type: text/markdown
-Requires-Dist: nvidia-nat~=1.2
+Requires-Dist: nvidia-nat==v1.2.0-rc6
 Requires-Dist: opentelemetry-api~=1.2
 Requires-Dist: opentelemetry-exporter-otlp~=1.3
 Requires-Dist: opentelemetry-sdk~=1.3
@@ -28,7 +28,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "NeMo Agent toolkit banner image"
+![NVIDIA NeMo Agent Toolkit](https://media.githubusercontent.com/media/NVIDIA/NeMo-Agent-Toolkit/refs/heads/main/docs/source/_static/banner.png "NeMo Agent toolkit banner image"
 
 # NVIDIA NeMo Agent Toolkit Subpackage
 This is a subpackage for OpenTelemetry integration for observability.

nvidia_nat_opentelemetry-1.2.0rc6.dist-info/RECORD

@@ -0,0 +1,14 @@
+nat/meta/pypi.md,sha256=_o1o1BLPY1pvjCkklWxlm7LlIDMPCk-2Rho85NUuN8U,1109
+nat/plugins/opentelemetry/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
+nat/plugins/opentelemetry/otel_span.py,sha256=dn-wI4iYS02z8WZWpamX4ISxhhmjRKWgPk9mAqIASZg,16554
+nat/plugins/opentelemetry/otel_span_exporter.py,sha256=zxlbj0kODcqPeI5zVs_XQR7HsvlOMFhw6iEWJOOLWVM,6559
+nat/plugins/opentelemetry/otlp_span_adapter_exporter.py,sha256=MtkmpLklvyIhCYXHvkY_sZRffwTiAvbQxBen02a89mo,3970
+nat/plugins/opentelemetry/register.py,sha256=ZwOL4aXsnX0bokfuRnYQLrCk27BrUGruTIy4um6lXW8,9354
+nat/plugins/opentelemetry/span_converter.py,sha256=lgdqXI0yn6OZ8xxo9wOwTmicUTJ_RA9f-uMnCAl2HIk,8846
+nat/plugins/opentelemetry/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
+nat/plugins/opentelemetry/mixin/otlp_span_exporter_mixin.py,sha256=va5TuEMz8raO6NeDCytYm5tL9uyMih3gYiP_H7KmNdU,2808
+nvidia_nat_opentelemetry-1.2.0rc6.dist-info/METADATA,sha256=9sW8949ypgqQRsClIm6YlbGLKgrcUiqKg2spnKxI9K0,1569
+nvidia_nat_opentelemetry-1.2.0rc6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nvidia_nat_opentelemetry-1.2.0rc6.dist-info/entry_points.txt,sha256=gmEKhCafyibUJLGxbn8luTK0UTgIvV2vAtr4uZ8M85I,72
+nvidia_nat_opentelemetry-1.2.0rc6.dist-info/top_level.txt,sha256=8-CJ2cP6-f0ZReXe5Hzqp-5pvzzHz-5Ds5H2bGqh1-U,4
+nvidia_nat_opentelemetry-1.2.0rc6.dist-info/RECORD,,

nvidia_nat_opentelemetry-1.2.0rc6.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[nat.components]
+nat_opentelemetry = nat.plugins.opentelemetry.register

nvidia_nat_opentelemetry-1.2.0rc6.dist-info/top_level.txt

@@ -0,0 +1 @@
+nat

nvidia_nat_opentelemetry-1.2.0rc5.dist-info/RECORD

@@ -1,14 +0,0 @@
-aiq/meta/pypi.md,sha256=lNE7pC7ozs7bGnoe6mNGsI8s-iu1lO2h4LJgNZpmcJQ,1120
-aiq/plugins/opentelemetry/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
-aiq/plugins/opentelemetry/otel_span.py,sha256=cQgKY12PNw6oo5OizbfZ22FPBbnHcrhq80_zExleR-w,16554
-aiq/plugins/opentelemetry/otel_span_exporter.py,sha256=sALT_XJt5jvmnchctONVZ0PkWv4Bt0Rnf8bO_FRyCQM,6565
-aiq/plugins/opentelemetry/otlp_span_adapter_exporter.py,sha256=4mZOd_uFutgRWV1sFDApRAGz29_wT7gC4dyScqSHtRY,3976
-aiq/plugins/opentelemetry/register.py,sha256=o24qdbNsgZ0j6kbm1Vpc3jN88HpScgwYh7K9BAL1JdU,9354
-aiq/plugins/opentelemetry/span_converter.py,sha256=XDGTeDVlHH-msCffMmFJ1vraR8HYeqph8-BW6L_9bpg,8847
-aiq/plugins/opentelemetry/mixin/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
-aiq/plugins/opentelemetry/mixin/otlp_span_exporter_mixin.py,sha256=HLPZOrE7okS0r0IXL0DKZUFnGwGZ747eqeLbwTEq3S0,2808
-nvidia_nat_opentelemetry-1.2.0rc5.dist-info/METADATA,sha256=DP71f0wvs-v9vxrq_WB6Wbb8UYveTC1V9PvFAOTbn8s,1565
-nvidia_nat_opentelemetry-1.2.0rc5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nvidia_nat_opentelemetry-1.2.0rc5.dist-info/entry_points.txt,sha256=4edEFrR3E_Ot_nXAGVh1GuSIA3NCh38nLPbmFIrShQ0,72
-nvidia_nat_opentelemetry-1.2.0rc5.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-nvidia_nat_opentelemetry-1.2.0rc5.dist-info/RECORD,,

nvidia_nat_opentelemetry-1.2.0rc5.dist-info/entry_points.txt

@@ -1,2 +0,0 @@
-[aiq.components]
-aiq_opentelemetry = aiq.plugins.opentelemetry.register

nvidia_nat_opentelemetry-1.2.0rc5.dist-info/top_level.txt

@@ -1 +0,0 @@
-aiq