From 88b3bf1621f88296fbbcfc83f6f02c8ec6fe02cc Mon Sep 17 00:00:00 2001
From: Jason Schleifer <jason.schleifer@gmail.com>
Date: Fri, 21 Jun 2024 10:08:17 +1200
Subject: [PATCH] checking for simple models

---
 nodes/tasks.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/nodes/tasks.py b/nodes/tasks.py
index dcbaa60..63087e9 100644
--- a/nodes/tasks.py
+++ b/nodes/tasks.py
@@ -37,6 +37,7 @@ from griptape.utils import load_file
 from schema import Schema
 
 from ..py.griptape_config import get_config
+from .agent import model_check
 from .base_audio_task import gtUIBaseAudioTask
 from .base_image_task import gtUIBaseImageTask
 from .base_task import gtUIBaseTask
@@ -531,6 +532,18 @@ class gtUIToolkitTask(gtUIBaseTask):
         if not agent:
             agent = Agent()
 
+        if model_check(agent):
+            return (
+                dedent(
+                    """
+                I'm sorry, this agent uses a simple model that can't handle ToolkitTasks.
+                You might want to try using a different Agent Configuration, or use a simple ToolTask instead.
+
+                Reach out for help on Discord (https://discord.gg/gnWRz88eym) if you would like some help.
+                """
+                ),
+                agent,
+            )
         task = ToolkitTask(prompt_text, tools=tools)
         try:
             agent.add_task(task)