media-agent-mcp 0.3.8__tar.gz → 0.3.10__tar.gz

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.3.8
+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.8 → media_agent_mcp-0.3.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "media-agent-mcp"
-version = "0.3.8"
+version = "0.3.10"
 description = "A Model Context Protocol server for media processing with AI tools"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -27,6 +27,7 @@ dependencies = [
     "python-dotenv>=1.0.0",
     "volcengine-python-sdk>=1.0.0",
     "volcengine>=1.0.194",
+    "tenacity>=8.2.3",
 ]
 
 [project.scripts]

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.10}/src/media_agent_mcp/__init__.py

@@ -2,6 +2,8 @@
 
 from . import ai_models, media_selectors, storage, video
 from .server import main
+from .async_server import main as async_main
+from . import async_wrapper
 
 __version__ = "0.1.0"
-__all__ = ['ai_models', 'media_selectors', 'storage', 'video', 'main']
+__all__ = ['ai_models', 'media_selectors', 'storage', 'video', 'main', 'async_main', 'async_wrapper']

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.10}/src/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_async(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_async(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_async(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_async(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_async(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_async(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 def seedance_generate_video_tool_async(prompt: str, first_frame_image: str, 
+@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_async(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 def seededit_tool_async(image_url: str, prompt: str, seed: int = -1, 
-                             scale: float = 0.5, charactor_keep: bool = False) -> str:
+@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) -> dict:
     """
     Asynchronously edit an image using Seededit model.
     
@@ -125,13 +141,14 @@ async def seededit_tool_async(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_async(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_async(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_async(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_async(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_async(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_async(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 def tos_save_content_tool_async(content: str, file_extension: str = "txt", 
-                                     object_key: Optional[str] = None) -> str:
+@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) -> dict:
     """
     Asynchronously save content to TOS and return URL.
     
@@ -186,7 +206,7 @@ async def tos_save_content_tool_async(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)
 
@@ -205,21 +225,6 @@ async def run_multiple_tools_concurrently(*coroutines):
     return await asyncio.gather(*coroutines, return_exceptions=True)
 
 
-# Example usage function
-async def example_concurrent_usage():
-    """
-    Example of how to use multiple tools concurrently.
-    """
-    # Example: Generate image and process video concurrently
-    image_task = seedream_generate_image_tool_async("A beautiful sunset", "1024x1024")
-    video_task = video_last_frame_tool_async("https://example.com/video.mp4")
-    
-    # Run both tasks concurrently
-    results = await run_multiple_tools_concurrently(image_task, video_task)
-    
-    return results
-
-
 def main():
     """Main entry point for the Async MCP server."""
     # Parse command line arguments
@@ -229,7 +234,7 @@ def main():
     parser.add_argument('--host', type=str, default='127.0.0.1',
                         help='Host for SSE transport (default: 127.0.0.1)')
     parser.add_argument('--port', type=int, default=8000,
-                        help='Port for SSE transport (default: 8001)')
+                        help='Port for SSE transport (default: 8000)')
     parser.add_argument('--version', action='store_true',
                         help='Show version information')
     

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.10}/src/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 video_concat_tool(video_urls: List[str]) -> str:
+def _sync_video_concat(video_urls: List[str]) -> str:
     """Synchronous video concatenation wrapper."""
     return concat_videos(video_urls)
 
 
 @async_wrapper
 @json_response_wrapper
-def video_last_frame_tool(video_url: str) -> str:
+def _sync_video_last_frame(video_url: str) -> str:
     """Synchronous video last frame extraction wrapper."""
     return extract_last_frame(video_url)
 
 
 @async_wrapper
 @json_response_wrapper
-def seedream_generate_image_tool(prompt: str, size: str = "1024x1024") -> str:
+def _sync_seedream_generate_image(prompt: str, size: str = "1024x1024") -> str:
     """Synchronous image generation wrapper."""
     return generate_image(prompt, size=size)
 
 
 @async_wrapper
 @json_response_wrapper
-def seedance_generate_video_tool(prompt: str, first_frame_image: str, 
+def _sync_seedance_generate_video(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 seedance_generate_video_tool(prompt: str, first_frame_image: str,
 
 @async_wrapper
 @json_response_wrapper
-def seededit_tool(image_url: str, prompt: str, seed: int = -1, 
+def _sync_seededit(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 seededit_tool(image_url: str, prompt: str, seed: int = -1,
 
 @async_wrapper
 @json_response_wrapper
-def vlm_vision_task_tool(messages: List) -> str:
+def _sync_vlm_vision_task(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 image_selector_tool(image_paths: List[str], prompt: str) -> str:
+def _sync_image_selector(image_paths: List[str], prompt: str) -> str:
     """Synchronous image selector wrapper."""
     try:
         result = select_best_image(image_paths, prompt)
@@ -168,7 +168,7 @@ def image_selector_tool(image_paths: List[str], prompt: str) -> str:
 
 
 @async_wrapper
-def video_selector_tool(video_paths: List[str], prompt: str) -> str:
+def _sync_video_selector(video_paths: List[str], prompt: str) -> str:
     """Synchronous video selector wrapper."""
     try:
         result = select_best_video(video_paths, prompt)
@@ -186,7 +186,7 @@ def video_selector_tool(video_paths: List[str], prompt: str) -> str:
 
 
 @async_wrapper
-def tos_save_content_tool(content: str, file_extension: str = "txt", 
+def _sync_tos_save_content(content: str, file_extension: str = "txt", 
                           object_key: Optional[str] = None) -> str:
     """Synchronous TOS content save wrapper."""
     import tempfile

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.10}/src/media_agent_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.3.8
+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.8 → media_agent_mcp-0.3.10}/src/media_agent_mcp.egg-info/requires.txt

@@ -9,3 +9,4 @@ numpy>=1.24.0
 python-dotenv>=1.0.0
 volcengine-python-sdk>=1.0.0
 volcengine>=1.0.194
+tenacity>=8.2.3