mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-07-26 17:23:34 +00:00
d3a81ac65b
Raise RepairableException when response lacks a top-level text or message string. The existing recovery path now warns the model to correct the call instead of exposing a KeyError, while preserving legacy message arguments.
26 lines
1.1 KiB
Python
26 lines
1.1 KiB
Python
from helpers.errors import RepairableException
|
|
from helpers.tool import Tool, Response
|
|
|
|
|
|
class ResponseTool(Tool):
|
|
|
|
async def execute(self, **kwargs):
|
|
for key in ("text", "message"):
|
|
message = self.args.get(key)
|
|
if isinstance(message, str):
|
|
return Response(message=message, break_loop=True)
|
|
raise RepairableException(
|
|
"response tool requires a top-level text or message string argument"
|
|
)
|
|
|
|
async def before_execution(self, **kwargs):
|
|
# self.log = self.agent.context.log.log(type="response", heading=f"{self.agent.agent_name}: Responding", content=self.args.get("text", ""))
|
|
# don't log here anymore, we have the live_response extension now
|
|
pass
|
|
|
|
async def after_execution(self, response, **kwargs):
|
|
# do not add anything to the history or output
|
|
|
|
if self.loop_data and "log_item_response" in self.loop_data.params_temporary:
|
|
log = self.loop_data.params_temporary["log_item_response"]
|
|
log.update(finished=True) # mark the message as finished
|