media-agent-mcp 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

media_agent_mcp/async_server.py

@@ -27,15 +27,15 @@ from mcp.server.fastmcp import FastMCP
 
 # Import async wrappers
 from media_agent_mcp.async_wrapper import (
-    async_video_concat_tool,
-    async_video_last_frame_tool,
-    async_seedream_generate_image_tool,
-    async_seedance_generate_video_tool,
-    async_seededit_tool,
-    async_vlm_vision_task_tool,
-    async_image_selector_tool,
-    async_video_selector_tool,
-    async_tos_save_content_tool,
+    video_concat_tool,
+    video_last_frame_tool,
+    seedream_generate_image_tool,
+    seedance_generate_video_tool,
+    seededit_tool,
+    vlm_vision_task_tool,
+    image_selector_tool,
+    video_selector_tool,
+    tos_save_content_tool,
     cleanup_executor
 )
 
@@ -49,7 +49,7 @@ mcp = FastMCP("Media-Agent-MCP-Async")
 
 
 @mcp.tool()
-async def video_concat_tool_async(video_urls: List[str]) -> str:
+async def video_concat_tool(video_urls: List[str]) -> str:
     """
     Asynchronously concatenate multiple videos from URLs and upload to TOS.
     
@@ -59,11 +59,11 @@ async def video_concat_tool_async(video_urls: List[str]) -> str:
     Returns:
         JSON string with status, data, and message
     """
-    return await async_video_concat_tool(video_urls)
+    return await video_concat_tool(video_urls)
 
 
 @mcp.tool()
-async def video_last_frame_tool_async(video_url: str) -> str:
+async def video_last_frame_tool(video_url: str) -> str:
     """
     Asynchronously extract the last frame from a video file and upload to TOS.
     
@@ -73,11 +73,11 @@ async def video_last_frame_tool_async(video_url: str) -> str:
     Returns:
         JSON string with status, data, and message
     """
-    return await async_video_last_frame_tool(video_url)
+    return await video_last_frame_tool(video_url)
 
 
 @mcp.tool()
-async def seedream_generate_image_tool_async(prompt: str, size: str = "1024x1024") -> str:
+async def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> str:
     """
     Asynchronously generate an image using Seedream AI model.
     
@@ -88,11 +88,11 @@ async def seedream_generate_image_tool_async(prompt: str, size: str = "1024x1024
     Returns:
         JSON string with status, data, and message
     """
-    return await async_seedream_generate_image_tool(prompt, size)
+    return await seedream_generate_image_tool(prompt, size)
 
 
 @mcp.tool()
-async def seedance_generate_video_tool_async(prompt: str, first_frame_image: str, 
+async def seedance_generate_video_tool(prompt: str, first_frame_image: str, 
                                             last_frame_image: str = None, duration: int = 5, 
                                             resolution: str = "720p") -> str:
     """
@@ -108,11 +108,11 @@ async def seedance_generate_video_tool_async(prompt: str, first_frame_image: str
     Returns:
         JSON string with status, data, and message
     """
-    return await async_seedance_generate_video_tool(prompt, first_frame_image, last_frame_image, duration, resolution)
+    return await seedance_generate_video_tool(prompt, first_frame_image, last_frame_image, duration, resolution)
 
 
 @mcp.tool()
-async def seededit_tool_async(image_url: str, prompt: str, seed: int = -1, 
+async def seededit_tool(image_url: str, prompt: str, seed: int = -1, 
                              scale: float = 0.5, charactor_keep: bool = False) -> str:
     """
     Asynchronously edit an image using Seededit model.
@@ -127,11 +127,11 @@ async def seededit_tool_async(image_url: str, prompt: str, seed: int = -1,
     Returns:
         JSON string with status, data, and message
     """
-    return await async_seededit_tool(image_url, prompt, seed, scale, charactor_keep)
+    return await seededit_tool(image_url, prompt, seed, scale, charactor_keep)
 
 
 @mcp.tool()
-async def vlm_vision_task_tool_async(messages: List) -> str:
+async def vlm_vision_task_tool(messages: List) -> str:
     """
     Asynchronously perform vision-language tasks using VLM model.
     
@@ -141,11 +141,11 @@ async def vlm_vision_task_tool_async(messages: List) -> str:
     Returns:
         JSON string with status, data, and message
     """
-    return await async_vlm_vision_task_tool(messages)
+    return await vlm_vision_task_tool(messages)
 
 
 @mcp.tool()
-async def image_selector_tool_async(image_paths: List[str], prompt: str) -> str:
+async def image_selector_tool(image_paths: List[str], prompt: str) -> str:
     """
     Asynchronously select the best image from multiple options using VLM model.
     
@@ -156,11 +156,11 @@ async def image_selector_tool_async(image_paths: List[str], prompt: str) -> str:
     Returns:
         JSON string with status, data, and message
     """
-    return await async_image_selector_tool(image_paths, prompt)
+    return await image_selector_tool(image_paths, prompt)
 
 
 @mcp.tool()
-async def video_selector_tool_async(video_paths: List[str], prompt: str) -> str:
+async def video_selector_tool(video_paths: List[str], prompt: str) -> str:
     """
     Asynchronously select the best video from multiple options using VLM model.
     
@@ -171,11 +171,11 @@ async def video_selector_tool_async(video_paths: List[str], prompt: str) -> str:
     Returns:
         JSON string with status, data, and message
     """
-    return await async_video_selector_tool(video_paths, prompt)
+    return await video_selector_tool(video_paths, prompt)
 
 
 @mcp.tool()
-async def tos_save_content_tool_async(content: str, file_extension: str = "txt", 
+async def tos_save_content_tool(content: str, file_extension: str = "txt", 
                                      object_key: Optional[str] = None) -> str:
     """
     Asynchronously save content to TOS and return URL.
@@ -188,7 +188,7 @@ async def tos_save_content_tool_async(content: str, file_extension: str = "txt",
     Returns:
         JSON string with status, data, and message
     """
-    return await async_tos_save_content_tool(content, file_extension, object_key)
+    return await tos_save_content_tool(content, file_extension, object_key)
 
 
 # Utility function for concurrent execution

media_agent_mcp/async_wrapper.py

@@ -85,28 +85,28 @@ def json_response_wrapper(func: Callable) -> Callable:
 # Async wrapped functions
 @async_wrapper
 @json_response_wrapper
-def _sync_video_concat(video_urls: List[str]) -> str:
+def video_concat_tool(video_urls: List[str]) -> str:
     """Synchronous video concatenation wrapper."""
     return concat_videos(video_urls)
 
 
 @async_wrapper
 @json_response_wrapper
-def _sync_video_last_frame(video_url: str) -> str:
+def video_last_frame_tool(video_url: str) -> str:
     """Synchronous video last frame extraction wrapper."""
     return extract_last_frame(video_url)
 
 
 @async_wrapper
 @json_response_wrapper
-def _sync_seedream_generate_image(prompt: str, size: str = "1024x1024") -> str:
+def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> str:
     """Synchronous image generation wrapper."""
     return generate_image(prompt, size=size)
 
 
 @async_wrapper
 @json_response_wrapper
-def _sync_seedance_generate_video(prompt: str, first_frame_image: str, 
+def seedance_generate_video_tool(prompt: str, first_frame_image: str, 
                                  last_frame_image: str = None, duration: int = 5, 
                                  resolution: str = "720p") -> str:
     """Synchronous video generation wrapper."""
@@ -128,7 +128,7 @@ def _sync_seedance_generate_video(prompt: str, first_frame_image: str,
 
 @async_wrapper
 @json_response_wrapper
-def _sync_seededit(image_url: str, prompt: str, seed: int = -1, 
+def seededit_tool(image_url: str, prompt: str, seed: int = -1, 
                   scale: float = 0.5, charactor_keep: bool = False) -> str:
     """Synchronous image editing wrapper."""
     return seededit(
@@ -143,14 +143,14 @@ def _sync_seededit(image_url: str, prompt: str, seed: int = -1,
 
 @async_wrapper
 @json_response_wrapper
-def _sync_vlm_vision_task(messages: List) -> str:
+def vlm_vision_task_tool(messages: List) -> str:
     """Synchronous VLM vision task wrapper."""
     from media_agent_mcp.ai_models.seed16 import process_vlm_task
     return process_vlm_task(messages)
 
 
 @async_wrapper
-def _sync_image_selector(image_paths: List[str], prompt: str) -> str:
+def image_selector_tool(image_paths: List[str], prompt: str) -> str:
     """Synchronous image selector wrapper."""
     try:
         result = select_best_image(image_paths, prompt)
@@ -168,7 +168,7 @@ def _sync_image_selector(image_paths: List[str], prompt: str) -> str:
 
 
 @async_wrapper
-def _sync_video_selector(video_paths: List[str], prompt: str) -> str:
+def video_selector_tool(video_paths: List[str], prompt: str) -> str:
     """Synchronous video selector wrapper."""
     try:
         result = select_best_video(video_paths, prompt)
@@ -186,7 +186,7 @@ def _sync_video_selector(video_paths: List[str], prompt: str) -> str:
 
 
 @async_wrapper
-def _sync_tos_save_content(content: str, file_extension: str = "txt", 
+def tos_save_content_tool(content: str, file_extension: str = "txt", 
                           object_key: Optional[str] = None) -> str:
     """Synchronous TOS content save wrapper."""
     import tempfile

{media_agent_mcp-0.3.4.dist-info → media_agent_mcp-0.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.3.4
+Version: 0.3.5
 Summary: A Model Context Protocol server for media processing with AI tools
 Author-email: Media Agent Team <team@mediaagent.com>
 Keywords: mcp,ai,media,video,image,processing

{media_agent_mcp-0.3.4.dist-info → media_agent_mcp-0.3.5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 media_agent_mcp/__init__.py,sha256=4kV5u8kHrsdjpMHDNUhc8h4U6AwOyw1mNptFPd2snrQ,365
-media_agent_mcp/async_server.py,sha256=ZvUKAezbsMTmXTe-SvffoGkn_GLiP9mSqSxGXVEMI0s,9793
-media_agent_mcp/async_wrapper.py,sha256=bHi0cCKxv6R9oHcNcNq8tV0AyKrTDV47s8BuWetqlIg,8992
+media_agent_mcp/async_server.py,sha256=600PJ9pkpxSSQPPnw-xFf51O7XYuRn2eRfuxvo2GNsQ,9631
+media_agent_mcp/async_wrapper.py,sha256=pBYtOPrAVavPhWDW4WIE9m5hCmlyoCMiDV8Gv6N7ZeI,8983
 media_agent_mcp/server.py,sha256=-mTeNcJXIlgByCG2HhaqV6zSbhjrQIEv_ssLN47KAzM,13759
 media_agent_mcp/ai_models/__init__.py,sha256=WN8JDSJbj2-sNwaQg3MDwmm9mvJTC1mqpWliac3sHoc,427
 media_agent_mcp/ai_models/seed16.py,sha256=P3KHNF_SfLryJz6qFP7UUtBfw_28ubdrj2lJkt3WsY8,4989
@@ -14,8 +14,8 @@ media_agent_mcp/storage/__init__.py,sha256=eio7ZiSeLjCxICSZwZisiR7wKJfXlT2PV7aDE
 media_agent_mcp/storage/tos_client.py,sha256=9c3GPmQe2stvxgZzsWYjVqKzVyvluZqYneCXYj-FQ3M,3174
 media_agent_mcp/video/__init__.py,sha256=4ILnqYaUaYKI4GWOSg2SNZqt10UM1Y-8Q2SVvjoijqY,277
 media_agent_mcp/video/processor.py,sha256=5ABPyj1IoD2xyIiB8dkEx4ZssPTXKHtuoFjy-dmnFks,11866
-media_agent_mcp-0.3.4.dist-info/METADATA,sha256=jgvdLlrqaELoPb23mLBycd5l9876vp-atRUtkrZKbp4,10975
-media_agent_mcp-0.3.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-media_agent_mcp-0.3.4.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
-media_agent_mcp-0.3.4.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
-media_agent_mcp-0.3.4.dist-info/RECORD,,
+media_agent_mcp-0.3.5.dist-info/METADATA,sha256=F0tsG5aIwLkGV9B5j0-On8F5yBtb4GAbu0I2BkiP-cw,10975
+media_agent_mcp-0.3.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+media_agent_mcp-0.3.5.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
+media_agent_mcp-0.3.5.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
+media_agent_mcp-0.3.5.dist-info/RECORD,,