From ec40a9c6ad239ce74f6bed451fd6924d4730bb51 Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Fri, 16 Sep 2022 08:34:46 -0400 Subject: [PATCH] Common: Inherit from InfectionMonkeyModelConfig in CredsStolenEvent --- monkey/common/events/credentials_stolen_events.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monkey/common/events/credentials_stolen_events.py b/monkey/common/events/credentials_stolen_events.py index 8c7dfdd1e..84f4b44f3 100644 --- a/monkey/common/events/credentials_stolen_events.py +++ b/monkey/common/events/credentials_stolen_events.py @@ -2,6 +2,7 @@ from typing import Sequence from pydantic import Field +from common.base_models import InfectionMonkeyModelConfig from common.credentials import Credentials from ..credentials.encoding import SecretEncodingConfig @@ -18,5 +19,5 @@ class CredentialsStolenEvent(AbstractAgentEvent): stolen_credentials: Sequence[Credentials] = Field(default_factory=list) - class Config(SecretEncodingConfig): + class Config(SecretEncodingConfig, InfectionMonkeyModelConfig): pass