Fix: KeyError 'message' in response.py when both keys missing

[hurtdidit]

Bug Description

response.py execute() crashes with KeyError: 'message' when neither text nor message keys exist in self.args dictionary.

Current Buggy Code (line 7)

return Response(message=self.args["text"] if "text" in self.args else self.args["message"], break_loop=True)

When both keys are missing, this throws KeyError: 'message' and causes the agent to completely stall.

Proposed Fix

message_text = self.args.get("text") or self.args.get("message", "")
return Response(message=message_text, break_loop=True)

Using .get() with a safe fallback prevents the crash.

Impact

• Causes agents to completely stall when edge case occurs
• Confirmed on multiple Agent Zero instances
• Works fine in normal cases (happy path) but fails on error handling edge cases

Testing

1. Trigger response tool with missing text/message args (edge case)
2. With fix: Returns empty response safely instead of crashing
3. Normal usage: No change in behavior