feat(event): add session_id to Event class and methods

agentops/event.py

@@ -25,6 +25,7 @@ class Event:
     end_timestamp(str): A timestamp indicating when the event ended. Defaults to the time when this Event was instantiated.
     agent_id(UUID, optional): The unique identifier of the agent that triggered the event.
     id(UUID): A unique identifier for the event. Defaults to a new UUID.
+    session_id(UUID, optional): The unique identifier of the session that the event belongs to.
 
     foo(x=1) {
         ...
@@ -43,6 +44,7 @@ class Event:
     end_timestamp: Optional[str] = None
     agent_id: Optional[UUID] = field(default_factory=check_call_stack_for_agent_id)
     id: UUID = field(default_factory=uuid4)
+    session_id: Optional[UUID] = None
 
 
 @dataclass

agentops/session.py

@@ -340,6 +340,9 @@ def record(self, event: Union[Event, ErrorEvent], flush_now=False):
         if not self.is_running:
             return
 
+        # Set session_id on the event itself
+        event.session_id = self.session_id
+
         # Create session context
         token = set_value("session.id", str(self.session_id))
 
@@ -355,6 +358,11 @@ def record(self, event: Union[Event, ErrorEvent], flush_now=False):
                 span_def.attributes.update({
                     AgentOpsAttributes.SESSION_ID: str(self.session_id),
                     AgentOpsAttributes.SESSION_TAGS: ",".join(self.tags) if self.tags else "",
+                    # Add session_id to event data
+                    AgentOpsAttributes.EVENT_DATA: json.dumps({
+                        "session_id": str(self.session_id),
+                        **span_def.attributes.get(AgentOpsAttributes.EVENT_DATA, {})
+                    })
                 })
 
                 with self._otel_tracer.start_as_current_span(

agentops/telemetry/converter.py

@@ -183,6 +183,7 @@ def _get_span_attributes(event: Event) -> Dict[str, AttributeValue]:
             AgentOpsAttributes.EVENT_START_TIME: event.init_timestamp if hasattr(event, 'init_timestamp') else event.timestamp,
             AgentOpsAttributes.EVENT_END_TIME: getattr(event, 'end_timestamp', None),
             AgentOpsAttributes.EVENT_ID: str(event.id),
+            AgentOpsAttributes.SESSION_ID: str(event.session_id) if event.session_id else None,
         })
 
         # Add agent ID if present