@cjsqbn/resume-screening-mcp 0.1.2 → 0.2.0

package/index.js

@@ -1,7 +1,11 @@
 #!/usr/bin/env node
 
+import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { z } from "zod";
+
 const SERVER_NAME = "resume-screening-mcp";
-const SERVER_VERSION = "0.1.0";
+const SERVER_VERSION = "0.2.0";
 
 const DEFAULT_CONFIG = {
   trust_threshold: 60,
@@ -531,179 +535,76 @@ function generateReport({ screening_result }) {
   return { markdown: lines.join("\n") };
 }
 
-const TOOLS = [
-  {
-    name: "parse_jd",
-    description: "从岗位 JD 文本中提取工作年限、技能、行业、职责和加分项要求。",
-    inputSchema: {
-      type: "object",
-      properties: {
-        job_description: { type: "string", description: "岗位 JD 原文。" },
-      },
-      required: ["job_description"],
-    },
-  },
-  {
-    name: "analyze_candidate",
-    description: "分析单个候选人，输出结构化简历信息、可信度、匹配度、综合得分和面试追问。",
-    inputSchema: {
-      type: "object",
-      properties: {
-        resume: {
-          type: "object",
-          properties: {
-            file_name: { type: "string" },
-            text: { type: "string" },
-          },
-          required: ["text"],
-        },
-        job_description: { type: "string" },
-        config: { type: "object" },
-      },
-      required: ["resume"],
-    },
-  },
-  {
-    name: "screen_resumes",
-    description: "批量分析多份简历和岗位 JD，输出淘汰名单、晋级排序榜、疑点追问和 Markdown 报告。",
-    inputSchema: {
-      type: "object",
-      properties: {
-        resumes: {
-          type: "array",
-          items: {
-            type: "object",
-            properties: {
-              file_name: { type: "string" },
-              text: { type: "string" },
-            },
-            required: ["text"],
-          },
-        },
-        job_description: { type: "string" },
-        config: {
-          type: "object",
-          properties: {
-            trust_threshold: { type: "number" },
-            match_weight: { type: "number" },
-            competitiveness_weight: { type: "number" },
-          },
-        },
-      },
-      required: ["resumes", "job_description"],
-    },
-  },
-  {
-    name: "generate_report",
-    description: "把 screen_resumes 的结构化结果转换为中文 Markdown 报告。",
-    inputSchema: {
-      type: "object",
-      properties: {
-        screening_result: { type: "object" },
+function asTextResult(value) {
+  return {
+    content: [
+      {
+        type: "text",
+        text: JSON.stringify(value, null, 2),
       },
-      required: ["screening_result"],
-    },
-  },
-];
-
-function callTool(name, args) {
-  if (name === "parse_jd") return parseJd(args || {});
-  if (name === "analyze_candidate") return analyzeCandidate(args || {});
-  if (name === "screen_resumes") return screenResumes(args || {});
-  if (name === "generate_report") return generateReport(args || {});
-  throw new Error(`未知工具：${name}`);
+    ],
+  };
 }
 
-let inputBuffer = Buffer.alloc(0);
-const keepAlive = setInterval(() => {}, 1 << 30);
-
-process.stdin.on("data", (chunk) => {
-  inputBuffer = Buffer.concat([inputBuffer, chunk]);
-  processMessages();
+const resumeSchema = z.object({
+  file_name: z.string().optional(),
+  name: z.string().optional(),
+  text: z.string().min(1, "resume.text 不能为空"),
 });
 
-process.stdin.on("error", () => process.exit(1));
-process.stdin.resume();
-
-function processMessages() {
-  while (true) {
-    const headerEnd = inputBuffer.indexOf("\r\n\r\n");
-    if (headerEnd === -1) return;
-
-    const header = inputBuffer.slice(0, headerEnd).toString("utf8");
-    const match = header.match(/Content-Length:\s*(\d+)/i);
-    if (!match) {
-      inputBuffer = inputBuffer.slice(headerEnd + 4);
-      continue;
-    }
-
-    const length = Number(match[1]);
-    const messageStart = headerEnd + 4;
-    const messageEnd = messageStart + length;
-    if (inputBuffer.length < messageEnd) return;
-
-    const raw = inputBuffer.slice(messageStart, messageEnd).toString("utf8");
-    inputBuffer = inputBuffer.slice(messageEnd);
-
-    try {
-      handleMessage(JSON.parse(raw));
-    } catch (error) {
-      sendError(null, -32700, `解析 MCP 消息失败：${error.message}`);
-    }
-  }
-}
-
-function handleMessage(message) {
-  if (!message || typeof message !== "object") return;
-  if (message.id === undefined || message.id === null) return;
-
-  try {
-    if (message.method === "initialize") {
-      sendResult(message.id, {
-        protocolVersion: message.params?.protocolVersion || "2024-11-05",
-        capabilities: { tools: {} },
-        serverInfo: { name: SERVER_NAME, version: SERVER_VERSION },
-      });
-      return;
-    }
-
-    if (message.method === "tools/list") {
-      sendResult(message.id, { tools: TOOLS });
-      return;
-    }
+const configSchema = z.object({
+  trust_threshold: z.number().optional(),
+  high_trust_threshold: z.number().optional(),
+  final_score_priority_threshold: z.number().optional(),
+  final_score_interview_threshold: z.number().optional(),
+  match_weight: z.number().optional(),
+  competitiveness_weight: z.number().optional(),
+}).passthrough().optional();
+
+const server = new McpServer({
+  name: SERVER_NAME,
+  version: SERVER_VERSION,
+});
 
-    if (message.method === "tools/call") {
-      const toolName = message.params?.name;
-      const args = message.params?.arguments || {};
-      const result = callTool(toolName, args);
-      sendResult(message.id, {
-        content: [
-          {
-            type: "text",
-            text: JSON.stringify(result, null, 2),
-          },
-        ],
-        structuredContent: result,
-      });
-      return;
-    }
+server.tool(
+  "parse_jd",
+  "从岗位 JD 文本中提取工作年限、技能、行业、职责和加分项要求。",
+  {
+    job_description: z.string().min(1, "job_description 不能为空"),
+  },
+  async (args) => asTextResult(parseJd(args)),
+);
 
-    sendError(message.id, -32601, `不支持的方法：${message.method}`);
-  } catch (error) {
-    sendError(message.id, -32000, error.message || String(error));
-  }
-}
+server.tool(
+  "analyze_candidate",
+  "分析单个候选人，输出结构化简历信息、可信度、匹配度、综合得分和面试追问。",
+  {
+    resume: resumeSchema,
+    job_description: z.string().optional(),
+    config: configSchema,
+  },
+  async (args) => asTextResult(analyzeCandidate(args)),
+);
 
-function sendResult(id, result) {
-  writeMessage({ jsonrpc: "2.0", id, result });
-}
+server.tool(
+  "screen_resumes",
+  "批量分析多份简历和岗位 JD，输出淘汰名单、晋级排序榜、疑点追问和 Markdown 报告。",
+  {
+    resumes: z.array(resumeSchema).min(1, "resumes 必须是非空数组"),
+    job_description: z.string().min(1, "job_description 不能为空"),
+    config: configSchema,
+  },
+  async (args) => asTextResult(screenResumes(args)),
+);
 
-function sendError(id, code, message) {
-  writeMessage({ jsonrpc: "2.0", id, error: { code, message } });
-}
+server.tool(
+  "generate_report",
+  "把 screen_resumes 的结构化结果转换为中文 Markdown 报告。",
+  {
+    screening_result: z.record(z.string(), z.any()),
+  },
+  async (args) => asTextResult(generateReport(args)),
+);
 
-function writeMessage(payload) {
-  const json = JSON.stringify(payload);
-  const bytes = Buffer.byteLength(json, "utf8");
-  process.stdout.write(`Content-Length: ${bytes}\r\n\r\n${json}`);
-}
+const transport = new StdioServerTransport();
+await server.connect(transport);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cjsqbn/resume-screening-mcp",
-  "version": "0.1.2",
+  "version": "0.2.0",
   "description": "MCP service for resume screening, JD matching, trust checking, scoring and candidate ranking.",
   "type": "module",
   "bin": {
@@ -20,6 +20,10 @@
   "engines": {
     "node": ">=18"
   },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.18.1",
+    "zod": "^3.25.76"
+  },
   "publishConfig": {
     "access": "public"
   },