diff --git a/monkey/common/events/credentials_stolen_events.py b/monkey/common/events/credentials_stolen_events.py index 184ee81c3..c9f19e5d3 100644 --- a/monkey/common/events/credentials_stolen_events.py +++ b/monkey/common/events/credentials_stolen_events.py @@ -1,12 +1,14 @@ -from dataclasses import dataclass, field +from dataclasses import field from typing import Sequence +import pydantic + from common.credentials import Credentials from . import AbstractAgentEvent -@dataclass(frozen=True) +@pydantic.dataclasses.dataclass(frozen=True) class CredentialsStolenEvent(AbstractAgentEvent): """ An event that occurs when an agent collects credentials from the victim