feat: support video url

Author: 潘婉宁

volcenginesdkarkruntime/_utils/_utils.py

@@ -349,7 +349,7 @@ def removeprefix(string: str, prefix: str) -> str:
     Backport of `str.removeprefix` for Python < 3.9
     """
     if string.startswith(prefix):
-        return string[len(prefix):]
+        return string[len(prefix) :]
     return string
 
 

volcenginesdkarkruntime/resources/batch_chat/completions.py

@@ -71,6 +71,7 @@ def _process_messages(
                     )
                 )
 
+
 def _calculate_retry_timeout(retry_times) -> float:
     nbRetries = min(retry_times, MAX_RETRY_DELAY / INITIAL_RETRY_DELAY)
     sleep_seconds = min(INITIAL_RETRY_DELAY * pow(2, nbRetries), MAX_RETRY_DELAY)
@@ -134,7 +135,6 @@ def _process_messages(
                         )
                     )
 
-
     def _encrypt(
         self,
         model: str,

volcenginesdkarkruntime/resources/images/images.py

@@ -52,19 +52,19 @@ def generate(
 class AsyncImages(AsyncAPIResource):
     @async_apikey_required
     async def generate(
-            self,
-            *,
-            model: str,
-            prompt: str,
-            response_format: str | None = None,
-            size: str | None = None,
-            seed: int | None = None,
-            guidance_scale: float | None = None,
-            watermark: bool | None = None,
-            extra_headers: Headers | None = None,
-            extra_query: Query | None = None,
-            extra_body: Body | None = None,
-            timeout: float | httpx.Timeout | None = None,
+        self,
+        *,
+        model: str,
+        prompt: str,
+        response_format: str | None = None,
+        size: str | None = None,
+        seed: int | None = None,
+        guidance_scale: float | None = None,
+        watermark: bool | None = None,
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None = None,
     ) -> ImagesResponse:
         return await self._post(
             "/images/generations",

volcenginesdkarkruntime/types/bot_chat/__init__.py

@@ -9,4 +9,3 @@
     "BotChatCompletionChunk",
     "BotChatCompletion",
 ]
-

volcenginesdkarkruntime/types/chat/__init__.py

@@ -19,9 +19,6 @@
 from .chat_completion_stream_options_param import (
     ChatCompletionStreamOptionsParam as ChatCompletionStreamOptionsParam,
 )
-from .chat_completion_stream_options_param import (
-    ChatCompletionStreamOptionsParam as ChatCompletionStreamOptionsParam,
-)
 from .chat_completion_tool_param import (
     ChatCompletionToolParam as ChatCompletionToolParam,
 )

volcenginesdkarkruntime/types/chat/chat_completion_content_part_param.py

@@ -6,9 +6,14 @@
 from .chat_completion_content_part_image_param import (
     ChatCompletionContentPartImageParam,
 )
+from .chat_completion_content_part_video_param import (
+    ChatCompletionContentPartVideoParam,
+)
 
 __all__ = ["ChatCompletionContentPartParam"]
 
 ChatCompletionContentPartParam = Union[
-    ChatCompletionContentPartTextParam, ChatCompletionContentPartImageParam
+    ChatCompletionContentPartTextParam,
+    ChatCompletionContentPartImageParam,
+    ChatCompletionContentPartVideoParam,
 ]

volcenginesdkarkruntime/types/chat/chat_completion_content_part_video_param.py

@@ -0,0 +1,20 @@
+from __future__ import annotations
+
+from typing_extensions import Literal, Required, TypedDict
+
+__all__ = ["ChatCompletionContentPartVideoParam", "VideoURL"]
+
+
+class VideoURL(TypedDict, total=False):
+    url: Required[str]
+    """Either a URL of the video or the base64 encoded video data."""
+
+    fps: float
+    """The sampling fps of the video."""
+
+
+class ChatCompletionContentPartVideoParam(TypedDict, total=False):
+    video_url: Required[VideoURL]
+
+    type: Required[Literal["video_url"]]
+    """The type of the content part."""

volcenginesdkarkruntime/types/context/__init__.py

@@ -14,4 +14,3 @@
     "TruncationStrategy",
     "TTLTypes",
 ]
-

volcenginesdkarkruntime/types/images/__init__.py

@@ -1 +1 @@
-from .images import ImagesResponse
+from .images import ImagesResponse