wan26_Default_value_modificationtions_and_Fix_name_too_long_error

Author: lether

src/agentscope_bricks/components/generations/async_image_to_video_fl_wan22.py

@@ -95,17 +95,15 @@ 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"
     )
 
     @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,
@@ -226,9 +224,7 @@ class ImageToVideoByFirstAndLastFrameWan22Fetch(
         ImageToVideoByFirstAndLastFrameWan22FetchOutput,
     ],
 ):
-    name: str = (
-        "modelstudio_image_to_video_by_first_and_last_frame_wan22_fetch_result"
-    )
+    name: str = "modelstudio_image_to_video_fl_wan22_fetch_result"
     description: str = (
         "查询通义万相 wan2.2-kf2v-flash 首尾帧生视频任务的结果。\n"
         "输入 Task ID，返回生成的视频 URL 及任务状态。\n"
@@ -238,7 +234,7 @@ class ImageToVideoByFirstAndLastFrameWan22Fetch(
 
     @trace(
         trace_type="AIGC",
-        trace_name="image_to_video_by_first_and_last_frame_wan22_fetch",
+        trace_name="image_to_video_fl_wan22_fetch",
     )
     async def arun(
         self,

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(
         ...,
@@ -104,10 +104,8 @@ async def arun(
             raise ValueError("Please set valid DASHSCOPE_API_KEY!")
 
         model_name = "wan2.6-image"
-
-        # 构造多模态 content：文本 + 可选图像
         content = [{"text": args.prompt}]
-        images = args.images or []  # 安全处理 None
+        images = args.images or []
         for img_url in images:
             content.append({"image": img_url})
 
@@ -120,15 +118,15 @@ async def arun(
         parameters = {}
         if args.negative_prompt:
             parameters["negative_prompt"] = args.negative_prompt
-        if args.size and args.size != "1280*1280":
+        if args.size:
             parameters["size"] = args.size
         if args.seed is not None:
             parameters["seed"] = args.seed
         if args.watermark is not None:
             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(
@@ -156,7 +154,6 @@ async def arun(
                         message = getattr(choice, "message", {})
                         msg_content = getattr(message, "content", [])
                         if isinstance(msg_content, list):
-                            # 遍历当前 choice 的 content
                             for item in msg_content:
                                 if isinstance(item, dict) and "image" in item:
                                     results.append(item["image"])
@@ -176,7 +173,6 @@ async def arun(
                             and "image" in msg_content
                         ):
                             results.append(msg_content["image"])
-                    # --- 修改结束 ---
         except Exception as e:
             raise RuntimeError(
                 f"Failed to parse Wanx 2.6 API response: {str(e)}",

src/agentscope_bricks/components/generations/image_generation_wan26.py

@@ -120,9 +120,9 @@ async def arun(
         parameters = {}
         if args.negative_prompt:
             parameters["negative_prompt"] = args.negative_prompt
-        if args.size and args.size != "1024*1024":
+        if args.size:
             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
@@ -150,7 +150,7 @@ async def arun(
         try:
             if hasattr(response, "output") and response.output:
                 choices = getattr(response.output, "choices", [])
-                for choice in choices:  # ← 遍历所有 choices，而不是只取 [0]
+                for choice in choices:
                     message = getattr(choice, "message", {})
                     content = getattr(message, "content", [])
                     if isinstance(content, list):