diff --git a/phone_agent/video_learning.py b/phone_agent/video_learning.py index 587c1ce..caa6245 100644 --- a/phone_agent/video_learning.py +++ b/phone_agent/video_learning.py @@ -494,6 +494,10 @@ class VideoLearningAgent: description: Optional[str] = None, likes: Optional[int] = None, comments: Optional[int] = None, + shares: Optional[int] = None, + tags: Optional[List[str]] = None, + category: Optional[str] = None, + elements: Optional[List[str]] = None, ) -> VideoRecord: """ Record a watched video. @@ -528,6 +532,10 @@ class VideoLearningAgent: description=description, likes=likes, comments=comments, + shares=shares, + tags=tags or [], + category=category, + elements=elements or [], position_in_session=self.video_counter, )