From 405a9620e37afdd3ae8e17ef18bab6c46f181c60 Mon Sep 17 00:00:00 2001
From: Howard Gil <howardbgil@gmail.com>
Date: Fri, 15 Dec 2023 19:19:30 -0600
Subject: [PATCH] Refactoring video logic (#59)

* Adding support for multion screenshots and video

* Added support for screenshot images and screen recording videos

* Changed screenshot type from PIL Image to string (base64)

* Updated arg definition to reflect that screenshot can be base64 or url

* typo

* Refactored video logic

* Typo
---
 agentops/session.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

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):
         """