media-agent-mcp 0.1.1__py3-none-any.whl → 0.2.1__py3-none-any.whl

media_agent_mcp/ai_models/seedance.py

@@ -222,21 +222,21 @@ def _poll_video_status(task_id: str, api_key: str, max_wait_time: int = 90) -> D
             
             elif status in ['queued', 'running']:
                 # Continue polling
-                time.sleep(5)
+                time.sleep(0.5)
                 continue
             
             else:
                 logger.warning(f"Unknown status: {status}")
-                time.sleep(5)
+                time.sleep(0.5)
                 continue
                 
         except requests.exceptions.RequestException as e:
             logger.error(f"Error polling status: {str(e)}")
-            time.sleep(5)
+            time.sleep(0.5)
             continue
         except Exception as e:
             logger.error(f"Unexpected error polling status: {str(e)}")
-            time.sleep(5)
+            time.sleep(0.5)
             continue
     
     return f"Error: Video generation timed out after {max_wait_time} seconds"

media_agent_mcp/ai_models/seedream.py

@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
 
 
 def generate_image(prompt: str, size: str = "1024x1024", guidance_scale: float = 2.5, 
-                  watermark: bool = True, seed: int = -1, **kwargs) -> Dict[str, Any]:
+                  watermark: bool = False, seed: int = -1, **kwargs) -> Dict[str, Any]:
     """Generate an image using Seedream model.
     
     Args:

media_agent_mcp/media_selectors/image_selector.py

@@ -70,6 +70,7 @@ def select_best_image(image_urls: List[str], prompt: str) -> dict:
 
         try:
             response = process_vlm_task(messages, is_json=True)
+            response = response['data']['response']
             logger.info(f"model response: {response}")
 
         except Exception as e:

media_agent_mcp/media_selectors/video_selector.py

@@ -110,6 +110,7 @@ def select_best_video(video_urls: List[str], prompt: str) -> dict:
 
         try:
             response = process_vlm_task(messages, is_json=True)
+            response = response['data']['response']
             logger.info(f"model response: {response}")
 
         except Exception as e:

media_agent_mcp/server.py

@@ -17,6 +17,7 @@ import argparse
 import logging
 from typing import Optional, Dict, Any
 import json
+from dotenv import load_dotenv
 
 from mcp.server.fastmcp import FastMCP
 
@@ -34,6 +35,7 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # Initialize FastMCP server (will be configured in main function)
+load_dotenv()
 mcp = FastMCP("Media Agent MCP")
 
 
@@ -361,7 +363,7 @@ def main():
     """Main entry point for the MCP server."""
     # Parse command line arguments
     parser = argparse.ArgumentParser(description='Media Agent MCP Server')
-    parser.add_argument('--transport', type=str, choices=['sse', 'stdio'], default='stdio',
+    parser.add_argument('--transport', type=str, choices=['sse', 'stdio'], default='sse',
                         help='Transport method: sse or stdio (default: stdio)')
     parser.add_argument('--host', type=str, default='127.0.0.1',
                         help='Host for SSE transport (default: 127.0.0.1)')

{media_agent_mcp-0.1.1.dist-info → media_agent_mcp-0.2.1.dist-info}/METADATA

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

{media_agent_mcp-0.1.1.dist-info → media_agent_mcp-0.2.1.dist-info}/RECORD

@@ -1,18 +1,18 @@
 media_agent_mcp/__init__.py,sha256=xOmCIpPlIRJH3RiA-dlBOhKlwyJp51XQRt2eeyU5EtI,255
-media_agent_mcp/server.py,sha256=Uf7RjuLfiohLsLoToo1QPi3wQ0Q38UDS5_X0AesA114,13671
+media_agent_mcp/server.py,sha256=dQvc_hAdUTpvq8OrJmD6d83VTKfTh4AJv19VVraAQBc,13714
 media_agent_mcp/ai_models/__init__.py,sha256=WN8JDSJbj2-sNwaQg3MDwmm9mvJTC1mqpWliac3sHoc,427
 media_agent_mcp/ai_models/seed16.py,sha256=P3KHNF_SfLryJz6qFP7UUtBfw_28ubdrj2lJkt3WsY8,4989
-media_agent_mcp/ai_models/seedance.py,sha256=2kj1Qqc-_1I-Ub23FMNXR-gpxExhGCEQe2XacBdV3sY,9016
+media_agent_mcp/ai_models/seedance.py,sha256=fv7hdxlwRtFqMnOG9o7Kx_-k2ihB6VBgN3wCoo-Y-Qs,9024
 media_agent_mcp/ai_models/seededit.py,sha256=eLOox-8lZudTYe0CpWyONGHC8D12r7nakhkXIXmE7H4,3245
-media_agent_mcp/ai_models/seedream.py,sha256=91xFVfC6nh_jYfuz0FK6LL2fHreLd-pRQ4mWHpjQlvg,4466
+media_agent_mcp/ai_models/seedream.py,sha256=lKWMRzV-ZjZ2JB93A3XYUOF7QfzP3zFMQleBtEOkuW8,4467
 media_agent_mcp/media_selectors/__init__.py,sha256=eN3wyTPdHTsbW_xDyPXBLDaMA15YuI9qeiy-zeY7ZBQ,260
-media_agent_mcp/media_selectors/image_selector.py,sha256=tcg-yz1lEex3Q-cAh1bzvngwNBz3rM5264TUnZVLLQ8,4573
-media_agent_mcp/media_selectors/video_selector.py,sha256=NwwpZqZeW99aQznlU8vloTNbcn7epXlmYZgxY1gnuaA,6216
+media_agent_mcp/media_selectors/image_selector.py,sha256=jVPgk3LiruMK2SJ4gPqCGkRmum2ahObTRbQ7q1sjvhg,4625
+media_agent_mcp/media_selectors/video_selector.py,sha256=xgTyvuVSBTeZHQGFB6mjLvpYPGTodlHGuGZT3n8D2gA,6268
 media_agent_mcp/storage/__init__.py,sha256=eio7ZiSeLjCxICSZwZisiR7wKJfXlT2PV7aDEQKBUQQ,215
 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.1.1.dist-info/METADATA,sha256=fsNW1mKkHS0z-bpnKyoAOfvzse47XV-RZ_XNugTxoL0,10974
-media_agent_mcp-0.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-media_agent_mcp-0.1.1.dist-info/entry_points.txt,sha256=kHB6iLdmnTsqibMZGfsGJzbj_ZU2uJiOfAMA8rYX8bg,64
-media_agent_mcp-0.1.1.dist-info/RECORD,,
+media_agent_mcp-0.2.1.dist-info/METADATA,sha256=t1XiDuJhEgQjyjYIKtxLzXtQNKDe1YgmGuptpudnwAI,10974
+media_agent_mcp-0.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+media_agent_mcp-0.2.1.dist-info/entry_points.txt,sha256=kHB6iLdmnTsqibMZGfsGJzbj_ZU2uJiOfAMA8rYX8bg,64
+media_agent_mcp-0.2.1.dist-info/RECORD,,