feat: Add multiple new parameters to Generation API

Author: kevin

dashscope/aigc/video_synthesis.py

@@ -1,7 +1,7 @@
 # -*- coding: utf-8 -*-
 # Copyright (c) Alibaba, Inc. and its affiliates.
 
-from typing import Any, Dict, Union, List
+from typing import Any, Dict, Union, List, Optional
 
 from concurrent.futures import ThreadPoolExecutor, as_completed
 
@@ -75,15 +75,15 @@ def call(  # type: ignore[override]
         first_frame_url: str = None,
         last_frame_url: str = None,
         media: List[Dict] = None,
-        size: str = None,
-        duration: int = None,
-        seed: int = None,
-        prompt_extend: bool = None,
-        watermark: bool = None,
-        resolution: str = None,
-        ratio: str = None,
-        shot_type: str = None,
-        audio_setting: str = None,
+        size: Optional[str] = None,
+        duration: Optional[int] = None,
+        seed: Optional[int] = None,
+        prompt_extend: Optional[bool] = None,
+        watermark: Optional[bool] = None,
+        resolution: Optional[str] = None,
+        ratio: Optional[str] = None,
+        shot_type: Optional[str] = None,
+        audio_setting: Optional[str] = None,
         **kwargs,
     ) -> VideoSynthesisResponse:
         """Call video synthesis service and get result.
@@ -388,15 +388,15 @@ def async_call(  # pylint: disable=arguments-renamed  # type: ignore[override] #
         first_frame_url: str = None,
         last_frame_url: str = None,
         media: List[Dict] = None,
-        size: str = None,
-        duration: int = None,
-        seed: int = None,
-        prompt_extend: bool = None,
-        watermark: bool = None,
-        resolution: str = None,
-        ratio: str = None,
-        shot_type: str = None,
-        audio_setting: str = None,
+        size: Optional[str] = None,
+        duration: Optional[int] = None,
+        seed: Optional[int] = None,
+        prompt_extend: Optional[bool] = None,
+        watermark: Optional[bool] = None,
+        resolution: Optional[str] = None,
+        ratio: Optional[str] = None,
+        shot_type: Optional[str] = None,
+        audio_setting: Optional[str] = None,
         **kwargs,
     ) -> VideoSynthesisResponse:
         """Create a video synthesis task, and return task information.

tests/unit/test_video_synthesis_params.py

@@ -183,7 +183,7 @@ def test_with_all_new_parameters(self, mock_server: MockServer):
 
         req = mock_server.requests.get(block=True)
         params = req["parameters"]
-        
+
         assert params["size"] == "1280*720"
         assert params["duration"] == 10
         assert params["seed"] == 42
@@ -193,7 +193,7 @@ def test_with_all_new_parameters(self, mock_server: MockServer):
         assert params["ratio"] == "16:9"
         assert params["shot_type"] == "single"
         assert params["audio_setting"] == "origin"
-        
+
         assert response.status_code == HTTPStatus.OK
 
     def test_with_none_parameters_not_included(self, mock_server: MockServer):
@@ -211,10 +211,10 @@ def test_with_none_parameters_not_included(self, mock_server: MockServer):
 
         req = mock_server.requests.get(block=True)
         params = req["parameters"]
-        
+
         # None parameters should not be in the request
         assert "size" not in params
         assert "duration" not in params
         assert "seed" not in params
-        
+
         assert response.status_code == HTTPStatus.OK