support lang in video mashup task (#38)

01-ai · Dec 4, 2024 · 8214c6b · 8214c6b
1 parent fe03cc2
commit 8214c6b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/client/py/pyproject.toml b/client/py/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "yidong"
-version = "0.0.10"
+version = "0.0.11"
 description = ""
 authors = ["Yi Team <[email protected]>"]
 license = "MIT"

diff --git a/client/py/yidong/client.py b/client/py/yidong/client.py
@@ -372,6 +372,7 @@ def video_mashup(
         voice_style_id: str,
         voice_style_text: str,
         chapters: list[Chapter] | None = None,
+        lang: str = "en",
     ) -> TaskRef[VideoMashupTask, VideoMashupTaskResult]:
         """Create a new video based on the given videos and other elements.
 
@@ -382,6 +383,7 @@ def video_mashup(
             bgm_id: The background music id. Make sure it exists first.
             voice_style_id: The voice style resource id. Make sure it exists first by uploading your voice sample.
             voice_style_text: The transcript of the voice style.
+            lang: The language of the voice over text. We'll choose appropriate font and style based on the language. Make sure the voice style matches the language specified here.
         """
         return self._submit_task(locals())
 

diff --git a/client/py/yidong/model.py b/client/py/yidong/model.py
@@ -175,6 +175,7 @@ class VideoMashupTask(BaseModel):
     bgm_id: str
     voice_style_id: str
     voice_style_text: str
+    lang: str = "en"
 
 
 class VideoMashupTaskResult(BaseModel):