Skip to content

Commit 3843ed0

Browse files
authored
fix: donot use tuple when sequential_image_generation==auot (#393)
1 parent 987d4df commit 3843ed0

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

veadk/tools/builtin_tools/image_generate.py

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -122,25 +122,23 @@ def handle_single_task_sync(
122122
and sequential_image_generation == "auto"
123123
and max_images
124124
):
125-
response = (
126-
client.images.generate(
127-
model=getenv(
128-
"MODEL_IMAGE_NAME",
129-
DEFAULT_IMAGE_GENERATE_MODEL_NAME,
130-
),
131-
**inputs,
132-
sequential_image_generation_options=SequentialImageGenerationOptions(
133-
max_images=max_images
134-
),
135-
extra_headers={
136-
"veadk-source": "veadk",
137-
"veadk-version": VERSION,
138-
"User-Agent": f"VeADK/{VERSION}",
139-
"X-Client-Request-Id": getenv(
140-
"MODEL_AGENT_CLIENT_REQ_ID", f"veadk/{VERSION}"
141-
),
142-
},
125+
response = client.images.generate(
126+
model=getenv(
127+
"MODEL_IMAGE_NAME",
128+
DEFAULT_IMAGE_GENERATE_MODEL_NAME,
143129
),
130+
**inputs,
131+
sequential_image_generation_options=SequentialImageGenerationOptions(
132+
max_images=max_images
133+
),
134+
extra_headers={
135+
"veadk-source": "veadk",
136+
"veadk-version": VERSION,
137+
"User-Agent": f"VeADK/{VERSION}",
138+
"X-Client-Request-Id": getenv(
139+
"MODEL_AGENT_CLIENT_REQ_ID", f"veadk/{VERSION}"
140+
),
141+
},
144142
)
145143
else:
146144
response = client.images.generate(

0 commit comments

Comments
 (0)