fix(api-nodes): bad indentation in Recraft API node function (#10175)

This commit is contained in:
Alexander Piskun 2025-10-03 21:41:06 +03:00 committed by GitHub
parent 8a293372ec
commit c2c5a7d5f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 39 additions and 40 deletions

View File

@ -38,48 +38,48 @@ from PIL import UnidentifiedImageError
async def handle_recraft_file_request( async def handle_recraft_file_request(
image: torch.Tensor, image: torch.Tensor,
path: str, path: str,
mask: torch.Tensor=None, mask: torch.Tensor=None,
total_pixels=4096*4096, total_pixels=4096*4096,
timeout=1024, timeout=1024,
request=None, request=None,
auth_kwargs: dict[str,str] = None, auth_kwargs: dict[str,str] = None,
) -> list[BytesIO]: ) -> list[BytesIO]:
""" """
Handle sending common Recraft file-only request to get back file bytes. Handle sending common Recraft file-only request to get back file bytes.
""" """
if request is None: if request is None:
request = EmptyRequest() request = EmptyRequest()
files = { files = {
'image': tensor_to_bytesio(image, total_pixels=total_pixels).read() 'image': tensor_to_bytesio(image, total_pixels=total_pixels).read()
} }
if mask is not None: if mask is not None:
files['mask'] = tensor_to_bytesio(mask, total_pixels=total_pixels).read() files['mask'] = tensor_to_bytesio(mask, total_pixels=total_pixels).read()
operation = SynchronousOperation( operation = SynchronousOperation(
endpoint=ApiEndpoint( endpoint=ApiEndpoint(
path=path, path=path,
method=HttpMethod.POST, method=HttpMethod.POST,
request_model=type(request), request_model=type(request),
response_model=RecraftImageGenerationResponse, response_model=RecraftImageGenerationResponse,
), ),
request=request, request=request,
files=files, files=files,
content_type="multipart/form-data", content_type="multipart/form-data",
auth_kwargs=auth_kwargs, auth_kwargs=auth_kwargs,
multipart_parser=recraft_multipart_parser, multipart_parser=recraft_multipart_parser,
) )
response: RecraftImageGenerationResponse = await operation.execute() response: RecraftImageGenerationResponse = await operation.execute()
all_bytesio = [] all_bytesio = []
if response.image is not None: if response.image is not None:
all_bytesio.append(await download_url_to_bytesio(response.image.url, timeout=timeout)) all_bytesio.append(await download_url_to_bytesio(response.image.url, timeout=timeout))
else: else:
for data in response.data: for data in response.data:
all_bytesio.append(await download_url_to_bytesio(data.url, timeout=timeout)) all_bytesio.append(await download_url_to_bytesio(data.url, timeout=timeout))
return all_bytesio return all_bytesio
def recraft_multipart_parser(data, parent_key=None, formatter: callable=None, converted_to_check: list[list]=None, is_list=False) -> dict: def recraft_multipart_parser(data, parent_key=None, formatter: callable=None, converted_to_check: list[list]=None, is_list=False) -> dict:

View File

@ -65,7 +65,6 @@ messages_control.disable = [
"unnecessary-pass", "unnecessary-pass",
"unidiomatic-typecheck", "unidiomatic-typecheck",
"unnecessary-lambda-assignment", "unnecessary-lambda-assignment",
"bad-indentation",
"no-else-return", "no-else-return",
"no-else-raise", "no-else-raise",
"invalid-overridden-method", "invalid-overridden-method",