danvics/haiku.rag
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

handle ActivitySnapshotEvent for skill sub-agent tool calls

Browse source
This commit is contained in:
Yiorgis Gozadinos 2026-03-13 11:37:24 +02:00
parent b09b45e8fa
commit 1d108b42ec
No known key found for this signature in database
1 changed files with 25 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
haiku_rag_slim/haiku/rag/chat/app.py
View file

@ -31,6 +31,7 @@ except ImportError:
try:
import textual_image.widget # noqa: F401 - import early for renderer detection
from ag_ui.core import (
ActivitySnapshotEvent,
AssistantMessage,
EventType,
RunAgentInput,
@ -271,6 +272,30 @@ class ChatApp(App):
elif event.type == EventType.TOOL_CALL_END:
assert isinstance(event, ToolCallEndEvent)
chat_history.mark_tool_complete(event.tool_call_id)
elif event.type == EventType.ACTIVITY_SNAPSHOT:
assert isinstance(event, ActivitySnapshotEvent)
content = event.content
if event.activity_type == "skill_tool_call":
tool_call_id = content["tool_call_id"]
skill_name = content.get("skill", "")
tool_name = content["tool_name"]
display_name = (
f"{skill_name} → {tool_name}"
if skill_name
else tool_name
)
args_str = content.get("args", "{}")
chat_history.hide_thinking()
await chat_history.add_tool_call(tool_call_id, display_name)
try:
args = json.loads(args_str)
chat_history.update_tool_args(tool_call_id, args)
except json.JSONDecodeError:
pass
await chat_history.show_thinking("Working...")
elif event.activity_type == "skill_tool_result":
tool_call_id = content["tool_call_id"]
chat_history.mark_tool_complete(tool_call_id)
elif event.type == EventType.STATE_DELTA:
assert isinstance(event, StateDeltaEvent)
patch = JsonPatch(event.delta)