feat: add MediaType class

Author: ankit-v2-3

videodb/__init__.py

@@ -5,7 +5,7 @@
 
 from typing import Optional
 from videodb._utils._video import play_stream
-from videodb._constants import VIDEO_DB_API
+from videodb._constants import VIDEO_DB_API, MediaType
 from videodb.client import Connection
 from videodb.exceptions import (
     VideodbError,
@@ -25,6 +25,7 @@
     "InvalidRequestError",
     "SearchError",
     "play_stream",
+    "MediaType",
 ]
 
 

videodb/_constants.py

@@ -4,6 +4,11 @@
 VIDEO_DB_API: str = "https://api.videodb.io"
 
 
+class MediaType:
+    video = "video"
+    audio = "audio"
+
+
 class SearchType:
     semantic = "semantic"
 

videodb/audio.py

@@ -21,4 +21,4 @@ def __repr__(self) -> str:
         )
 
     def delete(self) -> None:
-        self._connection.delete(f"{ApiPath.AUDIO}/{self.id}")
+        self._connection.delete(f"{ApiPath.audio}/{self.id}")