Skip to content

Commit 7cc4fc7

Browse files
committed
Adding taskUUID support to modelUpload
1 parent e164edf commit 7cc4fc7

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

runware/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1700,7 +1700,7 @@ async def _modelUpload(
17001700
) -> Optional[IUploadModelResponse]:
17011701
await self.ensureConnection()
17021702

1703-
task_uuid = getUUID()
1703+
task_uuid = requestModel.taskUUID or getUUID()
17041704
base_fields = {
17051705
"taskType": ETaskType.MODEL_UPLOAD.value,
17061706
"taskUUID": task_uuid,

runware/types.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -958,6 +958,7 @@ class IUploadModelBaseType:
958958
format: str
959959
private: bool
960960
category: str
961+
taskUUID: Optional[str] = None
961962
heroImageURL: Optional[str] = None
962963
tags: Optional[List[str]] = field(default_factory=list)
963964
shortDescription: Optional[str] = None

0 commit comments

Comments
 (0)