Don't create subtasks when not using tools

griptape-ai · Dec 26, 2024 · 1948bff · 1948bff
1 parent af1e5eb
commit 1948bff
Showing 1 changed file with 16 additions and 13 deletions.
diff --git a/griptape/tasks/prompt_task.py b/griptape/tasks/prompt_task.py
@@ -203,21 +203,24 @@ def try_run(self) -> BaseArtifact:
             self.prompt_driver.tokenizer.stop_sequences.extend([self.response_stop_sequence])
 
         result = self.prompt_driver.run(self.prompt_stack)
-        subtask = self.add_subtask(ActionsSubtask(result.to_artifact()))
-
-        while True:
-            if subtask.output is None:
-                if len(self.subtasks) >= self.max_subtasks:
-                    subtask.output = ErrorArtifact(f"Exceeded tool limit of {self.max_subtasks} subtasks per task")
+        if self.tools:
+            subtask = self.add_subtask(ActionsSubtask(result.to_artifact()))
+
+            while True:
+                if subtask.output is None:
+                    if len(self.subtasks) >= self.max_subtasks:
+                        subtask.output = ErrorArtifact(f"Exceeded tool limit of {self.max_subtasks} subtasks per task")
+                    else:
+                        subtask.run()
+
+                        result = self.prompt_driver.run(self.prompt_stack)
+                        subtask = self.add_subtask(ActionsSubtask(result.to_artifact()))
                 else:
-                    subtask.run()
-
-                    result = self.prompt_driver.run(self.prompt_stack)
-                    subtask = self.add_subtask(ActionsSubtask(result.to_artifact()))
-            else:
-                break
+                    break
 
-        self.output = subtask.output
+            self.output = subtask.output
+        else:
+            self.output = result.to_artifact()
 
         return self.output