youling-agent-mcp/bin/cli.js

MCP服务器实现POMDP研发任务拆解

#!/usr/bin/env node

/**
 * CLI入口文件
 * 支持通过npx或全局安装运行
 * 支持HTTP模式和stdio模式
 */

import { logger, isStdioMode } from './logger.js';

// 包版本信息
const PACKAGE_VERSION = '0.0.3'; // 当前包版本
const SERVER_NAME = 'youling-agent-mcp';

// 检查是否为MCP stdio模式 - 重要：此检查必须放在任何可能产生输出的代码之前
const runtimeIsStdioMode = process.argv.includes('--stdio') || 
                         process.argv.includes('--mcp') || 
                         process.argv.includes('--transportType=stdio') || 
                         process.env.transportType === 'stdio';

process.env.MCP_TRANSPORT_MODE = runtimeIsStdioMode ? 'stdio' : 'http';

// 导入MCP服务器，它将根据环境变量选择正确的传输模式
import('../dist/mcp-server.js')
  .then(() => {
    // MCP服务器初始化完成
    // 注意：HTTP和stdio模式的具体处理都在mcp-server.js中实现
    logger.debug(`MCP服务器初始化成功，模式：${runtimeIsStdioMode ? 'stdio' : 'http'}`);
  })
  .catch((error) => {
    logger.error('启动服务器失败:', error instanceof Error ? error.message : String(error));
    process.exit(1);
  });