feat: unify debug runtime timelines
This commit is contained in:
@@ -521,6 +521,24 @@ class PromptBrainTests(unittest.IsolatedAsyncioTestCase):
|
||||
self.assertIn("点击“确认”", confirmation_messages[0]["content"])
|
||||
self.assertIn("弹窗标题:重要提示", confirmation_messages[0]["content"])
|
||||
self.assertIn("弹窗内容:请确认已阅读。", confirmation_messages[0]["content"])
|
||||
message_traces = [
|
||||
frame.message
|
||||
for frame in queued
|
||||
if isinstance(frame, OutputTransportMessageUrgentFrame)
|
||||
and frame.message.get("type") == "workflow-event"
|
||||
and frame.message.get("event")
|
||||
in {"message_started", "message_completed"}
|
||||
]
|
||||
self.assertEqual(
|
||||
[event["event"] for event in message_traces],
|
||||
["message_started", "message_completed"],
|
||||
)
|
||||
self.assertEqual(message_traces[0]["messageTitle"], "重要提示")
|
||||
self.assertEqual(
|
||||
message_traces[0]["messageContent"],
|
||||
"请确认已阅读。",
|
||||
)
|
||||
self.assertEqual(message_traces[1]["action"], "confirmed")
|
||||
self.assertEqual(
|
||||
sum(isinstance(frame, LLMRunFrame) for frame in queued),
|
||||
1,
|
||||
|
||||
Reference in New Issue
Block a user