Xây dựng Máy chủ MCP Kết nối với OpenAI: Hướng dẫn Hoàn chỉnh cho Nhà phát triển

mkdir mcp-server-openai
cd mcp-server-openai
npm init -y
npm install @modelcontextprotocol/sdk openai dotenv express cors
npm install --save-dev typescript @types/node ts-node

{
  "compilerOptions": {
    "target": "ES2020",
    "module": "commonjs",
    "lib": ["ES2020"],
    "outDir": "./dist",
    "rootDir": "./src",
    "strict": true,
    "esModuleInterop": true,
    "skipLibCheck": true,
    "forceConsistentCasingInFileNames": true
  }
}

python -m venv venv
source venv/bin/activate  # Trên Windows: venv\Scripts\activate
pip install mcp openai python-dotenv fastapi uvicorn

import { Server } from "@modelcontextprotocol/sdk/server/index.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import {
  CallToolRequestSchema,
  ListToolsRequestSchema,
} from "@modelcontextprotocol/sdk/types.js";

const server = new Server({
  name: "openai-mcp-server",
  version: "1.0.0",
});

// Định nghĩa các công cụ
const tools = [
  {
    name: "get_customer_info",
    description: "Truy xuất thông tin khách hàng theo ID",
    inputSchema: {
      type: "object",
      properties: {
        customer_id: {
          type: "string",
          description: "Định danh khách hàng duy nhất",
        },
      },
      required: ["customer_id"],
    },
  },
  // Thêm công cụ khác tại đây
];

// Xử lý yêu cầu liệt kê công cụ
server.setRequestHandler(ListToolsRequestSchema, async () => {
  return { tools };
});

// Xử lý yêu cầu thực thi công cụ
server.setRequestHandler(CallToolRequestSchema, async (request) => {
  const { name, arguments: args } = request.params;

  if (name === "get_customer_info") {
    // Thực thi logic công cụ
    const customerId = args.customer_id;
    // Truy vấn database hoặc API
    return {
      content: [
        {
          type: "text",
          text: JSON.stringify({
            id: customerId,
            name: "John Doe",
            email: "john@example.com",
          }),
        },
      ],
    };
  }

  return {
    content: [{ type: "text", text: `Unknown tool: ${name}` }],
    isError: true,
  };
});

// Khởi động server
const transport = new StdioServerTransport();
server.connect(transport);

from mcp.server import Server
from mcp.types import Tool, TextContent, ToolResult
import json

app = Server("openai-mcp-server")

# Định nghĩa các công cụ
TOOLS = [
    Tool(
        name="get_customer_info",
        description="Truy xuất thông tin khách hàng theo ID",
        inputSchema={
            "type": "object",
            "properties": {
                "customer_id": {
                    "type": "string",
                    "description": "Định danh khách hàng duy nhất"
                }
            },
            "required": ["customer_id"]
        }
    )
]

@app.list_tools()
async def list_tools():
    return TOOLS

@app.call_tool()
async def call_tool(name: str, arguments: dict):
    if name == "get_customer_info":
        customer_id = arguments.get("customer_id")
        # Thực thi logic công cụ
        result = {
            "id": customer_id,
            "name": "John Doe",
            "email": "john@example.com"
        }
        return ToolResult(
            content=[TextContent(type="text", text=json.dumps(result))]
        )

    return ToolResult(
        content=[TextContent(type="text", text=f"Unknown tool: {name}")],
        isError=True
    )

if __name__ == "__main__":
    import uvicorn
    uvicorn.run(app, host="0.0.0.0", port=8000)

import { Pool } from "pg"; // Dùng PostgreSQL

const pool = new Pool({
  connectionString: process.env.DATABASE_URL,
});

async function getCustomerInfo(customerId: string) {
  try {
    const result = await pool.query(
      "SELECT id, name, email, account_status FROM customers WHERE id = $1",
      [customerId]
    );

    if (result.rows.length === 0) {
      return {
        error: "Không tìm thấy khách hàng",
        status: 404,
      };
    }

    return result.rows[0];
  } catch (error) {
    return {
      error: "Lỗi truy vấn cơ sở dữ liệu",
      details: error.message,
      status: 500,
    };
  }
}

import axios from "axios";

async function searchExternalDatabase(query: string) {
  try {
    const response = await axios.get(
      "https://api.external-service.com/search",
      {
        params: { q: query },
        headers: {
          Authorization: `Bearer ${process.env.EXTERNAL_API_KEY}`,
        },
      }
    );

    return response.data;
  } catch (error) {
    return {
      error: "Lỗi gọi API bên ngoài",
      details: error.message,
    };
  }
}

function validateInput(args: any, schema: any): { valid: boolean; error?: string } {
  // Kiểm tra trường bắt buộc
  for (const required of schema.required || []) {
    if (!(required in args)) {
      return { valid: false, error: `Thiếu tham số bắt buộc: ${required}` };
    }
  }

  // Kiểm tra kiểu dữ liệu
  for (const [key, property] of Object.entries(schema.properties || {})) {
    if (key in args) {
      const value = args[key];
      const expectedType = (property as any).type;

      if (typeof value !== expectedType) {
        return {
          valid: false,
          error: `Tham số ${key} phải thuộc kiểu ${expectedType}`,
        };
      }
    }
  }

  return { valid: true };
}

import { Client } from "@modelcontextprotocol/sdk/client/index.js";
import { StdioClientTransport } from "@modelcontextprotocol/sdk/client/stdio.js";
import OpenAI from "openai";

class MCPOpenAIBridge {
  private mcpClient: Client;
  private openaiClient: OpenAI;
  private availableTools: any[] = [];

  constructor() {
    this.openaiClient = new OpenAI({
      apiKey: process.env.OPENAI_API_KEY,
    });
  }

  async initialize() {
    // Kết nối tới máy chủ MCP
    const transport = new StdioClientTransport({
      command: "node",
      args: ["dist/server.js"],
    });

    this.mcpClient = new Client({
      name: "openai-mcp-client",
      version: "1.0.0",
    });

    await this.mcpClient.connect(transport);

    // Phát hiện công cụ khả dụng
    const toolsResponse = await this.mcpClient.listTools();
    this.availableTools = toolsResponse.tools;
  }

  async executeWithOpenAI(userMessage: string) {
    const messages = [{ role: "user" as const, content: userMessage }];

    // Chuyển đổi công cụ MCP sang định dạng function của OpenAI
    const tools = this.availableTools.map((tool) => ({
      type: "function" as const,
      function: {
        name: tool.name,
        description: tool.description,
        parameters: tool.inputSchema,
      },
    }));

    let response = await this.openaiClient.chat.completions.create({
      model: "gpt-4",
      messages,
      tools,
    });

    // Xử lý vòng lặp gọi công cụ
    while (response.choices[0].finish_reason === "tool_calls") {
      const toolCalls = response.choices[0].message.tool_calls || [];

      for (const toolCall of toolCalls) {
        const toolName = toolCall.function.name;
        const toolArgs = JSON.parse(toolCall.function.arguments);

        // Thực thi công cụ trên máy chủ MCP
        const toolResult = await this.mcpClient.callTool({
          name: toolName,
          arguments: toolArgs,
        });

        // Thêm kết quả vào message
        messages.push({
          role: "assistant",
          content: response.choices[0].message.content || "",
        });

        messages.push({
          role: "tool",
          content: JSON.stringify(toolResult),
          tool_call_id: toolCall.id,
        });
      }

      // Gửi lại cho OpenAI
      response = await this.openaiClient.chat.completions.create({
        model: "gpt-4",
        messages,
        tools,
      });
    }

    return response.choices[0].message.content;
  }
}

// Sử dụng
const bridge = new MCPOpenAIBridge();
await bridge.initialize();
const result = await bridge.executeWithOpenAI(
  "Lịch sử đơn hàng của khách hàng 12345 là gì?"
);
console.log(result);

import crypto from "crypto";

class APIKeyManager {
  private validKeys: Set<string> = new Set();

  constructor() {
    // Nạp danh sách API key hợp lệ từ môi trường hoặc lưu trữ an toàn
    const keys = process.env.VALID_API_KEYS?.split(",") || [];
    this.validKeys = new Set(keys);
  }

  validateRequest(apiKey: string): boolean {
    return this.validKeys.has(apiKey);
  }

  generateNewKey(): string {
    return crypto.randomBytes(32).toString("hex");
  }
}

import crypto from "crypto";

function signRequest(payload: any, secret: string): string {
  return crypto
    .createHmac("sha256", secret)
    .update(JSON.stringify(payload))
    .digest("hex");
}

function verifySignature(payload: any, signature: string, secret: string): boolean {
  const expectedSignature = signRequest(payload, secret);
  return crypto.timingSafeEqual(
    Buffer.from(signature),
    Buffer.from(expectedSignature)
  );
}

import rateLimit from "express-rate-limit";

const limiter = rateLimit({
  windowMs: 15 * 60 * 1000, // 15 phút
  max: 100, // mỗi IP tối đa 100 request mỗi cửa sổ
  message: "Quá nhiều yêu cầu từ IP này, vui lòng thử lại sau.",
});

app.use("/api/", limiter);

function sanitizeInput(input: string): string {
  // Loại bỏ ký tự nguy hiểm
  return input
    .replace(/[<>]/g, "")
    .trim()
    .substring(0, 1000); // Giới hạn độ dài
}

function validateCustomerId(customerId: string): boolean {
  // Chỉ cho phép chữ, số, dấu gạch ngang
  return /^[a-zA-Z0-9-]+$/.test(customerId);
}

import { getCustomerInfo } from "../tools/customer";

describe("Công cụ khách hàng", () => {
  test("trả về thông tin khách hàng với ID hợp lệ", async () => {
    const result = await getCustomerInfo("cust_123");
    expect(result).toHaveProperty("id");
    expect(result).toHaveProperty("name");
    expect(result).toHaveProperty("email");
  });

  test("trả về lỗi với ID không hợp lệ", async () => {
    const result = await getCustomerInfo("invalid");
    expect(result).toHaveProperty("error");
  });

  test("xử lý lỗi database một cách an toàn", async () => {
    // Giả lập lỗi database
    const result = await getCustomerInfo("cust_error");
    expect(result).toHaveProperty("error");
    expect(result.status).toBe(500);
  });
});

describe("Tích hợp Máy chủ MCP", () => {
  let server: Server;

  beforeAll(async () => {
    server = new Server({ name: "test-server", version: "1.0.0" });
    // Khởi tạo server
  });

  test("liệt kê đầy đủ công cụ khả dụng", async () => {
    const tools = await server.listTools();
    expect(tools.length).toBeGreaterThan(0);
    expect(tools[0]).toHaveProperty("name");
    expect(tools[0]).toHaveProperty("description");
  });

  test("thực thi công cụ và trả về kết quả", async () => {
    const result = await server.callTool({
      name: "get_customer_info",
      arguments: { customer_id: "cust_123" },
    });
    expect(result).toBeDefined();
  });
});

FROM node:18-alpine

WORKDIR /app

COPY package*.json ./
RUN npm ci --only=production

COPY dist ./dist

EXPOSE 8000

CMD ["node", "dist/server.js"]

docker build -t my-mcp-server:1.0.0 .
docker push myregistry.azurecr.io/my-mcp-server:1.0.0

apiVersion: apps/v1
kind: Deployment
metadata:
  name: mcp-server
spec:
  replicas: 3
  selector:
    matchLabels:
      app: mcp-server
  template:
    metadata:
      labels:
        app: mcp-server
    spec:
      containers:
      - name: mcp-server
        image: myregistry.azurecr.io/my-mcp-server:1.0.0
        ports:
        - containerPort: 8000
        env:
        - name: OPENAI_API_KEY
          valueFrom:
            secretKeyRef:
              name: openai-secrets
              key: api-key
        - name: DATABASE_URL
          valueFrom:
            secretKeyRef:
              name: db-secrets
              key: connection-string
        resources:
          requests:
            memory: "256Mi"
            cpu: "250m"
          limits:
            memory: "512Mi"
            cpu: "500m"

# .env.production
OPENAI_API_KEY=sk-...
DATABASE_URL=postgresql://user:pass@host:5432/db
MCP_SERVER_PORT=8000
LOG_LEVEL=info
ENABLE_METRICS=true

async function processCustomerRefund(customerId: string, orderId: string, amount: number) {
  // Lấy thông tin khách hàng
  const customer = await getCustomerInfo(customerId);
  
  // Lấy chi tiết đơn hàng
  const order = await getOrderDetails(orderId);
  
  // Kiểm tra đơn hàng thuộc khách hàng
  if (order.customerId !== customerId) {
    throw new Error("Đơn hàng không thuộc về khách hàng này");
  }
  
  // Xử lý hoàn tiền
  const refund = await createRefund(orderId, amount);
  
  // Gửi thông báo
  await sendNotification(customer.email, `Đã hoàn tiền $${amount}`);
  
  return refund;
}

import NodeCache from "node-cache";

const cache = new NodeCache({ stdTTL: 600 }); // 10 phút

async function getCustomerInfoWithCache(customerId: string) {
  const cacheKey = `customer_${customerId}`;
  
  // Kiểm tra cache trước
  const cached = cache.get(cacheKey);
  if (cached) {
    return cached;
  }
  
  // Lấy dữ liệu từ database
  const customer = await getCustomerInfo(customerId);
  
  // Lưu vào cache
  cache.set(cacheKey, customer);
  
  return customer;
}

import Bull from "bull";

const jobQueue = new Bull("mcp-jobs");

jobQueue.process(async (job) => {
  const { toolName, arguments: args } = job.data;
  
  // Thực thi công cụ
  const result = await executeTool(toolName, args);
  
  return result;
});

async function executeToolAsync(toolName: string, args: any) {
  const job = await jobQueue.add(
    { toolName, arguments: args },
    { attempts: 3, backoff: { type: "exponential", delay: 2000 } }
  );
  
  return job.id;
}

import winston from "winston";

const logger = winston.createLogger({
  format: winston.format.json(),
  transports: [
    new winston.transports.File({ filename: "error.log", level: "error" }),
    new winston.transports.File({ filename: "combined.log" }),
  ],
});

logger.info("Thực thi công cụ", {
  toolName: "get_customer_info",
  customerId: "cust_123",
  duration: 145,
  status: "success",
});

import prometheus from "prom-client";

const toolExecutionDuration = new prometheus.Histogram({
  name: "mcp_tool_execution_duration_ms",
  help: "Thời gian thực thi công cụ (ms)",
  labelNames: ["tool_name", "status"],
});

const toolExecutionCounter = new prometheus.Counter({
  name: "mcp_tool_executions_total",
  help: "Tổng số lần thực thi công cụ",
  labelNames: ["tool_name", "status"],
});

async function executeToolWithMetrics(toolName: string, args: any) {
  const startTime = Date.now();
  
  try {
    const result = await executeTool(toolName, args);
    const duration = Date.now() - startTime;
    
    toolExecutionDuration.labels(toolName, "success").observe(duration);
    toolExecutionCounter.labels(toolName, "success").inc();
    
    return result;
  } catch (error) {
    const duration = Date.now() - startTime;
    
    toolExecutionDuration.labels(toolName, "error").observe(duration);
    toolExecutionCounter.labels(toolName, "error").inc