nvidia-nat 1.2.0rc5__py3-none-any.whl → 1.2.0rc7__py3-none-any.whl

nat/meta/pypi.md

@@ -0,0 +1,58 @@
+<!--
+SPDX-FileCopyrightText: Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+SPDX-License-Identifier: Apache-2.0
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+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/banner.png "NeMo Agent toolkit banner image")
+
+# NVIDIA NeMo Agent Toolkit
+
+NeMo Agent toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, NeMo Agent toolkit enables true composability: build once and reuse anywhere.
+
+## Key Features
+
+- [**Framework Agnostic:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/extend/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
+- [**Reusability:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/extend/sharing-components.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
+- [**Rapid Development:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/tutorials/index.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
+- [**Profiling:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
+- [**Observability:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool, with examples using [Phoenix](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-phoenix.html) and [W&B Weave](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/observe/observe-workflow-with-weave.html).
+- [**Evaluation System:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
+- [**User Interface:**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/quick-start/launching-ui.html) Use the NeMo Agent toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
+- [**MCP Compatibility**](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/workflows/mcp/mcp-client.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as NeMo Agent toolkit functions.
+
+With NeMo Agent toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
+
+## Links
+ * [Documentation](https://docs.nvidia.com/nemo/agent-toolkit/1.2.0/index.html): Explore the full documentation for NeMo Agent toolkit.
+
+## First time user?
+ If this is your first time using NeMo Agent toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/NeMo-Agent-Toolkit?tab=readme-ov-file#quick-start) on GitHub. This package is intended for users who are familiar with NeMo Agent toolkit applications and need to add NeMo Agent toolkit as a dependency to their project.
+
+## Feedback
+
+We would love to hear from you! Please file an issue on [GitHub](https://github.com/NVIDIA/NeMo-Agent-Toolkit/issues) if you have any feedback or feature requests.
+
+## Acknowledgements
+
+We would like to thank the following open source projects that made NeMo Agent toolkit possible:
+
+- [CrewAI](https://github.com/crewAIInc/crewAI)
+- [FastAPI](https://github.com/tiangolo/fastapi)
+- [LangChain](https://github.com/langchain-ai/langchain)
+- [Llama-Index](https://github.com/run-llama/llama_index)
+- [Mem0ai](https://github.com/mem0ai/mem0)
+- [Ragas](https://github.com/explodinggradients/ragas)
+- [Semantic Kernel](https://github.com/microsoft/semantic-kernel)
+- [uv](https://github.com/astral-sh/uv)

{aiq → nat}/object_store/__init__.py

@@ -13,8 +13,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """
-AIQ Toolkit Object Store Module
+NAT Object Store Module
 
 This package provides foundational classes and interfaces
-for managing object storage in AIQ Toolkit's LLM-based agents.
+for managing object storage in NAT's LLM-based agents.
 """

{aiq → nat}/object_store/in_memory_object_store.py

@@ -15,12 +15,12 @@
 
 import asyncio
 
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_object_store
-from aiq.data_models.object_store import KeyAlreadyExistsError
-from aiq.data_models.object_store import NoSuchKeyError
-from aiq.data_models.object_store import ObjectStoreBaseConfig
-from aiq.utils.type_utils import override
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_object_store
+from nat.data_models.object_store import KeyAlreadyExistsError
+from nat.data_models.object_store import NoSuchKeyError
+from nat.data_models.object_store import ObjectStoreBaseConfig
+from nat.utils.type_utils import override
 
 from .interfaces import ObjectStore
 from .models import ObjectStoreItem

{aiq → nat}/observability/exporter/base_exporter.py

@@ -26,11 +26,11 @@ from typing import Generic
 from typing import TypeVar
 from typing import overload
 
-from aiq.builder.context import AIQContextState
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.observability.exporter.exporter import Exporter
-from aiq.utils.reactive.subject import Subject
-from aiq.utils.type_utils import override
+from nat.builder.context import ContextState
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.observability.exporter.exporter import Exporter
+from nat.utils.reactive.subject import Subject
+from nat.utils.type_utils import override
 
 logger = logging.getLogger(__name__)
 
@@ -117,7 +117,7 @@ class BaseExporter(Exporter):
         - This keeps observability "off the hot path" for optimal performance
 
     Args:
-        context_state (AIQContextState, optional): The context state to use for the exporter. Defaults to None.
+        context_state (ContextState, optional): The context state to use for the exporter. Defaults to None.
     """
 
     # Class-level tracking for debugging and monitoring
@@ -130,10 +130,10 @@ class BaseExporter(Exporter):
     _ready_event: IsolatedAttribute[asyncio.Event] = IsolatedAttribute(asyncio.Event)
     _shutdown_event: IsolatedAttribute[asyncio.Event] = IsolatedAttribute(asyncio.Event)
 
-    def __init__(self, context_state: AIQContextState | None = None):
+    def __init__(self, context_state: ContextState | None = None):
         """Initialize the BaseExporter."""
         if context_state is None:
-            context_state = AIQContextState.get()
+            context_state = ContextState.get()
 
         self._context_state = context_state
         self._subscription = None
@@ -413,7 +413,7 @@ class BaseExporter(Exporter):
         """
         await self._ready_event.wait()
 
-    def create_isolated_instance(self, context_state: AIQContextState) -> "BaseExporter":
+    def create_isolated_instance(self, context_state: ContextState) -> "BaseExporter":
         """Create an isolated copy with automatic descriptor-based state reset.
 
         This method creates a shallow copy that shares expensive resources

{aiq → nat}/observability/exporter/exporter.py

@@ -18,7 +18,7 @@ from abc import ABC
 from abc import abstractmethod
 from collections.abc import AsyncGenerator
 
-from aiq.data_models.intermediate_step import IntermediateStep
+from nat.data_models.intermediate_step import IntermediateStep
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/observability/exporter/file_exporter.py

@@ -15,11 +15,11 @@
 
 import logging
 
-from aiq.builder.context import AIQContextState
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.observability.exporter.raw_exporter import RawExporter
-from aiq.observability.mixin.file_mixin import FileExportMixin
-from aiq.observability.processor.intermediate_step_serializer import IntermediateStepSerializer
+from nat.builder.context import ContextState
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.observability.exporter.raw_exporter import RawExporter
+from nat.observability.mixin.file_mixin import FileExportMixin
+from nat.observability.processor.intermediate_step_serializer import IntermediateStepSerializer
 
 logger = logging.getLogger(__name__)
 
@@ -27,7 +27,7 @@ logger = logging.getLogger(__name__)
 class FileExporter(FileExportMixin, RawExporter[IntermediateStep, str]):  # pylint: disable=R0901
     """A File exporter that exports telemetry traces to a local file."""
 
-    def __init__(self, context_state: AIQContextState | None = None, **file_kwargs):
+    def __init__(self, context_state: ContextState | None = None, **file_kwargs):
         super().__init__(context_state=context_state, **file_kwargs)
         self._processor = IntermediateStepSerializer()
         self.add_processor(self._processor)

{aiq → nat}/observability/exporter/processing_exporter.py

@@ -21,14 +21,14 @@ from typing import Any
 from typing import Generic
 from typing import TypeVar
 
-from aiq.builder.context import AIQContextState
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.observability.exporter.base_exporter import BaseExporter
-from aiq.observability.mixin.type_introspection_mixin import TypeIntrospectionMixin
-from aiq.observability.processor.callback_processor import CallbackProcessor
-from aiq.observability.processor.processor import Processor
-from aiq.utils.type_utils import DecomposedType
-from aiq.utils.type_utils import override
+from nat.builder.context import ContextState
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.observability.exporter.base_exporter import BaseExporter
+from nat.observability.mixin.type_introspection_mixin import TypeIntrospectionMixin
+from nat.observability.processor.callback_processor import CallbackProcessor
+from nat.observability.processor.processor import Processor
+from nat.utils.type_utils import DecomposedType
+from nat.utils.type_utils import override
 
 PipelineInputT = TypeVar("PipelineInputT")
 PipelineOutputT = TypeVar("PipelineOutputT")
@@ -53,7 +53,7 @@ class ProcessingExporter(Generic[PipelineInputT, PipelineOutputT], BaseExporter,
     - Automatic type validation before export
     """
 
-    def __init__(self, context_state: AIQContextState | None = None):
+    def __init__(self, context_state: ContextState | None = None):
         """Initialize the processing exporter.
 
         Args:

{aiq → nat}/observability/exporter/raw_exporter.py

@@ -17,9 +17,9 @@ import logging
 from abc import abstractmethod
 from typing import TypeVar
 
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.observability.exporter.processing_exporter import ProcessingExporter
-from aiq.utils.type_utils import override
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.observability.exporter.processing_exporter import ProcessingExporter
+from nat.utils.type_utils import override
 
 logger = logging.getLogger(__name__)
 
@@ -37,7 +37,7 @@ class RawExporter(ProcessingExporter[InputT, OutputT]):
     The flow is: IntermediateStep -> [Processing Pipeline] -> OutputT -> Export
 
     Args:
-        context_state (AIQContextState, optional): The context state to use for the exporter. Defaults to None.
+        context_state (ContextState, optional): The context state to use for the exporter. Defaults to None.
     """
 
     @abstractmethod

{aiq → nat}/observability/exporter/span_exporter.py

@@ -14,24 +14,29 @@
 # limitations under the License.
 
 import logging
+import os
 import re
+import typing
 from abc import abstractmethod
 from typing import TypeVar
 
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.data_models.intermediate_step import IntermediateStepState
-from aiq.data_models.intermediate_step import TraceMetadata
-from aiq.data_models.span import MimeTypes
-from aiq.data_models.span import Span
-from aiq.data_models.span import SpanAttributes
-from aiq.data_models.span import SpanContext
-from aiq.data_models.span import event_type_to_span_kind
-from aiq.observability.exporter.base_exporter import IsolatedAttribute
-from aiq.observability.exporter.processing_exporter import ProcessingExporter
-from aiq.observability.mixin.serialize_mixin import SerializeMixin
-from aiq.observability.utils.dict_utils import merge_dicts
-from aiq.observability.utils.time_utils import ns_timestamp
-from aiq.utils.type_utils import override
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.data_models.intermediate_step import IntermediateStepState
+from nat.data_models.intermediate_step import TraceMetadata
+from nat.data_models.span import MimeTypes
+from nat.data_models.span import Span
+from nat.data_models.span import SpanAttributes
+from nat.data_models.span import SpanContext
+from nat.data_models.span import event_type_to_span_kind
+from nat.observability.exporter.base_exporter import IsolatedAttribute
+from nat.observability.exporter.processing_exporter import ProcessingExporter
+from nat.observability.mixin.serialize_mixin import SerializeMixin
+from nat.observability.utils.dict_utils import merge_dicts
+from nat.observability.utils.time_utils import ns_timestamp
+from nat.utils.type_utils import override
+
+if typing.TYPE_CHECKING:
+    from nat.builder.context import ContextState
 
 logger = logging.getLogger(__name__)
 
@@ -67,8 +72,13 @@ class SpanExporter(ProcessingExporter[InputSpanT, OutputSpanT], SerializeMixin):
     1. IntermediateStep (START) → Create Span → Add to tracking
     2. IntermediateStep (END) → Complete Span → Process through pipeline → Export
 
-    Args:
-        context_state (AIQContextState, optional): The context state to use for the exporter. Defaults to None.
+    Parameters
+    ----------
+        context_state: `ContextState`, optional
+            The context state to use for the exporter. Defaults to None.
+        span_prefix: `str`, optional
+            The prefix name to use for span attributes. If `None` the value of the `NAT_SPAN_PREFIX` environment
+            variable is used. Defaults to `"nat"` if neither are defined.
     """
 
     # Use descriptors for automatic isolation of span-specific state
@@ -76,6 +86,13 @@ class SpanExporter(ProcessingExporter[InputSpanT, OutputSpanT], SerializeMixin):
     _span_stack: IsolatedAttribute[dict] = IsolatedAttribute(dict)
     _metadata_stack: IsolatedAttribute[dict] = IsolatedAttribute(dict)
 
+    def __init__(self, context_state: "ContextState | None" = None, span_prefix: str | None = None):
+        super().__init__(context_state=context_state)
+        if span_prefix is None:
+            span_prefix = os.getenv("NAT_SPAN_PREFIX", "nat").strip() or "nat"
+
+        self._span_prefix = span_prefix
+
     @abstractmethod
     async def export_processed(self, item: OutputSpanT) -> None:
         """Export the processed span.
@@ -137,22 +154,27 @@ class SpanExporter(ProcessingExporter[InputSpanT, OutputSpanT], SerializeMixin):
         else:
             sub_span_name = f"{event.payload.event_type}"
 
-        sub_span = Span(
-            name=sub_span_name,
-            parent=parent_span,
-            context=span_ctx,
-            attributes={
-                "aiq.event_type": event.payload.event_type.value,
-                "aiq.function.id": event.function_ancestry.function_id if event.function_ancestry else "unknown",
-                "aiq.function.name": event.function_ancestry.function_name if event.function_ancestry else "unknown",
-                "aiq.subspan.name": event.payload.name or "",
-                "aiq.event_timestamp": event.event_timestamp,
-                "aiq.framework": event.payload.framework.value if event.payload.framework else "unknown",
-            },
-            start_time=start_ns)
+        sub_span = Span(name=sub_span_name,
+                        parent=parent_span,
+                        context=span_ctx,
+                        attributes={
+                            f"{self._span_prefix}.event_type":
+                                event.payload.event_type.value,
+                            f"{self._span_prefix}.function.id":
+                                event.function_ancestry.function_id if event.function_ancestry else "unknown",
+                            f"{self._span_prefix}.function.name":
+                                event.function_ancestry.function_name if event.function_ancestry else "unknown",
+                            f"{self._span_prefix}.subspan.name":
+                                event.payload.name or "",
+                            f"{self._span_prefix}.event_timestamp":
+                                event.event_timestamp,
+                            f"{self._span_prefix}.framework":
+                                event.payload.framework.value if event.payload.framework else "unknown",
+                        },
+                        start_time=start_ns)
 
         span_kind = event_type_to_span_kind(event.event_type)
-        sub_span.set_attribute("aiq.span.kind", span_kind.value)
+        sub_span.set_attribute(f"{self._span_prefix}.span.kind", span_kind.value)
 
         if event.payload.data and event.payload.data.input:
             match = re.search(r"Human:\s*Question:\s*(.*)", str(event.payload.data.input))
@@ -204,9 +226,9 @@ class SpanExporter(ProcessingExporter[InputSpanT, OutputSpanT], SerializeMixin):
         # Optionally add more attributes from usage_info or data
         usage_info = event.payload.usage_info
         if usage_info:
-            sub_span.set_attribute(SpanAttributes.AIQ_USAGE_NUM_LLM_CALLS.value,
+            sub_span.set_attribute(SpanAttributes.NAT_USAGE_NUM_LLM_CALLS.value,
                                    usage_info.num_llm_calls if usage_info.num_llm_calls else 0)
-            sub_span.set_attribute(SpanAttributes.AIQ_USAGE_SECONDS_BETWEEN_CALLS.value,
+            sub_span.set_attribute(SpanAttributes.NAT_USAGE_SECONDS_BETWEEN_CALLS.value,
                                    usage_info.seconds_between_calls if usage_info.seconds_between_calls else 0)
             sub_span.set_attribute(SpanAttributes.LLM_TOKEN_COUNT_PROMPT.value,
                                    usage_info.token_usage.prompt_tokens if usage_info.token_usage else 0)
@@ -239,8 +261,9 @@ class SpanExporter(ProcessingExporter[InputSpanT, OutputSpanT], SerializeMixin):
 
         merged_metadata = merge_dicts(start_metadata, end_metadata)
         serialized_metadata, is_json = self._serialize_payload(merged_metadata)
-        sub_span.set_attribute("aiq.metadata", serialized_metadata)
-        sub_span.set_attribute("aiq.metadata.mime_type", MimeTypes.JSON.value if is_json else MimeTypes.TEXT.value)
+        sub_span.set_attribute(f"{self._span_prefix}.metadata", serialized_metadata)
+        sub_span.set_attribute(f"{self._span_prefix}.metadata.mime_type",
+                               MimeTypes.JSON.value if is_json else MimeTypes.TEXT.value)
 
         end_ns = ns_timestamp(event.payload.event_timestamp)
 

{aiq → nat}/observability/exporter_manager.py

@@ -17,8 +17,8 @@ import asyncio
 import logging
 from contextlib import asynccontextmanager
 
-from aiq.builder.context import AIQContextState
-from aiq.observability.exporter.base_exporter import BaseExporter
+from nat.builder.context import ContextState
+from nat.observability.exporter.base_exporter import BaseExporter
 
 logger = logging.getLogger(__name__)
 
@@ -133,7 +133,7 @@ class ExporterManager:
         """
         return self._exporter_registry
 
-    def create_isolated_exporters(self, context_state: AIQContextState | None = None) -> dict[str, BaseExporter]:
+    def create_isolated_exporters(self, context_state: ContextState | None = None) -> dict[str, BaseExporter]:
         """
         Create isolated copies of all exporters for concurrent execution.
 
@@ -141,7 +141,7 @@ class ExporterManager:
         expensive resources but have separate mutable state.
 
         Args:
-            context_state (AIQContextState | None, optional): The isolated context state for the new exporter instances.
+            context_state (ContextState | None, optional): The isolated context state for the new exporter instances.
                 If not provided, a new context state will be created.
 
         Returns:
@@ -149,7 +149,7 @@ class ExporterManager:
         """
         # Provide default context state if None
         if context_state is None:
-            context_state = AIQContextState.get()
+            context_state = ContextState.get()
 
         isolated_exporters = {}
         for name, exporter in self._exporter_registry.items():
@@ -198,7 +198,7 @@ class ExporterManager:
             logger.error("Error stopping isolated exporter '%s': %s", name, e)
 
     @asynccontextmanager
-    async def start(self, context_state: AIQContextState | None = None):
+    async def start(self, context_state: ContextState | None = None):
         """
         Start all registered exporters concurrently.
 

{aiq → nat}/observability/mixin/file_mixin.py

@@ -19,8 +19,8 @@ from datetime import datetime
 from pathlib import Path
 from typing import Any
 
-from aiq.observability.mixin.file_mode import FileMode
-from aiq.observability.mixin.resource_conflict_mixin import ResourceConflictMixin
+from nat.observability.mixin.file_mode import FileMode
+from nat.observability.mixin.resource_conflict_mixin import ResourceConflictMixin
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/observability/processor/batching_processor.py

@@ -23,7 +23,7 @@ from typing import Any
 from typing import Generic
 from typing import TypeVar
 
-from aiq.observability.processor.callback_processor import CallbackProcessor
+from nat.observability.processor.callback_processor import CallbackProcessor
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/observability/processor/callback_processor.py

@@ -19,7 +19,7 @@ from collections.abc import Callable
 from typing import Any
 from typing import TypeVar
 
-from aiq.observability.processor.processor import Processor
+from nat.observability.processor.processor import Processor
 
 InputT = TypeVar('InputT')
 OutputT = TypeVar('OutputT')

{aiq → nat}/observability/processor/intermediate_step_serializer.py

@@ -13,10 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.observability.mixin.serialize_mixin import SerializeMixin
-from aiq.observability.processor.processor import Processor
-from aiq.utils.type_utils import override
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.observability.mixin.serialize_mixin import SerializeMixin
+from nat.observability.processor.processor import Processor
+from nat.utils.type_utils import override
 
 
 class IntermediateStepSerializer(SerializeMixin, Processor[IntermediateStep, str]):

{aiq → nat}/observability/processor/processor.py

@@ -18,7 +18,7 @@ from abc import abstractmethod
 from typing import Generic
 from typing import TypeVar
 
-from aiq.observability.mixin.type_introspection_mixin import TypeIntrospectionMixin
+from nat.observability.mixin.type_introspection_mixin import TypeIntrospectionMixin
 
 InputT = TypeVar('InputT')
 OutputT = TypeVar('OutputT')

{aiq → nat}/observability/register.py

@@ -17,12 +17,12 @@ import logging
 
 from pydantic import Field
 
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_logging_method
-from aiq.cli.register_workflow import register_telemetry_exporter
-from aiq.data_models.logging import LoggingBaseConfig
-from aiq.data_models.telemetry_exporter import TelemetryExporterBaseConfig
-from aiq.observability.mixin.file_mode import FileMode
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_logging_method
+from nat.cli.register_workflow import register_telemetry_exporter
+from nat.data_models.logging import LoggingBaseConfig
+from nat.data_models.telemetry_exporter import TelemetryExporterBaseConfig
+from nat.observability.mixin.file_mode import FileMode
 
 logger = logging.getLogger(__name__)
 
@@ -50,7 +50,7 @@ async def file_telemetry_exporter(config: FileTelemetryExporterConfig, builder:
     Build and return a FileExporter for file-based telemetry export with optional rolling.
     """
 
-    from aiq.observability.exporter.file_exporter import FileExporter
+    from nat.observability.exporter.file_exporter import FileExporter
 
     yield FileExporter(output_path=config.output_path,
                        project=config.project,

{aiq → nat}/profiler/calc/calc_runner.py

@@ -22,23 +22,23 @@ from pathlib import Path
 
 from pydantic import ValidationError
 
-from aiq.eval.config import EvaluationRunConfig
-from aiq.eval.runners.config import MultiEvaluationRunConfig
-from aiq.eval.runners.multi_eval_runner import MultiEvaluationRunner
-from aiq.profiler.calc.calculations import LinearFitResult
-from aiq.profiler.calc.calculations import calc_gpu_estimate_based_on_slope
-from aiq.profiler.calc.calculations import calc_gpu_estimate_for_single_concurrency
-from aiq.profiler.calc.calculations import compute_slope
-from aiq.profiler.calc.data_models import CalcAlerts
-from aiq.profiler.calc.data_models import CalcData
-from aiq.profiler.calc.data_models import CalcRunnerConfig
-from aiq.profiler.calc.data_models import CalcRunnerOutput
-from aiq.profiler.calc.data_models import FitConfig
-from aiq.profiler.calc.data_models import FitResults
-from aiq.profiler.calc.data_models import GPUEstimates
-from aiq.profiler.calc.data_models import SizingMetricPerItem
-from aiq.profiler.calc.data_models import SizingMetrics
-from aiq.profiler.calc.data_models import SizingMetricsAlerts
+from nat.eval.config import EvaluationRunConfig
+from nat.eval.runners.config import MultiEvaluationRunConfig
+from nat.eval.runners.multi_eval_runner import MultiEvaluationRunner
+from nat.profiler.calc.calculations import LinearFitResult
+from nat.profiler.calc.calculations import calc_gpu_estimate_based_on_slope
+from nat.profiler.calc.calculations import calc_gpu_estimate_for_single_concurrency
+from nat.profiler.calc.calculations import compute_slope
+from nat.profiler.calc.data_models import CalcAlerts
+from nat.profiler.calc.data_models import CalcData
+from nat.profiler.calc.data_models import CalcRunnerConfig
+from nat.profiler.calc.data_models import CalcRunnerOutput
+from nat.profiler.calc.data_models import FitConfig
+from nat.profiler.calc.data_models import FitResults
+from nat.profiler.calc.data_models import GPUEstimates
+from nat.profiler.calc.data_models import SizingMetricPerItem
+from nat.profiler.calc.data_models import SizingMetrics
+from nat.profiler.calc.data_models import SizingMetricsAlerts
 
 logger = logging.getLogger(__name__)
 
@@ -416,7 +416,7 @@ class CalcRunner:
 
     def plot_concurrency_vs_time_metrics(self, output_dir: Path):
         """Plots concurrency vs. time metrics using pre-computed fits."""
-        from aiq.profiler.calc.plot import plot_concurrency_vs_time_metrics as plot_metrics
+        from nat.profiler.calc.plot import plot_concurrency_vs_time_metrics as plot_metrics
 
         # Only plot if we have valid metrics and at least one fit
         if not self.metrics_per_concurrency:

{aiq → nat}/profiler/calc/calculations.py

@@ -17,9 +17,9 @@ import logging
 
 import numpy as np
 
-from aiq.profiler.calc.data_models import FitConfig
-from aiq.profiler.calc.data_models import GPUEstimates
-from aiq.profiler.calc.data_models import LinearFitResult
+from nat.profiler.calc.data_models import FitConfig
+from nat.profiler.calc.data_models import GPUEstimates
+from nat.profiler.calc.data_models import LinearFitResult
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/profiler/calc/plot.py

@@ -20,8 +20,8 @@ import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
 
-from aiq.profiler.calc.data_models import LinearFitResult
-from aiq.profiler.calc.data_models import SizingMetrics
+from nat.profiler.calc.data_models import LinearFitResult
+from nat.profiler.calc.data_models import SizingMetrics
 
 logger = logging.getLogger(__name__)