diff --git a/src/agentscope_bricks/components/generations/async_image_to_video_fl_wan22.py b/src/agentscope_bricks/components/generations/async_image_to_video_fl_wan22.py index 7bd2477..856bb66 100644 --- a/src/agentscope_bricks/components/generations/async_image_to_video_fl_wan22.py +++ b/src/agentscope_bricks/components/generations/async_image_to_video_fl_wan22.py @@ -95,9 +95,7 @@ class ImageToVideoByFirstAndLastFrameWan22Submit( task using the wan2.2-kf2v-flash model. """ - name: str = ( - "modelstudio_image_to_video_by_first_and_last_frame_wan22_submit_task" - ) + name: str = "modelstudio_image_to_video_fl_wan22_submit_task" description: str = ( "[版本: wan2.2] 通义万相首尾帧生视频模型(wan2.2-kf2v-flash)异步任务提交工具。\n" "基于首帧与尾帧图像及文本提示,生成一段流畅的无声视频(当前不支持音频输出)。\n" @@ -105,7 +103,7 @@ class ImageToVideoByFirstAndLastFrameWan22Submit( @trace( trace_type="AIGC", - trace_name="image_to_video_by_first_and_last_frame_wan22_submit", + trace_name="image_to_video_fl_wan22_submit_task", ) async def arun( self, diff --git a/src/agentscope_bricks/components/generations/image_edit_wan26.py b/src/agentscope_bricks/components/generations/image_edit_wan26.py index 855aa2f..2c2001d 100644 --- a/src/agentscope_bricks/components/generations/image_edit_wan26.py +++ b/src/agentscope_bricks/components/generations/image_edit_wan26.py @@ -41,8 +41,8 @@ class ImageGenInput(BaseModel): description="是否添加水印,false:默认值,不添加水印,true:添加水印。", ) n: Optional[int] = Field( - default=4, - description="生成图片的数量。取值范围为1~4张 默认4", + default=1, + description="生成图片的数量。取值范围为1~4张 默认1", ) images: list[str] = Field( ..., @@ -128,7 +128,7 @@ async def arun( parameters["watermark"] = args.watermark if args.prompt_extend is not None: parameters["prompt_extend"] = args.prompt_extend - if args.n is not None and args.n != 4: + if args.n is not None: parameters["n"] = args.n try: response = await AioMultiModalConversation.call( diff --git a/src/agentscope_bricks/components/generations/image_generation_wan26.py b/src/agentscope_bricks/components/generations/image_generation_wan26.py index efe20cb..005b18b 100644 --- a/src/agentscope_bricks/components/generations/image_generation_wan26.py +++ b/src/agentscope_bricks/components/generations/image_generation_wan26.py @@ -79,7 +79,7 @@ class ImageGenerationWan26( name: str = "modelstudio_wanx26_image_generation" description: str = ( - "[版本: wan2.6] 通义万相文生图模型(wanx2.6-t2i)。AI绘画服务,根据文本描述生成高质量图像,并返回图片URL。\n" + "[版本: wan2.6] 通义万相文生图模型(wan2.6-t2i)。AI绘画服务,根据文本描述生成高质量图像,并返回图片URL。\n" "新功能包括图像编辑和图文混合输出,满足更多样化的生成与集成需求。\n" "支持自定义分辨率:图像面积介于 768×768 至 1440×1440 像素之间," "允许在该范围内自由调整宽高比(例如 768×2700)。\n" @@ -122,7 +122,7 @@ async def arun( parameters["negative_prompt"] = args.negative_prompt if args.size and args.size != "1024*1024": parameters["size"] = args.size - if args.n is not None and args.n != 1: + if args.n is not None: parameters["n"] = args.n if args.seed is not None: parameters["seed"] = args.seed @@ -140,18 +140,18 @@ async def arun( ) except Exception as e: raise RuntimeError( - f"Failed to call Wanx 2.6 image generation API: {str(e)}", + f"Failed to call Wan 2.6 image generation API: {str(e)}", ) from e if response.status_code != 200 or not response.output: - raise RuntimeError(f"Wanx 2.6 image generation failed: {response}") + raise RuntimeError(f"Wan 2.6 image generation failed: {response}") results = [] try: if hasattr(response, "output") and response.output: choices = getattr(response.output, "choices", []) - if choices: - message = getattr(choices[0], "message", {}) + for choice in choices: # ← 遍历所有 choices,而不是只取 [0] + message = getattr(choice, "message", {}) content = getattr(message, "content", []) if isinstance(content, list): for item in content: @@ -163,7 +163,7 @@ async def arun( results.append(content["image"]) except Exception as e: raise RuntimeError( - f"Failed to parse Wanx 2.6 API response: {str(e)}", + f"Failed to parse Wan 2.6 API response: {str(e)}", ) from e if not results: