63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
#!/usr/bin/env python
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
FLUX风格塑形MCP服务器启动脚本
|
||
--------------------------
|
||
启动带有FLUX风格塑形功能的MCP服务器
|
||
"""
|
||
|
||
import logging
|
||
import argparse
|
||
import sys
|
||
import asyncio
|
||
|
||
# 设置日志
|
||
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
||
logger = logging.getLogger(__name__)
|
||
|
||
# 导入MCP服务器
|
||
from mcp_server import MCPServer
|
||
from flux_style_resource import register_flux_resources
|
||
|
||
async def main_async():
|
||
"""异步主函数"""
|
||
# 解析命令行参数
|
||
parser = argparse.ArgumentParser(description='启动带有FLUX风格塑形功能的MCP服务器')
|
||
parser.add_argument('--host', type=str, default='127.0.0.1', help='服务器监听地址')
|
||
parser.add_argument('--port', type=int, default=8189, help='服务器监听端口')
|
||
parser.add_argument('--name', type=str, default='FLUX-MCP', help='服务器名称')
|
||
args = parser.parse_args()
|
||
|
||
try:
|
||
# 创建MCP服务器实例
|
||
logger.info(f"正在初始化MCP服务器: {args.name}...")
|
||
server = MCPServer(server_name=args.name)
|
||
|
||
# 注册FLUX风格塑形资源和工具
|
||
logger.info("正在注册FLUX风格塑形资源...")
|
||
register_flux_resources(server)
|
||
|
||
# 启动服务器
|
||
logger.info(f"正在启动MCP服务器，监听地址: {args.host}:{args.port}")
|
||
await server.start(host=args.host, port=args.port)
|
||
except KeyboardInterrupt:
|
||
logger.info("收到中断信号，服务器关闭")
|
||
except Exception as e:
|
||
logger.error(f"服务器启动失败: {e}")
|
||
return 1
|
||
|
||
return 0
|
||
|
||
def main():
|
||
"""主函数"""
|
||
try:
|
||
return asyncio.run(main_async())
|
||
except KeyboardInterrupt:
|
||
logger.info("收到中断信号，服务器关闭")
|
||
return 0
|
||
except Exception as e:
|
||
logger.error(f"服务器运行出错: {e}")
|
||
return 1
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main()) |