feat(session): force flush pending spans on session end

AgentOps-AI · Nov 26, 2024 · 894c846 · 894c846
1 parent 259c045
commit 894c846
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/agentops/session.py b/agentops/session.py
@@ -285,6 +285,10 @@ def end_session(
                 return None
 
             try:
+                # Force flush any pending spans before ending session
+                if hasattr(self, "_span_processor"):
+                    self._span_processor.force_flush(timeout_millis=5000)
+
                 # 1. Set shutdown flag on exporter first
                 if hasattr(self, "_otel_exporter"):
                     self._otel_exporter.shutdown()