send end timestamp if not set before record

AgentOps-AI · May 2, 2024 · d397547 · d397547
1 parent 139488a
commit d397547
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agentops/client.py b/agentops/client.py
@@ -135,7 +135,8 @@ def record(self, event: Event | ErrorEvent):
             Args:
                 event (Event): The event to record.
         """
-
+        if not event.end_timestamp or event.init_timestamp == event.end_timestamp:
+            event.end_timestamp = get_ISO_time()
         if self._session is not None and not self._session.has_ended:
             if isinstance(event, ErrorEvent):
                 if event.trigger_event: