From bc9c36e94247d8067f058f8a3e07b2f367bb3275 Mon Sep 17 00:00:00 2001
From: Jason Schleifer <jason.schleifer@gmail.com>
Date: Tue, 16 Jul 2024 06:43:46 +1200
Subject: [PATCH] added max_subtasks

---
 nodes/tasks.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/nodes/tasks.py b/nodes/tasks.py
index 8957c19..76f6cd5 100644
--- a/nodes/tasks.py
+++ b/nodes/tasks.py
@@ -578,6 +578,11 @@ class gtUIToolkitTask(gtUIBaseTask):
         inputs = super().INPUT_TYPES()
 
         # Update optional inputs to include 'tool' and adjust others as necessary
+        inputs["required"].update(
+            {
+                "max_subtasks": ("INT", {"default": 20}),
+            }
+        )
         inputs["optional"].update(
             {
                 "tools": ("TOOL_LIST",),
@@ -590,6 +595,7 @@ class gtUIToolkitTask(gtUIBaseTask):
         tools = kwargs.get("tools", [])
         input_string = kwargs.get("input_string", None)
         agent = kwargs.get("agent", None)
+        max_subtasks = kwargs.get("max_subtasks", 5)
         prompt_text = self.get_prompt_text(STRING, input_string)
 
         if len(tools) == 0:
@@ -606,7 +612,7 @@ class gtUIToolkitTask(gtUIBaseTask):
             response = agent.model_response(model)
             return (response, agent)
 
-        task = ToolkitTask(prompt_text, tools=tools)
+        task = ToolkitTask(prompt_text, tools=tools, max_subtasks=max_subtasks)
         # if deferred_evaluation:
         #     return ("Toolkit Task Created.", task)
         try: