media-agent-mcp 0.1.0__py3-none-any.whl → 0.2.0__py3-none-any.whl

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)')
@@ -395,6 +397,9 @@ def main():
     # Start the server with specified transport
     if args.transport == 'sse':
         logger.info(f"Starting SSE server on {args.host}:{args.port}")
+        mcp.settings.host = args.host
+        mcp.settings.port = args.port
+
         mcp.run(transport="sse")
     else:
         # Default stdio transport

{media_agent_mcp-0.1.0.dist-info → media_agent_mcp-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 0.1.0
+Version: 0.2.0
 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.0.dist-info → media_agent_mcp-0.2.0.dist-info}/RECORD

@@ -1,18 +1,18 @@
 media_agent_mcp/__init__.py,sha256=xOmCIpPlIRJH3RiA-dlBOhKlwyJp51XQRt2eeyU5EtI,255
-media_agent_mcp/server.py,sha256=beN1Zp2gg-VUgXH9tD7Tu5SUMnzXfUU2KT-yp6t66hw,13594
+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/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.0.dist-info/METADATA,sha256=d7vKEHQDq7owJ_FMKWUcZEPxb2WeQqFNc_ARYl75bPM,10974
-media_agent_mcp-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-media_agent_mcp-0.1.0.dist-info/entry_points.txt,sha256=kHB6iLdmnTsqibMZGfsGJzbj_ZU2uJiOfAMA8rYX8bg,64
-media_agent_mcp-0.1.0.dist-info/RECORD,,
+media_agent_mcp-0.2.0.dist-info/METADATA,sha256=JgL4ATkyCVctZXk054BSEli0YGS4o_gxaxIQ_tm-zOk,10974
+media_agent_mcp-0.2.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+media_agent_mcp-0.2.0.dist-info/entry_points.txt,sha256=kHB6iLdmnTsqibMZGfsGJzbj_ZU2uJiOfAMA8rYX8bg,64
+media_agent_mcp-0.2.0.dist-info/RECORD,,