feat(realtime): support lucy_v2v_14b_rt (#20)

Author: AdirAmsalem

decart/models.py

@@ -4,7 +4,9 @@
 from .types import FileInput, MotionTrajectoryInput
 
 
-RealTimeModels = Literal["mirage", "mirage_v2", "lucy_v2v_720p_rt", "avatar-live"]
+RealTimeModels = Literal[
+    "mirage", "mirage_v2", "lucy_v2v_720p_rt", "lucy_v2v_14b_rt", "avatar-live"
+]
 VideoModels = Literal[
     "lucy-dev-i2v",
     "lucy-fast-v2v",
@@ -175,6 +177,14 @@ class ImageToImageInput(DecartBaseModel):
             height=704,
             input_schema=BaseModel,
         ),
+        "lucy_v2v_14b_rt": ModelDefinition(
+            name="lucy_v2v_14b_rt",
+            url_path="/v1/stream",
+            fps=15,
+            width=1280,
+            height=704,
+            input_schema=BaseModel,
+        ),
         "avatar-live": ModelDefinition(
             name="avatar-live",
             url_path="/v1/avatar-live/stream",

tests/test_realtime_unit.py

@@ -39,6 +39,17 @@ def test_realtime_models_available():
 
     model2 = models.realtime("lucy_v2v_720p_rt")
     assert model2.name == "lucy_v2v_720p_rt"
+    assert model2.fps == 25
+    assert model2.width == 1280
+    assert model2.height == 704
+    assert model2.url_path == "/v1/stream"
+
+    model2 = models.realtime("lucy_v2v_14b_rt")
+    assert model2.name == "lucy_v2v_14b_rt"
+    assert model2.fps == 15
+    assert model2.width == 1280
+    assert model2.height == 704
+    assert model2.url_path == "/v1/stream"
 
 
 @pytest.mark.asyncio