fix: preprocess configs

Author: lmh

volcenginesdkarkruntime/resources/files.py

@@ -18,7 +18,7 @@
 import httpx
 
 from .. import _legacy_response
-from ..types import FilePurpose, file_list_params, file_create_params
+from ..types import FilePurpose, PreprocessConfigs, file_list_params, file_create_params
 from .._types import Body, Query, Headers, FileTypes
 from .._utils import (
     extract_files,
@@ -66,6 +66,7 @@ def create(
         file: FileTypes,
         purpose: FilePurpose,
         expires_at: datetime.datetime | int | None = None,
+        preprocess_configs: PreprocessConfigs | None = None,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -97,6 +98,7 @@ def create(
                 "file": file,
                 "purpose": purpose,
                 "expires_at": expires_at,
+                "preprocess_configs": preprocess_configs,
             }
         )
         files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])
@@ -304,6 +306,7 @@ async def create(
         file: FileTypes,
         purpose: FilePurpose,
         expire_at: int | None = None,
+        preprocess_configs: PreprocessConfigs | None = None,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -335,6 +338,7 @@ async def create(
                 "file": file,
                 "purpose": purpose,
                 "expire_at": expire_at,
+                "preprocess_configs": preprocess_configs,
             }
         )
         files = extract_files(cast(Mapping[str, object], body), paths=[["file"]])

volcenginesdkarkruntime/resources/responses/responses.py

@@ -53,7 +53,7 @@
 RESPONSES_MULTIMODAL_CONTENT_DATA_KEYS = {
     "input_image": "image_url",
     "input_video": "video_url",
-    "input_file": "file_data",
+    "input_file": "file_url",
 }
 
 FILE_PATH_SCHEME = "file"
@@ -367,7 +367,9 @@ async def _prepare_responses_input_file(self, content: ResponseInputContentParam
         file = await self._client.files.create(file=file_path, purpose="user_data")
         file = await self._client.files.wait_for_processing(id=file.id)
         if file.status != "active":
-            raise ArkAPIError(f"File path: {full_path},id: {file.id} processing failed with status {file.status}.")
+            raise ArkAPIError(
+                f"File path: {full_path},id: {file.id} processing failed with status {file.status}."
+            )
 
         # replace with file id
         content[content_data_key] = None

volcenginesdkarkruntime/types/__init__.py

@@ -25,3 +25,4 @@
 )
 from .file_purpose import FilePurpose as FilePurpose
 from .file_list_params import FileListParams as FileListParams
+from .file_create_params import PreprocessConfigs as PreprocessConfigs

volcenginesdkarkruntime/types/file_create_params.py

@@ -16,7 +16,7 @@
 from .._types import FileTypes
 from .file_purpose import FilePurpose
 
-__all__ = ["FileCreateParams"]
+__all__ = ["FileCreateParams", "PreprocessConfigs"]
 
 
 class FileCreateParams(TypedDict, total=False):
@@ -31,6 +31,9 @@ class FileCreateParams(TypedDict, total=False):
     """The expiration timestamp for the file.
     """
 
+    preprocess_configs: Optional[PreprocessConfigs]
+    """The preprocess configs of the file."""
+
 
 class Video(TypedDict, total=False):
     """The video configs of the file."""

volcenginesdkexamples/volcenginesdkarkruntime/async_responses_video.py

@@ -12,7 +12,12 @@ async def main():
     file = await client.files.create(
         # replace with your local video path
         file=open("ark_vlm_video_input.mp4", "rb"),
-        purpose="user_data"
+        purpose="user_data",
+        preprocess_configs={
+            "video": {
+                "fps": 0.3,  # define the sampling fps of the video, default is 1.0
+            }
+        }
     )
     print(f"File uploaded: {file.id}")