From b83e7bffc0291709512c4413c332c04aad9ffa2f Mon Sep 17 00:00:00 2001 From: worldmozara Date: Tue, 14 Jul 2026 22:11:25 +0800 Subject: [PATCH] core/agent: add ErrorEvent and CancelledEvent --- src/plyngent/agent/__init__.py | 2 ++ src/plyngent/agent/events.py | 18 +++++++++++++++++- 2 files changed, 19 insertions(+), 1 deletion(-) diff --git a/src/plyngent/agent/__init__.py b/src/plyngent/agent/__init__.py index 3b6fd6d..f3d9c50 100644 --- a/src/plyngent/agent/__init__.py +++ b/src/plyngent/agent/__init__.py @@ -2,6 +2,8 @@ from .chat import ChatAgent as ChatAgent from .client import ChatClient as ChatClient from .events import AgentEvent as AgentEvent from .events import AssistantMessageEvent as AssistantMessageEvent +from .events import CancelledEvent as CancelledEvent +from .events import ErrorEvent as ErrorEvent from .events import MaxRoundsEvent as MaxRoundsEvent from .events import TextDeltaEvent as TextDeltaEvent from .events import ToolCallEvent as ToolCallEvent diff --git a/src/plyngent/agent/events.py b/src/plyngent/agent/events.py index 41d2f4c..855accc 100644 --- a/src/plyngent/agent/events.py +++ b/src/plyngent/agent/events.py @@ -28,4 +28,20 @@ class MaxRoundsEvent(Struct, tag_field="type", tag="max_rounds"): continued: bool = False -type AgentEvent = TextDeltaEvent | AssistantMessageEvent | ToolCallEvent | ToolResultEvent | MaxRoundsEvent +class ErrorEvent(Struct, tag_field="type", tag="error"): + message: str + + +class CancelledEvent(Struct, tag_field="type", tag="cancelled"): + pass + + +type AgentEvent = ( + TextDeltaEvent + | AssistantMessageEvent + | ToolCallEvent + | ToolResultEvent + | MaxRoundsEvent + | ErrorEvent + | CancelledEvent +)