mirror of
https://github.com/agent0ai/agent-zero.git
synced 2026-08-10 08:55:39 +00:00
87bafe10e0
Guard partially parsed tool arguments while streaming and render both text and legacy message response fields. Keep live response logs synchronized without allowing incomplete code arguments to raise.
51 lines
1.8 KiB
Python
51 lines
1.8 KiB
Python
from helpers import persist_chat, tokens
|
|
from helpers import extract_tools
|
|
from helpers.extension import Extension
|
|
from agent import LoopData
|
|
import asyncio
|
|
from helpers.log import LogItem
|
|
from helpers import log
|
|
|
|
|
|
class LiveResponse(Extension):
|
|
|
|
async def execute(
|
|
self,
|
|
loop_data: LoopData = LoopData(),
|
|
text: str = "",
|
|
parsed: dict = {},
|
|
**kwargs,
|
|
):
|
|
if not self.agent:
|
|
return
|
|
|
|
try:
|
|
tool_name, tool_args = extract_tools.normalize_tool_request(parsed)
|
|
message = tool_args.get("text")
|
|
if not isinstance(message, str) or not message.strip():
|
|
message = tool_args.get("message")
|
|
if (
|
|
tool_name != "response"
|
|
or not isinstance(message, str)
|
|
or not message.strip()
|
|
):
|
|
return # not a response
|
|
|
|
# create log message and store it in loop data temporary params
|
|
if "log_item_response" not in loop_data.params_temporary:
|
|
# Share id with the agent log item so branching covers the response bubble
|
|
gen_item = loop_data.params_temporary.get("log_item_generating")
|
|
shared_id = gen_item.id if gen_item and gen_item.id else ""
|
|
loop_data.params_temporary["log_item_response"] = (
|
|
self.agent.context.log.log(
|
|
type="response",
|
|
heading=f"icon://chat {self.agent.agent_name}: Responding",
|
|
id=shared_id,
|
|
)
|
|
)
|
|
|
|
# update log message
|
|
log_item = loop_data.params_temporary["log_item_response"]
|
|
log_item.update(content=message)
|
|
except Exception as e:
|
|
pass
|