mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-12-16 09:42:29 +08:00
fix(api-nodes): bad indentation in Recraft API node function (#10175)
This commit is contained in:
parent
8a293372ec
commit
c2c5a7d5f8
@ -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:
|
||||||
|
|||||||
@ -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",
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user