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

media_agent_mcp/async_server.py

@@ -22,6 +22,27 @@ from typing import List, Optional
 import json
 from dotenv import load_dotenv
 import uvicorn
+from functools import wraps
+
+def async_retry(max_retries=3, delay=2):
+    def decorator(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            for attempt in range(max_retries):
+                try:
+                    result = await func(*args, **kwargs)
+                    if isinstance(result, dict) and result.get('status') == 'error':
+                        logger.warning(f"Attempt {attempt + 1} of {max_retries} failed for {func.__name__}. Error: {result.get('message')}. Retrying in {delay}s...")
+                        await asyncio.sleep(delay)
+                        continue
+                    return result
+                except Exception as e:
+                    logger.error(f"Attempt {attempt + 1} of {max_retries} failed for {func.__name__} with exception: {e}. Retrying in {delay}s...")
+                    await asyncio.sleep(delay)
+            logger.error(f"Function {func.__name__} failed after {max_retries} retries.")
+            return {"status": "error", "data": None, "message": f"Function {func.__name__} failed after {max_retries} retries."}
+        return wrapper
+    return decorator
 
 from mcp.server.fastmcp import FastMCP
 
@@ -49,7 +70,8 @@ mcp = FastMCP("Media-Agent-MCP-Async")
 
 
 @mcp.tool()
-async def video_concat_tool(video_urls: List[str]) -> str:
+@async_retry()
+async def video_concat_tool(video_urls: List[str]) -> dict:
     """
     Asynchronously concatenate multiple videos from URLs and upload to TOS.
     
@@ -57,13 +79,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:
+@async_retry()
+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 +94,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:
+@async_retry()
+async def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> dict:
     """
     Asynchronously generate an image using Seedream AI model.
     
@@ -86,15 +110,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()
+@async_retry()
 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 +131,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()
+@async_retry()
 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 +151,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:
+@async_retry()
+async def vlm_vision_task_tool(messages: List) -> dict:
     """
     Asynchronously perform vision-language tasks using VLM model.
     
@@ -139,13 +166,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:
+@async_retry()
+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 +182,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:
+@async_retry()
+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 +198,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()
+@async_retry()
 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 +216,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.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.3.9
+Version: 0.3.11
 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.9.dist-info → media_agent_mcp-0.3.11.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=IpXezeefilGukutgBqOU7RvCDp-t9_j11A2OOqiu2z0,10524
 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.11.dist-info/METADATA,sha256=92jf-Ti8koiMfQqYbpcFNwNVQ4sntuCz0XrRTnBe0rI,10976
+media_agent_mcp-0.3.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+media_agent_mcp-0.3.11.dist-info/entry_points.txt,sha256=qhOUwR-ORVf9GO7emhhl7Lgd6MISgqbZr8bEuSH_VdA,70
+media_agent_mcp-0.3.11.dist-info/top_level.txt,sha256=WEa0YfchpTxZgiKn8gdxYgs-dir5HepJaTOrxAGx9nY,16
+media_agent_mcp-0.3.11.dist-info/RECORD,,