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

nat/data_models/discovery_metadata.py

@@ -15,12 +15,10 @@
 
 import importlib.metadata
 import inspect
-import json
 import logging
 import typing
 from enum import Enum
 from functools import lru_cache
-from pathlib import Path
 from types import ModuleType
 from typing import TYPE_CHECKING
 
@@ -56,11 +54,11 @@ class DiscoveryMetadata(BaseModel):
     """A data model representing metadata about each registered component to faciliate its discovery.
 
     Args:
-        package (str): The name of the package containing the AIQ Toolkit component.
-        version (str): The version number of the package containing the AIQ Toolkit component.
-        component_type (AIQComponentEnum): The type of AIQ Toolkit component this metadata represents.
-        component_name (str): The registered name of the AIQ Toolkit component.
-        description (str): Description of the AIQ Toolkit component pulled from its config objects docstrings.
+        package (str): The name of the package containing the NAT component.
+        version (str): The version number of the package containing the NAT component.
+        component_type (ComponentEnum): The type of NAT component this metadata represents.
+        component_name (str): The registered name of the NAT component.
+        description (str): Description of the NAT component pulled from its config objects docstrings.
         developer_notes (str): Other notes to a developers to aid in the use of the component.
         status (DiscoveryStatusEnum): Provides the status of the metadata discovery process.
     """
@@ -95,27 +93,12 @@ class DiscoveryMetadata(BaseModel):
         mapping = importlib.metadata.packages_distributions()
         try:
             distro_names = mapping.get(root_package_name, [None])
-            distro_name = DiscoveryMetadata.get_preferred_item(distro_names, "aiqtoolkit")
+            distro_name = DiscoveryMetadata.get_preferred_item(distro_names, "nvidia-nat")
         except KeyError:
             return root_package_name
 
         return distro_name if distro_name else root_package_name
 
-    @staticmethod
-    @lru_cache
-    def get_distribution_name_from_private_data(root_package: str) -> str | None:
-        # Locate distibution mapping stored in the packages private data
-        module = __import__(root_package)
-        for path in module.__path__:
-            package_dir = Path(path).resolve()
-            distinfo_path = package_dir / "meta" / "module_to_distro.json"
-
-            if distinfo_path.exists():
-                with distinfo_path.open("r") as f:
-                    data = json.load(f)
-                    return data.get(root_package, None)
-        return None
-
     @staticmethod
     @lru_cache
     def get_distribution_name_from_module(module: ModuleType | None) -> str:
@@ -125,22 +108,22 @@ class DiscoveryMetadata(BaseModel):
             module (ModuleType): A registered component's module.
 
         Returns:
-            str: The distribution name of the AIQ Toolkit component.
+            str: The distribution name of the NAT component.
         """
         from nat.runtime.loader import get_all_entrypoints_distro_mapping
 
         if module is None:
-            return "aiqtoolkit"
+            return "nvidia-nat"
 
         # Get the mapping of module names to distro names
         mapping = get_all_entrypoints_distro_mapping()
         module_package = module.__package__
 
         if module_package is None:
-            return "aiqtoolkit"
+            return "nvidia-nat"
 
         # Traverse the module package parts in reverse order to find the distro name
-        # This is because the module package is the root package for the AIQ Toolkit component
+        # This is because the module package is the root package for the NAT component
         # and the distro name is the name of the package that contains the component
         module_package_parts = module_package.split(".")
         for part_idx in range(len(module_package_parts), 0, -1):
@@ -149,7 +132,7 @@ class DiscoveryMetadata(BaseModel):
             if candidate_distro_name is not None:
                 return candidate_distro_name
 
-        return "aiqtoolkit"
+        return "nvidia-nat"
 
     @staticmethod
     @lru_cache
@@ -160,32 +143,20 @@ class DiscoveryMetadata(BaseModel):
             config_type (type[TypedBaseModelT]): A registered component's configuration object.
 
         Returns:
-            str: The distribution name of the AIQ Toolkit component.
+            str: The distribution name of the NAT component.
         """
         module = inspect.getmodule(config_type)
         return DiscoveryMetadata.get_distribution_name_from_module(module)
 
-    @staticmethod
-    @lru_cache
-    def get_distribution_name(root_package: str) -> str:
-        """
-        The aiq library packages use a distro name 'aiqtoolkit[]' and
-        root package name 'aiq'. They provide mapping in a metadata file
-        for optimized installation.
-        """
-
-        distro_name = DiscoveryMetadata.get_distribution_name_from_private_data(root_package)
-        return distro_name if distro_name else root_package
-
     @staticmethod
     def from_config_type(config_type: type["TypedBaseModelT"],
                          component_type: ComponentEnum = ComponentEnum.UNDEFINED) -> "DiscoveryMetadata":
-        """Generates discovery metadata from an AIQ Toolkit config object.
+        """Generates discovery metadata from a NAT config object.
 
         Args:
             config_type (type[TypedBaseModelT]): A registered component's configuration object.
-            component_type (AIQComponentEnum, optional): The type of the registered component. Defaults to
-            AIQComponentEnum.UNDEFINED.
+            component_type (ComponentEnum, optional): The type of the registered component. Defaults to
+            ComponentEnum.UNDEFINED.
 
         Returns:
             DiscoveryMetadata: A an object containing component metadata to facilitate discovery and reuse.
@@ -228,8 +199,8 @@ class DiscoveryMetadata(BaseModel):
             wrapper_type (LLMFrameworkEnum): The wrapper to apply to the callable to faciliate inter-framwork
             interoperability.
 
-            component_type (AIQComponentEnum, optional): The type of the registered component. Defaults to
-            AIQComponentEnum.TOOL_WRAPPER.
+            component_type (ComponentEnum, optional): The type of the registered component. Defaults to
+            ComponentEnum.TOOL_WRAPPER.
 
         Returns:
             DiscoveryMetadata: A an object containing component metadata to facilitate discovery and reuse.
@@ -263,7 +234,7 @@ class DiscoveryMetadata(BaseModel):
         """Generates discovery metadata from an installed package name.
 
         Args:
-            package_name (str): The name of the AIQ Toolkit plugin package containing registered components.
+            package_name (str): The name of the NAT plugin package containing registered components.
             package_version (str, optional): The version of the package, Defaults to None.
 
         Returns:
@@ -302,9 +273,9 @@ class DiscoveryMetadata(BaseModel):
             wrapper_type (LLMFrameworkEnum | str): The wrapper to apply to the callable to faciliate inter-framwork
             interoperability.
 
-            provider_type (AIQComponentEnum): The type of provider the registered component supports.
-            component_type (AIQComponentEnum, optional): The type of the registered component. Defaults to
-            AIQComponentEnum.UNDEFINED.
+            provider_type (ComponentEnum): The type of provider the registered component supports.
+            component_type (ComponentEnum, optional): The type of the registered component. Defaults to
+            ComponentEnum.UNDEFINED.
 
         Returns:
             DiscoveryMetadata: A an object containing component metadata to facilitate discovery and reuse.

nat/data_models/evaluate.py

@@ -54,7 +54,7 @@ class JobManagementConfig(BaseModel):
 
 class EvalOutputConfig(BaseModel):
     # Output directory for the workflow and evaluation results
-    dir: Path = Path("/tmp/aiq/examples/default/")
+    dir: Path = Path("./.tmp/nat/examples/default/")
     # S3 prefix for the workflow and evaluation results
     remote_dir: str | None = None
     # Custom scripts to run after the workflow and evaluation results are saved
@@ -77,7 +77,7 @@ class EvalGeneralConfig(BaseModel):
     workflow_alias: str | None = None
 
     # Output directory for the workflow and evaluation results
-    output_dir: Path = Path("/tmp/aiq/examples/default/")
+    output_dir: Path = Path("./.tmp/nat/examples/default/")
 
     # If present overrides output_dir
     output: EvalOutputConfig | None = None

nat/data_models/intermediate_step.py

@@ -66,7 +66,7 @@ class IntermediateStepState(str, Enum):
 
 class StreamEventData(BaseModel):
     """
-    AIQStreamEventData is a data model that represents the data field in an streaming event.
+    StreamEventData is a data model that represents the data field in an streaming event.
     """
 
     # Allow extra fields in the model_config to support derived models
@@ -121,7 +121,7 @@ class TraceMetadata(BaseModel):
 
 class IntermediateStepPayload(BaseModel):
     """
-    AIQIntermediateStep is a data model that represents an intermediate step in the AIQ Toolkit. Intermediate steps are
+    IntermediateStep is a data model that represents an intermediate step in the NAT. Intermediate steps are
     captured while a request is running and can be used to show progress or to evaluate the path a workflow took to get
     a response.
     """
@@ -226,7 +226,7 @@ class IntermediateStepPayload(BaseModel):
 
 class IntermediateStep(BaseModel):
     """
-    AIQIntermediateStep is a data model that represents an intermediate step in the AIQ Toolkit. Intermediate steps are
+    IntermediateStep is a data model that represents an intermediate step in the NAT. Intermediate steps are
     captured while a request is running and can be used to show progress or to evaluate the path a workflow took to get
     a response.
     """
@@ -243,7 +243,7 @@ class IntermediateStep(BaseModel):
 
     function_ancestry: InvocationNode
     """
-    The function ancestry for the current step showing the current AIQ function that was being executed when the step
+    The function ancestry for the current step showing the current NAT function that was being executed when the step
     was created.
     """
 

nat/embedder/register.py

@@ -20,5 +20,3 @@
 # Import any providers which need to be automatically registered here
 from . import nim_embedder
 from . import openai_embedder
-# Import any clients which need to be automatically registered here
-from . import langchain_client

nat/eval/dataset_handler/dataset_handler.py

@@ -15,10 +15,12 @@
 
 import json
 import math
+from pathlib import Path
 
 import pandas as pd
 
 from nat.data_models.dataset_handler import EvalDatasetConfig
+from nat.data_models.dataset_handler import EvalDatasetCustomConfig
 from nat.data_models.dataset_handler import EvalDatasetJsonConfig
 from nat.data_models.intermediate_step import IntermediateStep
 from nat.data_models.intermediate_step import IntermediateStepType
@@ -38,7 +40,7 @@ class DatasetHandler:
                  dataset_config: EvalDatasetConfig,
                  reps: int,
                  concurrency: int,
-                 num_passes: int | None = None,
+                 num_passes: int = 1,
                  adjust_dataset_size: bool = False):
         from nat.eval.intermediate_step_adapter import IntermediateStepAdapter
 
@@ -184,6 +186,10 @@ class DatasetHandler:
         # if a dataset file has been provided in the command line, use that
         dataset_config = EvalDatasetJsonConfig(file_path=dataset) if dataset else self.dataset_config
 
+        # Handle custom dataset type with special processing
+        if isinstance(self.dataset_config, EvalDatasetCustomConfig):
+            return self._handle_custom_dataset(dataset)
+
         # Download the dataset if it is remote
         downloader = DatasetDownloader(dataset_config=dataset_config)
         downloader.download_dataset()
@@ -192,6 +198,19 @@ class DatasetHandler:
         # Parse the dataset into a DataFrame
         input_df = parser(dataset_config.file_path, **kwargs)
 
+        # Apply standard preprocessing and convert to EvalInput
+        return self._preprocess_eval_dataframe(input_df)
+
+    def _preprocess_dataframe(self, input_df: pd.DataFrame) -> pd.DataFrame:
+        """
+        Apply standard preprocessing to a DataFrame: filters, deduplication, repetitions, and size adjustment.
+
+        Args:
+            input_df: DataFrame to preprocess
+
+        Returns:
+            Preprocessed DataFrame
+        """
         # Apply filters and deduplicate
         input_df = self.dataset_filter.apply_filters(input_df)
         input_df.drop_duplicates(subset=[self.dataset_config.id_key], inplace=True)
@@ -205,12 +224,104 @@ class DatasetHandler:
         elif self.adjust_dataset_size:
             input_df = self.adjust_dataset(input_df)
 
-        # Convert the DataFrame to a list of EvalInput objects
-        return self.get_eval_input_from_df(input_df)
+        return input_df
+
+    def _preprocess_eval_dataframe(self, input_df: pd.DataFrame) -> EvalInput:
+        """
+        Apply standard preprocessing to a DataFrame and convert to EvalInput.
+
+        Args:
+            input_df: DataFrame to preprocess
+
+        Returns:
+            Preprocessed EvalInput object
+        """
+        processed_df = self._preprocess_dataframe(input_df)
+        return self.get_eval_input_from_df(processed_df)
+
+    def _preprocess_eval_input(self, eval_input: EvalInput) -> EvalInput:
+        """
+        Apply standard preprocessing to an EvalInput object.
+
+        Thin wrapper that converts EvalInput to DataFrame, processes it, and converts back.
+
+        Args:
+            eval_input: EvalInput object to preprocess
+
+        Returns:
+            Preprocessed EvalInput object
+        """
+        if not eval_input.eval_input_items:
+            return eval_input
+
+        input_df = self._eval_input_to_dataframe(eval_input)
+        return self._preprocess_eval_dataframe(input_df)
+
+    def _handle_custom_dataset(self, dataset: str | None) -> EvalInput:
+        """
+        Handle custom dataset type by calling the user-defined function
+        and applying standard preprocessing to the result.
+
+        Args:
+            dataset: Optional dataset file path from command line
+
+        Returns:
+            Preprocessed EvalInput object
+        """
+        # Determine input path - use command line dataset or config file_path
+        input_path = Path(dataset) if dataset else Path(self.dataset_config.file_path)
+
+        # Download the dataset if it is remote (for custom datasets too)
+        downloader = DatasetDownloader(dataset_config=self.dataset_config)
+        downloader.download_dataset()
+
+        # Load and call custom function
+        custom_function, kwargs = self.dataset_config.parser()
+
+        try:
+            # Call the custom function with file_path and kwargs
+            eval_input = custom_function(file_path=input_path, **kwargs)
+
+            if not isinstance(eval_input, EvalInput):
+                raise ValueError(f"Custom function must return an EvalInput object, "
+                                 f"but returned {type(eval_input)}")
+
+        except Exception as e:
+            raise RuntimeError(f"Error calling custom dataset function: {e}") from e
+
+        # Apply standard preprocessing (filters, deduplication, repetitions)
+        return self._preprocess_eval_input(eval_input)
+
+    def _eval_input_to_dataframe(self, eval_input: EvalInput) -> pd.DataFrame:
+        """
+        Convert an EvalInput object to a pandas DataFrame for processing.
+
+        Args:
+            eval_input: EvalInput object to convert
+
+        Returns:
+            DataFrame representation of the EvalInput
+        """
+        data = []
+        for item in eval_input.eval_input_items:
+            row = item.full_dataset_entry.copy() if item.full_dataset_entry else {}
+
+            # Ensure key fields are present
+            row[self.id_key] = item.id
+            if self.is_structured_input():
+                row[self.question_key] = item.input_obj
+                row[self.answer_key] = item.expected_output_obj
+                row[self.generated_answer_key] = item.output_obj
+                row[self.trajectory_key] = item.trajectory
+                row[self.expected_trajectory_key] = item.expected_trajectory
+
+            data.append(row)
+
+        return pd.DataFrame(data)
 
     def filter_intermediate_steps(self,
                                   intermediate_steps: list[IntermediateStep],
-                                  event_filter: list[IntermediateStepType] = None) -> list[dict]:
+                                  event_filter: list[IntermediateStepType] | None = None) -> list[dict]:
         """
         Filter out the intermediate steps that are not relevant for evaluation.
         The output is written with with the intention of re-running the evaluation using the original config file.
@@ -220,7 +331,9 @@ class DatasetHandler:
         filtered_steps = self.intermediate_step_adapter.filter_intermediate_steps(intermediate_steps, event_filter)
         return self.intermediate_step_adapter.serialize_intermediate_steps(filtered_steps)
 
-    def publish_eval_input(self, eval_input, workflow_output_step_filter: list[IntermediateStepType] = None) -> str:
+    def publish_eval_input(self,
+                           eval_input,
+                           workflow_output_step_filter: list[IntermediateStepType] | None = None) -> str:
         """
         Convert the EvalInput object to a JSON output for storing in a file. Use the orginal keys to
         allow re-running evaluation using the orignal config file and '--skip_workflow' option.

nat/eval/evaluate.py

@@ -397,7 +397,7 @@ class EvaluationRun:  # pylint: disable=too-many-public-methods
             return self.eval_config.general.workflow_alias
 
         if not workflow_type or workflow_type == "EmptyFunctionConfig":
-            return "aiqtoolkit-eval"
+            return "nat-eval"
 
         return workflow_type
 

nat/eval/evaluator/evaluator_model.py

@@ -24,9 +24,9 @@ class EvalInputItem(BaseModel):
     id: typing.Any
     input_obj: typing.Any
     expected_output_obj: typing.Any
-    output_obj: typing.Any
-    expected_trajectory: list[IntermediateStep]
-    trajectory: list[IntermediateStep]
+    output_obj: typing.Any = None  # populated by the workflow
+    expected_trajectory: list[IntermediateStep] = []
+    trajectory: list[IntermediateStep] = []  # populated by the workflow
     full_dataset_entry: typing.Any
 
 

nat/eval/rag_evaluator/evaluate.py

@@ -99,7 +99,7 @@ class RAGEvaluator:
         return EvaluationDataset(samples=samples)
 
     def ragas_to_eval_output(self, eval_input: EvalInput, results_dataset: EvaluationResult | None) -> EvalOutput:
-        """Converts the ragas EvaluationResult to aiq EvalOutput"""
+        """Converts the ragas EvaluationResult to nat EvalOutput"""
 
         if not results_dataset:
             logger.error("Ragas evaluation failed with no results")

nat/eval/swe_bench_evaluator/evaluate.py

@@ -123,7 +123,7 @@ class SweBenchEvaluator:
                             for s in swebench_inputs if s not in supported_inputs})
 
         # Write SWEBenchInput to file
-        workflow_input_file = self.output_dir / "aiq_workflow_input.json"
+        workflow_input_file = self.output_dir / "nat_workflow_input.json"
         workflow_input_file.parent.mkdir(parents=True, exist_ok=True)
         Path(workflow_input_file).write_text(json.dumps([swebench.model_dump() for swebench in supported_inputs],
                                                         indent=2),
@@ -139,7 +139,7 @@ class SweBenchEvaluator:
             return None, None
 
         # Write SWEBenchOutput to file
-        workflow_output_file = self.output_dir / "aiq_workflow_output.json"
+        workflow_output_file = self.output_dir / "nat_workflow_output.json"
         Path(workflow_output_file).write_text(json.dumps([output.model_dump() for output in filtered_outputs],
                                                          indent=2),
                                               encoding="utf-8")

nat/experimental/test_time_compute/editing/motivation_aware_summarization.py

@@ -65,7 +65,7 @@ class MotivationAwareSummarization(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is required for MultiQueryRetrievalSearch. "
-                              "Install aiqtoolkit-langchain or similar.")
+                              "Install nvidia-nat-langchain or similar.")
 
         new_ttc_items: list[TTCItem] = []
 

nat/experimental/test_time_compute/functions/plan_select_execute_function.py

@@ -35,7 +35,7 @@ logger = logging.getLogger(__name__)
 
 class PlanSelectExecuteFunctionConfig(FunctionBaseConfig, name="plan_select_execute_function"):
     """
-    Defines an aiqtoolkit function that performs reasoning on the input data.
+    Defines a NAT function that performs reasoning on the input data.
     Output is passed to the next function in the workflow.
 
     Designed to be used with an InterceptingFunction.
@@ -83,7 +83,7 @@ async def plan_select_execute_function(config: PlanSelectExecuteFunctionConfig,
         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.")
 
     # Get the augmented function's description
     augmented_function = builder.get_function(config.augmented_fn)
@@ -159,7 +159,7 @@ async def plan_select_execute_function(config: PlanSelectExecuteFunctionConfig,
             Perform reasoning on the input text.
 
             Args:
-                input_message (AIQChatRequest): The input text to reason on.
+                input_message (ChatRequest): The input text to reason on.
             """
 
             input_text = "".join([str(message.model_dump()) + "\n" for message in input_message.messages])
@@ -192,7 +192,7 @@ async def plan_select_execute_function(config: PlanSelectExecuteFunctionConfig,
             Perform reasoning on the input text.
 
             Args:
-                input_message (AIQChatRequest): The input text to reason on.
+                input_message (ChatRequest): The input text to reason on.
             """
 
             input_text = "".join([str(message.model_dump()) + "\n" for message in input_message.messages])

nat/experimental/test_time_compute/functions/ttc_tool_wrapper_function.py

@@ -78,7 +78,7 @@ async def register_ttc_tool_wrapper_function(
         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.")
 
     augmented_function: Function = builder.get_function(config.augmented_fn)
     input_llm: BaseChatModel = await builder.get_llm(config.input_llm, wrapper_type=LLMFrameworkEnum.LANGCHAIN)

nat/experimental/test_time_compute/scoring/llm_based_agent_scorer.py

@@ -67,7 +67,7 @@ class LLMBasedAgentScorer(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 `scoring_llm` must be an instance of `BaseChatModel`.")

nat/experimental/test_time_compute/scoring/llm_based_plan_scorer.py

@@ -67,7 +67,7 @@ class LLMBasedPlanScorer(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 `scoring_llm` must be an instance of `BaseChatModel`.")

nat/experimental/test_time_compute/search/multi_llm_planner.py

@@ -96,7 +96,7 @@ class MultiLLMPlanner(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is not installed. Please install it to use MultiLLMPlanner.\n"
-                              "This error can be resolve by installing aiqtoolkit-langchain.")
+                              "This error can be resolve by installing nvidia-nat-langchain.")
 
         # Create a single PromptTemplate
         planning_template = PromptTemplate(template=self.config.planning_template,

nat/experimental/test_time_compute/search/multi_query_retrieval_search.py

@@ -68,7 +68,7 @@ class MultiQueryRetrievalSearch(StrategyBase):
             from langchain_core.prompts import PromptTemplate
         except ImportError:
             raise ImportError("langchain-core is required for MultiQueryRetrievalSearch. "
-                              "Install aiqtoolkit-langchain or similar.")
+                              "Install nvidia-nat-langchain or similar.")
 
         new_ttc_items: list[TTCItem] = []
 

nat/experimental/test_time_compute/search/single_shot_multi_plan_planner.py

@@ -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"],

nat/experimental/test_time_compute/selection/llm_based_agent_output_selector.py

@@ -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
 

nat/experimental/test_time_compute/selection/llm_based_output_merging_selector.py

@@ -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
 

nat/experimental/test_time_compute/selection/llm_based_plan_selector.py

@@ -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`.")

nat/front_ends/console/console_front_end_config.py

@@ -22,7 +22,7 @@ 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,

nat/front_ends/fastapi/fastapi_front_end_config.py

@@ -119,7 +119,7 @@ class AsyncGenerationStatusResponse(BaseAsyncStatusResponse):
 
 class FastApiFrontEndConfig(FrontEndBaseConfig, name="fastapi"):
     """
-    A FastAPI based front end that allows an NAT 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."),
     )