pull/160/head
Kye 2 years ago
parent 800b77d489
commit dbec973677

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

Loading…
Cancel
Save