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")