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

{aiq → nat}/registry_handlers/local/local_handler.py

@@ -18,19 +18,19 @@ import subprocess
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 
-from aiq.registry_handlers.package_utils import build_package_metadata
-from aiq.registry_handlers.registry_handler_base import AbstractRegistryHandler
-from aiq.registry_handlers.schemas.package import PackageNameVersionList
-from aiq.registry_handlers.schemas.publish import AIQArtifact
-from aiq.registry_handlers.schemas.publish import PublishResponse
-from aiq.registry_handlers.schemas.pull import PullRequestPackages
-from aiq.registry_handlers.schemas.pull import PullResponse
-from aiq.registry_handlers.schemas.remove import RemoveResponse
-from aiq.registry_handlers.schemas.search import SearchFields
-from aiq.registry_handlers.schemas.search import SearchQuery
-from aiq.registry_handlers.schemas.search import SearchResponse
-from aiq.registry_handlers.schemas.status import ActionEnum
-from aiq.registry_handlers.schemas.status import StatusEnum
+from nat.registry_handlers.package_utils import build_package_metadata
+from nat.registry_handlers.registry_handler_base import AbstractRegistryHandler
+from nat.registry_handlers.schemas.package import PackageNameVersionList
+from nat.registry_handlers.schemas.publish import Artifact
+from nat.registry_handlers.schemas.publish import PublishResponse
+from nat.registry_handlers.schemas.pull import PullRequestPackages
+from nat.registry_handlers.schemas.pull import PullResponse
+from nat.registry_handlers.schemas.remove import RemoveResponse
+from nat.registry_handlers.schemas.search import SearchFields
+from nat.registry_handlers.schemas.search import SearchQuery
+from nat.registry_handlers.schemas.search import SearchResponse
+from nat.registry_handlers.schemas.status import ActionEnum
+from nat.registry_handlers.schemas.status import StatusEnum
 
 logger = logging.getLogger(__name__)
 
@@ -41,11 +41,11 @@ class LocalRegistryHandler(AbstractRegistryHandler):
     search_fields: list[SearchFields] = [field for field in SearchFields if field != SearchFields.ALL]
 
     @asynccontextmanager
-    async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AIQ Toolkit artifact to a local registry.
+    async def publish(self, artifact: Artifact) -> AsyncGenerator[PublishResponse]:
+        """Publishes a NAT artifact to a local registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
+            artifact (Artifact): An artifact that contain NAT plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -62,10 +62,10 @@ class LocalRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AIQ Toolkit artifacts from a local registry.
+        """Download and install NAT artifacts from a local registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
+            packages (PullRequestPackages): Parameters used to pull the NAT artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
@@ -83,7 +83,7 @@ class LocalRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches the local aiq registry for relevant AIQ Toolkit components.
+        """Searches the local nat registry for relevant NAT components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.

{aiq → nat}/registry_handlers/local/register_local.py

@@ -15,14 +15,14 @@
 
 import logging
 
-from aiq.cli.register_workflow import register_registry_handler
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.cli.register_workflow import register_registry_handler
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
 
 logger = logging.getLogger(__name__)
 
 
 class LocalRegistryHandlerConfig(RegistryHandlerBaseConfig, name="local"):
-    """Interact with the local AIQ Toolkit environment to search and uninstall AIQ Toolkit components."""
+    """Interact with the local NAT environment to search and uninstall NAT components."""
 
     pass
 
@@ -30,7 +30,7 @@ class LocalRegistryHandlerConfig(RegistryHandlerBaseConfig, name="local"):
 @register_registry_handler(config_type=LocalRegistryHandlerConfig)
 async def local_registry_handler(config: LocalRegistryHandlerConfig):
 
-    from aiq.registry_handlers.local.local_handler import LocalRegistryHandler
+    from nat.registry_handlers.local.local_handler import LocalRegistryHandler
 
     registry_handler = LocalRegistryHandler()
 

{aiq → nat}/registry_handlers/metadata_factory.py

@@ -15,24 +15,24 @@
 
 import logging
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.data_models.discovery_metadata import DiscoveryMetadata
-from aiq.data_models.discovery_metadata import DiscoveryStatusEnum
-from aiq.registry_handlers.schemas.package import WheelData
+from nat.data_models.component import ComponentEnum
+from nat.data_models.discovery_metadata import DiscoveryMetadata
+from nat.data_models.discovery_metadata import DiscoveryStatusEnum
+from nat.registry_handlers.schemas.package import WheelData
 
 logger = logging.getLogger(__name__)
 
 
 class ComponentDiscoveryMetadata:
 
-    def __init__(self, component_type: AIQComponentEnum, wheel_data: WheelData | None = None):
+    def __init__(self, component_type: ComponentEnum, wheel_data: WheelData | None = None):
         self._component_type = component_type
         self._metadata_items: list[dict | DiscoveryMetadata] = []
         self._wheel_data: WheelData = wheel_data
 
     def load_metadata(self):
 
-        from aiq.cli.type_registry import GlobalTypeRegistry
+        from nat.cli.type_registry import GlobalTypeRegistry
 
         registry = GlobalTypeRegistry.get()
 
@@ -55,6 +55,6 @@ class ComponentDiscoveryMetadata:
         return self._metadata_items
 
     @staticmethod
-    def from_package_component_type(component_type: AIQComponentEnum,
+    def from_package_component_type(component_type: ComponentEnum,
                                     wheel_data: WheelData | None = None) -> "ComponentDiscoveryMetadata":
         return ComponentDiscoveryMetadata(component_type=component_type, wheel_data=wheel_data)

{aiq → nat}/registry_handlers/package_utils.py

@@ -22,12 +22,12 @@ from functools import lru_cache
 
 from packaging.requirements import Requirement
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.data_models.discovery_metadata import DiscoveryMetadata
-from aiq.registry_handlers.schemas.package import WheelData
-from aiq.registry_handlers.schemas.publish import AIQArtifact
-from aiq.runtime.loader import PluginTypes
-from aiq.runtime.loader import discover_entrypoints
+from nat.data_models.component import ComponentEnum
+from nat.data_models.discovery_metadata import DiscoveryMetadata
+from nat.registry_handlers.schemas.package import WheelData
+from nat.registry_handlers.schemas.publish import Artifact
+from nat.runtime.loader import PluginTypes
+from nat.runtime.loader import discover_entrypoints
 
 # pylint: disable=redefined-outer-name
 logger = logging.getLogger(__name__)
@@ -89,13 +89,13 @@ def resolve_extras_to_packages(package_name: str, extras: list[str]) -> set[str]
     """Resolve package extras to their actual package dependencies.
 
     Args:
-        package_name (str): The base package name (e.g., 'aiqtoolkit')
+        package_name (str): The base package name (e.g., 'nvidia-nat')
         extras (list[str]): List of extra names (e.g., ['langchain', 'telemetry'])
 
     Returns:
         set[str]: Set of additional package names that the extras resolve to
-        (e.g., {'aiqtoolkit-langchain', 'aiqtoolkit-opentelemetry', 'aiqtoolkit-phoenix',
-        'aiqtoolkit-weave', 'aiqtoolkit-ragaai'})
+        (e.g., {'nvidia-nat-langchain', 'nvidia-nat-opentelemetry', 'nvidia-nat-phoenix',
+        'nvidia-nat-weave', 'nvidia-nat-ragaai'})
     """
     resolved_packages = set()
 
@@ -149,8 +149,8 @@ def extract_dependencies_with_extras_resolved(pyproject_path: str) -> set[str]:
         set[str]: Set of all dependency names including those resolved from extras
 
     Example:
-        For a dependency like "aiqtoolkit[langchain,telemetry]~=1.2", this will return:
-        {'aiqtoolkit', 'aiqtoolkit-langchain', 'aiqtoolkit-opentelemetry', 'aiqtoolkit-phoenix', ...}
+        For a dependency like "nat[langchain,telemetry]~=1.2", this will return:
+        {'nvidia-nat', 'nvidia-nat-langchain', 'nvidia-nat-opentelemetry', 'nvidia-nat-phoenix', ...}
 
     Raises:
         FileNotFoundError: If the pyproject.toml file doesn't exist
@@ -287,7 +287,7 @@ def get_transitive_dependencies(distribution_names: list[str]) -> dict[str, set[
     This function recursively resolves all dependencies for the given distribution names,
     returning a mapping of each package to its complete set of transitive dependencies.
     This is useful when publishing plugins to remote registries that contain with nested dependencies,
-    ensuring that all dependencies are included in the AIQArtifact's metadata.
+    ensuring that all dependencies are included in the Artifact's metadata.
 
     Args:
         distribution_names (list[str]): List of Python distribution names (package names) to analyze.
@@ -408,7 +408,7 @@ def get_all_transitive_dependencies(distribution_names: list[str]) -> set[str]:
 
     Returns a flattened set of all unique dependencies across all the provided distribution names.
     This is useful when publishing plugins to remote registries that contain with nested dependencies,
-    ensuring that all dependencies are included in the AIQArtifact's metadata.
+    ensuring that all dependencies are included in the Artifact's metadata.
 
     Args:
         distribution_names: List of Python distribution names (package names) to analyze
@@ -449,7 +449,7 @@ def build_wheel(package_root: str) -> WheelData:
 
     toml_project: dict = data.get("project", {})
     toml_project_name = toml_project.get("name", None)
-    toml_packages = set(i for i in data.get("project", {}).get("entry-points", {}).get("aiq.plugins", {}))
+    toml_packages = set(i for i in data.get("project", {}).get("entry-points", {}).get("nat.plugins", {}))
 
     # Extract dependencies using the robust requirement parser with extras resolution
     try:
@@ -488,43 +488,43 @@ def build_wheel(package_root: str) -> WheelData:
 
     os.chdir(working_dir)
 
-    whl_version = Wheel(whl_path).version
+    whl_version = Wheel(whl_path).version or "unknown"
 
     return WheelData(package_root=package_root,
                      package_name=toml_project_name,
                      toml_project=toml_project,
                      toml_dependencies=toml_dependencies,
-                     toml_aiq_packages=toml_packages,
+                     toml_nat_packages=toml_packages,
                      union_dependencies=union_dependencies,
                      whl_path=whl_path,
                      whl_base64=whl_base64,
                      whl_version=whl_version)
 
 
-def build_package_metadata(wheel_data: WheelData | None) -> dict[AIQComponentEnum, list[dict | DiscoveryMetadata]]:
-    """Loads discovery metadata for all registered AIQ Toolkit components included in this Python package.
+def build_package_metadata(wheel_data: WheelData | None) -> dict[ComponentEnum, list[dict | DiscoveryMetadata]]:
+    """Loads discovery metadata for all registered NAT components included in this Python package.
 
     Args:
         wheel_data (WheelData): Data model containing a built python wheel and its corresponding metadata.
 
     Returns:
-        dict[AIQComponentEnum, list[typing.Union[dict, DiscoveryMetadata]]]: List containing each components discovery
+        dict[ComponentEnum, list[typing.Union[dict, DiscoveryMetadata]]]: List containing each components discovery
         metadata.
     """
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.metadata_factory import ComponentDiscoveryMetadata
-    from aiq.runtime.loader import discover_and_register_plugins
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.metadata_factory import ComponentDiscoveryMetadata
+    from nat.runtime.loader import discover_and_register_plugins
 
     discover_and_register_plugins(PluginTypes.ALL)
 
     registry = GlobalTypeRegistry.get()
 
-    aiq_plugins = discover_entrypoints(PluginTypes.ALL)
+    nat_plugins = discover_entrypoints(PluginTypes.ALL)
 
     if (wheel_data is not None):
         registry.register_package(package_name=wheel_data.package_name, package_version=wheel_data.whl_version)
-        for entry_point in aiq_plugins:
+        for entry_point in nat_plugins:
             package_name = entry_point.dist.name
             if (package_name == wheel_data.package_name):
                 continue
@@ -532,13 +532,13 @@ def build_package_metadata(wheel_data: WheelData | None) -> dict[AIQComponentEnu
                 registry.register_package(package_name=package_name)
 
     else:
-        for entry_point in aiq_plugins:
+        for entry_point in nat_plugins:
             registry.register_package(package_name=entry_point.dist.name)
 
     discovery_metadata = {}
-    for component_type in AIQComponentEnum:
+    for component_type in ComponentEnum:
 
-        if (component_type == AIQComponentEnum.UNDEFINED):
+        if (component_type == ComponentEnum.UNDEFINED):
             continue
         component_metadata = ComponentDiscoveryMetadata.from_package_component_type(wheel_data=wheel_data,
                                                                                     component_type=component_type)
@@ -548,20 +548,24 @@ def build_package_metadata(wheel_data: WheelData | None) -> dict[AIQComponentEnu
     return discovery_metadata
 
 
-def build_aiq_artifact(package_root: str) -> AIQArtifact:
-    """Builds a complete AIQ Toolkit Artifact that can be published for discovery and reuse.
+def build_artifact(package_root: str) -> Artifact:
+    """Builds a complete NeMo Agent toolkit Artifact that can be published for discovery and reuse.
 
     Args:
         package_root (str): Path to root of python package
 
     Returns:
-        AIQArtifact: An publishabla AIQArtifact containing package wheel and discovery metadata.
+        Artifact: A publishable Artifact containing package wheel and discovery metadata.
     """
 
-    from aiq.registry_handlers.schemas.publish import BuiltAIQArtifact
+    from nat.registry_handlers.schemas.publish import BuiltArtifact
 
     wheel_data = build_wheel(package_root=package_root)
     metadata = build_package_metadata(wheel_data=wheel_data)
-    built_artifact = BuiltAIQArtifact(whl=wheel_data.whl_base64, metadata=metadata)
+    built_artifact = BuiltArtifact(whl=wheel_data.whl_base64, metadata=metadata)
 
-    return AIQArtifact(artifact=built_artifact, whl_path=wheel_data.whl_path)
+    return Artifact(artifact=built_artifact, whl_path=wheel_data.whl_path)
+
+
+# Compatibility alias
+build_aiq_artifact = build_artifact

{aiq → nat}/registry_handlers/pypi/pypi_handler.py

@@ -18,20 +18,20 @@ import subprocess
 from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.registry_handlers.registry_handler_base import AbstractRegistryHandler
-from aiq.registry_handlers.schemas.package import PackageNameVersionList
-from aiq.registry_handlers.schemas.publish import AIQArtifact
-from aiq.registry_handlers.schemas.publish import PublishResponse
-from aiq.registry_handlers.schemas.pull import PackageNameVersion
-from aiq.registry_handlers.schemas.pull import PullRequestPackages
-from aiq.registry_handlers.schemas.pull import PullResponse
-from aiq.registry_handlers.schemas.remove import RemoveResponse
-from aiq.registry_handlers.schemas.search import SearchQuery
-from aiq.registry_handlers.schemas.search import SearchResponse
-from aiq.registry_handlers.schemas.search import SearchResponseItem
-from aiq.registry_handlers.schemas.status import ActionEnum
-from aiq.registry_handlers.schemas.status import StatusEnum
+from nat.data_models.component import ComponentEnum
+from nat.registry_handlers.registry_handler_base import AbstractRegistryHandler
+from nat.registry_handlers.schemas.package import PackageNameVersionList
+from nat.registry_handlers.schemas.publish import Artifact
+from nat.registry_handlers.schemas.publish import PublishResponse
+from nat.registry_handlers.schemas.pull import PackageNameVersion
+from nat.registry_handlers.schemas.pull import PullRequestPackages
+from nat.registry_handlers.schemas.pull import PullResponse
+from nat.registry_handlers.schemas.remove import RemoveResponse
+from nat.registry_handlers.schemas.search import SearchQuery
+from nat.registry_handlers.schemas.search import SearchResponse
+from nat.registry_handlers.schemas.search import SearchResponseItem
+from nat.registry_handlers.schemas.status import ActionEnum
+from nat.registry_handlers.schemas.status import StatusEnum
 
 logger = logging.getLogger(__name__)
 
@@ -59,11 +59,11 @@ class PypiRegistryHandler(AbstractRegistryHandler):
         self._search_route = search_route.strip("/")
 
     @asynccontextmanager
-    async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AIQ Toolkit artifact to a PyPI remote registry.
+    async def publish(self, artifact: Artifact) -> AsyncGenerator[PublishResponse]:
+        """Publishes a NAT artifact to a PyPI remote registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
+            artifact (Artifact): An artifact that contain NAT plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -101,10 +101,10 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AIQ Toolkit artifacts from a remote PyPI remote registry.
+        """Download and install NAT artifacts from a remote PyPI remote registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
+            packages (PullRequestPackages): Parameters used to pull the NAT artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse, None]]: A response message that includes a the pulled packages and a
@@ -160,7 +160,7 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches a remote PyPI registry for relevant AIQ Toolkit components.
+        """Searches a remote PyPI registry for relevant NAT components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.
@@ -192,7 +192,7 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
                 search_resp_item = SearchResponseItem(package=package,
                                                       version=version,
-                                                      component_type=AIQComponentEnum.PACKAGE,
+                                                      component_type=ComponentEnum.PACKAGE,
                                                       component_name=package,
                                                       description="",
                                                       developer_notes="")

{aiq → nat}/registry_handlers/pypi/register_pypi.py

@@ -15,8 +15,8 @@
 
 from pydantic import Field
 
-from aiq.cli.register_workflow import register_registry_handler
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.cli.register_workflow import register_registry_handler
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
 
 
 class PypiRegistryHandlerConfig(RegistryHandlerBaseConfig, name="pypi"):
@@ -25,15 +25,15 @@ class PypiRegistryHandlerConfig(RegistryHandlerBaseConfig, name="pypi"):
     endpoint: str = Field(description="A string representing the remote endpoint.")
     token: str | None = Field(default=None,
                               description="The authentication token to use when interacting with the registry.")
-    publish_route: str = Field(description="The route to the AIQ Toolkit publish service.")
-    pull_route: str = Field(description="The route to the AIQ Toolkit pull service.")
-    search_route: str = Field(default="simple", description="The route to the AIQ Toolkit search service.")
+    publish_route: str = Field(description="The route to the NAT publish service.")
+    pull_route: str = Field(description="The route to the NAT pull service.")
+    search_route: str = Field(default="simple", description="The route to the NAT search service.")
 
 
 @register_registry_handler(config_type=PypiRegistryHandlerConfig)
 async def pypi_publish_registry_handler(config: PypiRegistryHandlerConfig):
 
-    from aiq.registry_handlers.pypi.pypi_handler import PypiRegistryHandler
+    from nat.registry_handlers.pypi.pypi_handler import PypiRegistryHandler
 
     registry_handler = PypiRegistryHandler(endpoint=config.endpoint, token=config.token)
 

{aiq → nat}/registry_handlers/registry_handler_base.py

@@ -19,36 +19,36 @@ from collections.abc import AsyncGenerator
 from contextlib import asynccontextmanager
 from enum import Enum
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.data_models.discovery_metadata import DiscoveryMetadata
-from aiq.registry_handlers.schemas.package import PackageNameVersionList
-from aiq.registry_handlers.schemas.publish import AIQArtifact
-from aiq.registry_handlers.schemas.publish import PublishResponse
-from aiq.registry_handlers.schemas.pull import PullRequestPackages
-from aiq.registry_handlers.schemas.pull import PullResponse
-from aiq.registry_handlers.schemas.remove import RemoveResponse
-from aiq.registry_handlers.schemas.search import SearchQuery
-from aiq.registry_handlers.schemas.search import SearchResponse
-from aiq.registry_handlers.schemas.search import VisualizeFields
+from nat.data_models.component import ComponentEnum
+from nat.data_models.discovery_metadata import DiscoveryMetadata
+from nat.registry_handlers.schemas.package import PackageNameVersionList
+from nat.registry_handlers.schemas.publish import Artifact
+from nat.registry_handlers.schemas.publish import PublishResponse
+from nat.registry_handlers.schemas.pull import PullRequestPackages
+from nat.registry_handlers.schemas.pull import PullResponse
+from nat.registry_handlers.schemas.remove import RemoveResponse
+from nat.registry_handlers.schemas.search import SearchQuery
+from nat.registry_handlers.schemas.search import SearchResponse
+from nat.registry_handlers.schemas.search import VisualizeFields
 
 
 class AbstractRegistryHandler(ABC):
-    """Base class outlining the interfaces for remote AIQ Toolkit registry interactions."""
+    """Base class outlining the interfaces for remote NAT registry interactions."""
 
     def __init__(self):
-        self._discovery_metadata: dict[AIQComponentEnum, list[dict | DiscoveryMetadata]] = {}
-        self._aiq_artifact: AIQArtifact | None = None
+        self._discovery_metadata: dict[ComponentEnum, list[dict | DiscoveryMetadata]] = {}
+        self._nat_artifact: Artifact | None = None
         self._whl_bytes: bytes
         self._whl_path: str
         self._whl_base64: str
 
     @abstractmethod
     @asynccontextmanager
-    async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AIQ Toolkit artifact to a remote registry.
+    async def publish(self, artifact: Artifact) -> AsyncGenerator[PublishResponse]:
+        """Publishes a NAT artifact to a remote registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
+            artifact (Artifact): An artifact that contain NAT plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -61,10 +61,10 @@ class AbstractRegistryHandler(ABC):
     @abstractmethod
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AIQ Toolkit artifacts from a remote registry.
+        """Download and install NAT artifacts from a remote registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
+            packages (PullRequestPackages): Parameters used to pull the NAT artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
@@ -76,7 +76,7 @@ class AbstractRegistryHandler(ABC):
     @abstractmethod
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches the local aiq registry for relevant AIQ Toolkit components.
+        """Searches the local nat registry for relevant NAT components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.
@@ -118,7 +118,7 @@ class AbstractRegistryHandler(ABC):
         from rich.table import Table
         from rich.text import Text
 
-        table = Table(title="AIQ Toolkit Search Results", padding=(0, 1), show_lines=True)
+        table = Table(title="NAT Search Results", padding=(0, 1), show_lines=True)
         for column in VisualizeFields:
             table.add_column(column.value)
 

{aiq → nat}/registry_handlers/rest/register_rest.py

@@ -17,8 +17,8 @@ import os
 
 from pydantic import Field
 
-from aiq.cli.register_workflow import register_registry_handler
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.cli.register_workflow import register_registry_handler
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
 
 
 class RestRegistryHandlerConfig(RegistryHandlerBaseConfig, name="rest"):
@@ -27,16 +27,16 @@ class RestRegistryHandlerConfig(RegistryHandlerBaseConfig, name="rest"):
     endpoint: str = Field(description="A string representing the remote endpoint.")
     token: str | None = Field(default=None,
                               description="The authentication token to use when interacting with the registry.")
-    publish_route: str = Field(default="", description="The route to the AIQ Toolkit publish service.")
-    pull_route: str = Field(default="", description="The route to the AIQ Toolkit pull service.")
-    search_route: str = Field(default="", description="The route to the AIQ Toolkit search service")
-    remove_route: str = Field(default="", description="The route to the AIQ Toolkit remove service")
+    publish_route: str = Field(default="", description="The route to the NAT publish service.")
+    pull_route: str = Field(default="", description="The route to the NAT pull service.")
+    search_route: str = Field(default="", description="The route to the NAT search service")
+    remove_route: str = Field(default="", description="The route to the NAT remove service")
 
 
 @register_registry_handler(config_type=RestRegistryHandlerConfig)
 async def rest_search_handler(config: RestRegistryHandlerConfig):
 
-    from aiq.registry_handlers.rest.rest_handler import RestRegistryHandler
+    from nat.registry_handlers.rest.rest_handler import RestRegistryHandler
 
     if (config.token is None):
         registry_token = os.getenv("REGISTRY_TOKEN")

{aiq → nat}/registry_handlers/rest/rest_handler.py

@@ -23,18 +23,18 @@ from contextlib import asynccontextmanager
 
 import httpx
 
-from aiq.registry_handlers.registry_handler_base import AbstractRegistryHandler
-from aiq.registry_handlers.schemas.headers import RequestHeaders
-from aiq.registry_handlers.schemas.package import PackageNameVersionList
-from aiq.registry_handlers.schemas.publish import AIQArtifact
-from aiq.registry_handlers.schemas.publish import PublishResponse
-from aiq.registry_handlers.schemas.pull import PullRequestPackages
-from aiq.registry_handlers.schemas.pull import PullResponse
-from aiq.registry_handlers.schemas.remove import RemoveResponse
-from aiq.registry_handlers.schemas.search import SearchQuery
-from aiq.registry_handlers.schemas.search import SearchResponse
-from aiq.registry_handlers.schemas.status import ActionEnum
-from aiq.registry_handlers.schemas.status import StatusEnum
+from nat.registry_handlers.registry_handler_base import AbstractRegistryHandler
+from nat.registry_handlers.schemas.headers import RequestHeaders
+from nat.registry_handlers.schemas.package import PackageNameVersionList
+from nat.registry_handlers.schemas.publish import Artifact
+from nat.registry_handlers.schemas.publish import PublishResponse
+from nat.registry_handlers.schemas.pull import PullRequestPackages
+from nat.registry_handlers.schemas.pull import PullResponse
+from nat.registry_handlers.schemas.remove import RemoveResponse
+from nat.registry_handlers.schemas.search import SearchQuery
+from nat.registry_handlers.schemas.search import SearchResponse
+from nat.registry_handlers.schemas.status import ActionEnum
+from nat.registry_handlers.schemas.status import StatusEnum
 
 logger = logging.getLogger(__name__)
 
@@ -61,11 +61,11 @@ class RestRegistryHandler(AbstractRegistryHandler):
         self._headers = RequestHeaders(Authorization=f"Bearer: {token}").model_dump(by_alias=True)
 
     @asynccontextmanager
-    async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AIQ Toolkit artifact to a remote REST registry.
+    async def publish(self, artifact: Artifact) -> AsyncGenerator[PublishResponse]:
+        """Publishes a NAT artifact to a remote REST registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
+            artifact (Artifact): An artifact that contain NAT plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -98,17 +98,17 @@ class RestRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AIQ Toolkit artifacts from a remote REST registry.
+        """Download and install NAT artifacts from a remote REST registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
+            packages (PullRequestPackages): Parameters used to pull the NAT artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
                 completion status message.
         """
 
-        tmp_dir = ".tmp-aiq-pull"
+        tmp_dir = "./.tmp/nat-pull"
 
         try:
             async with httpx.AsyncClient(headers=self._headers, timeout=self._timeout) as client:
@@ -165,7 +165,7 @@ class RestRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches a remote REST registry for relevant AIQ Toolkit components.
+        """Searches a remote REST registry for relevant NAT components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.

{aiq → nat}/registry_handlers/schemas/package.py

@@ -28,8 +28,8 @@ class WheelData(BaseModel):
         package_name (str): The name of the python package.
         toml_project (dict): A dictionary containing data about the python project.
         toml_dependencies (set): The list of dependencies provided in the pyproject.toml file.
-        toml_aiq_packages (set): The AIQ Toolkit plugins listed in the pyproject.toml.
-        union_dependencies (set): The union of toml_dependencies and toml_aiq_packages.
+        toml_nat_packages (set): The NAT plugins listed in the pyproject.toml.
+        union_dependencies (set): The union of toml_dependencies and toml_nat_packages.
         whl_path (str): The path to the package wheel file.
         whl_base64 (str): Base64 encoded string of the wheel file.
         whl_version (str): The version representing the wheel file.
@@ -39,7 +39,7 @@ class WheelData(BaseModel):
     package_name: str
     toml_project: dict
     toml_dependencies: set
-    toml_aiq_packages: set
+    toml_nat_packages: set
     union_dependencies: set
     whl_path: str
     whl_base64: str