diff --git a/griptape/events/event_listener.py b/griptape/events/event_listener.py index 704e20d32..df4a2668a 100644 --- a/griptape/events/event_listener.py +++ b/griptape/events/event_listener.py @@ -49,7 +49,7 @@ def __exit__(self, type, value, traceback) -> None: # noqa: ANN001, A002 def publish_event(self, event: T, *, flush: bool = False) -> None: event_types = self.event_types - if event_types is None or type(event) in event_types: + if event_types is None or any(isinstance(event, event_type) for event_type in event_types): handled_event = event if self.handler is not None: handled_event = self.handler(event)