pollo-mcp/dist/lib/index.js

Pollo AI Model Context Protocol (MCP) Server for video generation

var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
import MCPServer from './server.js';
import { GetTaskStatusTool, Img2VideoTool, Text2VideoTool } from './tools.js';
import * as resources from './resources.js';
import { env } from './env.js';
export default function MCPServerFactory(name, version) {
    return __awaiter(this, void 0, void 0, function* () {
        const generationModels = yield resources.getGenerationModels();
        const t2vTools = generationModels
            .filter((m) => m.capability === 't2v')
            .filter((m) => env.POLLO_AI_VIDEO_MODEL_TEXT.includes(m.modelAlias))
            .map((m) => new Text2VideoTool(m.modelBrand, m.modelAlias, m.inputSchema));
        const i2vTools = generationModels
            .filter((m) => m.capability === 'i2v')
            .filter((m) => env.POLLO_AI_VIDEO_MODEL_IMG.includes(m.modelAlias))
            .map((m) => new Img2VideoTool(m.modelBrand, m.modelAlias, m.inputSchema));
        return () => new MCPServer({
            name,
            version,
        }, [
            new GetTaskStatusTool(),
            ...t2vTools,
            ...i2vTools,
        ]);
    });
}
//# sourceMappingURL=index.js.map