update translate text

Author: winzamark123

jigsawstack/translate.py

@@ -5,6 +5,8 @@
 from typing import List, Union
 from ._config import ClientConfig
 from .helpers import build_path
+from ._types import BaseResponse
+
 
 class TranslateImageParams(TypedDict):
     target_language: str
@@ -22,12 +24,13 @@ class TranslateImageParams(TypedDict):
 
     return_type: NotRequired[Literal["url", "binary", "base64"]]
 
+
 class TranslateParams(TypedDict):
     target_language: str
     """
     Target langauge to translate to.
     """
-    current_language: str
+    current_language: NotRequired[str]
     """
     Language to translate from.
     """
@@ -36,16 +39,14 @@ class TranslateParams(TypedDict):
     The text to translate.
     """
 
-class TranslateResponse(TypedDict):
-    success: bool
-    """
-    Indicates whether the translation was successful.
-    """
-    translated_text: str
+
+class TranslateResponse(BaseResponse):
+    translated_text: Union[str, List[str]]
     """
     The translated text.
     """
 
+
 class TranslateImageResponse(TypedDict):
     success: bool
     """
@@ -56,6 +57,7 @@ class TranslateImageResponse(TypedDict):
     The image data that was translated.
     """
 
+
 class TranslateListResponse(TypedDict):
     success: bool
     """
@@ -94,18 +96,22 @@ def text(
             verb="post",
         ).perform()
         return resp
-    
+
     @overload
     def image(self, params: TranslateImageParams) -> TranslateImageResponse: ...
     @overload
-    def image(self, blob: bytes, options: TranslateImageParams = None) -> TranslateImageParams: ...
+    def image(
+        self, blob: bytes, options: TranslateImageParams = None
+    ) -> TranslateImageParams: ...
 
     def image(
         self,
         blob: Union[TranslateImageParams, bytes],
         options: TranslateImageParams = None,
     ) -> TranslateImageResponse:
-        if isinstance(blob, dict): # If params is provided as a dict, we assume it's the first argument
+        if isinstance(
+            blob, dict
+        ):  # If params is provided as a dict, we assume it's the first argument
             resp = Request(
                 config=self.config,
                 path="/ai/translate/image",
@@ -156,12 +162,14 @@ async def text(
             verb="post",
         ).perform()
         return resp
-    
+
     @overload
     async def image(self, params: TranslateImageParams) -> TranslateImageResponse: ...
     @overload
-    async def image(self, blob: bytes, options: TranslateImageParams = None) -> TranslateImageParams: ...
-    
+    async def image(
+        self, blob: bytes, options: TranslateImageParams = None
+    ) -> TranslateImageParams: ...
+
     async def image(
         self,
         blob: Union[TranslateImageParams, bytes],
@@ -189,4 +197,4 @@ async def image(
             headers=headers,
             verb="post",
         ).perform_with_content()
-        return resp
+        return resp