media-agent-mcp 0.1.1__tar.gz → 0.2.0__tar.gz

{media_agent_mcp-0.1.1 → media_agent_mcp-0.2.0}/.env.template

@@ -7,8 +7,8 @@ TOS_REGION=ap-southeast-1
 
 # Volcengine Visual Intelligence SDK Configuration
 # 用于SeedEdit等视觉AI服务
-VOLC_ACCESS_KEY=your_volcengine_access_key_here
-VOLC_SECRET_KEY=your_volcengine_secret_key_here
+VOLC_AK=your_volcengine_access_key_here
+VOLC_SK=your_volcengine_secret_key_here
 
 # BytePlus ModelArk API Configuration
 # 获取API密钥: https://console.byteplus.com/
@@ -17,18 +17,13 @@ ARK_BASE_URL=https://ark.ap-southeast.bytepluses.com
 
 # Seedance Video Generation (首尾帧视频生成)
 # 支持文本生成视频、首帧生成视频、首尾帧生成视频
-SEEDANCE_EP=seedance-1-0-lite
+# SEEDANCE_EP=
 
 # Seedream Image Generation (图像生成)
-SEEDREAM_EP=seedream-1-0
+# SEEDREAM_EP=
 
 # SeedEdit Image Editing (图像编辑)
-SEEDEDIT_EP=seededit-3-0
+# SEEDEDIT_EP=
 
 # Seed1.6 VLM (视觉语言模型)
-VLM_EP=seed-1-6-chat
-
-# Server Configuration
-# MCP_HOST=127.0.0.1
-# MCP_PORT=8000
-# LOG_LEVEL=INFO
+# VLM_EP=

{media_agent_mcp-0.1.1 → media_agent_mcp-0.2.0}/.idea/workspace.xml

@@ -24,10 +24,11 @@
   <component name="PropertiesComponent"><![CDATA[{
   "keyToString": {
     "Python.image_selector.executor": "Run",
+    "Python.server.executor": "Debug",
     "RunOnceActivity.ShowReadmeOnStart": "true",
     "RunOnceActivity.git.unshallow": "true",
     "git-widget-placeholder": "main",
-    "last_opened_file_path": "/Users/bytedance/CodeQ3/Meida_Agent_MCP/media-agent-mcp/src/media_agent_mcp/selectors",
+    "last_opened_file_path": "/Users/bytedance/CodeQ3/Meida_Agent_MCP/media-agent-mcp/src/media_agent_mcp",
     "node.js.detected.package.eslint": "true",
     "node.js.detected.package.stylelint": "true",
     "node.js.detected.package.tslint": "true",
@@ -41,10 +42,11 @@
 }]]></component>
   <component name="RecentsManager">
     <key name="CopyFile.RECENT_KEYS">
+      <recent name="$PROJECT_DIR$/src/media_agent_mcp" />
       <recent name="$PROJECT_DIR$/src/media_agent_mcp/selectors" />
     </key>
   </component>
-  <component name="RunManager">
+  <component name="RunManager" selected="Python.server">
     <configuration name="image_selector" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true">
       <module name="media-agent-mcp" />
       <option name="ENV_FILES" value="" />
@@ -68,8 +70,32 @@
       <option name="INPUT_FILE" value="" />
       <method v="2" />
     </configuration>
+    <configuration name="server" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true">
+      <module name="media-agent-mcp" />
+      <option name="ENV_FILES" value="" />
+      <option name="INTERPRETER_OPTIONS" value="" />
+      <option name="PARENT_ENVS" value="true" />
+      <envs>
+        <env name="PYTHONUNBUFFERED" value="1" />
+      </envs>
+      <option name="SDK_HOME" value="" />
+      <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/src/media_agent_mcp" />
+      <option name="IS_MODULE_SDK" value="true" />
+      <option name="ADD_CONTENT_ROOTS" value="true" />
+      <option name="ADD_SOURCE_ROOTS" value="true" />
+      <EXTENSION ID="PythonCoverageRunConfigurationExtension" runner="coverage.py" />
+      <option name="SCRIPT_NAME" value="$PROJECT_DIR$/src/media_agent_mcp/server.py" />
+      <option name="PARAMETERS" value="" />
+      <option name="SHOW_COMMAND_LINE" value="false" />
+      <option name="EMULATE_TERMINAL" value="false" />
+      <option name="MODULE_MODE" value="false" />
+      <option name="REDIRECT_INPUT" value="false" />
+      <option name="INPUT_FILE" value="" />
+      <method v="2" />
+    </configuration>
     <recent_temporary>
       <list>
+        <item itemvalue="Python.server" />
         <item itemvalue="Python.image_selector" />
       </list>
     </recent_temporary>
@@ -90,14 +116,26 @@
       <option name="number" value="Default" />
       <option name="presentableId" value="Default" />
       <updated>1753446872370</updated>
-      <workItem from="1753446874397" duration="7330000" />
+      <workItem from="1753446874397" duration="8632000" />
     </task>
     <servers />
   </component>
   <component name="TypeScriptGeneratedFilesManager">
     <option name="version" value="3" />
   </component>
+  <component name="XDebuggerManager">
+    <breakpoint-manager>
+      <breakpoints>
+        <line-breakpoint enabled="true" suspend="THREAD" type="python-line">
+          <url>file://$PROJECT_DIR$/src/media_agent_mcp/media_selectors/image_selector.py</url>
+          <line>73</line>
+          <option name="timeStamp" value="2" />
+        </line-breakpoint>
+      </breakpoints>
+    </breakpoint-manager>
+  </component>
   <component name="com.intellij.coverage.CoverageDataManagerImpl">
+    <SUITE FILE_PATH="coverage/media_agent_mcp$server.coverage" NAME="server Coverage Results" MODIFIED="1753503806895" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/src/media_agent_mcp" />
     <SUITE FILE_PATH="coverage/media_agent_mcp$image_selector.coverage" NAME="image_selector Coverage Results" MODIFIED="1753447327399" SOURCE_PROVIDER="com.intellij.coverage.DefaultCoverageFileProvider" RUNNER="coverage.py" COVERAGE_BY_TEST_ENABLED="false" COVERAGE_TRACING_ENABLED="false" WORKING_DIRECTORY="$PROJECT_DIR$/src/media_agent_mcp/selectors" />
   </component>
 </project>

{media_agent_mcp-0.1.1 → media_agent_mcp-0.2.0}/PKG-INFO

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

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

{media_agent_mcp-0.1.1 → media_agent_mcp-0.2.0}/src/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-0.1.1 → media_agent_mcp-0.2.0}/src/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-0.1.1 → media_agent_mcp-0.2.0}/src/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-0.1.1 → media_agent_mcp-0.2.0}/src/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)')