Spaces:
Running
Running
File size: 1,829 Bytes
bc1af83 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 | 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
|