aiqtoolkit 1.2.0a20250801__py3-none-any.whl → 1.2.0a20250802__py3-none-any.whl

aiq/agent/react_agent/register.py

@@ -18,7 +18,6 @@ import logging
 from pydantic import AliasChoices
 from pydantic import Field
 
-from aiq.agent.base import AGENT_LOG_PREFIX
 from aiq.builder.builder import Builder
 from aiq.builder.framework_enum import LLMFrameworkEnum
 from aiq.builder.function_info import FunctionInfo
@@ -79,6 +78,7 @@ async def react_agent_workflow(config: ReActAgentWorkflowConfig, builder: Builde
     from langchain_core.messages import trim_messages
     from langgraph.graph.graph import CompiledGraph
 
+    from aiq.agent.base import AGENT_LOG_PREFIX
     from aiq.agent.react_agent.agent import ReActAgentGraph
     from aiq.agent.react_agent.agent import ReActGraphState
     from aiq.agent.react_agent.agent import create_react_agent_prompt

aiq/agent/reasoning_agent/reasoning_agent.py

@@ -19,7 +19,6 @@ from collections.abc import AsyncGenerator
 
 from pydantic import Field
 
-from aiq.agent.base import AGENT_LOG_PREFIX
 from aiq.builder.builder import Builder
 from aiq.builder.framework_enum import LLMFrameworkEnum
 from aiq.builder.function_info import FunctionInfo
@@ -86,6 +85,8 @@ async def build_reasoning_function(config: ReasoningFunctionConfig, builder: Bui
     from langchain_core.language_models import BaseChatModel
     from langchain_core.prompts import PromptTemplate
 
+    from aiq.agent.base import AGENT_LOG_PREFIX
+
     def remove_r1_think_tags(text: str):
         pattern = r'(<think>)?.*?</think>\s*(.*)'
 

aiq/agent/tool_calling_agent/register.py

@@ -17,7 +17,6 @@ import logging
 
 from pydantic import Field
 
-from aiq.agent.base import AGENT_LOG_PREFIX
 from aiq.builder.builder import Builder
 from aiq.builder.framework_enum import LLMFrameworkEnum
 from aiq.builder.function_info import FunctionInfo
@@ -49,6 +48,8 @@ async def tool_calling_agent_workflow(config: ToolCallAgentWorkflowConfig, build
     from langchain_core.messages.human import HumanMessage
     from langgraph.graph.graph import CompiledGraph
 
+    from aiq.agent.base import AGENT_LOG_PREFIX
+
     from .agent import ToolCallAgentGraph
     from .agent import ToolCallAgentGraphState
 

aiq/cli/commands/start.py

@@ -190,11 +190,6 @@ class StartCommandGroup(click.Group):
         # Override default front end config with values from the config file for serverless execution modes.
         # Check that we have the right kind of front end
         if (not isinstance(config.general.front_end, front_end.config_type)):
-            logger.warning(
-                "The front end type in the config file (%s) does not match the command name (%s). "
-                "Overwriting the config file front end.",
-                config.general.front_end.type,
-                cmd_name)
 
             # Set the front end config
             config.general.front_end = front_end.config_type()

aiq/observability/processor/callback_processor.py

@@ -17,7 +17,6 @@ from abc import abstractmethod
 from collections.abc import Awaitable
 from collections.abc import Callable
 from typing import Any
-from typing import Generic
 from typing import TypeVar
 
 from aiq.observability.processor.processor import Processor
@@ -26,7 +25,7 @@ InputT = TypeVar('InputT')
 OutputT = TypeVar('OutputT')
 
 
-class CallbackProcessor(Processor[InputT, OutputT], Generic[InputT, OutputT]):
+class CallbackProcessor(Processor[InputT, OutputT]):
     """Abstract base class for processors that support done callbacks.
 
     Processors inheriting from this class can register callbacks that are

aiq/runtime/loader.py

@@ -175,8 +175,8 @@ def discover_and_register_plugins(plugin_type: PluginTypes):
                 # Log a warning if the plugin took a long time to load. This can be useful for debugging slow imports.
                 # The threshold is 300 ms if no plugins have been loaded yet, and 100 ms otherwise. Triple the threshold
                 # if a debugger is attached.
-                if (elapsed_time > (300.0 if count == 0 else 100.0) * (3 if is_debugger_attached() else 1)):
-                    logger.warning(
+                if (elapsed_time > (300.0 if count == 0 else 150.0) * (3 if is_debugger_attached() else 1)):
+                    logger.debug(
                         "Loading module '%s' from entry point '%s' took a long time (%f ms). "
                         "Ensure all imports are inside your registered functions.",
                         entry_point.module,

{aiqtoolkit-1.2.0a20250801.dist-info → aiqtoolkit-1.2.0a20250802.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiqtoolkit
-Version: 1.2.0a20250801
+Version: 1.2.0a20250802
 Summary: NVIDIA Agent Intelligence toolkit
 Author: NVIDIA Corporation
 Maintainer: NVIDIA Corporation

{aiqtoolkit-1.2.0a20250801.dist-info → aiqtoolkit-1.2.0a20250802.dist-info}/RECORD

@@ -6,16 +6,16 @@ aiq/agent/react_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3h
 aiq/agent/react_agent/agent.py,sha256=w3IkPuU13JElSmmGMbYBo3YZF4LrDXvPHbEZ_ZL-L8s,19424
 aiq/agent/react_agent/output_parser.py,sha256=m7K6wRwtckBBpAHqOf3BZ9mqZLwrP13Kxz5fvNxbyZE,4219
 aiq/agent/react_agent/prompt.py,sha256=iGPBU6kh1xbp4QsU1p3o4A0JDov23J1EVM3HSAX6S0A,1713
-aiq/agent/react_agent/register.py,sha256=O0IYtUQ_TVtmaBleYKdpthkspX2fAhB0l8DGYLoE6F0,8128
+aiq/agent/react_agent/register.py,sha256=Qz7KO6tZKMB9TGmy-eJzAPmsJEbEl2lu7trXI4a2CmY,8132
 aiq/agent/reasoning_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-aiq/agent/reasoning_agent/reasoning_agent.py,sha256=1grbjv8c3neFgcTAl8qYeyPJ3B6gcEqkR6I68LOFPT8,9453
+aiq/agent/reasoning_agent/reasoning_agent.py,sha256=lutxHz3T0HUiFyuQfWmSg-MVRw2YTKQJrYCABtHV6cs,9458
 aiq/agent/rewoo_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/agent/rewoo_agent/agent.py,sha256=C8zUpbEFJ0De1tzMNvpxY66Qll2ekjAqSKCIcL0ftLA,19009
 aiq/agent/rewoo_agent/prompt.py,sha256=2XsuI-db_qmH02ypx_IDvi6jTak15cqt_4pZkUv9TFk,3929
 aiq/agent/rewoo_agent/register.py,sha256=krm0dUqM5RZpojiLZRpTgAsE0KGqa2NS2QCtlG70EJE,8128
 aiq/agent/tool_calling_agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/agent/tool_calling_agent/agent.py,sha256=e6VOeBnX_cHXUbnGW8N-ByRtHg4GrbsJecFhA9w0Ksk,5718
-aiq/agent/tool_calling_agent/register.py,sha256=kqcN2uovVBQxrIx5MszBS65opbhBrCRlAw00TlG2i30,5408
+aiq/agent/tool_calling_agent/register.py,sha256=Vf_7tOYYDwotxPoPWMuMMr4ZJrPLeLjqBEQ-qVsHbzU,5413
 aiq/authentication/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/authentication/interfaces.py,sha256=jfsbVx0MTrxZonyTEH0YFcSJoyFHVkfVHFhm9v3jMrY,3317
 aiq/authentication/register.py,sha256=dYChd2HRv-uv4m8Ebo2sLfbVnksT8D3jEWA-QT-MzX0,947
@@ -63,7 +63,7 @@ aiq/cli/cli_utils/config_override.py,sha256=WuX9ki1W0Z6jTqjm553U_owWFxbVzjbKRWGJ
 aiq/cli/cli_utils/validation.py,sha256=GlKpoi3HfE5HELjmz5wk8ezGbb5iZeY0zmA3uxmCrBU,1302
 aiq/cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/cli/commands/evaluate.py,sha256=_pqAuvrNKBf0DvGpZFO28vAKBWp6izMpaLbAVnP57_4,4783
-aiq/cli/commands/start.py,sha256=8mnJYSToQ1XuYuRBRSonuOsjSL1wqSm8nwj3tH9cnAM,10097
+aiq/cli/commands/start.py,sha256=zQJdSW9xQbK1oahM-mWaIXm-B2E25n2SZFNv6bSkQ6Y,9838
 aiq/cli/commands/uninstall.py,sha256=D3PGizMGy-c3DLQ-HBcVYPOv0X8eyxFFw27jJW4kxig,3195
 aiq/cli/commands/validate.py,sha256=YfYNRK7m5te_jkKp1klhGp4PdUVCuDyVG4LRW1YXZk0,1669
 aiq/cli/commands/configure/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -274,7 +274,7 @@ aiq/observability/mixin/serialize_mixin.py,sha256=DgRHJpXCz9qHFYzhlTTx8Dkj297Eyl
 aiq/observability/mixin/type_introspection_mixin.py,sha256=VCb68SY_hitWrWLaK2UHQLkjn2jsgxSn9593T2N3zC0,6637
 aiq/observability/processor/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/observability/processor/batching_processor.py,sha256=dE5E_0oHwISXfCxGXpOJ5fEM-YKIfuPz5h9j9gpJoMQ,13857
-aiq/observability/processor/callback_processor.py,sha256=CTa9FVcoGXaYBRfarF7Ig9jymgmKHNgWRIy-OXtDZ1k,1600
+aiq/observability/processor/callback_processor.py,sha256=U1IhQq5x5H6FaoqWCjZZqOFb5RcTdd4ORoQZCHgkdC8,1547
 aiq/observability/processor/intermediate_step_serializer.py,sha256=UQgcHauq568TqVmF_FrInsE5Q_Piryrf_fDnuJbRtAc,1249
 aiq/observability/processor/processor.py,sha256=kfYe1EiQZkOPSnqIwQ6Rjz44j-EpreRAe2uIspmHK9w,2593
 aiq/observability/utils/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
@@ -355,7 +355,7 @@ aiq/retriever/nemo_retriever/__init__.py,sha256=GUJrgGtpvyMUCjUBvR3faAdv-tZzbU9W
 aiq/retriever/nemo_retriever/register.py,sha256=ODV-TZfXzDs1VJHHLdj2kC05odirtlQZSeh9c1zw8AQ,2893
 aiq/retriever/nemo_retriever/retriever.py,sha256=IvScUr9XuDLiMR__I3QsboLaM52N5D5Qu94qtTOGQw8,6958
 aiq/runtime/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
-aiq/runtime/loader.py,sha256=CqZzeIpN_aWUEASr4JBtQjmVtSv2DNpu6dFha_4iRI8,7119
+aiq/runtime/loader.py,sha256=drLE-OnrAPjFgWrQ3ZsnhHSbon8GkRsUk0p7Oe8Jn2I,7117
 aiq/runtime/runner.py,sha256=CqmlVAYfrBh3ml3t2n3V693RaNyxtK9ScWT4S-Isbr8,6365
 aiq/runtime/session.py,sha256=i1pIqopZCBgGJqVUskKLiBnZYH-lTdMhvFu56dXAU5A,6206
 aiq/runtime/user_metadata.py,sha256=9EiBc-EEJzOdpf3Q1obHqAdY_kRlJ1T0TVvY0Jonk6o,3692
@@ -428,10 +428,10 @@ aiq/utils/reactive/base/observer_base.py,sha256=UAlyAY_ky4q2t0P81RVFo2Bs_R7z5Nde
 aiq/utils/reactive/base/subject_base.py,sha256=Ed-AC6P7cT3qkW1EXjzbd5M9WpVoeN_9KCe3OM3FLU4,2521
 aiq/utils/settings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/utils/settings/global_settings.py,sha256=U9TCLdoZsKq5qOVGjREipGVv9e-FlStzqy5zv82_VYk,7454
-aiqtoolkit-1.2.0a20250801.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
-aiqtoolkit-1.2.0a20250801.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-aiqtoolkit-1.2.0a20250801.dist-info/METADATA,sha256=BrsA79HoQ5RbxW4dd9v8o5Ym7CqbsDV8ywi_GSbg9x8,21564
-aiqtoolkit-1.2.0a20250801.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit-1.2.0a20250801.dist-info/entry_points.txt,sha256=iZR3yrf1liXfbcLqn5_pUkLhZyr1bUw_Qh1d2i7gsv4,625
-aiqtoolkit-1.2.0a20250801.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit-1.2.0a20250801.dist-info/RECORD,,
+aiqtoolkit-1.2.0a20250802.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
+aiqtoolkit-1.2.0a20250802.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+aiqtoolkit-1.2.0a20250802.dist-info/METADATA,sha256=MNhTmfIeb6bpYVMTDj9EIj9kS8IWaf8JLfwBMzH6WQ0,21564
+aiqtoolkit-1.2.0a20250802.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit-1.2.0a20250802.dist-info/entry_points.txt,sha256=iZR3yrf1liXfbcLqn5_pUkLhZyr1bUw_Qh1d2i7gsv4,625
+aiqtoolkit-1.2.0a20250802.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit-1.2.0a20250802.dist-info/RECORD,,