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

aiq/front_ends/fastapi/response_helpers.py

@@ -117,16 +117,26 @@ async def generate_single_response(
         return await runner.result(to_type=result_type)
 
 
-async def generate_streaming_response_raw(payload: typing.Any,
-                                          *,
-                                          session_manager: AIQSessionManager,
-                                          streaming: bool,
-                                          result_type: type | None = None,
-                                          output_type: type | None = None) -> AsyncGenerator[AIQResponseSerializable]:
+async def generate_streaming_response_full(payload: typing.Any,
+                                           *,
+                                           session_manager: AIQSessionManager,
+                                           streaming: bool,
+                                           result_type: type | None = None,
+                                           output_type: type | None = None,
+                                           filter_steps: str | None = None) -> AsyncGenerator[AIQResponseSerializable]:
     """
     Similar to generate_streaming_response but provides raw AIQResponseIntermediateStep objects
     without any step adaptor translations.
     """
+    # Parse filter_steps into a set of allowed types if provided
+    # Special case: if filter_steps is "none", suppress all steps
+    allowed_types = None
+    if filter_steps:
+        if filter_steps.lower() == "none":
+            allowed_types = set()  # Empty set means no steps allowed
+        else:
+            allowed_types = set(filter_steps.split(','))
+
     async with session_manager.run(payload) as runner:
         q: AsyncIOProducerConsumerQueue[AIQResponseSerializable] = AsyncIOProducerConsumerQueue()
 
@@ -150,7 +160,9 @@ async def generate_streaming_response_raw(payload: typing.Any,
 
             async for item in q:
                 if (isinstance(item, AIQResponseIntermediateStep)):
-                    yield item
+                    # Filter intermediate steps if filter_steps is provided
+                    if allowed_types is None or item.type in allowed_types:
+                        yield item
                 else:
                     yield AIQResponsePayloadOutput(payload=item)
         except Exception as e:
@@ -160,20 +172,22 @@ async def generate_streaming_response_raw(payload: typing.Any,
             await q.close()
 
 
-async def generate_streaming_response_raw_as_str(payload: typing.Any,
-                                                 *,
-                                                 session_manager: AIQSessionManager,
-                                                 streaming: bool,
-                                                 result_type: type | None = None,
-                                                 output_type: type | None = None) -> AsyncGenerator[str]:
+async def generate_streaming_response_full_as_str(payload: typing.Any,
+                                                  *,
+                                                  session_manager: AIQSessionManager,
+                                                  streaming: bool,
+                                                  result_type: type | None = None,
+                                                  output_type: type | None = None,
+                                                  filter_steps: str | None = None) -> AsyncGenerator[str]:
     """
-    Similar to generate_streaming_response_raw but converts the response to a string format.
+    Similar to generate_streaming_response but converts the response to a string format.
     """
-    async for item in generate_streaming_response_raw(payload,
-                                                      session_manager=session_manager,
-                                                      streaming=streaming,
-                                                      result_type=result_type,
-                                                      output_type=output_type):
+    async for item in generate_streaming_response_full(payload,
+                                                       session_manager=session_manager,
+                                                       streaming=streaming,
+                                                       result_type=result_type,
+                                                       output_type=output_type,
+                                                       filter_steps=filter_steps):
         if (isinstance(item, AIQResponseIntermediateStep) or isinstance(item, AIQResponsePayloadOutput)):
             yield item.get_stream_data()
         else:

aiq/memory/__init__.py

@@ -13,8 +13,8 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 """
-AgentIQ Memory Module
+AIQ Toolkit Memory Module
 
 This package provides foundational classes and interfaces
-for managing text-based memory in AgentIQ's LLM-based agents.
+for managing text-based memory in AIQ Toolkit's LLM-based agents.
 """

aiq/meta/pypi.md

@@ -6,7 +6,7 @@ Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at
 
-http:/www.apache.org/licenses/LICENSE-2.0
+http://www.apache.org/licenses/LICENSE-2.0
 
 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
@@ -15,39 +15,39 @@ See the License for the specific language governing permissions and
 limitations under the License.
 -->
 
-![NVIDIA AgentIQ](https://media.githubusercontent.com/media/NVIDIA/AgentIQ/refs/heads/main/docs/source/_static/agentiq_banner.png "AgentIQ banner image")
+![NVIDIA Agent Intelligence Toolkit](https://media.githubusercontent.com/media/NVIDIA/AIQToolkit/refs/heads/main/docs/source/_static/aiqtoolkit_banner.png "AIQ Toolkit banner image")
 
-# NVIDIA AgentIQ
+# NVIDIA Agent Intelligence Toolkit
 
-AgentIQ is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, AgentIQ enables true composability: build once and reuse anywhere.
+AIQ Toolkit is a flexible library designed to seamlessly integrate your enterprise agents—regardless of framework—with various data sources and tools. By treating agents, tools, and agentic workflows as simple function calls, AIQ Toolkit enables true composability: build once and reuse anywhere.
 
 ## Key Features
 
-- [**Framework Agnostic:**](https://docs.nvidia.com/agentiq/latest/concepts/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
-- [**Reusability:**](https://docs.nvidia.com/agentiq/latest/guides/sharing-workflows-and-tools.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
-- [**Rapid Development:**](https://docs.nvidia.com/agentiq/latest/guides/create-customize-workflows.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
-- [**Profiling:**](https://docs.nvidia.com/agentiq/latest/guides/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
-- [**Observability:**](https://docs.nvidia.com/agentiq/latest/guides/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool.
-- [**Evaluation System:**](https://docs.nvidia.com/agentiq/latest/guides/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
-- [**User Interface:**](https://docs.nvidia.com/agentiq/latest/guides/using-agentiq-ui-and-server.html) Use the AgentIQ UI chat interface to interact with your agents, visualize output, and debug workflows.
-- [**MCP Compatibility**](https://docs.nvidia.com/agentiq/latest/components/mcp.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AgentIQ functions.
+- [**Framework Agnostic:**](https://docs.nvidia.com/aiqtoolkit/latest/concepts/plugins.html) Works with any agentic framework, so you can use your current technology stack without replatforming.
+- [**Reusability:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/sharing-workflows-and-tools.html) Every agent, tool, or workflow can be combined and repurposed, allowing developers to leverage existing work in new scenarios.
+- [**Rapid Development:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/create-customize-workflows.html) Start with a pre-built agent, tool, or workflow, and customize it to your needs.
+- [**Profiling:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/profiler.html) Profile entire workflows down to the tool and agent level, track input/output tokens and timings, and identify bottlenecks.
+- [**Observability:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/observe-workflow-with-phoenix.html) Monitor and debug your workflows with any OpenTelemetry-compatible observability tool.
+- [**Evaluation System:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/evaluate.html) Validate and maintain accuracy of agentic workflows with built-in evaluation tools.
+- [**User Interface:**](https://docs.nvidia.com/aiqtoolkit/latest/guides/using-aiqtoolkit-ui-and-server.html) Use the AIQ Toolkit UI chat interface to interact with your agents, visualize output, and debug workflows.
+- [**MCP Compatibility**](https://docs.nvidia.com/aiqtoolkit/latest/components/mcp.html) Compatible with Model Context Protocol (MCP), allowing tools served by MCP Servers to be used as AIQ Toolkit functions.
 
-With AgentIQ, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
+With AIQ Toolkit, you can move quickly, experiment freely, and ensure reliability across all your agent-driven projects.
 
 ## Links
- * [Documentation](https://docs.nvidia.com/agentiq/latest/index.html): Explore the full documentation for AgentIQ.
- * [About AgentIQ](https://docs.nvidia.com/agentiq/latest/intro/why-agentiq.html): Learn more about the benefits of using AgentIQ.
+ * [Documentation](https://docs.nvidia.com/aiqtoolkit/latest/index.html): Explore the full documentation for AIQ Toolkit.
+ * [About AIQ Toolkit](https://docs.nvidia.com/aiqtoolkit/latest/intro/why-aiqtoolkit.html): Learn more about the benefits of using AIQ Toolkit.
 
 ## First time user?
- If this is your first time using AgentIQ, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/AgentIQ?tab=readme-ov-file#get-started) on GitHub. This package is intended for users who are familiar with AgentIQ applications and need to add AgentIQ as a dependency to their project.
+ If this is your first time using AIQ Toolkit, it is recommended to install the latest version from the [source repository](https://github.com/NVIDIA/AIQToolkit?tab=readme-ov-file#get-started) on GitHub. This package is intended for users who are familiar with AIQ Toolkit applications and need to add AIQ Toolkit as a dependency to their project.
 
 ## Feedback
 
-We would love to hear from you! Please file an issue on [GitHub](https://github.com/NVIDIA/AgentIQ/issues) if you have any feedback or feature requests.
+We would love to hear from you! Please file an issue on [GitHub](https://github.com/NVIDIA/AIQToolkit/issues) if you have any feedback or feature requests.
 
 ## Acknowledgements
 
-We would like to thank the following open source projects that made AgentIQ possible:
+We would like to thank the following open source projects that made AIQ Toolkit possible:
 
 - [CrewAI](https://github.com/crewAIInc/crewAI)
 - [FastAPI](https://github.com/tiangolo/fastapi)

aiq/observability/async_otel_listener.py

@@ -37,7 +37,7 @@ OPENINFERENCE_SPAN_KIND = SpanAttributes.OPENINFERENCE_SPAN_KIND
 
 def _ns_timestamp(seconds_float: float) -> int:
     """
-    Convert AgentIQ’s float `event_timestamp` (in seconds) into an integer number
+    Convert AIQ Toolkit’s float `event_timestamp` (in seconds) into an integer number
     of nanoseconds, as OpenTelemetry expects.
     """
     return int(seconds_float * 1e9)
@@ -45,14 +45,14 @@ def _ns_timestamp(seconds_float: float) -> int:
 
 class AsyncOtelSpanListener:
     """
-    A separate, async class that listens to the AgentIQ intermediate step
+    A separate, async class that listens to the AIQ Toolkit intermediate step
     event stream and creates proper Otel spans:
 
     - On FUNCTION_START => open a new top-level span
     - On any other intermediate step => open a child subspan (immediate open/close)
     - On FUNCTION_END => close the function’s top-level span
 
-    This runs fully independently from the normal AgentIQ workflow, so that
+    This runs fully independently from the normal AIQ Toolkit workflow, so that
     the workflow is not blocking or entangled by OTel calls.
     """
 
@@ -70,7 +70,7 @@ class AsyncOtelSpanListener:
         self._outstanding_spans: dict[str, Span] = {}
 
         # Stack of spans, for when we need to create a child span
-        self._span_stack: list[Span] = []
+        self._span_stack: dict[str, Span] = {}
 
         self._running = False
 
@@ -109,11 +109,11 @@ class AsyncOtelSpanListener:
 
             otel_listener = AsyncOtelSpanListener()
             async with otel_listener.start():
-                # run your AgentIQ workflow
+                # run your AIQ Toolkit workflow
                 ...
             # cleans up
 
-        This sets up the subscription to the AgentIQ event stream and starts the background loop.
+        This sets up the subscription to the AIQ Toolkit event stream and starts the background loop.
         """
         try:
             # Subscribe to the event stream
@@ -152,7 +152,7 @@ class AsyncOtelSpanListener:
 
         self._outstanding_spans.clear()
 
-        if self._span_stack:
+        if len(self._span_stack) > 0:
             logger.error(
                 "Not all spans were closed. Ensure all start events have a corresponding end event. Remaining: %s",
                 self._span_stack)
@@ -175,7 +175,10 @@ class AsyncOtelSpanListener:
         parent_ctx = None
 
         if (len(self._span_stack) > 0):
-            parent_span = self._span_stack[-1]
+            parent_span = self._span_stack.get(step.function_ancestry.parent_id, None)
+            if parent_span is None:
+                logger.warning("No parent span found for step %s", step.UUID)
+                return
 
             parent_ctx = set_span_in_context(parent_span)
 
@@ -230,7 +233,7 @@ class AsyncOtelSpanListener:
                 sub_span.set_attribute(SpanAttributes.INPUT_VALUE, serialized_input)
                 sub_span.set_attribute(SpanAttributes.INPUT_MIME_TYPE, "application/json" if is_json else "text/plain")
 
-        self._span_stack.append(sub_span)
+        self._span_stack[step.UUID] = sub_span
 
         self._outstanding_spans[step.UUID] = sub_span
 
@@ -243,7 +246,7 @@ class AsyncOtelSpanListener:
             logger.warning("No subspan found for step %s", step.UUID)
             return
 
-        self._span_stack.pop()
+        self._span_stack.pop(step.UUID, None)
 
         # Optionally add more attributes from usage_info or data
         usage_info = step.payload.usage_info

aiq/profiler/callbacks/agno_callback_handler.py

@@ -44,7 +44,7 @@ class AgnoProfilerHandler(BaseProfilerCallback):
       - LLM Calls
 
     to collect usage statistics (tokens, inputs, outputs, time intervals, etc.)
-    and store them in AgentIQ's usage_stats queue for subsequent analysis.
+    and store them in AIQ Toolkit's usage_stats queue for subsequent analysis.
     """
 
     def __init__(self) -> None:
@@ -73,7 +73,7 @@ class AgnoProfilerHandler(BaseProfilerCallback):
 
         # Note: Agno doesn't have a class-based tool structure to patch directly.
         # Instead, it uses decorators to convert functions to tools.
-        # In AgentIQ, tool executions are captured at the execute_agno_tool level
+        # In AIQ Toolkit, tool executions are captured at the execute_agno_tool level
         # in packages/aiqtoolkit_agno/src/aiq/plugins/agno/tool_wrapper.py
 
         # To properly monitor Agno tool executions, we would need to either:

aiq/profiler/callbacks/langchain_callback_handler.py

@@ -50,7 +50,7 @@ class LangchainProfilerHandler(AsyncCallbackHandler, BaseProfilerCallback):  # p
     completion_tokens: int = 0
     successful_requests: int = 0
     raise_error = True  # Override to raise error and run inline
-    run_inline = False
+    run_inline = True
 
     def __init__(self) -> None:
         super().__init__()

aiq/profiler/callbacks/llama_index_callback_handler.py

@@ -167,7 +167,7 @@ class LlamaIndexProfilerHandler(BaseCallbackHandler, BaseProfilerCallback):
                 except Exception as e:
                     logger.exception("Error getting model name: %s", e, exc_info=True)
 
-                # Append usage data to AgentIQ usage stats
+                # Append usage data to AIQ Toolkit usage stats
                 with self._lock:
                     stats = IntermediateStepPayload(
                         event_type=IntermediateStepType.LLM_END,

aiq/profiler/callbacks/semantic_kernel_callback_handler.py

@@ -55,7 +55,7 @@ class SemanticKernelProfilerHandler(BaseProfilerCallback):
     - Tool calls
 
     to collect usage statistics (tokens, inputs, outputs, time intervals, etc.)
-    and store them in AgentIQ's usage_stats queue for subsequent analysis.
+    and store them in AIQ Toolkit's usage_stats queue for subsequent analysis.
     """
 
     def __init__(self, workflow_llms: dict) -> None:

aiq/profiler/decorators/function_tracking.py

@@ -61,7 +61,7 @@ def push_intermediate_step(step_manager: IntermediateStepManager,
                            kwargs: Any = None,
                            output: Any = None,
                            metadata: dict[str, Any] | None = None) -> None:
-    """Push an intermediate step to the AgentIQ Event Stream."""
+    """Push an intermediate step to the AIQ Toolkit Event Stream."""
 
     payload = IntermediateStepPayload(UUID=identifier,
                                       event_type=event_type,

aiq/profiler/profile_runner.py

@@ -48,7 +48,7 @@ class InferenceOptimizationHolder(BaseModel):
 
 class ProfilerRunner:
     """
-    A utility to run a series of prompts through an AgentIQ workflow for profiling:
+    A utility to run a series of prompts through an AIQ Toolkit workflow for profiling:
 
     - can load prompts from a file
     - or generate them via an LLM

aiq/registry_handlers/local/local_handler.py

@@ -42,10 +42,10 @@ class LocalRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AgentIQ artifact to a local registry.
+        """Publishes an AIQ Toolkit artifact to a local registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AgentIQ plugin wheel and it's corrosponding discovery
+            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -62,10 +62,10 @@ class LocalRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AgentIQ artifacts from a local registry.
+        """Download and install AIQ Toolkit artifacts from a local registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AgentIQ artifact.
+            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
@@ -83,7 +83,7 @@ class LocalRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches the local aiq registry for relevant AgentIQ components.
+        """Searches the local aiq registry for relevant AIQ Toolkit components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.

aiq/registry_handlers/local/register_local.py

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
 
 
 class LocalRegistryHandlerConfig(RegistryHandlerBaseConfig, name="local"):
-    """Interact with the local AgentIQ environment to search and uninstall AgentIQ components."""
+    """Interact with the local AIQ Toolkit environment to search and uninstall AIQ Toolkit components."""
 
     pass
 

aiq/registry_handlers/package_utils.py

@@ -132,7 +132,7 @@ def build_wheel(package_root: str) -> WheelData:
 
 
 def build_package_metadata(wheel_data: WheelData | None) -> dict[AIQComponentEnum, list[dict | DiscoveryMetadata]]:
-    """Loads discovery metadata for all registered AgentIQ components included in this Python package.
+    """Loads discovery metadata for all registered AIQ Toolkit components included in this Python package.
 
     Args:
         wheel_data (WheelData): Data model containing a built python wheel and its corresponding metadata.
@@ -180,7 +180,7 @@ def build_package_metadata(wheel_data: WheelData | None) -> dict[AIQComponentEnu
 
 
 def build_aiq_artifact(package_root: str) -> AIQArtifact:
-    """Builds a complete AgentIQ Artifact that can be published for discovery and reuse.
+    """Builds a complete AIQ Toolkit Artifact that can be published for discovery and reuse.
 
     Args:
         package_root (str): Path to root of python package

aiq/registry_handlers/pypi/pypi_handler.py

@@ -60,10 +60,10 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AgentIQ artifact to a PyPI remote registry.
+        """Publishes an AIQ Toolkit artifact to a PyPI remote registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AgentIQ plugin wheel and it's corrosponding discovery
+            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -101,10 +101,10 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AgentIQ artifacts from a remote PyPI remote registry.
+        """Download and install AIQ Toolkit artifacts from a remote PyPI remote registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AgentIQ artifact.
+            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse, None]]: A response message that includes a the pulled packages and a
@@ -160,7 +160,7 @@ class PypiRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches a remote PyPI registry for relevant AgentIQ components.
+        """Searches a remote PyPI registry for relevant AIQ Toolkit components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.

aiq/registry_handlers/pypi/register_pypi.py

@@ -25,9 +25,9 @@ class PypiRegistryHandlerConfig(RegistryHandlerBaseConfig, name="pypi"):
     endpoint: str = Field(description="A string representing the remote endpoint.")
     token: str | None = Field(default=None,
                               description="The authentication token to use when interacting with the registry.")
-    publish_route: str = Field(description="The route to the AgentIQ publish service.")
-    pull_route: str = Field(description="The route to the AgentIQ pull service.")
-    search_route: str = Field(default="simple", description="The route to the AgentIQ search service.")
+    publish_route: str = Field(description="The route to the AIQ Toolkit publish service.")
+    pull_route: str = Field(description="The route to the AIQ Toolkit pull service.")
+    search_route: str = Field(default="simple", description="The route to the AIQ Toolkit search service.")
 
 
 @register_registry_handler(config_type=PypiRegistryHandlerConfig)

aiq/registry_handlers/registry_handler_base.py

@@ -33,7 +33,7 @@ from aiq.registry_handlers.schemas.search import VisualizeFields
 
 
 class AbstractRegistryHandler(ABC):
-    """Base class outlining the interfaces for remote AgentIQ registry interactions."""
+    """Base class outlining the interfaces for remote AIQ Toolkit registry interactions."""
 
     def __init__(self):
         self._discovery_metadata: dict[AIQComponentEnum, list[dict | DiscoveryMetadata]] = {}
@@ -45,10 +45,10 @@ class AbstractRegistryHandler(ABC):
     @abstractmethod
     @asynccontextmanager
     async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AgentIQ artifact to a remote registry.
+        """Publishes an AIQ Toolkit artifact to a remote registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AgentIQ plugin wheel and it's corrosponding discovery
+            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -61,10 +61,10 @@ class AbstractRegistryHandler(ABC):
     @abstractmethod
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AgentIQ artifacts from a remote registry.
+        """Download and install AIQ Toolkit artifacts from a remote registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AgentIQ artifact.
+            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
@@ -76,7 +76,7 @@ class AbstractRegistryHandler(ABC):
     @abstractmethod
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches the local aiq registry for relevant AgentIQ components.
+        """Searches the local aiq registry for relevant AIQ Toolkit components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.
@@ -118,7 +118,7 @@ class AbstractRegistryHandler(ABC):
         from rich.table import Table
         from rich.text import Text
 
-        table = Table(title="AgentIQ Search Results", padding=(0, 1), show_lines=True)
+        table = Table(title="AIQ Toolkit Search Results", padding=(0, 1), show_lines=True)
         for column in VisualizeFields:
             table.add_column(column.value)
 

aiq/registry_handlers/rest/register_rest.py

@@ -27,10 +27,10 @@ class RestRegistryHandlerConfig(RegistryHandlerBaseConfig, name="rest"):
     endpoint: str = Field(description="A string representing the remote endpoint.")
     token: str | None = Field(default=None,
                               description="The authentication token to use when interacting with the registry.")
-    publish_route: str = Field(default="", description="The route to the AgentIQ publish service.")
-    pull_route: str = Field(default="", description="The route to the AgentIQ pull service.")
-    search_route: str = Field(default="", description="The route to the AgentIQ search service")
-    remove_route: str = Field(default="", description="The route to the AgentIQ remove service")
+    publish_route: str = Field(default="", description="The route to the AIQ Toolkit publish service.")
+    pull_route: str = Field(default="", description="The route to the AIQ Toolkit pull service.")
+    search_route: str = Field(default="", description="The route to the AIQ Toolkit search service")
+    remove_route: str = Field(default="", description="The route to the AIQ Toolkit remove service")
 
 
 @register_registry_handler(config_type=RestRegistryHandlerConfig)

aiq/registry_handlers/rest/rest_handler.py

@@ -62,10 +62,10 @@ class RestRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def publish(self, artifact: AIQArtifact) -> AsyncGenerator[PublishResponse]:
-        """Publishes an AgentIQ artifact to a remote REST registry.
+        """Publishes an AIQ Toolkit artifact to a remote REST registry.
 
         Args:
-            artifact (AIQArtifact): An artifact that contain AgentIQ plugin wheel and it's corrosponding discovery
+            artifact (AIQArtifact): An artifact that contain AIQ Toolkit plugin wheel and it's corrosponding discovery
             metadata.
 
         Yields:
@@ -98,10 +98,10 @@ class RestRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def pull(self, packages: PullRequestPackages) -> AsyncGenerator[PullResponse]:
-        """Download and install AgentIQ artifacts from a remote REST registry.
+        """Download and install AIQ Toolkit artifacts from a remote REST registry.
 
         Args:
-            packages (PullRequestPackages): Parameters used to pull the AgentIQ artifact.
+            packages (PullRequestPackages): Parameters used to pull the AIQ Toolkit artifact.
 
         Yields:
             Iterator[AsyncGenerator[PullResponse]]: A response message that includes a the pulled packages and a
@@ -165,7 +165,7 @@ class RestRegistryHandler(AbstractRegistryHandler):
 
     @asynccontextmanager
     async def search(self, query: SearchQuery) -> AsyncGenerator[SearchResponse]:
-        """Searches a remote REST registry for relevant AgentIQ components.
+        """Searches a remote REST registry for relevant AIQ Toolkit components.
 
         Args:
             query (SearchQuery): Parameters of the search to be performed.

aiq/registry_handlers/schemas/package.py

@@ -28,7 +28,7 @@ class WheelData(BaseModel):
         package_name (str): The name of the python package.
         toml_project (dict): A dictionary containing data about the python project.
         toml_dependencies (set): The list of dependencies provided in the pyproject.toml file.
-        toml_aiq_packages (set): The AgentIQ plugins listed in the pyproject.toml.
+        toml_aiq_packages (set): The AIQ Toolkit plugins listed in the pyproject.toml.
         union_dependencies (set): The union of toml_dependencies and toml_aiq_packages.
         whl_path (str): The path to the package wheel file.
         whl_base64 (str): Base64 encoded string of the wheel file.

aiq/registry_handlers/schemas/publish.py

@@ -25,10 +25,10 @@ logger = logging.getLogger(__name__)
 
 
 class BuiltAIQArtifact(BaseModel):
-    """An AgentIQ artifact including base64 encoded string of wheel package and corrosponding discovery metadata.
+    """An AIQ Toolkit artifact including base64 encoded string of wheel package and corrosponding discovery metadata.
 
     Args:
-        whl (str): A base64 encoded string of an AgentIQ package wheel (.whl).
+        whl (str): A base64 encoded string of an AIQ Toolkit package wheel (.whl).
 
         metadata (dict[AIQComponentEnum, list[DiscoveryMetadata]]): Provides rich discover metadata for developers to
         quickly find useful components.
@@ -39,10 +39,10 @@ class BuiltAIQArtifact(BaseModel):
 
 
 class AIQArtifact(BaseModel):
-    """An AgentIQ artifact including base64 encoded string of wheel package and corrosponding discovery metadata.
+    """An AIQ Toolkit artifact including base64 encoded string of wheel package and corrosponding discovery metadata.
 
     Args:
-        artifact (BuildAIQArtifact): An AgentIQ artifact including base64 encoded string of wheel package and
+        artifact (BuildAIQArtifact): An AIQ Toolkit artifact including base64 encoded string of wheel package and
         corrosponding discovery metadata.
 
         whl_path (str): A local path to the built wheel package.

aiq/registry_handlers/schemas/pull.py

@@ -27,7 +27,7 @@ class PulledPackage(BaseModel):
     """Represents a data model of a pulled package containing the package wheel and its name.
 
     Args:
-        whl (str): Base64 encoded string of the AgentIQ python package wheel (.whl).
+        whl (str): Base64 encoded string of the AIQ Toolkit python package wheel (.whl).
         whl_name (str): A string representing the wheel filename.
     """
 
@@ -37,7 +37,8 @@ class PulledPackage(BaseModel):
 
 class PullResponse(BaseModel):
     """
-    Represents a data model of the expected respones from a AgentIQ pull request, including detailed status information.
+    Represents a data model of the expected respones from a AIQ Toolkit pull request, including detailed status
+    information.
 
     Args:
         packages (list[PulledPackage]): A list of pulled packages included in the pull request.
@@ -71,11 +72,11 @@ class PullRequestPackage(BaseModel):
 
 
 class PullRequestPackages(BaseModel):
-    """Represents a list of all packages th download and install in the local AgentIQ environment.
+    """Represents a list of all packages th download and install in the local AIQ Toolkit environment.
 
     Args:
         packages (list[typing.Union[PackageNameVersion, PullPackageWhl]]): A list of packages that can be
-            downloaded and installed in the local AgentIQ environment.
+            downloaded and installed in the local AIQ Toolkit environment.
     """
 
     packages: list[PackageNameVersion | PullPackageWhl]