media-agent-mcp 0.3.9__py3-none-any.whl → 0.3.10__py3-none-any.whl

media_agent_mcp/async_server.py

@@ -22,6 +22,17 @@ from typing import List, Optional
 import json
 from dotenv import load_dotenv
 import uvicorn
+from tenacity import retry, stop_after_attempt, wait_fixed
+
+def _is_result_error(result):
+    """Return True if the result is a dictionary with status 'error'."""
+    if isinstance(result, str):
+        try:
+            data = json.loads(result)
+            return isinstance(data, dict) and data.get('status') == 'error'
+        except json.JSONDecodeError:
+            return False
+    return isinstance(result, dict) and result.get('status') == 'error'
 
 from mcp.server.fastmcp import FastMCP
 
@@ -49,7 +60,8 @@ mcp = FastMCP("Media-Agent-MCP-Async")
 
 
 @mcp.tool()
-async def video_concat_tool(video_urls: List[str]) -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def video_concat_tool(video_urls: List[str]) -> dict:
     """
     Asynchronously concatenate multiple videos from URLs and upload to TOS.
     
@@ -57,13 +69,14 @@ async def video_concat_tool(video_urls: List[str]) -> str:
         video_urls: List of video URLs to concatenate in order
     
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_video_concat_tool(video_urls)
 
 
 @mcp.tool()
-async def video_last_frame_tool(video_url: str) -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def video_last_frame_tool(video_url: str) -> dict:
     """
     Asynchronously extract the last frame from a video file and upload to TOS.
     
@@ -71,13 +84,14 @@ async def video_last_frame_tool(video_url: str) -> str:
         video_url: URL or path to the video file
         
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_video_last_frame_tool(video_url)
 
 
 @mcp.tool()
-async def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> dict:
     """
     Asynchronously generate an image using Seedream AI model.
     
@@ -86,15 +100,16 @@ async def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") ->
         size: Size of the image (e.g., "1024x1024")
     
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_seedream_generate_image_tool(prompt, size)
 
 
 @mcp.tool()
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
 async def seedance_generate_video_tool(prompt: str, first_frame_image: str, 
                                             last_frame_image: str = None, duration: int = 5, 
-                                            resolution: str = "720p") -> str:
+                                            resolution: str = "720p") -> dict:
     """
     Asynchronously generate a video using Seedance AI model with first/last frame images.
     
@@ -106,14 +121,15 @@ async def seedance_generate_video_tool(prompt: str, first_frame_image: str,
         resolution: Video resolution (480p, 720p)
     
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_seedance_generate_video_tool(prompt, first_frame_image, last_frame_image, duration, resolution)
 
 
 @mcp.tool()
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
 async def seededit_tool(image_url: str, prompt: str, seed: int = -1, 
-                             scale: float = 0.5, charactor_keep: bool = False) -> str:
+                             scale: float = 0.5, charactor_keep: bool = False) -> dict:
     """
     Asynchronously edit an image using Seededit model.
     
@@ -125,13 +141,14 @@ async def seededit_tool(image_url: str, prompt: str, seed: int = -1,
         charactor_keep: whether to keep the main character in this image, if you wanna change the main character, please keep False
         
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_seededit_tool(image_url, prompt, seed, scale, charactor_keep)
 
 
 @mcp.tool()
-async def vlm_vision_task_tool(messages: List) -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def vlm_vision_task_tool(messages: List) -> dict:
     """
     Asynchronously perform vision-language tasks using VLM model.
     
@@ -139,13 +156,14 @@ async def vlm_vision_task_tool(messages: List) -> str:
         messages: OpenAI-compatible messages format
         
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_vlm_vision_task_tool(messages)
 
 
 @mcp.tool()
-async def image_selector_tool(image_paths: List[str], prompt: str) -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def image_selector_tool(image_paths: List[str], prompt: str) -> dict:
     """
     Asynchronously select the best image from multiple options using VLM model.
     
@@ -154,13 +172,14 @@ async def image_selector_tool(image_paths: List[str], prompt: str) -> str:
         prompt: Selection criteria prompt
         
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_image_selector_tool(image_paths, prompt)
 
 
 @mcp.tool()
-async def video_selector_tool(video_paths: List[str], prompt: str) -> str:
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
+async def video_selector_tool(video_paths: List[str], prompt: str) -> dict:
     """
     Asynchronously select the best video from multiple options using VLM model.
     
@@ -169,14 +188,15 @@ async def video_selector_tool(video_paths: List[str], prompt: str) -> str:
         prompt: Selection criteria prompt
     
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_video_selector_tool(video_paths, prompt)
 
 
 @mcp.tool()
+@retry(stop=stop_after_attempt(3), wait=wait_fixed(2), retry_if_result=_is_result_error)
 async def tos_save_content_tool(content: str, file_extension: str = "txt", 
-                                     object_key: Optional[str] = None) -> str:
+                                     object_key: Optional[str] = None) -> dict:
     """
     Asynchronously save content to TOS and return URL.
     
@@ -186,7 +206,7 @@ async def tos_save_content_tool(content: str, file_extension: str = "txt",
         object_key: Optional key to use for the object in TOS
         
     Returns:
-        JSON string with status, data, and message
+        Dictionary with status, data, and message
     """
     return await async_tos_save_content_tool(content, file_extension, object_key)
 

{media_agent_mcp-0.3.9.dist-info → media_agent_mcp-0.3.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.3.9
+Version: 0.3.10
 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
@@ -22,6 +22,7 @@ Requires-Dist: numpy>=1.24.0
 Requires-Dist: python-dotenv>=1.0.0
 Requires-Dist: volcengine-python-sdk>=1.0.0
 Requires-Dist: volcengine>=1.0.194
+Requires-Dist: tenacity>=8.2.3
 
 # Media Agent MCP
 

{media_agent_mcp-0.3.9.dist-info → media_agent_mcp-0.3.10.dist-info}/RECORD

@@ -1,5 +1,5 @@
 media_agent_mcp/__init__.py,sha256=4kV5u8kHrsdjpMHDNUhc8h4U6AwOyw1mNptFPd2snrQ,365
-media_agent_mcp/async_server.py,sha256=QYOcX0X3vFJ4elOwb1FzFItHC9YVpjA9DCg9CBanSiw,9238
+media_agent_mcp/async_server.py,sha256=vpOMvN804eWz-FKf00Y_H3pE5FTl2C52SeXvFtfDJ8U,10495
 media_agent_mcp/async_wrapper.py,sha256=bHi0cCKxv6R9oHcNcNq8tV0AyKrTDV47s8BuWetqlIg,8992
 media_agent_mcp/server.py,sha256=-mTeNcJXIlgByCG2HhaqV6zSbhjrQIEv_ssLN47KAzM,13759
 media_agent_mcp/ai_models/__init__.py,sha256=WN8JDSJbj2-sNwaQg3MDwmm9mvJTC1mqpWliac3sHoc,427
@@ -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.9.dist-info/METADATA,sha256=HgLx9BJINu6TANDbAZT4qX4adNFegiJDR6BYqm9OwSM,10975
-media_agent_mcp-0.3.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-media_agent_mcp-0.3.9.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
-media_agent_mcp-0.3.9.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
-media_agent_mcp-0.3.9.dist-info/RECORD,,
+media_agent_mcp-0.3.10.dist-info/METADATA,sha256=F0VfYB3vYpzoj0vk9537QWNBFIsCjfzpGUrvd15QdA8,11007
+media_agent_mcp-0.3.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+media_agent_mcp-0.3.10.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
+media_agent_mcp-0.3.10.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
+media_agent_mcp-0.3.10.dist-info/RECORD,,