From 0d5e6765188dc39db31e4a634678a1d5e238194d Mon Sep 17 00:00:00 2001 From: Teo Date: Tue, 26 Nov 2024 19:25:47 -0600 Subject: [PATCH] refactor(event): update ActionEvent constructor for clarity --- agentops/event.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/agentops/event.py b/agentops/event.py index 70ec059c..10abf859 100644 --- a/agentops/event.py +++ b/agentops/event.py @@ -61,6 +61,10 @@ class ActionEvent(Event): logs: Optional[Union[str, Sequence[Any]]] = None screenshot: Optional[str] = None + def __init__(self, action_type: Optional[str] = None, **kwargs): + super().__init__(event_type=EventType.ACTION.value, **kwargs) + self.action_type = action_type + @dataclass class LLMEvent(Event):