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

{aiq → nat}/cli/commands/info/list_components.py

@@ -19,24 +19,24 @@ from contextlib import AsyncExitStack
 
 import click
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.registry_handlers.schemas.search import SearchFields
+from nat.data_models.component import ComponentEnum
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.registry_handlers.schemas.search import SearchFields
 
 logger = logging.getLogger(__name__)
 
 
 async def search_artifacts(  # pylint: disable=R0917
         registry_handler_config: RegistryHandlerBaseConfig,
-        component_types: list[AIQComponentEnum],
+        component_types: list[ComponentEnum],
         visualize: bool,
         query: str,
         num_results: int,
         query_fields: list[SearchFields],
         save_path: str | None) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.schemas.search import SearchQuery
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.schemas.search import SearchQuery
 
     registry = GlobalTypeRegistry.get()
 
@@ -46,7 +46,7 @@ async def search_artifacts(  # pylint: disable=R0917
         registry_handler = await stack.enter_async_context(registry_handler_info.build_fn(registry_handler_config))
 
         if (len(component_types) == 0):
-            component_types = [t.value for t in AIQComponentEnum]
+            component_types = [t.value for t in ComponentEnum]
 
         if (len(query_fields) == 0):
             query_fields = (SearchFields.ALL, )
@@ -60,15 +60,15 @@ async def search_artifacts(  # pylint: disable=R0917
             registry_handler.save_search_results(search_response=search_response, save_path=save_path)
 
 
-@click.group(name=__name__, invoke_without_command=True, help="List the locally registered AIQ Toolkit components.")
+@click.group(name=__name__, invoke_without_command=True, help="List the locally registered NAT components.")
 @click.option(
     "-t",
     "--types",
     "component_types",
     multiple=True,
-    type=click.Choice([e.value for e in AIQComponentEnum], case_sensitive=False),
+    type=click.Choice([e.value for e in ComponentEnum], case_sensitive=False),
     required=False,
-    help=("Filter the search by AIQ Toolkit component type."),
+    help=("Filter the search by NAT component type."),
 )
 @click.option(
     "-o",
@@ -104,12 +104,12 @@ async def search_artifacts(  # pylint: disable=R0917
 def list_components(fields: list[SearchFields],
                     query: str,
                     num_results: int,
-                    component_types: list[AIQComponentEnum],
+                    component_types: list[ComponentEnum],
                     output_path: str | None = None) -> None:
 
-    from aiq.runtime.loader import PluginTypes
-    from aiq.runtime.loader import discover_and_register_plugins
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.runtime.loader import PluginTypes
+    from nat.runtime.loader import discover_and_register_plugins
+    from nat.settings.global_settings import GlobalSettings
 
     discover_and_register_plugins(PluginTypes.ALL)
 

{aiq → nat}/cli/commands/info/list_mcp.py

@@ -13,16 +13,18 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
+import asyncio
 import json
 import logging
+import time
 from typing import Any
 
-import anyio
 import click
+from pydantic import BaseModel
 
-from aiq.tool.mcp.exceptions import MCPError
-from aiq.tool.mcp.mcp_client import MCPBuilder
-from aiq.utils.exception_handlers.mcp import format_mcp_error
+from nat.tool.mcp.exceptions import MCPError
+from nat.tool.mcp.mcp_client import MCPBuilder
+from nat.utils.exception_handlers.mcp import format_mcp_error
 
 # Suppress verbose logs from mcp.client.sse and httpx
 logging.getLogger("mcp.client.sse").setLevel(logging.WARNING)
@@ -31,6 +33,21 @@ logging.getLogger("httpx").setLevel(logging.WARNING)
 logger = logging.getLogger(__name__)
 
 
+class MCPPingResult(BaseModel):
+    """Result of an MCP server ping request.
+
+    Attributes:
+        url (str): The MCP server URL that was pinged
+        status (str): Health status - 'healthy', 'unhealthy', or 'unknown'
+        response_time_ms (float | None): Response time in milliseconds, None if request failed completely
+        error (str | None): Error message if the ping failed, None if successful
+    """
+    url: str
+    status: str
+    response_time_ms: float | None
+    error: str | None
+
+
 def format_tool(tool: Any) -> dict[str, str | None]:
     """Format an MCP tool into a dictionary for display.
 
@@ -103,9 +120,8 @@ async def list_tools_and_schemas(url: str, tool_name: str | None = None) -> list
         if tool_name:
             tool = await builder.get_tool(tool_name)
             return [format_tool(tool)]
-        else:
-            tools = await builder.get_tools()
-            return [format_tool(tool) for tool in tools.values()]
+        tools = await builder.get_tools()
+        return [format_tool(tool) for tool in tools.values()]
     except MCPError as e:
         format_mcp_error(e, include_traceback=False)
         return []
@@ -152,8 +168,8 @@ async def list_tools_direct(url: str, tool_name: str | None = None) -> list[dict
                 return tools
     except Exception as e:
         # Convert raw exceptions to structured MCPError for consistency
-        from aiq.utils.exception_handlers.mcp import convert_to_mcp_error
-        from aiq.utils.exception_handlers.mcp import extract_primary_exception
+        from nat.utils.exception_handlers.mcp import convert_to_mcp_error
+        from nat.utils.exception_handlers.mcp import extract_primary_exception
 
         if isinstance(e, ExceptionGroup):  # noqa: F821
             primary_exception = extract_primary_exception(list(e.exceptions))
@@ -165,6 +181,49 @@ async def list_tools_direct(url: str, tool_name: str | None = None) -> list[dict
         return []
 
 
+async def ping_mcp_server(url: str, timeout: int) -> MCPPingResult:
+    """Ping an MCP server to check if it's responsive.
+
+    Args:
+        url (str): MCP server URL to ping
+        timeout (int): Timeout in seconds for the ping request
+
+    Returns:
+        MCPPingResult: Structured result with status, response_time, and any error info
+    """
+    from mcp.client.session import ClientSession
+    from mcp.client.sse import sse_client
+
+    async def _ping_operation():
+        async with sse_client(url) as (read, write):
+            async with ClientSession(read, write) as session:
+                # Initialize the session
+                await session.initialize()
+
+                # Record start time just before ping
+                start_time = time.time()
+                # Send ping request
+                await session.send_ping()
+
+                end_time = time.time()
+                response_time_ms = round((end_time - start_time) * 1000, 2)
+
+                return MCPPingResult(url=url, status="healthy", response_time_ms=response_time_ms, error=None)
+
+    try:
+        # Apply timeout to the entire ping operation
+        return await asyncio.wait_for(_ping_operation(), timeout=timeout)
+
+    except asyncio.TimeoutError:
+        return MCPPingResult(url=url,
+                             status="unreachable",
+                             response_time_ms=None,
+                             error=f"Timeout after {timeout} seconds")
+
+    except Exception as e:
+        return MCPPingResult(url=url, status="unhealthy", response_time_ms=None, error=str(e))
+
+
 @click.group(invoke_without_command=True, help="List tool names (default), or show details with --detail or --tool.")
 @click.option('--direct', is_flag=True, help='Bypass MCPBuilder and use direct MCP protocol')
 @click.option('--url', default='http://localhost:9901/sse', show_default=True, help='MCP server URL')
@@ -189,16 +248,16 @@ def list_mcp(ctx: click.Context, direct: bool, url: str, tool: str | None, detai
         json_output (bool): Whether to output tool metadata in JSON format instead of text
 
     Examples:
-        aiq info mcp                           # List tool names only
-        aiq info mcp --detail                  # Show all tools with full details
-        aiq info mcp --tool my_tool            # Show details for specific tool
-        aiq info mcp --json-output             # Get JSON format output
-        aiq info mcp --direct --url http://...  # Use direct protocol with custom URL
+        nat info mcp                           # List tool names only
+        nat info mcp --detail                  # Show all tools with full details
+        nat info mcp --tool my_tool            # Show details for specific tool
+        nat info mcp --json-output             # Get JSON format output
+        nat info mcp --direct --url http://...  # Use direct protocol with custom URL
     """
     if ctx.invoked_subcommand is not None:
         return
     fetcher = list_tools_direct if direct else list_tools_and_schemas
-    tools = anyio.run(fetcher, url, tool)
+    tools = asyncio.run(fetcher(url, tool))
 
     if json_output:
         click.echo(json.dumps(tools, indent=2))
@@ -211,3 +270,35 @@ def list_mcp(ctx: click.Context, direct: bool, url: str, tool: str | None, detai
     else:
         for tool_dict in tools:
             click.echo(tool_dict.get('name', 'Unknown tool'))
+
+
+@list_mcp.command()
+@click.option('--url', default='http://localhost:9901/sse', show_default=True, help='MCP server URL')
+@click.option('--timeout', default=60, show_default=True, help='Timeout in seconds for ping request')
+@click.option('--json-output', is_flag=True, help='Output ping result in JSON format')
+def ping(url: str, timeout: int, json_output: bool) -> None:
+    """Ping an MCP server to check if it's responsive.
+
+    This command sends a ping request to the MCP server and measures the response time.
+    It's useful for health checks and monitoring server availability.
+
+    Args:
+        url (str): MCP server URL to ping (default: http://localhost:9901/sse)
+        timeout (int): Timeout in seconds for the ping request (default: 60)
+        json_output (bool): Whether to output the result in JSON format
+
+    Examples:
+        nat info mcp ping                                    # Ping default server
+        nat info mcp ping --url http://custom-server:9901/sse # Ping custom server
+        nat info mcp ping --timeout 10                      # Use 10 second timeout
+        nat info mcp ping --json-output                     # Get JSON format output
+    """
+    result = asyncio.run(ping_mcp_server(url, timeout))
+
+    if json_output:
+        click.echo(result.model_dump_json(indent=2))
+    else:
+        if result.status == "healthy":
+            click.echo(f"Server at {result.url} is healthy (response time: {result.response_time_ms}ms)")
+        else:
+            click.echo(f"Server at {result.url} {result.status}: {result.error}")

{aiq → nat}/cli/commands/registry/publish.py

@@ -20,16 +20,16 @@ from pathlib import Path
 
 import click
 
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.utils.data_models.schema_validator import validate_yaml
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.utils.data_models.schema_validator import validate_yaml
 
 logger = logging.getLogger(__name__)
 
 
 async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, package_root: str) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.package_utils import build_aiq_artifact
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.package_utils import build_artifact
 
     registry = GlobalTypeRegistry.get()
 
@@ -38,7 +38,7 @@ async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, p
         registry_handler_info = registry.get_registry_handler(type(registry_handler_config))
         registry_handler = await stack.enter_async_context(registry_handler_info.build_fn(registry_handler_config))
         try:
-            artifact = build_aiq_artifact(package_root=package_root)
+            artifact = build_artifact(package_root=package_root)
         except Exception as e:
             logger.exception("Error building artifact: %s", e, exc_info=True)
             return
@@ -47,7 +47,7 @@ async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, p
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Publish local AIQ Toolkit artifacts to a remote "
+             help=("Publish local NAT artifacts to a remote "
                    "registry from package repository."))
 @click.option(
     "--config_file",
@@ -61,14 +61,14 @@ async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, p
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel to use when publishing the AIQ Toolkit artifact."),
+    help=("The remote registry channel to use when publishing the NAT artifact."),
 )
 @click.argument("package_root", type=str)
 def publish(channel: str, config_file: str, package_root: str) -> None:
     """
-    Publish AIQ Toolkit artifacts with the specified configuration
+    Publish NAT artifacts with the specified configuration
     """
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.settings.global_settings import GlobalSettings
 
     settings = GlobalSettings().get()
 

{aiq → nat}/cli/commands/registry/pull.py

@@ -20,18 +20,18 @@ from pathlib import Path
 
 import click
 
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.utils.data_models.schema_validator import validate_yaml
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.utils.data_models.schema_validator import validate_yaml
 
 logger = logging.getLogger(__name__)
 
 
 async def pull_artifact(registry_handler_config: RegistryHandlerBaseConfig, packages: list[str]) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.schemas.package import PackageNameVersion
-    from aiq.registry_handlers.schemas.pull import PullPackageWhl
-    from aiq.registry_handlers.schemas.pull import PullRequestPackages
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.schemas.package import PackageNameVersion
+    from nat.registry_handlers.schemas.pull import PullPackageWhl
+    from nat.registry_handlers.schemas.pull import PullRequestPackages
 
     registry = GlobalTypeRegistry.get()
 
@@ -74,7 +74,7 @@ async def pull_artifact(registry_handler_config: RegistryHandlerBaseConfig, pack
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Pull AIQ Toolkit artifacts from a remote registry "
+             help=("Pull NAT artifacts from a remote registry "
                    "by package name."))
 @click.option(
     "--config_file",
@@ -88,15 +88,15 @@ async def pull_artifact(registry_handler_config: RegistryHandlerBaseConfig, pack
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel to use when pulling the AIQ Toolkit artifact."),
+    help=("The remote registry channel to use when pulling the NAT artifact."),
 )
 @click.argument("packages", type=str)
 def pull(channel: str, config_file: str, packages: str) -> None:
     """
-    Pull AIQ Toolkit artifacts from a remote registry channel.
+    Pull NAT artifacts from a remote registry channel.
     """
 
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.settings.global_settings import GlobalSettings
 
     packages = packages.split()
 

{aiq → nat}/cli/commands/registry/registry.py

@@ -17,17 +17,15 @@ import logging
 
 import click
 
-from aiq.cli.commands.registry.publish import publish
-from aiq.cli.commands.registry.pull import pull
-from aiq.cli.commands.registry.remove import remove
-from aiq.cli.commands.registry.search import search
+from nat.cli.commands.registry.publish import publish
+from nat.cli.commands.registry.pull import pull
+from nat.cli.commands.registry.remove import remove
+from nat.cli.commands.registry.search import search
 
 logger = logging.getLogger(__name__)
 
 
-@click.group(name=__name__,
-             invoke_without_command=False,
-             help="Utility to configure AIQ Toolkit remote registry channels.")
+@click.group(name=__name__, invoke_without_command=False, help="Utility to configure NAT remote registry channels.")
 def registry_command(**kwargs):
     pass
 

{aiq → nat}/cli/commands/registry/remove.py

@@ -20,16 +20,16 @@ from pathlib import Path
 
 import click
 
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.utils.data_models.schema_validator import validate_yaml
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.utils.data_models.schema_validator import validate_yaml
 
 logger = logging.getLogger(__name__)
 
 
 async def remove_artifact(registry_handler_config: RegistryHandlerBaseConfig, packages: list[dict[str, str]]) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.schemas.package import PackageNameVersionList
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.schemas.package import PackageNameVersionList
 
     registry = GlobalTypeRegistry.get()
 
@@ -48,7 +48,7 @@ async def remove_artifact(registry_handler_config: RegistryHandlerBaseConfig, pa
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Remove AIQ Toolkit artifact from a remote registry by name and version."))
+             help=("Remove NAT artifact from a remote registry by name and version."))
 @click.argument("packages", type=str)
 @click.option(
     "--config_file",
@@ -62,14 +62,14 @@ async def remove_artifact(registry_handler_config: RegistryHandlerBaseConfig, pa
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel that will remove the AIQ Toolkit artifact."),
+    help=("The remote registry channel that will remove the NAT artifact."),
 )
 def remove(channel: str, config_file: str, packages: str) -> None:
     """
-    Remove AIQ Toolkit artifacts from a remote registry.
+    Remove NAT artifacts from a remote registry.
     """
 
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.settings.global_settings import GlobalSettings
 
     # Extract package name and version
     packages = packages.split()

{aiq → nat}/cli/commands/registry/search.py

@@ -20,11 +20,11 @@ from pathlib import Path
 
 import click
 
-from aiq.data_models.component import AIQComponentEnum
-from aiq.data_models.registry_handler import RegistryHandlerBaseConfig
-from aiq.registry_handlers.schemas.search import SearchFields
-from aiq.registry_handlers.schemas.status import StatusEnum
-from aiq.utils.data_models.schema_validator import validate_yaml
+from nat.data_models.component import ComponentEnum
+from nat.data_models.registry_handler import RegistryHandlerBaseConfig
+from nat.registry_handlers.schemas.search import SearchFields
+from nat.registry_handlers.schemas.status import StatusEnum
+from nat.utils.data_models.schema_validator import validate_yaml
 
 logger = logging.getLogger(__name__)
 
@@ -34,12 +34,12 @@ async def search_artifacts(  # pylint: disable=R0917
         query: str,
         search_fields: list[SearchFields],
         visualize: bool,
-        component_types: list[AIQComponentEnum],
+        component_types: list[ComponentEnum],
         save_path: str | None = None,
         n_results: int = 10) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.schemas.search import SearchQuery
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.schemas.search import SearchQuery
 
     registry = GlobalTypeRegistry.get()
 
@@ -49,7 +49,7 @@ async def search_artifacts(  # pylint: disable=R0917
         registry_handler = await stack.enter_async_context(registry_handler_info.build_fn(registry_handler_config))
 
         if (len(component_types) == 0):
-            component_types = [t.value for t in AIQComponentEnum]
+            component_types = [t.value for t in ComponentEnum]
 
         query = SearchQuery(query=query, fields=search_fields, top_k=n_results, component_types=component_types)
 
@@ -62,7 +62,7 @@ async def search_artifacts(  # pylint: disable=R0917
                 registry_handler.save_search_results(search_response=search_response, save_path=save_path)
 
 
-@click.group(name=__name__, invoke_without_command=True, help="Search for AIQ Toolkit artifacts from remote registry.")
+@click.group(name=__name__, invoke_without_command=True, help="Search for NAT artifacts from remote registry.")
 @click.option(
     "--config_file",
     type=click.Path(exists=True, file_okay=True, dir_okay=False, path_type=Path),
@@ -75,7 +75,7 @@ async def search_artifacts(  # pylint: disable=R0917
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel to use when pulling the AIQ Toolkit artifact."),
+    help=("The remote registry channel to use when pulling the NAT artifact."),
 )
 @click.option(
     "-o",
@@ -112,7 +112,7 @@ async def search_artifacts(  # pylint: disable=R0917
     "--types",
     "component_types",
     multiple=True,
-    type=click.Choice([e.value for e in AIQComponentEnum], case_sensitive=False),
+    type=click.Choice([e.value for e in ComponentEnum], case_sensitive=False),
     required=False,
     help=("The component types to include in search."),
 )
@@ -121,14 +121,14 @@ def search(  # pylint: disable=R0917
         channel: str,
         fields: list[str],
         query: str,
-        component_types: list[AIQComponentEnum],
+        component_types: list[ComponentEnum],
         n_results: int,
         output_path: str) -> None:
     """
-    Search for AIQ Toolkit artifacts with the specified configuration.
+    Search for NAT artifacts with the specified configuration.
     """
 
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.settings.global_settings import GlobalSettings
 
     settings = GlobalSettings().get()
 

{aiq → nat}/cli/commands/sizing/calc.py

@@ -20,9 +20,9 @@ from pathlib import Path
 import click
 from tabulate import tabulate
 
-from aiq.profiler.calc.calc_runner import CalcRunner
-from aiq.profiler.calc.data_models import CalcRunnerConfig
-from aiq.profiler.calc.data_models import CalcRunnerOutput
+from nat.profiler.calc.calc_runner import CalcRunner
+from nat.profiler.calc.data_models import CalcRunnerConfig
+from nat.profiler.calc.data_models import CalcRunnerOutput
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/cli/commands/start.py

@@ -23,12 +23,12 @@ from pathlib import Path
 import click
 from pydantic_core import SchemaValidator
 
-from aiq.cli.cli_utils.config_override import load_and_override_config
-from aiq.cli.type_registry import GlobalTypeRegistry
-from aiq.cli.type_registry import RegisteredFrontEndInfo
-from aiq.data_models.config import AIQConfig
-from aiq.utils.data_models.schema_validator import validate_schema
-from aiq.utils.type_utils import DecomposedType
+from nat.cli.cli_utils.config_override import load_and_override_config
+from nat.cli.type_registry import GlobalTypeRegistry
+from nat.cli.type_registry import RegisteredFrontEndInfo
+from nat.data_models.config import Config
+from nat.utils.data_models.schema_validator import validate_schema
+from nat.utils.type_utils import DecomposedType
 
 logger = logging.getLogger(__name__)
 
@@ -133,8 +133,8 @@ class StartCommandGroup(click.Group):
         if (self._commands is not None):
             return self._commands
 
-        from aiq.runtime.loader import PluginTypes
-        from aiq.runtime.loader import discover_and_register_plugins
+        from nat.runtime.loader import PluginTypes
+        from nat.runtime.loader import discover_and_register_plugins
 
         # Only load front ends here for performance. Ensures a responsive CLI
         discover_and_register_plugins(PluginTypes.FRONT_END)
@@ -149,7 +149,7 @@ class StartCommandGroup(click.Group):
 
             # Build the command parameters
             params: list[click.Parameter] = self._build_params(registered_front_end)
-            help_msg = f"Run an AIQ Toolkit workflow using the {registered_front_end.local_name} front end."
+            help_msg = f"Run a NAT workflow using the {registered_front_end.local_name} front end."
 
             cmd = click.Command(name=registered_front_end.local_name,
                                 params=params,
@@ -169,8 +169,8 @@ class StartCommandGroup(click.Group):
                           override: tuple[tuple[str, str], ...],
                           **kwargs) -> int | None:
 
-        from aiq.runtime.loader import PluginTypes
-        from aiq.runtime.loader import discover_and_register_plugins
+        from nat.runtime.loader import PluginTypes
+        from nat.runtime.loader import discover_and_register_plugins
 
         if (config_file is None):
             raise click.ClickException("No config file provided.")
@@ -178,14 +178,14 @@ class StartCommandGroup(click.Group):
         # Here we need to ensure all objects are loaded before we try to create the config object
         discover_and_register_plugins(PluginTypes.CONFIG_OBJECT)
 
-        logger.info("Starting AIQ Toolkit from config file: '%s'", config_file)
+        logger.info("Starting NAT from config file: '%s'", config_file)
 
         config_dict = load_and_override_config(config_file, override)
 
         # Get the front end for the command
         front_end: RegisteredFrontEndInfo = self._registered_front_ends[cmd_name]
 
-        config = validate_schema(config_dict, AIQConfig)
+        config = validate_schema(config_dict, Config)
 
         # Override default front end config with values from the config file for serverless execution modes.
         # Check that we have the right kind of front end
@@ -238,9 +238,9 @@ class StartCommandGroup(click.Group):
 
 @click.command(name=__name__,
                invoke_without_command=False,
-               help="Run an AIQ Toolkit workflow using a front end configuration.",
+               help="Run a NAT workflow using a front end configuration.",
                cls=StartCommandGroup)
 @click.pass_context
 def start_command(ctx: click.Context, **kwargs) -> None:
-    """Run an AIQ Toolkit workflow using a front end configuration."""
+    """Run a NAT workflow using a front end configuration."""
     pass

{aiq → nat}/cli/commands/uninstall.py

@@ -24,11 +24,11 @@ logger = logging.getLogger(__name__)
 
 async def uninstall_packages(packages: list[dict[str, str]]) -> None:
 
-    from aiq.cli.type_registry import GlobalTypeRegistry
-    from aiq.registry_handlers.schemas.package import PackageNameVersionList
-    from aiq.runtime.loader import PluginTypes
-    from aiq.runtime.loader import discover_and_register_plugins
-    from aiq.settings.global_settings import GlobalSettings
+    from nat.cli.type_registry import GlobalTypeRegistry
+    from nat.registry_handlers.schemas.package import PackageNameVersionList
+    from nat.runtime.loader import PluginTypes
+    from nat.runtime.loader import discover_and_register_plugins
+    from nat.settings.global_settings import GlobalSettings
 
     discover_and_register_plugins(PluginTypes.CONFIG_OBJECT)
 

{aiq → nat}/cli/commands/validate.py

@@ -28,7 +28,7 @@ def validate_command(config_file: Path):
     # load function level dependencies
     from io import StringIO
 
-    from aiq.runtime.loader import load_config
+    from nat.runtime.loader import load_config
 
     try:
         click.echo(f"Validating configuration file: {config_file}")

{aiq → nat}/cli/commands/workflow/templates/pyproject.toml.j2

@@ -9,14 +9,14 @@ root = "{{ rel_path_to_repo_root}}"{% else %}requires = ["setuptools >= 64"]{% e
 name = "{{ package_name }}"
 {% if editable %}dynamic = ["version"]{% else %}version = "0.1.0"{% endif %}
 dependencies = [
-  "aiqtoolkit[langchain]",
+  "nvidia-nat[langchain]",
 ]
 requires-python = ">=3.11,<3.13"
-description = "Custom AIQ Toolkit Workflow"
+description = "Custom NeMo Agent Toolkit Workflow"
 classifiers = ["Programming Language :: Python"]
 
 {% if editable %}[tool.uv.sources]
-aiqtoolkit = { path = "{{ rel_path_to_repo_root}}", editable = true }{% endif %}
+nvidia-nat = { path = "{{ rel_path_to_repo_root}}", editable = true }{% endif %}
 
-[project.entry-points.'aiq.components']
+[project.entry-points.'nat.components']
 {{ package_name }} = "{{ package_name }}.register"