add return type to params

Author: winzamark123

jigsawstack/audio.py

@@ -4,7 +4,7 @@
 from .async_request import AsyncRequest, AsyncRequestConfig
 from ._config import ClientConfig
 from typing import Any, Dict, List, cast
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict, Literal
 from .custom_typing import SupportedAccents
 from .helpers import build_path
 
@@ -14,6 +14,7 @@ class TextToSpeechParams(TypedDict):
     accent: NotRequired[SupportedAccents]
     speaker_clone_url: NotRequired[str]
     speaker_clone_file_store_key: NotRequired[str]
+    return_type: NotRequired[Literal["url", "binary", "base64"]]
 
 
 class TTSCloneParams(TypedDict):

jigsawstack/image_generation.py

@@ -53,6 +53,8 @@ class ImageGenerationParams(TypedDict):
     File store key to use as image input.
     """
 
+    return_type: NotRequired[Literal["url", "binary", "base64"]]
+
 class ImageGenerationResponse(TypedDict):
     success: bool
     """

jigsawstack/translate.py

@@ -1,5 +1,5 @@
 from typing import Any, Dict, List, Union, cast, overload
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import NotRequired, TypedDict, Literal
 from .request import Request, RequestConfig
 from .async_request import AsyncRequest
 from typing import List, Union
@@ -20,6 +20,8 @@ class TranslateImageParams(TypedDict):
     The file store key of the image to translate.
     """
 
+    return_type: NotRequired[Literal["url", "binary", "base64"]]
+
 class TranslateParams(TypedDict):
     target_language: str
     """

jigsawstack/web.py

@@ -42,9 +42,9 @@ class DNSResponse(TypedDict):
 # HTML to Any
 #
 class HTMLToAnyParams(TypedDict):
-    html: str
-    url: str
-    goto_options: NotRequired[object]
+    html: NotRequired[str]
+    url: NotRequired[str]
+    goto_options: NotRequired[Dict[str, Union[int, str]]]
     scale: NotRequired[int]
     full_page: NotRequired[bool]
     omit_background: NotRequired[bool]
@@ -59,6 +59,7 @@ class HTMLToAnyParams(TypedDict):
     is_mobile: NotRequired[bool]
     dark_mode: NotRequired[bool]
     use_graphic_renderer: NotRequired[bool]
+    return_type: NotRequired[Literal["url", "binary", "base64"]]
 
 
 class HTMLToAnyResponse(TypedDict):