Skip to content

Commit f319ced

Browse files
author
Lin Fangyuan
committed
feat: videogen return last frame
1 parent c9ee81d commit f319ced

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

volcenginesdkarkruntime/resources/content_generation/tasks.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@ def create(
3434
model: str,
3535
content: Iterable[CreateTaskContentParam],
3636
callback_url: Optional[str] = None,
37+
return_last_frame: Optional[bool] = None,
3738
extra_headers: Headers | None = None,
3839
extra_query: Query | None = None,
3940
extra_body: Body | None = None,
@@ -45,6 +46,7 @@ def create(
4546
"model": model,
4647
"content": content,
4748
"callback_url": callback_url,
49+
"return_last_frame": return_last_frame,
4850
},
4951
options=make_request_options(
5052
extra_headers=extra_headers,
@@ -153,6 +155,7 @@ async def create(
153155
model: str,
154156
content: Iterable[CreateTaskContentParam],
155157
callback_url: Optional[str] = None,
158+
return_last_frame: Optional[bool] = None,
156159
extra_headers: Headers | None = None,
157160
extra_query: Query | None = None,
158161
extra_body: Body | None = None,
@@ -164,6 +167,7 @@ async def create(
164167
"model": model,
165168
"content": content,
166169
"callback_url": callback_url,
170+
"return_last_frame": return_last_frame,
167171
},
168172
options=make_request_options(
169173
extra_headers=extra_headers,

volcenginesdkarkruntime/types/content_generation/content_generation_task.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,9 @@ class Content(BaseModel):
1212
video_url: str
1313
"""The URL of the generated video, if any."""
1414

15+
last_frame_url: str
16+
"""The URL of the last frame of the generated video, if any."""
17+
1518

1619
class ContentGenerationError(BaseModel):
1720
message: str

0 commit comments

Comments
 (0)