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

{aiq → nat}/authentication/oauth2/oauth2_auth_code_flow_provider.py

@@ -19,12 +19,12 @@ from datetime import timezone
 from authlib.integrations.httpx_client import OAuth2Client as AuthlibOAuth2Client
 from pydantic import SecretStr
 
-from aiq.authentication.interfaces import AuthProviderBase
-from aiq.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
-from aiq.builder.context import AIQContext
-from aiq.data_models.authentication import AuthFlowType
-from aiq.data_models.authentication import AuthResult
-from aiq.data_models.authentication import BearerTokenCred
+from nat.authentication.interfaces import AuthProviderBase
+from nat.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
+from nat.builder.context import Context
+from nat.data_models.authentication import AuthFlowType
+from nat.data_models.authentication import AuthResult
+from nat.data_models.authentication import BearerTokenCred
 
 
 class OAuth2AuthCodeFlowProvider(AuthProviderBase[OAuth2AuthCodeFlowProviderConfig]):
@@ -32,7 +32,7 @@ class OAuth2AuthCodeFlowProvider(AuthProviderBase[OAuth2AuthCodeFlowProviderConf
     def __init__(self, config: OAuth2AuthCodeFlowProviderConfig):
         super().__init__(config)
         self._authenticated_tokens: dict[str, AuthResult] = {}
-        self._context = AIQContext.get()
+        self._context = Context.get()
 
     async def _attempt_token_refresh(self, user_id: str, auth_result: AuthResult) -> AuthResult | None:
         refresh_token = auth_result.raw.get("refresh_token")
@@ -63,9 +63,9 @@ class OAuth2AuthCodeFlowProvider(AuthProviderBase[OAuth2AuthCodeFlowProviderConf
         return new_auth_result
 
     async def authenticate(self, user_id: str | None = None) -> AuthResult:
-        if user_id is None and hasattr(AIQContext.get(), "metadata") and hasattr(
-                AIQContext.get().metadata, "cookies") and AIQContext.get().metadata.cookies is not None:
-            session_id = AIQContext.get().metadata.cookies.get("aiqtoolkit-session", None)
+        if user_id is None and hasattr(Context.get(), "metadata") and hasattr(
+                Context.get().metadata, "cookies") and Context.get().metadata.cookies is not None:
+            session_id = Context.get().metadata.cookies.get("nat-session", None)
             if not session_id:
                 raise RuntimeError("Authentication failed. No session ID found. Cannot identify user.")
 
@@ -82,7 +82,7 @@ class OAuth2AuthCodeFlowProvider(AuthProviderBase[OAuth2AuthCodeFlowProviderConf
 
         auth_callback = self._context.user_auth_callback
         if not auth_callback:
-            raise RuntimeError("Authentication callback not set on AIQContext.")
+            raise RuntimeError("Authentication callback not set on Context.")
 
         try:
             authenticated_context = await auth_callback(self.config, AuthFlowType.OAUTH2_AUTHORIZATION_CODE)

{aiq → nat}/authentication/oauth2/oauth2_auth_code_flow_provider_config.py

@@ -15,7 +15,7 @@
 
 from pydantic import Field
 
-from aiq.data_models.authentication import AuthProviderBaseConfig
+from nat.data_models.authentication import AuthProviderBaseConfig
 
 
 class OAuth2AuthCodeFlowProviderConfig(AuthProviderBaseConfig, name="oauth2_auth_code_flow"):

{aiq → nat}/authentication/oauth2/register.py

@@ -13,13 +13,13 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
-from aiq.builder.builder import Builder
-from aiq.cli.register_workflow import register_auth_provider
+from nat.authentication.oauth2.oauth2_auth_code_flow_provider_config import OAuth2AuthCodeFlowProviderConfig
+from nat.builder.builder import Builder
+from nat.cli.register_workflow import register_auth_provider
 
 
 @register_auth_provider(config_type=OAuth2AuthCodeFlowProviderConfig)
 async def oauth2_client(authentication_provider: OAuth2AuthCodeFlowProviderConfig, builder: Builder):
-    from aiq.authentication.oauth2.oauth2_auth_code_flow_provider import OAuth2AuthCodeFlowProvider
+    from nat.authentication.oauth2.oauth2_auth_code_flow_provider import OAuth2AuthCodeFlowProvider
 
     yield OAuth2AuthCodeFlowProvider(authentication_provider)

{aiq → nat}/authentication/register.py

@@ -16,6 +16,6 @@
 # pylint: disable=unused-import
 # flake8: noqa
 
-from aiq.authentication.api_key import register as register_api_key
-from aiq.authentication.http_basic_auth import register as register_http_basic_auth
-from aiq.authentication.oauth2 import register as register_oauth2
+from nat.authentication.api_key import register as register_api_key
+from nat.authentication.http_basic_auth import register as register_http_basic_auth
+from nat.authentication.oauth2 import register as register_oauth2

{aiq → nat}/builder/builder.py

@@ -20,38 +20,38 @@ from abc import abstractmethod
 from collections.abc import Sequence
 from pathlib import Path
 
-from aiq.authentication.interfaces import AuthProviderBase
-from aiq.builder.context import AIQContext
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.function import Function
-from aiq.data_models.authentication import AuthProviderBaseConfig
-from aiq.data_models.component_ref import AuthenticationRef
-from aiq.data_models.component_ref import EmbedderRef
-from aiq.data_models.component_ref import FunctionRef
-from aiq.data_models.component_ref import LLMRef
-from aiq.data_models.component_ref import MemoryRef
-from aiq.data_models.component_ref import ObjectStoreRef
-from aiq.data_models.component_ref import RetrieverRef
-from aiq.data_models.component_ref import TTCStrategyRef
-from aiq.data_models.embedder import EmbedderBaseConfig
-from aiq.data_models.evaluator import EvaluatorBaseConfig
-from aiq.data_models.function import FunctionBaseConfig
-from aiq.data_models.function_dependencies import FunctionDependencies
-from aiq.data_models.llm import LLMBaseConfig
-from aiq.data_models.memory import MemoryBaseConfig
-from aiq.data_models.object_store import ObjectStoreBaseConfig
-from aiq.data_models.retriever import RetrieverBaseConfig
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
-from aiq.experimental.test_time_compute.models.stage_enums import StageTypeEnum
-from aiq.memory.interfaces import MemoryEditor
-from aiq.object_store.interfaces import ObjectStore
-from aiq.retriever.interface import AIQRetriever
+from nat.authentication.interfaces import AuthProviderBase
+from nat.builder.context import Context
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.builder.function import Function
+from nat.data_models.authentication import AuthProviderBaseConfig
+from nat.data_models.component_ref import AuthenticationRef
+from nat.data_models.component_ref import EmbedderRef
+from nat.data_models.component_ref import FunctionRef
+from nat.data_models.component_ref import LLMRef
+from nat.data_models.component_ref import MemoryRef
+from nat.data_models.component_ref import ObjectStoreRef
+from nat.data_models.component_ref import RetrieverRef
+from nat.data_models.component_ref import TTCStrategyRef
+from nat.data_models.embedder import EmbedderBaseConfig
+from nat.data_models.evaluator import EvaluatorBaseConfig
+from nat.data_models.function import FunctionBaseConfig
+from nat.data_models.function_dependencies import FunctionDependencies
+from nat.data_models.llm import LLMBaseConfig
+from nat.data_models.memory import MemoryBaseConfig
+from nat.data_models.object_store import ObjectStoreBaseConfig
+from nat.data_models.retriever import RetrieverBaseConfig
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.experimental.test_time_compute.models.stage_enums import PipelineTypeEnum
+from nat.experimental.test_time_compute.models.stage_enums import StageTypeEnum
+from nat.memory.interfaces import MemoryEditor
+from nat.object_store.interfaces import ObjectStore
+from nat.retriever.interface import Retriever
 
 
 class UserManagerHolder():
 
-    def __init__(self, context: AIQContext) -> None:
+    def __init__(self, context: Context) -> None:
         self._context = context
 
     def get_id(self):
@@ -214,11 +214,11 @@ class Builder(ABC):  # pylint: disable=too-many-public-methods
         ...
 
     @typing.overload
-    async def get_retriever(self, retriever_name: str | RetrieverRef, wrapper_type: None) -> AIQRetriever:
+    async def get_retriever(self, retriever_name: str | RetrieverRef, wrapper_type: None) -> Retriever:
         ...
 
     @typing.overload
-    async def get_retriever(self, retriever_name: str | RetrieverRef) -> AIQRetriever:
+    async def get_retriever(self, retriever_name: str | RetrieverRef) -> Retriever:
         ...
 
     @abstractmethod

{aiq → nat}/builder/component_utils.py

@@ -21,21 +21,21 @@ from collections.abc import Iterable
 import networkx as nx
 from pydantic import BaseModel
 
-from aiq.data_models.authentication import AuthProviderBaseConfig
-from aiq.data_models.common import TypedBaseModel
-from aiq.data_models.component import ComponentGroup
-from aiq.data_models.component_ref import ComponentRef
-from aiq.data_models.component_ref import ComponentRefNode
-from aiq.data_models.component_ref import generate_instance_id
-from aiq.data_models.config import AIQConfig
-from aiq.data_models.embedder import EmbedderBaseConfig
-from aiq.data_models.function import FunctionBaseConfig
-from aiq.data_models.llm import LLMBaseConfig
-from aiq.data_models.memory import MemoryBaseConfig
-from aiq.data_models.object_store import ObjectStoreBaseConfig
-from aiq.data_models.retriever import RetrieverBaseConfig
-from aiq.data_models.ttc_strategy import TTCStrategyBaseConfig
-from aiq.utils.type_utils import DecomposedType
+from nat.data_models.authentication import AuthProviderBaseConfig
+from nat.data_models.common import TypedBaseModel
+from nat.data_models.component import ComponentGroup
+from nat.data_models.component_ref import ComponentRef
+from nat.data_models.component_ref import ComponentRefNode
+from nat.data_models.component_ref import generate_instance_id
+from nat.data_models.config import Config
+from nat.data_models.embedder import EmbedderBaseConfig
+from nat.data_models.function import FunctionBaseConfig
+from nat.data_models.llm import LLMBaseConfig
+from nat.data_models.memory import MemoryBaseConfig
+from nat.data_models.object_store import ObjectStoreBaseConfig
+from nat.data_models.retriever import RetrieverBaseConfig
+from nat.data_models.ttc_strategy import TTCStrategyBaseConfig
+from nat.utils.type_utils import DecomposedType
 
 logger = logging.getLogger(__name__)
 
@@ -56,7 +56,7 @@ class ComponentInstanceData(BaseModel):
     """A data model to hold component runtime instance metadata to support generating build sequences.
 
     Args:
-        component_group (ComponentGroup): The component group in an AIQ Toolkit configuration object.
+        component_group (ComponentGroup): The component group in a NAT configuration object.
         name (ComponentRef): The name of the component runtime instance.
         config (TypedBaseModel): The runtime instance's configuration object.
         instance_id (str): Unique identifier for each runtime instance.
@@ -161,12 +161,12 @@ def recursive_componentref_discovery(cls: TypedBaseModel, value: typing.Any,
             yield from recursive_componentref_discovery(cls, value, arg)
 
 
-def update_dependency_graph(config: "AIQConfig", instance_config: TypedBaseModel,
+def update_dependency_graph(config: "Config", instance_config: TypedBaseModel,
                             dependency_graph: nx.DiGraph) -> nx.DiGraph:
     """Updates the hierarchical component instance dependency graph from a configuration runtime instance.
 
     Args:
-        config (AIQConfig): An AIQ Toolkit configuration object with runtime instance details.
+        config (Config): A NAT configuration object with runtime instance details.
         instance_config (TypedBaseModel): A component's runtime instance configuration object.
         dependency_graph (nx.DiGraph): A graph tracking runtime instance component dependencies.
 
@@ -192,11 +192,11 @@ def update_dependency_graph(config: "AIQConfig", instance_config: TypedBaseModel
     return dependency_graph
 
 
-def config_to_dependency_objects(config: "AIQConfig") -> tuple[dict[str, ComponentInstanceData], nx.DiGraph]:
+def config_to_dependency_objects(config: "Config") -> tuple[dict[str, ComponentInstanceData], nx.DiGraph]:
     """Generates a map of component runtime instance IDs to use when generating a build sequence.
 
     Args:
-        config (AIQConfig): The AIQ Toolkit workflow configuration object.
+        config (Config): The NAT workflow configuration object.
 
     Returns:
         tuple[dict[str, ComponentInstanceData], nx.DiGraph]: A tuple containing a map of component runtime instance
@@ -243,11 +243,11 @@ def config_to_dependency_objects(config: "AIQConfig") -> tuple[dict[str, Compone
     return dependency_map, dependency_graph
 
 
-def build_dependency_sequence(config: "AIQConfig") -> list[ComponentInstanceData]:
-    """Generates the depencency sequence from an AIQ Toolkit configuration object
+def build_dependency_sequence(config: "Config") -> list[ComponentInstanceData]:
+    """Generates the depencency sequence from a NAT configuration object
 
     Args:
-        config (AIQConfig): An AIQ Toolkit configuration object.
+        config (Config): A NAT configuration object.
 
     Returns:
         list[ComponentInstanceData]: A list representing the instatiation sequence to ensure all valid

{aiq → nat}/builder/context.py

@@ -20,20 +20,20 @@ from collections.abc import Callable
 from contextlib import contextmanager
 from contextvars import ContextVar
 
-from aiq.builder.intermediate_step_manager import IntermediateStepManager
-from aiq.builder.user_interaction_manager import AIQUserInteractionManager
-from aiq.data_models.authentication import AuthenticatedContext
-from aiq.data_models.authentication import AuthFlowType
-from aiq.data_models.authentication import AuthProviderBaseConfig
-from aiq.data_models.interactive import HumanResponse
-from aiq.data_models.interactive import InteractionPrompt
-from aiq.data_models.intermediate_step import IntermediateStep
-from aiq.data_models.intermediate_step import IntermediateStepPayload
-from aiq.data_models.intermediate_step import IntermediateStepType
-from aiq.data_models.intermediate_step import StreamEventData
-from aiq.data_models.invocation_node import InvocationNode
-from aiq.runtime.user_metadata import RequestAttributes
-from aiq.utils.reactive.subject import Subject
+from nat.builder.intermediate_step_manager import IntermediateStepManager
+from nat.builder.user_interaction_manager import UserInteractionManager
+from nat.data_models.authentication import AuthenticatedContext
+from nat.data_models.authentication import AuthFlowType
+from nat.data_models.authentication import AuthProviderBaseConfig
+from nat.data_models.interactive import HumanResponse
+from nat.data_models.interactive import InteractionPrompt
+from nat.data_models.intermediate_step import IntermediateStep
+from nat.data_models.intermediate_step import IntermediateStepPayload
+from nat.data_models.intermediate_step import IntermediateStepType
+from nat.data_models.intermediate_step import StreamEventData
+from nat.data_models.invocation_node import InvocationNode
+from nat.runtime.user_metadata import RequestAttributes
+from nat.utils.reactive.subject import Subject
 
 
 class Singleton(type):
@@ -61,7 +61,7 @@ class ActiveFunctionContextManager:
         self._output = output
 
 
-class AIQContextState(metaclass=Singleton):
+class ContextState(metaclass=Singleton):
 
     def __init__(self):
         self.conversation_id: ContextVar[str | None] = ContextVar("conversation_id", default=None)
@@ -78,19 +78,19 @@ class AIQContextState(metaclass=Singleton):
         self.user_input_callback: ContextVar[Callable[[InteractionPrompt], Awaitable[HumanResponse | None]]
                                              | None] = ContextVar(
                                                  "user_input_callback",
-                                                 default=AIQUserInteractionManager.default_callback_handler)
+                                                 default=UserInteractionManager.default_callback_handler)
         self.user_auth_callback: ContextVar[Callable[[AuthProviderBaseConfig, AuthFlowType],
                                                      Awaitable[AuthenticatedContext]]
                                             | None] = ContextVar("user_auth_callback", default=None)
 
     @staticmethod
-    def get() -> "AIQContextState":
-        return AIQContextState()
+    def get() -> "ContextState":
+        return ContextState()
 
 
-class AIQContext:
+class Context:
 
-    def __init__(self, context: AIQContextState):
+    def __init__(self, context: ContextState):
         self._context_state = context
 
     @property
@@ -134,12 +134,12 @@ class AIQContext:
         return self._context_state.metadata.get()
 
     @property
-    def user_interaction_manager(self) -> AIQUserInteractionManager:
+    def user_interaction_manager(self) -> UserInteractionManager:
         """
-        Return an instance of AIQUserInteractionManager that uses
+        Return an instance of UserInteractionManager that uses
         the current context's user_input_callback.
         """
-        return AIQUserInteractionManager(self._context_state)
+        return UserInteractionManager(self._context_state)
 
     @property
     def intermediate_step_manager(self) -> IntermediateStepManager:
@@ -251,14 +251,20 @@ class AIQContext:
         return callback
 
     @staticmethod
-    def get() -> "AIQContext":
+    def get() -> "Context":
         """
-        Static method to retrieve the current AIQContext instance.
+        Static method to retrieve the current Context instance.
 
-        This method creates and returns an instance of the AIQContext class
-        by obtaining the current state from the AIQContextState.
+        This method creates and returns an instance of the Context class
+        by obtaining the current state from the ContextState.
 
         Returns:
-            AIQContext: The created AIQContext instance.
+            Context: The created Context instance.
         """
-        return AIQContext(AIQContextState.get())
+        return Context(ContextState.get())
+
+
+# Compatibility aliases with previous releases
+
+AIQContextState = ContextState
+AIQContext = Context

{aiq → nat}/builder/embedder.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-from aiq.data_models.embedder import EmbedderBaseConfig
+from nat.data_models.embedder import EmbedderBaseConfig
 
 
 class EmbedderProviderInfo:

{aiq → nat}/builder/eval_builder.py

@@ -18,17 +18,17 @@ import logging
 from contextlib import asynccontextmanager
 from pathlib import Path
 
-from aiq.builder.builder import EvalBuilder
-from aiq.builder.evaluator import EvaluatorInfo
-from aiq.builder.framework_enum import LLMFrameworkEnum
-from aiq.builder.workflow_builder import WorkflowBuilder
-from aiq.cli.type_registry import TypeRegistry
-from aiq.data_models.config import AIQConfig
-from aiq.data_models.config import GeneralConfig
-from aiq.data_models.evaluate import EvalGeneralConfig
-from aiq.data_models.evaluator import EvaluatorBaseConfig
-from aiq.data_models.function import EmptyFunctionConfig
-from aiq.utils.type_utils import override
+from nat.builder.builder import EvalBuilder
+from nat.builder.evaluator import EvaluatorInfo
+from nat.builder.framework_enum import LLMFrameworkEnum
+from nat.builder.workflow_builder import WorkflowBuilder
+from nat.cli.type_registry import TypeRegistry
+from nat.data_models.config import Config
+from nat.data_models.config import GeneralConfig
+from nat.data_models.evaluate import EvalGeneralConfig
+from nat.data_models.evaluator import EvaluatorBaseConfig
+from nat.data_models.function import EmptyFunctionConfig
+from nat.utils.type_utils import override
 
 logger = logging.getLogger(__name__)
 
@@ -127,7 +127,7 @@ class WorkflowEvalBuilder(WorkflowBuilder, EvalBuilder):
                                 remaining_components,
                                 original_error)
 
-    async def populate_builder(self, config: AIQConfig):
+    async def populate_builder(self, config: Config):
         # Skip setting workflow if workflow config is EmptyFunctionConfig
         skip_workflow = isinstance(config.workflow, EmptyFunctionConfig)
 
@@ -154,7 +154,7 @@ class WorkflowEvalBuilder(WorkflowBuilder, EvalBuilder):
 
     @classmethod
     @asynccontextmanager
-    async def from_config(cls, config: AIQConfig):
+    async def from_config(cls, config: Config):
 
         async with cls(config.general, config.eval.general, registry=None) as builder:
             await builder.populate_builder(config)

{aiq → nat}/builder/evaluator.py

@@ -15,9 +15,9 @@
 
 from collections.abc import Callable
 
-from aiq.data_models.evaluator import EvaluatorBaseConfig
-from aiq.eval.evaluator.evaluator_model import EvalInput
-from aiq.eval.evaluator.evaluator_model import EvalOutput
+from nat.data_models.evaluator import EvaluatorBaseConfig
+from nat.eval.evaluator.evaluator_model import EvalInput
+from nat.eval.evaluator.evaluator_model import EvalOutput
 
 
 class EvaluatorInfo:

{aiq → nat}/builder/front_end.py

@@ -17,33 +17,33 @@ import typing
 from abc import ABC
 from abc import abstractmethod
 
-from aiq.data_models.front_end import FrontEndConfigT
+from nat.data_models.front_end import FrontEndConfigT
 
 if (typing.TYPE_CHECKING):
-    from aiq.data_models.config import AIQConfig
+    from nat.data_models.config import Config
 
 
 class FrontEndBase(typing.Generic[FrontEndConfigT], ABC):
 
-    def __init__(self, full_config: "AIQConfig"):
+    def __init__(self, full_config: "Config"):
         """
-        Initializes the FrontEndBase object with the specified AIQ Toolkit configuration.
+        Initializes the FrontEndBase object with the specified NAT configuration.
 
         Parameters
         ----------
-        full_config : AIQConfig
+        full_config : Config
             The configuration object to use for the front end.
         """
 
         super().__init__()
 
-        self._full_config: "AIQConfig" = full_config
+        self._full_config: "Config" = full_config
         self._front_end_config: FrontEndConfigT = typing.cast(FrontEndConfigT, full_config.general.front_end)
 
     @property
     def front_end_config(self) -> FrontEndConfigT:
         """
-        Returns the front end configuration object extracted from the AIQ Toolkit configuration.
+        Returns the front end configuration object extracted from the NAT configuration.
 
         Returns
         -------
@@ -53,14 +53,14 @@ class FrontEndBase(typing.Generic[FrontEndConfigT], ABC):
         return self._front_end_config
 
     @property
-    def full_config(self) -> "AIQConfig":
+    def full_config(self) -> "Config":
         """
-        Returns the full AIQ Toolkit configuration object.
+        Returns the full NAT configuration object.
 
         Returns
         -------
-        AIQConfig
-            The full AIQ Toolkit configuration object.
+        Config
+            The full NAT configuration object.
         """
 
         return self._full_config

{aiq → nat}/builder/function.py

@@ -23,13 +23,13 @@ from collections.abc import Callable
 
 from pydantic import BaseModel
 
-from aiq.builder.context import AIQContext
-from aiq.builder.function_base import FunctionBase
-from aiq.builder.function_base import InputT
-from aiq.builder.function_base import SingleOutputT
-from aiq.builder.function_base import StreamingOutputT
-from aiq.builder.function_info import FunctionInfo
-from aiq.data_models.function import FunctionBaseConfig
+from nat.builder.context import Context
+from nat.builder.function_base import FunctionBase
+from nat.builder.function_base import InputT
+from nat.builder.function_base import SingleOutputT
+from nat.builder.function_base import StreamingOutputT
+from nat.builder.function_info import FunctionInfo
+from nat.data_models.function import FunctionBaseConfig
 
 _InvokeFnT = Callable[[InputT], Awaitable[SingleOutputT]]
 _StreamFnT = Callable[[InputT], AsyncGenerator[StreamingOutputT]]
@@ -59,7 +59,7 @@ class Function(FunctionBase[InputT, StreamingOutputT, SingleOutputT], ABC):
         self.config = config
         self.description = description
         self.instance_name = instance_name or config.type
-        self._context = AIQContext.get()
+        self._context = Context.get()
 
     def convert(self, value: typing.Any, to_type: type[_T]) -> _T:
         """

{aiq → nat}/builder/function_base.py

@@ -12,10 +12,10 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-"""Base class for AIQ Toolkit functions providing type handling and schema management.
+"""Base class for NAT functions providing type handling and schema management.
 
 This module contains the FunctionBase abstract base class which provides core functionality
-for AIQ Toolkit functions including type handling via generics, schema management for inputs and outputs,
+for NAT functions including type handling via generics, schema management for inputs and outputs,
 and type conversion capabilities.
 """
 
@@ -28,8 +28,8 @@ from types import NoneType
 
 from pydantic import BaseModel
 
-from aiq.utils.type_converter import TypeConverter
-from aiq.utils.type_utils import DecomposedType
+from nat.utils.type_converter import TypeConverter
+from nat.utils.type_utils import DecomposedType
 
 InputT = typing.TypeVar("InputT")
 StreamingOutputT = typing.TypeVar("StreamingOutputT")
@@ -40,7 +40,7 @@ logger = logging.getLogger(__name__)
 
 class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC):
     """
-    Abstract base class providing core functionality for AIQ Toolkit functions.
+    Abstract base class providing core functionality for NAT functions.
 
     This class provides type handling via generics, schema management for inputs and outputs,
     and type conversion capabilities.
@@ -56,7 +56,7 @@ class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC)
 
     Notes
     -----
-    FunctionBase is the foundation of the AIQ Toolkit function system, providing:
+    FunctionBase is the foundation of the NAT function system, providing:
     - Type handling via generics
     - Schema management for inputs and outputs
     - Type conversion capabilities

{aiq → nat}/builder/function_info.py

@@ -29,8 +29,8 @@ from pydantic import Field
 from pydantic import create_model
 from pydantic_core import PydanticUndefined
 
-from aiq.data_models.streaming import Streaming
-from aiq.utils.type_utils import DecomposedType
+from nat.data_models.streaming import Streaming
+from nat.utils.type_utils import DecomposedType
 
 logger = logging.getLogger(__name__)
 
@@ -574,7 +574,7 @@ class FunctionInfo:
         Returns
         -------
         FunctionInfo
-            The created FunctionInfo object which can be used to create a Generic AIQ Toolkit function.
+            The created FunctionInfo object which can be used to create a Generic NAT function.
 
         """