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

{aiq → nat}/experimental/test_time_compute/search/single_shot_multi_plan_planner.py

@@ -17,16 +17,16 @@ import asyncio
 import logging
 import re
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.experimental.test_time_compute.models.search_config import SingleShotMultiPlanConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
-from aiq.utils.io.model_processing import remove_r1_think_tags
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.experimental.test_time_compute.models.search_config import SingleShotMultiPlanConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.utils.io.model_processing import remove_r1_think_tags
 
 logger = logging.getLogger(__name__)
 
@@ -64,7 +64,7 @@ class SingleShotMultiPlanPlanner(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is not installed. Please install it to use SingleShotMultiPlanPlanner.\n"
-                              "This error can be resolve by installing aiqtoolkit-langchain.")
+                              "This error can be resolve by installing nvidia-nat-langchain.")
 
         planning_template = PromptTemplate(template=self.config.planning_template,
                                            input_variables=["context", "prompt"],

{aiq → nat}/experimental/test_time_compute/selection/best_of_n_selector.py

@@ -15,13 +15,13 @@
 
 import logging
 
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.experimental.test_time_compute.models.selection_config import BestOfNSelectionConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.experimental.test_time_compute.models.selection_config import BestOfNSelectionConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/experimental/test_time_compute/selection/llm_based_agent_output_selector.py

@@ -16,16 +16,16 @@
 import logging
 import re
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.experimental.test_time_compute.models.selection_config import LLMBasedAgentOutputSelectionConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
-from aiq.utils.io.model_processing import remove_r1_think_tags
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.experimental.test_time_compute.models.selection_config import LLMBasedAgentOutputSelectionConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.utils.io.model_processing import remove_r1_think_tags
 
 logger = logging.getLogger(__name__)
 
@@ -70,7 +70,7 @@ class LLMBasedAgentOutputSelector(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is not installed. Please install it to use SingleShotMultiPlanPlanner.\n"
-                              "This error can be resolved by installing aiqtoolkit-langchain.")
+                              "This error can be resolved by installing nvidia-nat-langchain.")
 
         from pydantic import BaseModel
 

{aiq → nat}/experimental/test_time_compute/selection/llm_based_output_merging_selector.py

@@ -15,16 +15,16 @@
 
 import logging
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.experimental.test_time_compute.models.selection_config import LLMBasedOutputMergingConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
-from aiq.utils.io.model_processing import remove_r1_think_tags
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.experimental.test_time_compute.models.selection_config import LLMBasedOutputMergingConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.utils.io.model_processing import remove_r1_think_tags
 
 logger = logging.getLogger(__name__)
 
@@ -69,7 +69,7 @@ class LLMBasedOutputMergingSelector(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is not installed. Please install it to use SingleShotMultiPlanPlanner.\n"
-                              "This error can be resolved by installing aiqtoolkit-langchain.")
+                              "This error can be resolved by installing nvidia-nat-langchain.")
 
         from typing import Callable
 

{aiq → nat}/experimental/test_time_compute/selection/llm_based_plan_selector.py

@@ -16,16 +16,16 @@
 import logging
 import re
 
-from aiq.builder.builder import Builder
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.experimental.test_time_compute.models.selection_config import LLMBasedPlanSelectionConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
-from aiq.utils.io.model_processing import remove_r1_think_tags
+from nat.builder.builder import Builder
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.experimental.test_time_compute.models.selection_config import LLMBasedPlanSelectionConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.utils.io.model_processing import remove_r1_think_tags
 
 logger = logging.getLogger(__name__)
 
@@ -70,7 +70,7 @@ class LLMBasedPlanSelector(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is not installed. Please install it to use SingleShotMultiPlanPlanner.\n"
-                              "This error can be resolved by installing aiqtoolkit-langchain.")
+                              "This error can be resolved by installing nvidia-nat-langchain.")
 
         if not isinstance(self.llm_bound, BaseChatModel):
             raise ValueError("The `selection_llm` must be an instance of `BaseChatModel`.")

{aiq → nat}/experimental/test_time_compute/selection/threshold_selector.py

@@ -15,13 +15,13 @@
 
 import logging
 
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_ttc_strategy
-from aiq.experimental.test_time_compute.models.selection_config import ThresholdSelectionConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.experimental.test_time_compute.models.strategy_base import StrategyBase
-from aiq.experimental.test_time_compute.models.ttc_item import TTCItem
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_ttc_strategy
+from nat.experimental.test_time_compute.models.selection_config import ThresholdSelectionConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.experimental.test_time_compute.models.strategy_base import StrategyBase
+from nat.experimental.test_time_compute.models.ttc_item import TTCItem
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/front_ends/console/authentication_flow_handler.py

@@ -25,12 +25,12 @@ from authlib.integrations.httpx_client import AsyncOAuth2Client
 from fastapi import FastAPI
 from fastapi import Request
 
-from aiq.authentication.interfaces import FlowHandlerBase
-from aiq.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
-from aiq.data_models.authentication import AuthenticatedContext
-from aiq.data_models.authentication import AuthFlowType
-from aiq.data_models.authentication import AuthProviderBaseConfig
-from aiq.front_ends.fastapi.fastapi_front_end_controller import _FastApiFrontEndController
+from nat.authentication.interfaces import FlowHandlerBase
+from nat.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
+from nat.data_models.authentication import AuthenticatedContext
+from nat.data_models.authentication import AuthFlowType
+from nat.data_models.authentication import AuthProviderBaseConfig
+from nat.front_ends.fastapi.fastapi_front_end_controller import _FastApiFrontEndController
 
 
 # --------------------------------------------------------------------------- #

{aiq → nat}/front_ends/console/console_front_end_config.py

@@ -17,12 +17,12 @@ from pathlib import Path
 
 from pydantic import Field
 
-from aiq.data_models.front_end import FrontEndBaseConfig
+from nat.data_models.front_end import FrontEndBaseConfig
 
 
 class ConsoleFrontEndConfig(FrontEndBaseConfig, name="console"):
     """
-    A front end that allows an AIQ Toolkit workflow to be run from the console.
+    A front end that allows a NAT workflow to be run from the console.
     """
 
     input_query: list[str] | None = Field(default=None,

{aiq → nat}/front_ends/console/console_front_end_plugin.py

@@ -19,14 +19,14 @@ import logging
 import click
 from colorama import Fore
 
-from aiq.data_models.interactive import HumanPromptModelType
-from aiq.data_models.interactive import HumanResponse
-from aiq.data_models.interactive import HumanResponseText
-from aiq.data_models.interactive import InteractionPrompt
-from aiq.front_ends.console.authentication_flow_handler import ConsoleAuthenticationFlowHandler
-from aiq.front_ends.console.console_front_end_config import ConsoleFrontEndConfig
-from aiq.front_ends.simple_base.simple_front_end_plugin_base import SimpleFrontEndPluginBase
-from aiq.runtime.session import AIQSessionManager
+from nat.data_models.interactive import HumanPromptModelType
+from nat.data_models.interactive import HumanResponse
+from nat.data_models.interactive import HumanResponseText
+from nat.data_models.interactive import InteractionPrompt
+from nat.front_ends.console.authentication_flow_handler import ConsoleAuthenticationFlowHandler
+from nat.front_ends.console.console_front_end_config import ConsoleFrontEndConfig
+from nat.front_ends.simple_base.simple_front_end_plugin_base import SimpleFrontEndPluginBase
+from nat.runtime.session import SessionManager
 
 logger = logging.getLogger(__name__)
 
@@ -59,7 +59,7 @@ class ConsoleFrontEndPlugin(SimpleFrontEndPluginBase[ConsoleFrontEndConfig]):
         if (not self.front_end_config.input_query and not self.front_end_config.input_file):
             raise click.UsageError("Must specify either --input_query or --input_file")
 
-    async def run_workflow(self, session_manager: AIQSessionManager):
+    async def run_workflow(self, session_manager: SessionManager):
 
         assert session_manager is not None, "Session manager must be provided"
         runner_outputs = None

{aiq → nat}/front_ends/console/register.py

@@ -13,13 +13,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.cli.register_workflow import register_front_end
-from aiq.data_models.config import AIQConfig
-from aiq.front_ends.console.console_front_end_config import ConsoleFrontEndConfig
+from nat.cli.register_workflow import register_front_end
+from nat.data_models.config import Config
+from nat.front_ends.console.console_front_end_config import ConsoleFrontEndConfig
 
 
 @register_front_end(config_type=ConsoleFrontEndConfig)
-async def register_fastapi_front_end(config: ConsoleFrontEndConfig, full_config: AIQConfig):
-    from aiq.front_ends.console.console_front_end_plugin import ConsoleFrontEndPlugin
+async def register_fastapi_front_end(config: ConsoleFrontEndConfig, full_config: Config):
+    from nat.front_ends.console.console_front_end_plugin import ConsoleFrontEndPlugin
 
     yield ConsoleFrontEndPlugin(full_config=full_config)

{aiq → nat}/front_ends/fastapi/auth_flow_handlers/http_flow_handler.py

@@ -13,10 +13,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.authentication.interfaces import FlowHandlerBase
-from aiq.data_models.authentication import AuthenticatedContext
-from aiq.data_models.authentication import AuthFlowType
-from aiq.data_models.authentication import AuthProviderBaseConfig
+from nat.authentication.interfaces import FlowHandlerBase
+from nat.data_models.authentication import AuthenticatedContext
+from nat.data_models.authentication import AuthFlowType
+from nat.data_models.authentication import AuthProviderBaseConfig
 
 
 class HTTPAuthenticationFlowHandler(FlowHandlerBase):

{aiq → nat}/front_ends/fastapi/auth_flow_handlers/websocket_flow_handler.py

@@ -24,12 +24,12 @@ from dataclasses import field
 import pkce
 from authlib.integrations.httpx_client import AsyncOAuth2Client
 
-from aiq.authentication.interfaces import FlowHandlerBase
-from aiq.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
-from aiq.data_models.authentication import AuthenticatedContext
-from aiq.data_models.authentication import AuthFlowType
-from aiq.data_models.interactive import _HumanPromptOAuthConsent
-from aiq.front_ends.fastapi.message_handler import WebSocketMessageHandler
+from nat.authentication.interfaces import FlowHandlerBase
+from nat.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
+from nat.data_models.authentication import AuthenticatedContext
+from nat.data_models.authentication import AuthFlowType
+from nat.data_models.interactive import _HumanPromptOAuthConsent
+from nat.front_ends.fastapi.message_handler import WebSocketMessageHandler
 
 logger = logging.getLogger(__name__)
 

{aiq → nat}/front_ends/fastapi/fastapi_front_end_config.py

@@ -22,16 +22,16 @@ from pydantic import BaseModel
 from pydantic import Field
 from pydantic import field_validator
 
-from aiq.data_models.component_ref import ObjectStoreRef
-from aiq.data_models.front_end import FrontEndBaseConfig
-from aiq.data_models.step_adaptor import StepAdaptorConfig
+from nat.data_models.component_ref import ObjectStoreRef
+from nat.data_models.front_end import FrontEndBaseConfig
+from nat.data_models.step_adaptor import StepAdaptorConfig
 
 logger = logging.getLogger(__name__)
 
 YAML_EXTENSIONS = (".yaml", ".yml")
 
 
-class AIQEvaluateRequest(BaseModel):
+class EvaluateRequest(BaseModel):
     """Request model for the evaluate endpoint."""
     config_file: str = Field(description="Path to the configuration file for evaluation")
     job_id: str | None = Field(default=None, description="Unique identifier for the evaluation job")
@@ -84,12 +84,12 @@ class BaseAsyncResponse(BaseModel):
     status: str = Field(description="Current status of the job")
 
 
-class AIQEvaluateResponse(BaseAsyncResponse):
+class EvaluateResponse(BaseAsyncResponse):
     """Response model for the evaluate endpoint."""
     pass
 
 
-class AIQAsyncGenerateResponse(BaseAsyncResponse):
+class AsyncGenerateResponse(BaseAsyncResponse):
     """Response model for the async generation endpoint."""
     pass
 
@@ -104,14 +104,14 @@ class BaseAsyncStatusResponse(BaseModel):
     expires_at: datetime | None = Field(default=None, description="Timestamp when the job will expire")
 
 
-class AIQEvaluateStatusResponse(BaseAsyncStatusResponse):
+class EvaluateStatusResponse(BaseAsyncStatusResponse):
     """Response model for the evaluate status endpoint."""
     config_file: str = Field(description="Path to the configuration file used for evaluation")
     output_path: str | None = Field(default=None,
                                     description="Path to the output file if the job completed successfully")
 
 
-class AIQAsyncGenerationStatusResponse(BaseAsyncStatusResponse):
+class AsyncGenerationStatusResponse(BaseAsyncStatusResponse):
     output: dict | None = Field(
         default=None,
         description="Output of the generate request, this is only available if the job completed successfully.")
@@ -119,7 +119,7 @@ class AIQAsyncGenerationStatusResponse(BaseAsyncStatusResponse):
 
 class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
     """
-    A FastAPI based front end that allows an AIQ Toolkit workflow to be served as a microservice.
+    A FastAPI based front end that allows a NAT workflow to be served as a microservice.
     """
 
     class EndpointBase(BaseModel):
@@ -192,7 +192,7 @@ class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
         websocket_path="/websocket",
         openai_api_path="/chat",
         openai_api_v1_path="/v1/chat/completions",
-        description="Executes the default AIQ Toolkit workflow from the loaded configuration ",
+        description="Executes the default NAT workflow from the loaded configuration ",
     )
 
     evaluate: typing.Annotated[EndpointBase, Field(description="Endpoint for evaluating workflows.")] = EndpointBase(
@@ -207,9 +207,8 @@ class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
 
     endpoints: list[Endpoint] = Field(
         default_factory=list,
-        description=(
-            "Additional endpoints to add to the FastAPI app which run functions within the AIQ Toolkit configuration. "
-            "Each endpoint must have a unique path."))
+        description=("Additional endpoints to add to the FastAPI app which run functions within the NAT configuration. "
+                     "Each endpoint must have a unique path."))
 
     cors: CrossOriginResourceSharing = Field(
         default_factory=CrossOriginResourceSharing,
@@ -221,8 +220,8 @@ class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
     )
     runner_class: str | None = Field(
         default=None,
-        description=("The AIQ Toolkit runner class to use when launching the FastAPI app from multiple processes. "
-                     "Each runner is responsible for loading and running the AIQ Toolkit workflow. "
+        description=("The NAT runner class to use when launching the FastAPI app from multiple processes. "
+                     "Each runner is responsible for loading and running the NAT workflow. "
                      "Note: This is different from the worker class used by Gunicorn."),
     )
 
@@ -232,3 +231,11 @@ class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
             "Object store reference for the FastAPI app. If present, static files can be uploaded via a POST "
             "request to '/static' and files will be served from the object store. The files will be served from the "
             "object store at '/static/{file_name}'."))
+
+
+# Compatibility aliases with previous releases
+AIQEvaluateRequest = EvaluateRequest
+AIQEvaluateResponse = EvaluateResponse
+AIQAsyncGenerateResponse = AsyncGenerateResponse
+AIQEvaluateStatusResponse = EvaluateStatusResponse
+AIQAsyncGenerationStatusResponse = AsyncGenerationStatusResponse

{aiq → nat}/front_ends/fastapi/fastapi_front_end_plugin.py

@@ -18,11 +18,11 @@ import os
 import tempfile
 import typing
 
-from aiq.builder.front_end import FrontEndBase
-from aiq.front_ends.fastapi.fastapi_front_end_config import FastApiFrontEndConfig
-from aiq.front_ends.fastapi.fastapi_front_end_plugin_worker import FastApiFrontEndPluginWorkerBase
-from aiq.front_ends.fastapi.main import get_app
-from aiq.utils.io.yaml_tools import yaml_dump
+from nat.builder.front_end import FrontEndBase
+from nat.front_ends.fastapi.fastapi_front_end_config import FastApiFrontEndConfig
+from nat.front_ends.fastapi.fastapi_front_end_plugin_worker import FastApiFrontEndPluginWorkerBase
+from nat.front_ends.fastapi.main import get_app
+from nat.utils.io.yaml_tools import yaml_dump
 
 logger = logging.getLogger(__name__)
 
@@ -30,7 +30,7 @@ logger = logging.getLogger(__name__)
 class FastApiFrontEndPlugin(FrontEndBase[FastApiFrontEndConfig]):
 
     def get_worker_class(self) -> type[FastApiFrontEndPluginWorkerBase]:
-        from aiq.front_ends.fastapi.fastapi_front_end_plugin_worker import FastApiFrontEndPluginWorker
+        from nat.front_ends.fastapi.fastapi_front_end_plugin_worker import FastApiFrontEndPluginWorker
 
         return FastApiFrontEndPluginWorker
 
@@ -47,7 +47,7 @@ class FastApiFrontEndPlugin(FrontEndBase[FastApiFrontEndConfig]):
     async def run(self):
 
         # Write the entire config to a temporary file
-        with tempfile.NamedTemporaryFile(mode="w", prefix="aiq_config", suffix=".yml", delete=False) as config_file:
+        with tempfile.NamedTemporaryFile(mode="w", prefix="nat_config", suffix=".yml", delete=False) as config_file:
 
             # Get as dict
             config_dict = self.full_config.model_dump(mode="json", by_alias=True, round_trip=True)
@@ -59,10 +59,10 @@ class FastApiFrontEndPlugin(FrontEndBase[FastApiFrontEndConfig]):
             config_file_name = config_file.name
 
             # Set the config file in the environment
-            os.environ["AIQ_CONFIG_FILE"] = str(config_file.name)
+            os.environ["NAT_CONFIG_FILE"] = str(config_file.name)
 
             # Set the worker class in the environment
-            os.environ["AIQ_FRONT_END_WORKER"] = self.get_worker_class_name()
+            os.environ["NAT_FRONT_END_WORKER"] = self.get_worker_class_name()
 
         try:
             if not self.front_end_config.use_gunicorn:
@@ -70,7 +70,7 @@ class FastApiFrontEndPlugin(FrontEndBase[FastApiFrontEndConfig]):
 
                 reload_excludes = ["./.*"]
 
-                uvicorn.run("aiq.front_ends.fastapi.main:get_app",
+                uvicorn.run("nat.front_ends.fastapi.main:get_app",
                             host=self.front_end_config.host,
                             port=self.front_end_config.port,
                             workers=self.front_end_config.workers,