aiqtoolkit 1.2.0a20250730__py3-none-any.whl → 1.2.0a20250731__py3-none-any.whl

aiq/front_ends/console/console_front_end_plugin.py

@@ -15,12 +15,10 @@
 
 import asyncio
 import logging
-from io import StringIO
 
 import click
 from colorama import Fore
 
-from aiq.builder.workflow_builder import WorkflowBuilder
 from aiq.data_models.interactive import HumanPromptModelType
 from aiq.data_models.interactive import HumanResponse
 from aiq.data_models.interactive import HumanResponseText
@@ -61,27 +59,9 @@ class ConsoleFrontEndPlugin(SimpleFrontEndPluginBase[ConsoleFrontEndConfig]):
         if (not self.front_end_config.input_query and not self.front_end_config.input_file):
             raise click.UsageError("Must specify either --input_query or --input_file")
 
-    async def run(self):
-
-        # Must yield the workflow function otherwise it cleans up
-        async with WorkflowBuilder.from_config(config=self.full_config) as builder:
-
-            session_manager: AIQSessionManager = None
-
-            if logger.isEnabledFor(logging.INFO):
-                stream = StringIO()
-
-                self.full_config.print_summary(stream=stream)
-
-                click.echo(stream.getvalue())
-
-                workflow = builder.build()
-                session_manager = AIQSessionManager(workflow)
-
-            await self.run_workflow(session_manager)
-
-    async def run_workflow(self, session_manager: AIQSessionManager = None):
+    async def run_workflow(self, session_manager: AIQSessionManager):
 
+        assert session_manager is not None, "Session manager must be provided"
         runner_outputs = None
 
         if (self.front_end_config.input_query):

aiq/front_ends/simple_base/simple_front_end_plugin_base.py

@@ -45,8 +45,10 @@ class SimpleFrontEndPluginBase(FrontEndBase[FrontEndConfigT], ABC):
 
                 click.echo(stream.getvalue())
 
-            await self.run_workflow(builder.build())
+            workflow = builder.build()
+            session_manager = AIQSessionManager(workflow)
+            await self.run_workflow(session_manager)
 
     @abstractmethod
-    async def run_workflow(self, session_manager: AIQSessionManager = None):
+    async def run_workflow(self, session_manager: AIQSessionManager):
         pass

{aiqtoolkit-1.2.0a20250730.dist-info → aiqtoolkit-1.2.0a20250731.dist-info}/METADATA

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

{aiqtoolkit-1.2.0a20250730.dist-info → aiqtoolkit-1.2.0a20250731.dist-info}/RECORD

@@ -208,7 +208,7 @@ aiq/front_ends/register.py,sha256=OKv1xi-g8WHtUMuIPhwjG6wOYqaGDD-Q9vDtKtT9d1Y,88
 aiq/front_ends/console/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/front_ends/console/authentication_flow_handler.py,sha256=pKgaAnBfr3Zc5V3yPiyU-oVZyCC-dNwjvhLn1gSxnUE,9566
 aiq/front_ends/console/console_front_end_config.py,sha256=vI81IK9GJll0t9P-3yb1JcUda5PLfdvyKc-636ZSqYU,1327
-aiq/front_ends/console/console_front_end_plugin.py,sha256=3qOLXKI6JCE33oh2MylrxMm_EonZlUCGjxZrPvTJ_4k,4535
+aiq/front_ends/console/console_front_end_plugin.py,sha256=WXrB5ugakwsoGxbhXJPJjyb5Evoe-eDHBPOnJdtWO9A,3937
 aiq/front_ends/console/register.py,sha256=a84M0jWUFTgOQVyrUiS7UJcxx84i1zhCb1yRkjhapiQ,1159
 aiq/front_ends/cron/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
 aiq/front_ends/fastapi/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
@@ -235,7 +235,7 @@ aiq/front_ends/mcp/mcp_front_end_plugin.py,sha256=XXyu5A5zWLH-eZgLDGcdCELXkwC1YP
 aiq/front_ends/mcp/register.py,sha256=9u5AJVH5UXiniP9bmsOjpfWVzQLHFlUPfL5HZ10Qwnw,1179
 aiq/front_ends/mcp/tool_converter.py,sha256=Pgb06Gtb8MVWeV_dAaI4Tl0Hono_mSuJAHRxTvSONHQ,9840
 aiq/front_ends/simple_base/__init__.py,sha256=Xs1JQ16L9btwreh4pdGKwskffAw1YFO48jKrU4ib_7c,685
-aiq/front_ends/simple_base/simple_front_end_plugin_base.py,sha256=HzcJFHZUZFnZJdw4YcRTG6nQbTDoBQSgKO6vwifSomk,1684
+aiq/front_ends/simple_base/simple_front_end_plugin_base.py,sha256=HFGa3qJXEK_og4lwXid_YGArlm1lz0GOZ89dMWn152E,1774
 aiq/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiq/llm/aws_bedrock_llm.py,sha256=FFlwS7xZYSTRuPWU_vxvTbmPTbuZpN0ByVxYukEThM4,2836
 aiq/llm/nim_llm.py,sha256=_9WwR4Pt9RCg8RG9oBeYxSt94KcZsMEVoKOFjig_2Zo,2179
@@ -427,10 +427,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.0a20250730.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
-aiqtoolkit-1.2.0a20250730.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
-aiqtoolkit-1.2.0a20250730.dist-info/METADATA,sha256=9q3CQ5wquTJIE_YV5Faodf2wXduAaWXpOVwFy0m8EEI,21564
-aiqtoolkit-1.2.0a20250730.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiqtoolkit-1.2.0a20250730.dist-info/entry_points.txt,sha256=iZR3yrf1liXfbcLqn5_pUkLhZyr1bUw_Qh1d2i7gsv4,625
-aiqtoolkit-1.2.0a20250730.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
-aiqtoolkit-1.2.0a20250730.dist-info/RECORD,,
+aiqtoolkit-1.2.0a20250731.dist-info/licenses/LICENSE-3rd-party.txt,sha256=8o7aySJa9CBvFshPcsRdJbczzdNyDGJ8b0J67WRUQ2k,183936
+aiqtoolkit-1.2.0a20250731.dist-info/licenses/LICENSE.md,sha256=QwcOLU5TJoTeUhuIXzhdCEEDDvorGiC6-3YTOl4TecE,11356
+aiqtoolkit-1.2.0a20250731.dist-info/METADATA,sha256=n1FtKvKcVGn7LJs3KZa1UUOJWXMi2bNI0baIy6rNGM8,21564
+aiqtoolkit-1.2.0a20250731.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiqtoolkit-1.2.0a20250731.dist-info/entry_points.txt,sha256=iZR3yrf1liXfbcLqn5_pUkLhZyr1bUw_Qh1d2i7gsv4,625
+aiqtoolkit-1.2.0a20250731.dist-info/top_level.txt,sha256=fo7AzYcNhZ_tRWrhGumtxwnxMew4xrT1iwouDy_f0Kc,4
+aiqtoolkit-1.2.0a20250731.dist-info/RECORD,,