refactor: model to model_name

Author: ankit-v2-3

videodb/_constants.py

@@ -27,10 +27,6 @@ class SceneExtractionType:
     time_based = "time"
 
 
-class SceneModels:
-    gpt4_o = "gpt4-o"
-
-
 class Workflows:
     add_subtitles = "add_subtitles"
 

videodb/image.py

@@ -1,6 +1,5 @@
 from videodb._constants import (
     ApiPath,
-    SceneModels,
 )
 
 
@@ -63,7 +62,7 @@ def to_json(self):
             "description": self.description,
         }
 
-    def describe(self, prompt: str = None, model_name=SceneModels.gpt4_o):
+    def describe(self, prompt: str = None, model_name=None):
         description_data = self._connection.post(
             path=f"{ApiPath.video}/{self.video_id}/{ApiPath.frame}/{self.id}/{ApiPath.describe}",
             data={"prompt": prompt, "model_name": model_name},

videodb/scene.py

@@ -1,6 +1,6 @@
 from typing import List
 
-from videodb._constants import ApiPath, SceneModels
+from videodb._constants import ApiPath
 
 from videodb.image import Frame
 
@@ -45,7 +45,7 @@ def to_json(self):
             "description": self.description,
         }
 
-    def describe(self, prompt: str = None, model_name=SceneModels.gpt4_o) -> None:
+    def describe(self, prompt: str = None, model_name=None) -> None:
         if self._connection is None:
             raise ValueError("Connection is required to describe a scene")
         description_data = self._connection.post(

videodb/video.py

@@ -257,7 +257,7 @@ def index_scenes(
         extraction_type: SceneExtractionType = SceneExtractionType.shot_based,
         extraction_config: Dict = {},
         prompt: Optional[str] = None,
-        model: Optional[str] = None,
+        model_name: Optional[str] = None,
         model_config: Optional[Dict] = None,
         name: Optional[str] = None,
         scenes: Optional[List[Scene]] = None,
@@ -270,7 +270,7 @@ def index_scenes(
                 "extraction_type": extraction_type,
                 "extraction_config": extraction_config,
                 "prompt": prompt,
-                "model": model,
+                "model_name": model_name,
                 "model_config": model_config,
                 "name": name,
                 "force": force,