From 3b13ff2a47be619a6b296a32b8793bdbe8f3f1da Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Wed, 4 Oct 2023 20:37:09 -0700 Subject: [PATCH] custom entity event data --- pyproject.toml | 2 +- wyvern/components/events/events.py | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index d5ee43e..55a3f9f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "wyvern-ai" -version = "0.0.24" +version = "0.0.25" description = "" authors = ["Wyvern AI "] readme = "README.md" diff --git a/wyvern/components/events/events.py b/wyvern/components/events/events.py index 4014fc8..16881ae 100644 --- a/wyvern/components/events/events.py +++ b/wyvern/components/events/events.py @@ -61,6 +61,10 @@ class EntityEventData(BaseModel): entity_identifier_type: str +class CustomEntityEventData(EntityEventData): + event_name: str + + ENTITY_EVENT_DATA_TYPE = TypeVar("ENTITY_EVENT_DATA_TYPE", bound=EntityEventData)