Skip to content

Commit

Permalink
Refactoring video logic (#59)
Browse files Browse the repository at this point in the history
* 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
  • Loading branch information
HowieG committed Dec 16, 2023
1 parent 663223c commit 405a962
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions agentops/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
"""
Expand Down

0 comments on commit 405a962

Please sign in to comment.