#!/usr/bin/env node
import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import {
CallToolRequestSchema,
ListToolsRequestSchema,
} from "@modelcontextprotocol/sdk/types.js";
// 导入业务工具
import { convertStrategy } from "./tools/convertStrategy.js";
// 创建 MCP server
const server = new Server(
{
name: "Quant2Ptrader-MCP",
version: "1.0.0",
},
{
capabilities: {
tools: {},
},
}
);
// 列出所有工具
server.setRequestHandler(ListToolsRequestSchema, async () => {
return {
tools: [
{
name: convertStrategy.name,
description: convertStrategy.description,
inputSchema: convertStrategy.parameters
}
]
};
});
// 执行工具调用
server.setRequestHandler(CallToolRequestSchema, async (request) => {
const { name, arguments: args } = request.params;
switch (name) {
case "convert_joinquant_to_ptrade": {
return await convertStrategy.run({
code: args?.code ? String(args.code) : undefined,
filepath: args?.filepath ? String(args.filepath) : undefined,
output_dir: args?.output_dir ? String(args.output_dir) : undefined
});
}
default:
throw new Error(`Unknown tool: ${name}`);
}
});
// 启动 stdio 传输
async function main() {
const transport = new StdioServerTransport();
await server.connect(transport);
console.error("Quant2Ptrader MCP Server running on stdio");
}
main().catch((error) => {
console.error("Server error:", error);
process.exit(1);
});
