Fix UploadRequest file_name param (#98)

This commit is contained in:
Jedrzej Kosinski 2025-05-01 02:59:39 -05:00 committed by GitHub
parent 8fd2f0e71b
commit 24d0e9001f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 3 deletions

View File

@ -331,10 +331,10 @@ def upload_images_to_comfyapi(
img_binary = tensor_to_bytesio(curr_image, mime_type=mime_type) img_binary = tensor_to_bytesio(curr_image, mime_type=mime_type)
# first, request upload/download urls from comfy API # first, request upload/download urls from comfy API
if not mime_type: if not mime_type:
request_object = UploadRequest(filename=img_binary.name) request_object = UploadRequest(file_name=img_binary.name)
else: else:
request_object = UploadRequest( request_object = UploadRequest(
filename=img_binary.name, content_type=mime_type file_name=img_binary.name, content_type=mime_type
) )
operation = SynchronousOperation( operation = SynchronousOperation(
endpoint=ApiEndpoint( endpoint=ApiEndpoint(

View File

@ -128,7 +128,7 @@ class EmptyRequest(BaseModel):
class UploadRequest(BaseModel): class UploadRequest(BaseModel):
filename: str = Field(..., description="Filename to upload") file_name: str = Field(..., description="Filename to upload")
content_type: str | None = Field( content_type: str | None = Field(
None, None,
description="Mime type of the file. For example: image/png, image/jpeg, video/mp4, etc.", description="Mime type of the file. For example: image/png, image/jpeg, video/mp4, etc.",