fixed for threading

AgentOps-AI · Sep 14, 2024 · d81695d · d81695d
1 parent 3edb4e8
commit d81695d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/agentops/time_travel.py b/agentops/time_travel.py
@@ -4,6 +4,9 @@
 from .http_client import HttpClient
 from .exceptions import ApiServerException
 from .singleton import singleton
+import threading
+
+_time_travel_lock = threading.Lock()
 
 
 @singleton
@@ -160,8 +163,9 @@ def set_time_travel_active_state(is_active: bool):
 
 
 def add_time_travel_terminal_indicator():
-    if TimeTravel()._initialized is False:
-        print(f"🖇⏰ AgentOps: Time Travel Activated")
+    with _time_travel_lock:
+        if not TimeTravel()._initialized:
+            print(f"🖇⏰ AgentOps: Time Travel Activated")
 
 
 def reset_terminal():