media-agent-mcp 2.6.2__tar.gz → 2.6.3__tar.gz

{media_agent_mcp-2.6.2 → media_agent_mcp-2.6.3}/PKG-INFO

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

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

{media_agent_mcp-2.6.2 → media_agent_mcp-2.6.3}/src/media_agent_mcp/ai_models/openaiedit.py

@@ -183,33 +183,47 @@ def google_edit(image_urls: List[str], prompt: str) -> Dict[str, Any]:
 
         contents = pil_images + [prompt]
         
-        # model = genai.GenerativeModel("gemini-1.5-flash-latest")
-        # response = model.generate_content(contents)
         client = genai.Client(
             api_key=api_key,
         )
 
-        response = client.models.generate_content(
-            model="gemini-2.5-flash-image-preview",
-            contents=contents
-        )
+        # 最多尝试3次
+        max_retries = 3
+        for attempt in range(max_retries):
+            try:
+                response = client.models.generate_content(
+                    model="gemini-2.5-flash-image-preview",
+                    contents=contents
+                )
 
-        image_data = None
-        if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
-            image_parts = [
-                part.inline_data.data
-                for part in response.candidates[0].content.parts
-                if part.inline_data
-            ]
-            if image_parts:
-                image_data = image_parts[0]
+                image_data = None
+                if response.candidates and response.candidates[0].content and response.candidates[0].content.parts:
+                    image_parts = [
+                        part.inline_data.data
+                        for part in response.candidates[0].content.parts
+                        if part.inline_data
+                    ]
+                    if image_parts:
+                        image_data = image_parts[0]
 
-        if not image_data:
-            return {
-                "status": "error",
-                "data": None,
-                "message": "Google API response did not contain image data."
-            }
+                # 如果获取到了图片数据，跳出重试循环
+                if image_data:
+                    break
+                    
+                # 如果是最后一次尝试且仍然没有图片数据，返回错误
+                if attempt == max_retries - 1:
+                    return {
+                        "status": "error",
+                        "data": None,
+                        "message": f"Google API response did not contain image data after {max_retries} attempts."
+                    }
+                    
+            except Exception as api_error:
+                # 如果是最后一次尝试，抛出异常
+                if attempt == max_retries - 1:
+                    raise api_error
+                # 否则继续下一次尝试
+                continue
 
         with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_edited_file:
             temp_edited_file.write(image_data)

{media_agent_mcp-2.6.2 → media_agent_mcp-2.6.3}/src/media_agent_mcp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: media-agent-mcp
-Version: 2.6.2
+Version: 2.6.3
 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