From dbec973677befae81f25314c4e13bd7abd187a46 Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 7 Jul 2023 17:54:07 -0400 Subject: [PATCH] clean up --- swarms/agents/workers/worker.py | 1 + 1 file changed, 1 insertion(+) diff --git a/swarms/agents/workers/worker.py b/swarms/agents/workers/worker.py index b1fa1c02..138e01c7 100644 --- a/swarms/agents/workers/worker.py +++ b/swarms/agents/workers/worker.py @@ -44,6 +44,7 @@ class WorkerNode: raise TypeError("tool_input must be a dictionary") if 'prompt' not in tool_input: raise ValueError("tool_input must contain the key 'prompt'") + prompt = tool_input['prompt'] if prompt is None: raise ValueError("Prompt not found in tool_input")