feat(mcp): add wan26 generations

src/agentscope_bricks/components/__init__.py

@@ -1,148 +1,185 @@
-# -*- coding: utf-8 -*-
-from typing import Dict, Type, List
-
-from pydantic import BaseModel, Field
-
-from agentscope_bricks.base import Component
-from agentscope_bricks.components.generations.async_image_to_video_wan25 import (  # noqa
-    ImageToVideoWan25Fetch,
-    ImageToVideoWan25Submit,
-)
-from agentscope_bricks.components.generations.async_text_to_video_wan25 import (  # noqa
-    TextToVideoWan25Submit,
-    TextToVideoWan25Fetch,
-)
-from agentscope_bricks.components.generations.image_edit_wan25 import (
-    ImageEditWan25,
-)
-from agentscope_bricks.components.generations.multichannel_speech_to_text import (  # noqa
-    MultichannelSpeechToText,
-)
-from agentscope_bricks.components.generations.qwen_image_edit import (
-    QwenImageEdit,
-)
-from agentscope_bricks.components.generations.qwen_image_generation import (
-    QwenImageGen,
-)
-from agentscope_bricks.components.generations.qwen_text_to_speech import (
-    QwenTextToSpeech,
-)
-from agentscope_bricks.components.generations.text_to_video import TextToVideo
-from agentscope_bricks.components.generations.image_to_video import (
-    ImageToVideo,
-)
-from agentscope_bricks.components.generations.speech_to_video import (
-    SpeechToVideo,
-)
-from agentscope_bricks.components.searches.modelstudio_search_lite import (
-    ModelstudioSearchLite,
-)
-from agentscope_bricks.components.generations.image_generation import (
-    ImageGeneration,
-)
-from agentscope_bricks.components.generations.image_generation_wan25 import (
-    ImageGenerationWan25,
-)
-from agentscope_bricks.components.generations.image_edit import ImageEdit
-from agentscope_bricks.components.generations.image_style_repaint import (
-    ImageStyleRepaint,
-)
-from agentscope_bricks.components.generations.speech_to_text import (
-    SpeechToText,
-)
-
-from agentscope_bricks.components.generations.async_text_to_video import (
-    TextToVideoSubmit,
-    TextToVideoFetch,
-)
-from agentscope_bricks.components.generations.async_image_to_video import (
-    ImageToVideoSubmit,
-    ImageToVideoFetch,
-)
-from agentscope_bricks.components.generations.async_speech_to_video import (
-    SpeechToVideoSubmit,
-    SpeechToVideoFetch,
-)
-from agentscope_bricks.components.generations.async_text_to_video_wan26 import (  # noqa
-    TextToVideoWan26Submit,
-)
-from agentscope_bricks.components.generations.async_image_to_video_wan26 import (  # noqa
-    ImageToVideoWan26Submit,
-)
-from agentscope_bricks.components.generations.image_generation_wan26 import (  # noqa
-    ImageGenerationWan26,
-)
-from agentscope_bricks.components.generations.fetch_wan import WanVideoFetch
-from agentscope_bricks.components.generations.qwen_image_edit_new import (
-    QwenImageEditNew,
-)  # noqa
-
-
-class McpServerMeta(BaseModel):
-    instructions: str = Field(
-        ...,
-        description="服务描述",
-    )
-    components: List[Type[Component]] = Field(
-        ...,
-        description="组件列表",
-    )
-
-
-mcp_server_metas: Dict[str, McpServerMeta] = {
-    "modelstudio_wan_image": McpServerMeta(
-        instructions="基于通义万相大模型的智能图像生成服务，提供高质量的图像处理和编辑功能",
-        components=[ImageGeneration, ImageEdit, ImageStyleRepaint],
-    ),
-    "modelstudio_wan_video": McpServerMeta(
-        instructions="基于通义万相大模型提供AI视频生成服务，支持文本到视频、图像到视频和语音到视频的多模态生成功能",
-        components=[
-            TextToVideoSubmit,
-            TextToVideoFetch,
-            ImageToVideoSubmit,
-            ImageToVideoFetch,
-            SpeechToVideoSubmit,
-            SpeechToVideoFetch,
-        ],
-    ),
-    "modelstudio_wan25_media": McpServerMeta(
-        instructions="基于通义万相大模型2.5版本提供的图像和视频生成服务",
-        components=[
-            ImageGenerationWan25,
-            ImageEditWan25,
-            TextToVideoWan25Submit,
-            TextToVideoWan25Fetch,
-            ImageToVideoWan25Submit,
-            ImageToVideoWan25Fetch,
-        ],
-    ),
-    "modelstudio_qwen_image": McpServerMeta(
-        instructions="基于通义千问大模型的智能图像生成服务，提供高质量的图像处理和编辑功能",
-        components=[
-            QwenImageGen,
-            QwenImageEdit,
-            QwenImageEditNew,
-        ],  # ← 新增 QwenImageEditNew
-    ),
-    "modelstudio_web_search": McpServerMeta(
-        instructions="提供实时互联网搜索服务，提供准确及时的信息检索功能",
-        components=[ModelstudioSearchLite],
-    ),
-    "modelstudio_speech_to_text": McpServerMeta(
-        instructions="录音文件的语音识别服务，支持多种音频格式的语音转文字功能",
-        components=[SpeechToText, MultichannelSpeechToText],
-    ),
-    "modelstudio_qwen_text_to_speech": McpServerMeta(
-        instructions="基于通义千问大模型的语音合成服务，支持多种语言语音合成功能",
-        components=[QwenTextToSpeech],
-    ),
-    "modelstudio_wan26_media": McpServerMeta(
-        instructions="基于通义万相大模型2.6版本提供的图像和视频生成服务",
-        components=[
-            ImageGenerationWan26,
-            TextToVideoWan26Submit,
-            ImageToVideoWan26Submit,
-            WanVideoFetch,
-        ],
-    ),
-}
+# -*- coding: utf-8 -*-
+from typing import Dict, Type, List
+
+from pydantic import BaseModel, Field
+
+from agentscope_bricks.base import Component
+from agentscope_bricks.components.generations.async_image_to_video_wan25 import (  # noqa
+    ImageToVideoWan25Fetch,
+    ImageToVideoWan25Submit,
+)
+from agentscope_bricks.components.generations.async_text_to_video_wan25 import (  # noqa
+    TextToVideoWan25Submit,
+    TextToVideoWan25Fetch,
+)
+from agentscope_bricks.components.generations.image_edit_wan25 import (
+    ImageEditWan25,
+)
+from agentscope_bricks.components.generations.multichannel_speech_to_text import (  # noqa
+    MultichannelSpeechToText,
+)
+from agentscope_bricks.components.generations.qwen_image_edit import (
+    QwenImageEdit,
+)
+from agentscope_bricks.components.generations.qwen_image_generation import (
+    QwenImageGen,
+)
+from agentscope_bricks.components.generations.qwen_text_to_speech import (
+    QwenTextToSpeech,
+)
+from agentscope_bricks.components.generations.text_to_video import TextToVideo
+from agentscope_bricks.components.generations.image_to_video import (
+    ImageToVideo,
+)
+from agentscope_bricks.components.generations.speech_to_video import (
+    SpeechToVideo,
+)
+from agentscope_bricks.components.searches.modelstudio_search_lite import (
+    ModelstudioSearchLite,
+)
+from agentscope_bricks.components.generations.image_generation import (
+    ImageGeneration,
+)
+from agentscope_bricks.components.generations.image_generation_wan25 import (
+    ImageGenerationWan25,
+)
+from agentscope_bricks.components.generations.image_edit import ImageEdit
+from agentscope_bricks.components.generations.image_style_repaint import (
+    ImageStyleRepaint,
+)
+from agentscope_bricks.components.generations.speech_to_text import (
+    SpeechToText,
+)
+
+from agentscope_bricks.components.generations.async_text_to_video import (
+    TextToVideoSubmit,
+    TextToVideoFetch,
+)
+from agentscope_bricks.components.generations.async_image_to_video import (
+    ImageToVideoSubmit,
+    ImageToVideoFetch,
+)
+from agentscope_bricks.components.generations.async_speech_to_video import (
+    SpeechToVideoSubmit,
+    SpeechToVideoFetch,
+)
+from agentscope_bricks.components.generations.async_text_to_video_wan26 import (  # noqa
+    TextToVideoWan26Submit,
+)
+from agentscope_bricks.components.generations.async_image_to_video_wan26 import (  # noqa
+    ImageToVideoWan26Submit,
+)
+from agentscope_bricks.components.generations.image_generation_wan26 import (  # noqa
+    ImageGenerationWan26,
+)
+from agentscope_bricks.components.generations.fetch_wan import WanVideoFetch
+from agentscope_bricks.components.generations.qwen_image_edit_new import (
+    QwenImageEditNew,
+)
+from agentscope_bricks.components.generations.image_edit_wan26 import (
+    ImageEditWan26,
+)
+from agentscope_bricks.components.generations.image_generation_zimage import (
+    ZImageGeneration,
+)
+from agentscope_bricks.components.generations.async_image_out_painting import (
+    ImageOutPaintingSubmit,
+    ImageOutPaintingFetch,
+)
+from agentscope_bricks.components.generations.async_image_to_video_fl_wan22 import (  # noqa
+    ImageToVideoByFirstAndLastFrameWan22Submit,
+)
+from agentscope_bricks.components.generations.image_out_painting import (
+    ImageOutPaintingAuto,
+)
+from agentscope_bricks.components.generations.image_text_interleave_generation_wan26 import (  # noqa
+    WanImageInterleaveGeneration,
+)
+
+
+class McpServerMeta(BaseModel):
+    instructions: str = Field(
+        ...,
+        description="服务描述",
+    )
+    components: List[Type[Component]] = Field(
+        ...,
+        description="组件列表",
+    )
+
+
+mcp_server_metas: Dict[str, McpServerMeta] = {
+    "modelstudio_wan_image": McpServerMeta(
+        instructions="基于通义万相大模型的智能图像生成服务，提供高质量的图像处理和编辑功能",
+        components=[
+            ImageGeneration,
+            ImageEdit,
+            ImageStyleRepaint,
+            ImageOutPaintingSubmit,
+            ImageOutPaintingFetch,
+            ImageOutPaintingAuto,
+        ],
+    ),
+    "modelstudio_wan_video": McpServerMeta(
+        instructions="基于通义万相大模型提供AI视频生成服务，支持文本到视频、图像到视频和语音到视频的多模态生成功能",
+        components=[
+            TextToVideoSubmit,
+            TextToVideoFetch,
+            ImageToVideoSubmit,
+            ImageToVideoFetch,
+            SpeechToVideoSubmit,
+            SpeechToVideoFetch,
+            ImageToVideoByFirstAndLastFrameWan22Submit,
+            WanVideoFetch,
+        ],
+    ),
+    "modelstudio_wan25_media": McpServerMeta(
+        instructions="基于通义万相大模型2.5版本提供的图像和视频生成服务",
+        components=[
+            ImageGenerationWan25,
+            ImageEditWan25,
+            TextToVideoWan25Submit,
+            TextToVideoWan25Fetch,
+            ImageToVideoWan25Submit,
+            ImageToVideoWan25Fetch,
+        ],
+    ),
+    "modelstudio_qwen_image": McpServerMeta(
+        instructions="基于通义千问大模型的智能图像生成服务，提供高质量的图像处理和编辑功能",
+        components=[
+            QwenImageGen,
+            QwenImageEdit,
+            QwenImageEditNew,
+        ],  # ← 新增 QwenImageEditNew
+    ),
+    "modelstudio_web_search": McpServerMeta(
+        instructions="提供实时互联网搜索服务，提供准确及时的信息检索功能",
+        components=[ModelstudioSearchLite],
+    ),
+    "modelstudio_speech_to_text": McpServerMeta(
+        instructions="录音文件的语音识别服务，支持多种音频格式的语音转文字功能",
+        components=[SpeechToText, MultichannelSpeechToText],
+    ),
+    "modelstudio_qwen_text_to_speech": McpServerMeta(
+        instructions="基于通义千问大模型的语音合成服务，支持多种语言语音合成功能",
+        components=[QwenTextToSpeech],
+    ),
+    "modelstudio_wan26_media": McpServerMeta(
+        instructions="基于通义万相大模型2.6版本提供的图像和视频生成服务",
+        components=[
+            ImageGenerationWan26,
+            TextToVideoWan26Submit,
+            ImageToVideoWan26Submit,
+            WanVideoFetch,
+            ImageEditWan26,
+            WanImageInterleaveGeneration,
+        ],
+    ),
+    "modelstudio_z_image": McpServerMeta(
+        instructions="基于通义Z-Image大模型的智能图像生成服务，是一款轻量级文生图模型，"
+        "可快速生成图像，支持中英文字渲染，并灵活适配多种分辨率与宽高比例。",
+        components=[
+            ZImageGeneration,
+        ],
+    ),
+}