diff --git a/monkey/common/event_serializers/__init__.py b/monkey/common/event_serializers/__init__.py index 8701670ab..388edf6f3 100644 --- a/monkey/common/event_serializers/__init__.py +++ b/monkey/common/event_serializers/__init__.py @@ -1,3 +1,4 @@ from .i_event_serializer import IEventSerializer, JSONSerializable from .event_serializer_registry import EventSerializerRegistry from .pydantic_event_serializer import PydanticEventSerializer +from .register import register_common_agent_event_serializers diff --git a/monkey/common/event_serializers/register.py b/monkey/common/event_serializers/register.py new file mode 100644 index 000000000..6c533c9bf --- /dev/null +++ b/monkey/common/event_serializers/register.py @@ -0,0 +1,9 @@ +from common.events import CredentialsStolenEvent + +from . import EventSerializerRegistry, PydanticEventSerializer + + +def register_common_agent_event_serializers(event_serializer_registry: EventSerializerRegistry): + event_serializer_registry[CredentialsStolenEvent] = PydanticEventSerializer( + CredentialsStolenEvent + )