Skip to content

Commit 0e60c8b

Browse files
authored
fix: image generate tool (#355)
* fix: image generate tool * fix: update default param
1 parent 10b9e27 commit 0e60c8b

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

veadk/consts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919

2020
DEFAULT_AGENT_NAME = "veAgent"
2121

22-
DEFAULT_MODEL_AGENT_NAME = "doubao-seed-1-6-250615"
22+
DEFAULT_MODEL_AGENT_NAME = "doubao-seed-1-6-251015"
2323
DEFAULT_MODEL_AGENT_PROVIDER = "openai"
2424
DEFAULT_MODEL_AGENT_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
2525
DEFAULT_MODEL_EXTRA_CONFIG = {
@@ -68,7 +68,7 @@
6868
DEFAULT_VIDEO_MODEL_NAME = "doubao-seedance-1-0-pro-250528"
6969
DEFAULT_VIDEO_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
7070

71-
DEFAULT_IMAGE_GENERATE_MODEL_NAME = "doubao-seedream-4-0-250828"
71+
DEFAULT_IMAGE_GENERATE_MODEL_NAME = "doubao-seedream-4-5-251128"
7272
DEFAULT_IMAGE_GENERATE_MODEL_API_BASE = "https://ark.cn-beijing.volces.com/api/v3/"
7373

7474
VEFAAS_IAM_CRIDENTIAL_PATH = "/var/run/secrets/iam/credential"

veadk/tools/builtin_tools/image_generate.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,6 +109,8 @@ def handle_single_task_sync(
109109
inputs["watermark"] = watermark
110110
if sequential_image_generation:
111111
inputs["sequential_image_generation"] = sequential_image_generation
112+
if image_field is not None:
113+
inputs["image"] = [image_field]
112114

113115
with tracer.start_as_current_span(f"call_llm_task_{idx}") as span:
114116
try:

0 commit comments

Comments
 (0)