aiqtoolkit 1.1.0a20250429__py3-none-any.whl → 1.1.0a20250501__py3-none-any.whl

aiq/agent/react_agent/register.py

@@ -34,8 +34,8 @@ logger = logging.getLogger(__name__)
 
 class ReActAgentWorkflowConfig(FunctionBaseConfig, name="react_agent"):
     """
-    Defines an AgentIQ function that uses a ReAct Agent performs reasoning inbetween tool calls, and utilizes the tool
-    names and descriptions to select the optimal tool.
+    Defines an AIQ Toolkit function that uses a ReAct Agent performs reasoning inbetween tool calls, and utilizes the
+    tool names and descriptions to select the optimal tool.
     """
 
     tool_names: list[FunctionRef] = Field(default_factory=list,

aiq/agent/reasoning_agent/reasoning_agent.py

@@ -34,7 +34,7 @@ logger = logging.getLogger(__name__)
 
 class ReasoningFunctionConfig(FunctionBaseConfig, name="reasoning_agent"):
     """
-    Defines an AgentIQ function that performs reasoning on the input data.
+    Defines an AIQ Toolkit 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.

aiq/agent/rewoo_agent/register.py

@@ -33,8 +33,8 @@ logger = logging.getLogger(__name__)
 
 class ReWOOAgentWorkflowConfig(FunctionBaseConfig, name="rewoo_agent"):
     """
-    Defines an AgentIQ function that uses a ReWOO Agent performs reasoning inbetween tool calls, and utilizes the tool
-    names and descriptions to select the optimal tool.
+    Defines an AIQ Toolkit function that uses a ReWOO Agent performs reasoning inbetween tool calls, and utilizes the
+    tool names and descriptions to select the optimal tool.
     """
 
     tool_names: list[FunctionRef] = Field(default_factory=list,

aiq/builder/component_utils.py

@@ -50,7 +50,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 AgentIQ configuration object.
+        component_group (ComponentGroup): The component group in an AIQ Toolkit 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.
@@ -154,7 +154,7 @@ def update_dependency_graph(config: "AIQConfig", instance_config: TypedBaseModel
     """Updates the hierarchical component instance dependency graph from a configuration runtime instance.
 
     Args:
-        config (AIQConfig): An AgentIQ configuration object with runtime instance details.
+        config (AIQConfig): An AIQ Toolkit 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.
 
@@ -184,7 +184,7 @@ def config_to_dependency_objects(config: "AIQConfig") -> tuple[dict[str, Compone
     """Generates a map of component runtime instance IDs to use when generating a build sequence.
 
     Args:
-        config (AIQConfig): The AgentIQ workflow configuration object.
+        config (AIQConfig): The AIQ Toolkit workflow configuration object.
 
     Returns:
         tuple[dict[str, ComponentInstanceData], nx.DiGraph]: A tuple containing a map of component runtime instance
@@ -232,10 +232,10 @@ def config_to_dependency_objects(config: "AIQConfig") -> tuple[dict[str, Compone
 
 
 def build_dependency_sequence(config: "AIQConfig") -> list[ComponentInstanceData]:
-    """Generates the depencency sequence from an AgentIQ configuration object
+    """Generates the depencency sequence from an AIQ Toolkit configuration object
 
     Args:
-        config (AIQConfig): An AgentIQ configuration object.
+        config (AIQConfig): An AIQ Toolkit configuration object.
 
     Returns:
         list[ComponentInstanceData]: A list representing the instatiation sequence to ensure all valid

aiq/builder/front_end.py

@@ -27,7 +27,7 @@ class FrontEndBase(typing.Generic[FrontEndConfigT], ABC):
 
     def __init__(self, full_config: "AIQConfig"):
         """
-        Initializes the FrontEndBase object with the specified AgentIQ configuration.
+        Initializes the FrontEndBase object with the specified AIQ Toolkit configuration.
 
         Parameters
         ----------
@@ -43,7 +43,7 @@ class FrontEndBase(typing.Generic[FrontEndConfigT], ABC):
     @property
     def front_end_config(self) -> FrontEndConfigT:
         """
-        Returns the front end configuration object extracted from the AgentIQ configuration.
+        Returns the front end configuration object extracted from the AIQ Toolkit configuration.
 
         Returns
         -------
@@ -55,12 +55,12 @@ class FrontEndBase(typing.Generic[FrontEndConfigT], ABC):
     @property
     def full_config(self) -> "AIQConfig":
         """
-        Returns the full AgentIQ configuration object.
+        Returns the full AIQ Toolkit configuration object.
 
         Returns
         -------
         AIQConfig
-            The full AgentIQ configuration object.
+            The full AIQ Toolkit configuration object.
         """
 
         return self._full_config

aiq/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 AgentIQ functions providing type handling and schema management.
+"""Base class for AIQ Toolkit functions providing type handling and schema management.
 
 This module contains the FunctionBase abstract base class which provides core functionality
-for AgentIQ functions including type handling via generics, schema management for inputs and outputs,
+for AIQ Toolkit functions including type handling via generics, schema management for inputs and outputs,
 and type conversion capabilities.
 """
 
@@ -40,7 +40,7 @@ logger = logging.getLogger(__name__)
 
 class FunctionBase(typing.Generic[InputT, StreamingOutputT, SingleOutputT], ABC):
     """
-    Abstract base class providing core functionality for AgentIQ functions.
+    Abstract base class providing core functionality for AIQ Toolkit 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 AgentIQ function system, providing:
+    FunctionBase is the foundation of the AIQ Toolkit function system, providing:
     - Type handling via generics
     - Schema management for inputs and outputs
     - Type conversion capabilities

aiq/builder/function_info.py

@@ -574,7 +574,7 @@ class FunctionInfo:
         Returns
         -------
         FunctionInfo
-            The created FunctionInfo object which can be used to create a Generic AgentIQ function.
+            The created FunctionInfo object which can be used to create a Generic AIQ Toolkit function.
 
         """
 

aiq/builder/intermediate_step_manager.py

@@ -43,7 +43,7 @@ class OpenStep:
 
 class IntermediateStepManager:
     """
-    Manages updates to the AgentIQ Event Stream for intermediate steps
+    Manages updates to the AIQ Toolkit Event Stream for intermediate steps
     """
 
     def __init__(self, context_state: "AIQContextState"):  # noqa: F821
@@ -53,7 +53,7 @@ class IntermediateStepManager:
 
     def push_intermediate_step(self, payload: IntermediateStepPayload) -> None:
         """
-        Pushes an intermediate step to the AgentIQ Event Stream
+        Pushes an intermediate step to the AIQ Toolkit Event Stream
         """
 
         if not isinstance(payload, IntermediateStepPayload):
@@ -81,13 +81,15 @@ class IntermediateStepManager:
                 logger.warning("Step id %s not found in outstanding start steps", payload.UUID)
                 return
 
-            # If we are in the same coroutine, we should have the same parent step id. If so, unset the current step id.
-            if (parent_step_id == payload.UUID):
-                _current_open_step_id.reset(open_step.token)
-
+            # Restore the parent step ID directly instead of using a cross‑context token.
+            if parent_step_id == payload.UUID:
+                _current_open_step_id.set(open_step.step_parent_id)
             else:
-                # Manually set the parent step ID. This happens when running on the thread pool
+                # Different context (e.g. thread‑pool); safely restore the parent ID **without**
+                # trying to use a token that belongs to another Context.
+                _current_open_step_id.set(open_step.step_parent_id)
                 parent_step_id = open_step.step_parent_id
+
         elif (payload.event_state == IntermediateStepState.CHUNK):
 
             # Get the current step from the outstanding steps
@@ -119,7 +121,7 @@ class IntermediateStepManager:
                   on_error: OnError = None,
                   on_complete: OnComplete = None) -> Subscription:
         """
-        Subscribes to the AgentIQ Event Stream for intermediate steps
+        Subscribes to the AIQ Toolkit Event Stream for intermediate steps
         """
 
         return self._context_state.event_stream.get().subscribe(on_next, on_error, on_complete)

aiq/builder/workflow_builder.py

@@ -153,7 +153,7 @@ class WorkflowBuilder(Builder, AbstractAsyncContextManager):
             # Store them in a dict so we can un-register them if needed
             self._logging_handlers[key] = handler
 
-            # Now attach to AgentIQ's root logger
+            # Now attach to AIQ Toolkit's root logger
             logging.getLogger().addHandler(handler)
 
         provider = TracerProvider()

aiq/cli/cli_utils/validation.py

@@ -26,7 +26,7 @@ def validate_config(config_file: Path) -> AIQConfig:
     try:
         from aiq.runtime.loader import load_config
 
-        # Load using the AgentIQ loader functions. This performs validation
+        # Load using the AIQ Toolkit loader functions. This performs validation
         config = load_config(config_file)
 
         return config

aiq/cli/commands/configure/channel/add.py

@@ -20,7 +20,7 @@ import click
 logger = logging.getLogger(__name__)
 
 
-@click.group(name=__name__, invoke_without_command=True, help="Utility to add an AgentIQ remote registry channel.")
+@click.group(name=__name__, invoke_without_command=True, help="Utility to add an AIQ Toolkit remote registry channel.")
 @click.argument("channel_type", type=str)
 def add(channel_type: str) -> None:
     from aiq.utils.settings.global_settings import add_channel_interative

aiq/cli/commands/configure/channel/channel.py

@@ -24,7 +24,9 @@ from aiq.cli.commands.configure.channel.update import update
 logger = logging.getLogger(__name__)
 
 
-@click.group(name=__name__, invoke_without_command=False, help="Utility to configure AgentIQ remote registry channels.")
+@click.group(name=__name__,
+             invoke_without_command=False,
+             help="Utility to configure AIQ Toolkit remote registry channels.")
 def channel(**kwargs):
     pass
 

aiq/cli/commands/configure/channel/remove.py

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help="Utility to remove a configured AgentIQ remote registry channel.")
+             help="Utility to remove a configured AIQ Toolkit remote registry channel.")
 @click.argument("channel", type=str)
 def remove(channel: str):
     from aiq.utils.settings.global_settings import remove_channel_interactive

aiq/cli/commands/configure/channel/update.py

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
 
 @click.group(name="update",
              invoke_without_command=True,
-             help="Utility to update an AgentIQ remote registry channel's settings.")
+             help="Utility to update an AIQ Toolkit remote registry channel's settings.")
 @click.argument("channel", type=str)
 def update(channel):
     from aiq.utils.settings.global_settings import update_channel_interactive

aiq/cli/commands/configure/configure.py

@@ -22,10 +22,10 @@ from aiq.cli.commands.configure.channel.channel import channel
 logger = logging.getLogger(__name__)
 
 
-@click.group(name=__name__, invoke_without_command=False, help="Configure AgentIQ developer preferences.")
+@click.group(name=__name__, invoke_without_command=False, help="Configure AIQ Toolkit developer preferences.")
 def configure_command(**kwargs):
     """
-    Publish AgentIQ artifacts with the specified configuration
+    Publish AIQ Toolkit artifacts with the specified configuration
     """
     pass
 

aiq/cli/commands/info/info.py

@@ -25,10 +25,10 @@ logger = logging.getLogger(__name__)
 
 @click.group(name=__name__,
              invoke_without_command=False,
-             help="Provide information about the local AgentIQ environment.")
+             help="Provide information about the local AIQ Toolkit environment.")
 def info_command(**kwargs):
     """
-    Provide information about the local AgentIQ environment.
+    Provide information about the local AIQ Toolkit environment.
     """
     pass
 

aiq/cli/commands/info/list_components.py

@@ -60,7 +60,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="List the locally registered AgentIQ components.")
+@click.group(name=__name__, invoke_without_command=True, help="List the locally registered AIQ Toolkit components.")
 @click.option(
     "-t",
     "--types",
@@ -68,7 +68,7 @@ async def search_artifacts(  # pylint: disable=R0917
     multiple=True,
     type=click.Choice([e.value for e in AIQComponentEnum], case_sensitive=False),
     required=False,
-    help=("Filter the search by AgentIQ component type."),
+    help=("Filter the search by AIQ Toolkit component type."),
 )
 @click.option(
     "-o",

aiq/cli/commands/registry/publish.py

@@ -47,7 +47,7 @@ async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, p
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Publish local AgentIQ artifacts to a remote "
+             help=("Publish local AIQ Toolkit artifacts to a remote "
                    "registry from package repository."))
 @click.option(
     "--config_file",
@@ -61,12 +61,12 @@ async def publish_artifact(registry_handler_config: RegistryHandlerBaseConfig, p
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel to use when publishing the AgentIQ artifact."),
+    help=("The remote registry channel to use when publishing the AIQ Toolkit artifact."),
 )
 @click.argument("package_root", type=str)
 def publish(channel: str, config_file: str, package_root: str) -> None:
     """
-    Publish AgentIQ artifacts with the specified configuration
+    Publish AIQ Toolkit artifacts with the specified configuration
     """
     from aiq.settings.global_settings import GlobalSettings
 

aiq/cli/commands/registry/pull.py

@@ -74,7 +74,7 @@ async def pull_artifact(registry_handler_config: RegistryHandlerBaseConfig, pack
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Pull AgentIQ artifacts from a remote registry "
+             help=("Pull AIQ Toolkit artifacts from a remote registry "
                    "by package name."))
 @click.option(
     "--config_file",
@@ -88,12 +88,12 @@ async def pull_artifact(registry_handler_config: RegistryHandlerBaseConfig, pack
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel to use when pulling the AgentIQ artifact."),
+    help=("The remote registry channel to use when pulling the AIQ Toolkit artifact."),
 )
 @click.argument("packages", type=str)
 def pull(channel: str, config_file: str, packages: str) -> None:
     """
-    Pull AgentIQ artifacts from a remote registry channel.
+    Pull AIQ Toolkit artifacts from a remote registry channel.
     """
 
     from aiq.settings.global_settings import GlobalSettings

aiq/cli/commands/registry/registry.py

@@ -25,7 +25,9 @@ from aiq.cli.commands.registry.search import search
 logger = logging.getLogger(__name__)
 
 
-@click.group(name=__name__, invoke_without_command=False, help="Utility to configure AgentIQ remote registry channels.")
+@click.group(name=__name__,
+             invoke_without_command=False,
+             help="Utility to configure AIQ Toolkit remote registry channels.")
 def registry_command(**kwargs):
     pass
 

aiq/cli/commands/registry/remove.py

@@ -48,7 +48,7 @@ async def remove_artifact(registry_handler_config: RegistryHandlerBaseConfig, pa
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Remove AgentIQ artifact from a remote registry by name and version."))
+             help=("Remove AIQ Toolkit artifact from a remote registry by name and version."))
 @click.argument("packages", type=str)
 @click.option(
     "--config_file",
@@ -62,11 +62,11 @@ async def remove_artifact(registry_handler_config: RegistryHandlerBaseConfig, pa
     "--channel",
     type=str,
     required=True,
-    help=("The remote registry channel that will remove the AgentIQ artifact."),
+    help=("The remote registry channel that will remove the AIQ Toolkit artifact."),
 )
 def remove(channel: str, config_file: str, packages: str) -> None:
     """
-    Remove AgentIQ artifacts from a remote registry.
+    Remove AIQ Toolkit artifacts from a remote registry.
     """
 
     from aiq.settings.global_settings import GlobalSettings

aiq/cli/commands/registry/search.py

@@ -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 AgentIQ artifacts from remote registry.")
+@click.group(name=__name__, invoke_without_command=True, help="Search for AIQ Toolkit 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 AgentIQ artifact."),
+    help=("The remote registry channel to use when pulling the AIQ Toolkit artifact."),
 )
 @click.option(
     "-o",
@@ -125,7 +125,7 @@ def search(  # pylint: disable=R0917
         n_results: int,
         output_path: str) -> None:
     """
-    Search for AgentIQ artifacts with the specified configuration.
+    Search for AIQ Toolkit artifacts with the specified configuration.
     """
 
     from aiq.settings.global_settings import GlobalSettings

aiq/cli/commands/start.py

@@ -149,7 +149,7 @@ class StartCommandGroup(click.MultiCommand):
 
             # Build the command parameters
             params: list[click.Parameter] = self._build_params(registered_front_end)
-            help_msg = f"Run an AgentIQ workflow using the {registered_front_end.local_name} front end."
+            help_msg = f"Run an AIQ Toolkit workflow using the {registered_front_end.local_name} front end."
 
             cmd = click.Command(name=registered_front_end.local_name,
                                 params=params,
@@ -178,7 +178,7 @@ class StartCommandGroup(click.MultiCommand):
         # 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 AgentIQ from config file: '%s'", config_file)
+        logger.info("Starting AIQ Toolkit from config file: '%s'", config_file)
 
         config_dict = load_and_override_config(config_file, override)
 
@@ -242,9 +242,9 @@ class StartCommandGroup(click.MultiCommand):
 
 @click.command(name=__name__,
                invoke_without_command=False,
-               help="Run an AgentIQ workflow using a front end configuration.",
+               help="Run an AIQ Toolkit workflow using a front end configuration.",
                cls=StartCommandGroup)
 @click.pass_context
 def start_command(ctx: click.Context, **kwargs) -> None:
-    """Run an AgentIQ workflow using a front end configuration."""
+    """Run an AIQ Toolkit workflow using a front end configuration."""
     pass

aiq/cli/commands/uninstall.py

@@ -55,11 +55,11 @@ async def uninstall_packages(packages: list[dict[str, str]]) -> None:
 
 @click.group(name=__name__,
              invoke_without_command=True,
-             help=("Uninstall an AgentIQ plugin packages from the local environment."))
+             help=("Uninstall an AIQ Toolkit plugin packages from the local environment."))
 @click.argument("packages", type=str)
 def uninstall_command(packages: str) -> None:
     """
-    Uninstall AgentIQ plugin packages from the local environment.
+    Uninstall AIQ Toolkit plugin packages from the local environment.
     """
 
     packages = packages.split()

aiq/cli/commands/workflow/templates/pyproject.toml.j2

@@ -11,8 +11,8 @@ name = "{{ package_name }}"
 dependencies = [
   "aiqtoolkit[langchain]",
 ]
-requires-python = ">=3.11"
-description = "Custom AgentIQ Workflow"
+requires-python = ">=3.11,<3.13"
+description = "Custom AIQ Toolkit Workflow"
 classifiers = ["Programming Language :: Python"]
 
 {% if editable %}[tool.uv.sources]

aiq/cli/commands/workflow/workflow_commands.py

@@ -98,7 +98,7 @@ def find_package_root(package_name: str) -> Path | None:
 
 def get_workflow_path_from_name(workflow_name: str):
     """
-    Look up the location of an installed AgentIQ workflow and retrieve the root directory of the installed workflow.
+    Look up the location of an installed AIQ Toolkit workflow and retrieve the root directory of the installed workflow.
 
     Args:
         workflow_name: The name of the workflow.
@@ -123,17 +123,17 @@ def get_workflow_path_from_name(workflow_name: str):
 @click.option(
     "--workflow-dir",
     default=".",
-    help="Output directory for saving the created workflow. A new folder with the workflow name will be created within."
-    "Defaults to the present working directory.")
+    help="Output directory for saving the created workflow. A new folder with the workflow name will be created "
+    "within. Defaults to the present working directory.")
 @click.option(
     "--description",
-    default="AgentIQ function template. Please update the description.",
+    default="AIQ Toolkit function template. Please update the description.",
     help="""A description of the component being created. Will be used to populate the docstring and will describe the
          component when inspecting installed components using 'aiq info component'""")
 # pylint: disable=missing-param-doc
 def create_command(workflow_name: str, install: bool, workflow_dir: str, description: str):
     """
-    Create a new AgentIQ workflow using templates.
+    Create a new AIQ Toolkit workflow using templates.
 
     Args:
         workflow_name (str): The name of the new workflow.
@@ -170,6 +170,8 @@ def create_command(workflow_name: str, install: bool, workflow_dir: str, descrip
         (new_workflow_dir / 'src' / package_name).mkdir(parents=True)
         # Create config directory
         (new_workflow_dir / 'src' / package_name / 'configs').mkdir(parents=True)
+        # Create package level configs directory
+        (new_workflow_dir / 'configs').mkdir(parents=True)
 
         # Initialize Jinja2 environment
         env = Environment(loader=FileSystemLoader(str(template_dir)))
@@ -196,7 +198,6 @@ def create_command(workflow_name: str, install: bool, workflow_dir: str, descrip
             'python_safe_workflow_name': workflow_name.replace("-", "_"),
             'package_name': package_name,
             'rel_path_to_repo_root': rel_path_to_repo_root,
-            # 'workflow_class_name': f"{workflow_name.capitalize().replace("-", "").replace("_", "")}WorkflowConfig",
             'workflow_class_name': f"{_generate_valid_classname(workflow_name)}FunctionConfig",
             'workflow_description': description
         }
@@ -207,6 +208,11 @@ def create_command(workflow_name: str, install: bool, workflow_dir: str, descrip
             with open(output_path, 'w', encoding="utf-8") as f:
                 f.write(content)
 
+        # Create symlink for config.yml
+        config_source = new_workflow_dir / 'src' / package_name / 'configs' / 'config.yml'
+        config_link = new_workflow_dir / 'configs' / 'config.yml'
+        os.symlink(config_source, config_link)
+
         if install:
             # Install the new package without changing directories
             click.echo(f"Installing workflow '{workflow_name}'...")
@@ -228,7 +234,7 @@ def create_command(workflow_name: str, install: bool, workflow_dir: str, descrip
 @click.argument('workflow_name')
 def reinstall_command(workflow_name):
     """
-    Reinstall an AgentIQ workflow to update dependencies and code changes.
+    Reinstall an AIQ Toolkit workflow to update dependencies and code changes.
 
     Args:
         workflow_name (str): The name of the workflow to reinstall.
@@ -264,7 +270,7 @@ def reinstall_command(workflow_name):
 @click.argument('workflow_name')
 def delete_command(workflow_name: str):
     """
-    Delete an AgentIQ workflow and uninstall its package.
+    Delete an AIQ Toolkit workflow and uninstall its package.
 
     Args:
         workflow_name (str): The name of the workflow to delete.

aiq/cli/entrypoint.py

@@ -77,7 +77,7 @@ def get_version():
               help='Set the logging level')
 @click.pass_context
 def cli(ctx: click.Context, log_level: str):
-    """Main entrypoint for the AgentIQ CLI"""
+    """Main entrypoint for the AIQ Toolkit CLI"""
 
     ctx_dict = ctx.ensure_object(dict)
 

aiq/data_models/api_server.py

@@ -39,7 +39,7 @@ class Message(BaseModel):
 
 class AIQChatRequest(BaseModel):
     """
-    AIQChatRequest is a data model that represents a request to the AgentIQ chat API.
+    AIQChatRequest is a data model that represents a request to the AIQ Toolkit chat API.
     """
 
     # Allow extra fields in the model_config to support derived models
@@ -88,8 +88,8 @@ class AIQUsage(BaseModel):
 
 class AIQResponseSerializable(abc.ABC):
     """
-    AIQChatResponseSerializable is an abstract class that defines the interface for serializing output for the AgentIQ
-    chat streaming API.
+    AIQChatResponseSerializable is an abstract class that defines the interface for serializing output for the AIQ
+    Toolkit chat streaming API.
     """
 
     @abstractmethod
@@ -111,7 +111,7 @@ class AIQResponseBaseModelIntermediate(BaseModel, AIQResponseSerializable):
 
 class AIQChatResponse(AIQResponseBaseModelOutput):
     """
-    AIQChatResponse is a data model that represents a response from the AgentIQ chat API.
+    AIQChatResponse is a data model that represents a response from the AIQ Toolkit chat API.
     """
 
     # Allow extra fields in the model_config to support derived models
@@ -152,7 +152,7 @@ class AIQChatResponse(AIQResponseBaseModelOutput):
 
 class AIQChatResponseChunk(AIQResponseBaseModelOutput):
     """
-    AIQChatResponseChunk is a data model that represents a response chunk from the AgentIQ chat streaming API.
+    AIQChatResponseChunk is a data model that represents a response chunk from the AIQ Toolkit chat streaming API.
     """
 
     # Allow extra fields in the model_config to support derived models
@@ -191,7 +191,7 @@ class AIQChatResponseChunk(AIQResponseBaseModelOutput):
 
 class AIQResponseIntermediateStep(AIQResponseBaseModelIntermediate):
     """
-    AIQResponseSerializedStep is a data model that represents a serialized step in the AgentIQ chat streaming API.
+    AIQResponseSerializedStep is a data model that represents a serialized step in the AIQ Toolkit chat streaming API.
     """
 
     # Allow extra fields in the model_config to support derived models