@infimind/video-mcp-cli 1.0.2 → 1.0.4

package/dist/server.js

@@ -6,6 +6,8 @@
  */
 import { Server } from "@modelcontextprotocol/sdk/server/index.js";
 import { CallToolRequestSchema, ListToolsRequestSchema, } from "@modelcontextprotocol/sdk/types.js";
+import { readFileSync, existsSync } from "fs";
+import { extname, basename } from "path";
 /**
  * Create and configure the MCP Server
  */
@@ -44,7 +46,7 @@ export function createServer(token, apiBaseUrl) {
                         },
                         sourceImage: {
                             type: "string",
-                            description: "可选的参考图片 - 可以是本地文件路径（如 /path/to/image.jpg）或公网 URL",
+                            description: "可选的参考图片 - 可以是本地文件路径（如 /path/to/image.jpg）或公网 URL。本地文件会被自动转换为 base64",
                         },
                         model: {
                             type: "string",
@@ -98,13 +100,13 @@ export function createServer(token, apiBaseUrl) {
             {
                 name: "create_video_analysis_task",
                 description: "创建视频拆解任务，将视频拆解为脚本、风格、卖点话术、爆款公式，并生成用于视频创作的提示词。" +
-                    "支持本地文件路径或公网 URL 作为视频文件。",
+                    "支持本地文件路径或公网 URL 作为视频文件。本地文件会被自动转换为 base64。",
                 inputSchema: {
                     type: "object",
                     properties: {
                         videoFile: {
                             type: "string",
-                            description: "要分析的视频文件 - 可以是本地文件路径（如 /path/to/video.mp4）或公网 URL",
+                            description: "要分析的视频文件 - 可以是本地文件路径（如 /path/to/video.mp4）或公网 URL。本地文件会被自动转换为 base64",
                         },
                     },
                     required: ["videoFile"],
@@ -187,6 +189,54 @@ async function handleGetUserInfo(apiBaseUrl, token) {
         ],
     };
 }
+/**
+ * 处理文件输入：支持本地文件路径或公网 URL
+ * 对于本地文件，读取并转换为 base64
+ */
+async function processFileInput(fileInput) {
+    // 检查是否是公网 URL
+    if (fileInput.startsWith("http://") || fileInput.startsWith("https://")) {
+        return {
+            data: fileInput,
+            fileName: fileInput.split("/").pop() || "file",
+            mimeType: "url",
+            isUrl: true,
+        };
+    }
+    // 本地文件路径处理
+    const filePath = fileInput;
+    // 检查文件是否存在
+    if (!existsSync(filePath)) {
+        throw new Error(`文件不存在: ${filePath}`);
+    }
+    // 读取文件
+    const buffer = readFileSync(filePath);
+    // 获取文件扩展名并确定 MIME type
+    const ext = extname(filePath).toLowerCase();
+    let mimeType = "application/octet-stream";
+    const mimeTypes = {
+        ".jpg": "image/jpeg",
+        ".jpeg": "image/jpeg",
+        ".png": "image/png",
+        ".gif": "image/gif",
+        ".webp": "image/webp",
+        ".mp4": "video/mp4",
+        ".mov": "video/quicktime",
+        ".avi": "video/x-msvideo",
+        ".mkv": "video/x-matroska",
+    };
+    if (ext in mimeTypes) {
+        mimeType = mimeTypes[ext];
+    }
+    // 转换为 base64
+    const base64 = `data:${mimeType};base64,${buffer.toString("base64")}`;
+    return {
+        data: base64,
+        fileName: basename(filePath),
+        mimeType,
+        isUrl: false,
+    };
+}
 /**
  * Handle create_video_task tool call
  */
@@ -194,7 +244,7 @@ async function handleCreateVideoTask(apiBaseUrl, token, args) {
     const response = await callRestApi(apiBaseUrl, "/mcp/video/create", {
         token,
         prompt: args.prompt,
-        sourceImage: args.sourceImage,
+        sourceImage: args.sourceImage ? (await processFileInput(args.sourceImage)).data : undefined,
         model: args.model || "sora_video2-15s",
         size: args.size || "1280x704",
         seconds: args.seconds || 10,
@@ -272,7 +322,7 @@ async function handleGetVideoStatus(apiBaseUrl, token, args) {
 async function handleCreateVideoAnalysisTask(apiBaseUrl, token, args) {
     const response = await callRestApi(apiBaseUrl, "/mcp/video-analysis/create", {
         token,
-        videoFile: args.videoFile,
+        videoFile: (await processFileInput(args.videoFile)).data,
     }, "POST");
     const result = [
         `# 视频拆解任务已创建`,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@infimind/video-mcp-cli",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "MCP Server for Sora AI Video Generator - Access video generation features through Claude Desktop",
   "type": "module",
   "main": "dist/index.js",