langroid 0.56.0__py3-none-any.whl → 0.56.1__py3-none-any.whl

langroid/agent/tools/task_tool.py

@@ -88,10 +88,9 @@ class TaskTool(ToolMessage):
         description="Optional max iterations for the sub-agent to run the task",
     )
 
-    def handle(self, agent: ChatAgent) -> Optional[ChatDocument]:
+    def _set_up_task(self, agent: ChatAgent) -> Task:
         """
-        Handle the TaskTool by creating a sub-agent with specified tools
-        and running the task non-interactively.
+        Helper method to set up a task for the sub-agent.
 
         Args:
             agent: The parent ChatAgent that is handling this tool
@@ -144,6 +143,32 @@ class TaskTool(ToolMessage):
         # Create a non-interactive task
         task = Task(sub_agent, interactive=False)
 
+        return task
+
+    def handle(self, agent: ChatAgent) -> Optional[ChatDocument]:
+        """
+
+        Handle the TaskTool by creating a sub-agent with specified tools
+        and running the task non-interactively.
+
+        Args:
+            agent: The parent ChatAgent that is handling this tool
+        """
+
+        task = self._set_up_task(agent)
         # Run the task on the prompt, and return the result
         result = task.run(self.prompt, turns=self.max_iterations or 10)
         return result
+
+    async def handle_async(self, agent: ChatAgent) -> Optional[ChatDocument]:
+        """
+        Async method to handle the TaskTool by creating a sub-agent with specified tools
+        and running the task non-interactively.
+
+        Args:
+            agent: The parent ChatAgent that is handling this tool
+        """
+        task = self._set_up_task(agent)
+        # Run the task on the prompt, and return the result
+        result = await task.run_async(self.prompt, turns=self.max_iterations or 10)
+        return result

{langroid-0.56.0.dist-info → langroid-0.56.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: langroid
-Version: 0.56.0
+Version: 0.56.1
 Summary: Harness LLMs with Multi-Agent Programming
 Author-email: Prasad Chalasani <pchalasani@gmail.com>
 License: MIT

{langroid-0.56.0.dist-info → langroid-0.56.1.dist-info}/RECORD

@@ -54,7 +54,7 @@ langroid/agent/tools/recipient_tool.py,sha256=dr0yTxgNEIoxUYxH6TtaExC4G_8WdJ0xGo
 langroid/agent/tools/retrieval_tool.py,sha256=zcAV20PP_6VzSd-UE-IJcabaBseFL_QNz59Bnig8-lE,946
 langroid/agent/tools/rewind_tool.py,sha256=XAXL3BpNhCmBGYq_qi_sZfHJuIw7NY2jp4wnojJ7WRs,5606
 langroid/agent/tools/segment_extract_tool.py,sha256=__srZ_VGYLVOdPrITUM8S0HpmX4q7r5FHWMDdHdEv8w,1440
-langroid/agent/tools/task_tool.py,sha256=jM1HjhPU6gicTjPZ6TBEeAFgGMWGz-ylVS0M1tb_ELk,5705
+langroid/agent/tools/task_tool.py,sha256=O-3-tALocGuwfxXikBzRWFenFKZWIlOoRVvnZ5mf7io,6514
 langroid/agent/tools/tavily_search_tool.py,sha256=soI-j0HdgVQLf09wRQScaEK4b5RpAX9C4cwOivRFWWI,1903
 langroid/agent/tools/mcp/__init__.py,sha256=DJNM0VeFnFS3pJKCyFGggT8JVjVu0rBzrGzasT1HaSM,387
 langroid/agent/tools/mcp/decorators.py,sha256=h7dterhsmvWJ8q4mp_OopmuG2DF71ty8cZwOyzdDZuk,1127
@@ -137,7 +137,7 @@ langroid/vector_store/pineconedb.py,sha256=otxXZNaBKb9f_H75HTaU3lMHiaR2NUp5MqwLZ
 langroid/vector_store/postgres.py,sha256=wHPtIi2qM4fhO4pMQr95pz1ZCe7dTb2hxl4VYspGZoA,16104
 langroid/vector_store/qdrantdb.py,sha256=O6dSBoDZ0jzfeVBd7LLvsXu083xs2fxXtPa9gGX3JX4,18443
 langroid/vector_store/weaviatedb.py,sha256=Yn8pg139gOy3zkaPfoTbMXEEBCiLiYa1MU5d_3UA1K4,11847
-langroid-0.56.0.dist-info/METADATA,sha256=Ce8x84ihG8zSmHt62EzuTYlxxLH68R-CgT1ksrKzC7A,65366
-langroid-0.56.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-langroid-0.56.0.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
-langroid-0.56.0.dist-info/RECORD,,
+langroid-0.56.1.dist-info/METADATA,sha256=Yz9pHxVllXS-Sihx0AqWVjuXom9AjDbc_2DsU0wCgks,65366
+langroid-0.56.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+langroid-0.56.1.dist-info/licenses/LICENSE,sha256=EgVbvA6VSYgUlvC3RvPKehSg7MFaxWDsFuzLOsPPfJg,1065
+langroid-0.56.1.dist-info/RECORD,,