diff --git a/agentops/session.py b/agentops/session.py index e29c6cadf..bb5427c4d 100644 --- a/agentops/session.py +++ b/agentops/session.py @@ -27,6 +27,16 @@ def __init__(self, session_id: str, tags: Optional[List[str]] = None): self.session_id = session_id self.init_timestamp = get_ISO_time() self.tags = tags + self.video = None + + def set_session_video(self, video: str): + """ + Sets a url to the video recording of the session. + + Args: + video (str): The url of the video recording + """ + self.video = video def end_session(self, end_state: str = "Indeterminate", rating: Optional[str] = None): """