Update Wrapper:Request to handle binaries and json types gracefully to return types directly to user.

Author: Khurdhula-Harshavardhan

jigsawstack/request.py

@@ -50,13 +50,13 @@ def perform(self) -> Union[T, None]:
         """
         resp = self.make_request(url=f"{self.api_url}{self.path}")
 
-        # For binary responses or empty responses
+        #for binary responses
         if resp.status_code == 200:
             content_type = resp.headers.get("content-type", "")
-            if not resp.text or any(t in content_type for t in ["audio/", "image/", "application/octet-stream"]):
-                return cast(T, resp)
+            if not resp.text or any(t in content_type for t in ["audio/", "image/", "application/octet-stream", "image/png"]):
+                return cast(T, resp.content)
 
-        # Handle errors and JSON responses
+        #for json resposes.
         if resp.status_code != 200:
             try:
                 error = resp.json()
@@ -104,6 +104,12 @@ def perform_file(self) -> Union[T, None]:
                 message=error.get("message"),
                 err=error.get("error"),
             )
+
+        #for binary responses
+        if resp.status_code == 200:
+            content_type = resp.headers.get("content-type", "")
+            if "application/json" not in content_type:
+                resp = cast(T, resp.content)
         return resp
 
     def perform_with_content(self) -> T:

jigsawstack/translate.py

@@ -6,6 +6,20 @@
 from ._config import ClientConfig
 
 
+class TranslateImageParams(TypedDict):
+    target_language: str
+    """
+    Target langauge to translate to.
+    """
+    url: str
+    """
+    The URL of the image to translate.
+    """
+    file_store_key: NotRequired[str]
+    """
+    The file store key of the image to translate.
+    """
+
 class TranslateParams(TypedDict):
     target_language: str
     """
@@ -20,7 +34,6 @@ class TranslateParams(TypedDict):
     The text to translate.
     """
 
-
 class TranslateResponse(TypedDict):
     success: bool
     """
@@ -31,6 +44,15 @@ class TranslateResponse(TypedDict):
     The translated text.
     """
 
+class TranslateImageResponse(TypedDict):
+    success: bool
+    """
+    Indicates whether the translation was successful.
+    """
+    image: bytes
+    """
+    The image data that was translated.
+    """
 
 class TranslateListResponse(TypedDict):
     success: bool
@@ -61,15 +83,18 @@ def __init__(
         )
 
     def translate(
-        self, params: TranslateParams
-    ) -> Union[TranslateResponse, TranslateListResponse]:
-        path = "/ai/translate"
+        self, params: Union[TranslateParams, TranslateImageParams]
+    ) -> Union[TranslateResponse, TranslateListResponse, TranslateImageResponse]:
+        if "url" in params or "file_store_key" in params:
+            path = "/ai/translate/image"
+        else:
+            path = "/ai/translate"
         resp = Request(
             config=self.config,
             path=path,
             params=cast(Dict[Any, Any], params),
             verb="post",
-        ).perform_with_content()
+        ).perform()
         return resp
 
 
@@ -91,13 +116,16 @@ def __init__(
         )
 
     async def translate(
-        self, params: TranslateParams
-    ) -> Union[TranslateResponse, TranslateListResponse]:
-        path = "/ai/translate"
+        self, params: Union[TranslateParams, TranslateImageParams]
+    ) -> Union[TranslateResponse, TranslateListResponse, TranslateImageParams]:
+        if "url" in params or "file_store_key" in params:
+            path = "/ai/translate/image"
+        else:
+            path = "/ai/translate"
         resp = await AsyncRequest(
             config=self.config,
             path=path,
             params=cast(Dict[Any, Any], params),
             verb="post",
-        ).perform_with_content()
+        ).perform()
         return resp