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

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

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

@@ -1,6 +1,6 @@
 [project]
 name = "media-agent-mcp"
-version = "0.3.8"
+version = "0.3.9"
 description = "A Model Context Protocol server for media processing with AI tools"
 readme = "README.md"
 requires-python = ">=3.12"

{media_agent_mcp-0.3.8 → media_agent_mcp-0.3.9}/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.9}/src/media_agent_mcp/async_server.py

@@ -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.
     
@@ -63,7 +63,7 @@ async def video_concat_tool_async(video_urls: List[str]) -> str:
 
 
 @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.
     
@@ -77,7 +77,7 @@ async def video_last_frame_tool_async(video_url: str) -> str:
 
 
 @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.
     
@@ -92,7 +92,7 @@ async def seedream_generate_image_tool_async(prompt: str, size: str = "1024x1024
 
 
 @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:
     """
@@ -112,7 +112,7 @@ async def seedance_generate_video_tool_async(prompt: str, first_frame_image: str
 
 
 @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.
@@ -131,7 +131,7 @@ async def seededit_tool_async(image_url: str, prompt: str, seed: int = -1,
 
 
 @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.
     
@@ -145,7 +145,7 @@ async def vlm_vision_task_tool_async(messages: List) -> str:
 
 
 @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.
     
@@ -160,7 +160,7 @@ async def image_selector_tool_async(image_paths: List[str], prompt: str) -> str:
 
 
 @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.
     
@@ -175,7 +175,7 @@ async def video_selector_tool_async(video_paths: List[str], prompt: str) -> str:
 
 
 @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.
@@ -205,21 +205,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 +214,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.9}/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.9}/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.9
 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