From e39ea075a1d534510611bba530263b792e62081f Mon Sep 17 00:00:00 2001
From: Gabriel Baker <gabriel.baker@lydia-app.com>
Date: Fri, 18 Apr 2025 09:53:59 +0200
Subject: [PATCH] Allow step end_time overwrite

---
 literalai/observability/step.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/literalai/observability/step.py b/literalai/observability/step.py
index 39d7a0c..95e99d5 100644
--- a/literalai/observability/step.py
+++ b/literalai/observability/step.py
@@ -402,7 +402,7 @@ def start(self):
         active_steps_var.set(new_steps)
 
     def end(self):
-        self.end_time = utc_now()
+        self.end_time = utc_now() if self.end_time is None else self.end_time
 
         # Update active steps
         active_steps = active_steps_var.get()