fix: formatting for ruff.

Author: Khurdhula-Harshavardhan

jigsawstack/async_request.py

@@ -250,9 +250,7 @@ async def make_request(
                 form_data.add_field(
                     "file",
                     BytesIO(data),
-                    content_type=headers.get(
-                        "Content-Type", "application/octet-stream"
-                    ),
+                    content_type=headers.get("Content-Type", "application/octet-stream"),
                     filename="file",
                 )
 

jigsawstack/embedding.py

@@ -47,9 +47,7 @@ def __init__(
     @overload
     def execute(self, params: EmbeddingParams) -> EmbeddingResponse: ...
     @overload
-    def execute(
-        self, blob: bytes, options: EmbeddingParams = None
-    ) -> EmbeddingResponse: ...
+    def execute(self, blob: bytes, options: EmbeddingParams = None) -> EmbeddingResponse: ...
 
     def execute(
         self,
@@ -101,9 +99,7 @@ def __init__(
     @overload
     async def execute(self, params: EmbeddingParams) -> EmbeddingResponse: ...
     @overload
-    async def execute(
-        self, blob: bytes, options: EmbeddingParams = None
-    ) -> EmbeddingResponse: ...
+    async def execute(self, blob: bytes, options: EmbeddingParams = None) -> EmbeddingResponse: ...
 
     async def execute(
         self,

jigsawstack/embedding_v2.py

@@ -45,9 +45,7 @@ def __init__(
     @overload
     def execute(self, params: EmbeddingV2Params) -> EmbeddingV2Response: ...
     @overload
-    def execute(
-        self, blob: bytes, options: EmbeddingV2Params = None
-    ) -> EmbeddingV2Response: ...
+    def execute(self, blob: bytes, options: EmbeddingV2Params = None) -> EmbeddingV2Response: ...
 
     def execute(
         self,

jigsawstack/helpers.py

@@ -2,9 +2,7 @@
 from urllib.parse import urlencode
 
 
-def build_path(
-    base_path: str, params: Optional[Dict[str, Union[str, int, bool]]] = None
-) -> str:
+def build_path(base_path: str, params: Optional[Dict[str, Union[str, int, bool]]] = None) -> str:
     """
     Build an API endpoint path with query parameters.
 
@@ -20,9 +18,7 @@ def build_path(
 
     # remove None values from the parameters
     filtered_params = {
-        k: str(v).lower() if isinstance(v, bool) else v
-        for k, v in params.items()
-        if v is not None
+        k: str(v).lower() if isinstance(v, bool) else v for k, v in params.items() if v is not None
     }
 
     # encode the parameters

jigsawstack/image_generation.py

@@ -92,9 +92,7 @@ def __init__(
         api_url: str,
         disable_request_logging: Union[bool, None] = False,
     ):
-        super().__init__(
-            api_key, api_url, disable_request_logging=disable_request_logging
-        )
+        super().__init__(api_key, api_url, disable_request_logging=disable_request_logging)
         self.config = RequestConfig(
             api_url=api_url,
             api_key=api_key,
@@ -123,9 +121,7 @@ def __init__(
         api_url: str,
         disable_request_logging: Union[bool, None] = False,
     ):
-        super().__init__(
-            api_key, api_url, disable_request_logging=disable_request_logging
-        )
+        super().__init__(api_key, api_url, disable_request_logging=disable_request_logging)
         self.config = RequestConfig(
             api_url=api_url,
             api_key=api_key,

jigsawstack/prompt_engine.py

@@ -119,14 +119,10 @@ def create(self, params: PromptEngineCreateParams) -> PromptEngineCreateResponse
 
     def get(self, id: str) -> PromptEngineGetResponse:
         path = f"/prompt_engine/{id}"
-        resp = Request(
-            config=self.config, path=path, params={}, verb="get"
-        ).perform_with_content()
+        resp = Request(config=self.config, path=path, params={}, verb="get").perform_with_content()
         return resp
 
-    def list(
-        self, params: Union[PromptEngineListParams, None] = None
-    ) -> PromptEngineListResponse:
+    def list(self, params: Union[PromptEngineListParams, None] = None) -> PromptEngineListResponse:
         if params is None:
             params = {}
 
@@ -141,9 +137,7 @@ def list(
             base_path="/prompt_engine",
             params=params,
         )
-        resp = Request(
-            config=self.config, path=path, params={}, verb="get"
-        ).perform_with_content()
+        resp = Request(config=self.config, path=path, params={}, verb="get").perform_with_content()
         return resp
 
     def delete(self, id: str) -> PromptEngineDeleteResponse:
@@ -219,9 +213,7 @@ def __init__(
             disable_request_logging=disable_request_logging,
         )
 
-    async def create(
-        self, params: PromptEngineCreateParams
-    ) -> PromptEngineCreateResponse:
+    async def create(self, params: PromptEngineCreateParams) -> PromptEngineCreateResponse:
         path = "/prompt_engine"
         resp = await AsyncRequest(
             config=self.config,

jigsawstack/request.py

@@ -91,10 +91,7 @@ def perform_file(self) -> Union[T, None]:
         # handle error in case there is a statusCode attr present
         # and status != 200 and response is a json.
 
-        if (
-            "application/json" not in resp.headers["content-type"]
-            and resp.status_code != 200
-        ):
+        if "application/json" not in resp.headers["content-type"] and resp.status_code != 200:
             raise_for_code_and_type(
                 code=500,
                 message="Failed to parse JigsawStack API response. Please try again.",

jigsawstack/search.py

@@ -319,9 +319,7 @@ async def search(self, params: SearchParams) -> SearchResponse:
         ).perform_with_content()
         return resp
 
-    async def suggestions(
-        self, params: SearchSuggestionsParams
-    ) -> SearchSuggestionsResponse:
+    async def suggestions(self, params: SearchSuggestionsParams) -> SearchSuggestionsResponse:
         query = params["query"]
         path = f"/web/search/suggest?query={query}"
         resp = await AsyncRequest(

jigsawstack/store.py

@@ -23,9 +23,7 @@ class FileUploadResponse(TypedDict):
     key: str
     url: str
     size: int
-    temp_public_url: NotRequired[
-        str
-    ]  # Optional, only if temp_public_url is set to True in params
+    temp_public_url: NotRequired[str]  # Optional, only if temp_public_url is set to True in params
 
 
 class Store(ClientConfig):

jigsawstack/vision.py

@@ -218,9 +218,7 @@ def vocr(
         return resp
 
     @overload
-    def object_detection(
-        self, params: ObjectDetectionParams
-    ) -> ObjectDetectionResponse: ...
+    def object_detection(self, params: ObjectDetectionParams) -> ObjectDetectionResponse: ...
     @overload
     def object_detection(
         self, blob: bytes, options: ObjectDetectionParams = None
@@ -307,9 +305,7 @@ async def vocr(
         return resp
 
     @overload
-    async def object_detection(
-        self, params: ObjectDetectionParams
-    ) -> ObjectDetectionResponse: ...
+    async def object_detection(self, params: ObjectDetectionParams) -> ObjectDetectionResponse: ...
     @overload
     async def object_detection(
         self, blob: bytes, options: ObjectDetectionParams = None