From 73910da5d0c185de5ca14d1c09feaf4a4b4be660 Mon Sep 17 00:00:00 2001 From: Somesh Koli Date: Sun, 2 Feb 2025 01:55:47 +0530 Subject: [PATCH 1/2] fix: controlnet args structure to match upstream structure before : args {image: {image: "", mask: "" } now: args {image: "", mask: ""} Error with before: TypeError: string indices must be integers, not 'str' response now: {"task_id":"9dca9d76-142e-4c42-ae28-0fdf5f57dc35"} --- agent_scheduler/task_helpers.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/agent_scheduler/task_helpers.py b/agent_scheduler/task_helpers.py index b9e2ad4..4db3df1 100644 --- a/agent_scheduler/task_helpers.py +++ b/agent_scheduler/task_helpers.py @@ -227,11 +227,8 @@ def map_controlnet_args_to_api_task_args(args: Dict): args = args.__dict__ for k, v in args.items(): - if k == "image" and v is not None: - args[k] = { - "image": encode_image_to_base64(v["image"]), - "mask": encode_image_to_base64(v["mask"]) if v.get("mask", None) is not None else None, - } + if (k == "image" || k == "mask") and v is not None: + args[k] = encode_image_to_base64(v) if isinstance(v, Enum): args[k] = v.value From 91a2e29a95669c05a3064200531691ffdc3999db Mon Sep 17 00:00:00 2001 From: Somesh Koli Date: Sun, 2 Feb 2025 02:00:04 +0530 Subject: [PATCH 2/2] Update task_helpers.py --- agent_scheduler/task_helpers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agent_scheduler/task_helpers.py b/agent_scheduler/task_helpers.py index 4db3df1..8a603c2 100644 --- a/agent_scheduler/task_helpers.py +++ b/agent_scheduler/task_helpers.py @@ -227,7 +227,7 @@ def map_controlnet_args_to_api_task_args(args: Dict): args = args.__dict__ for k, v in args.items(): - if (k == "image" || k == "mask") and v is not None: + if (k == "image" or k == "mask") and v is not None: args[k] = encode_image_to_base64(v) if isinstance(v, Enum): args[k] = v.value