Add optional duration to FileInfo

edenai · Oct 17, 2024 · 50a49aa · 50a49aa
1 parent 289b1a2
commit 50a49aa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/edenai_apis/utils/files.py b/edenai_apis/utils/files.py
@@ -11,13 +11,15 @@ def __init__(
         self.file_extension = file_extension
         if args:
             self.file_frame_rate, self.file_channels = args
-
+        self.file_duration = kwargs.get("duration", None)
     file_size: int
     file_media_type: str
     supported_extensions: List[str]
     # for audio and video files
     file_frame_rate: Optional[str]
     file_channels: Optional[str]
+    file_duration: Optional[float]
+
 
 
 class FileWrapper: