beacon-mfg-mcp 1.3.3 → 1.4.0

package/mcp_adapter.py

@@ -0,0 +1,153 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+BeaconMFG MCP —— 标准协议适配层（adapter / shim）。
+
+为什么需要它：
+  beacon-mfg 的 mcp/server.py 用「换行分隔 JSON-RPC 子集」做传输
+  （`for raw in sys.stdin` + `json + "\n"`），并不是标准 MCP 的
+  `Content-Length` 分帧。标准 MCP 客户端（WorkBuddy 连接器、Claude Desktop
+  等）按 LSP 分帧跟它握手会解析失败。
+
+本适配层做什么：
+  - 对外（父进程 = MCP host）：讲标准 MCP stdio（Content-Length 分帧）。
+  - 对内（子进程 = server.py）：把请求转成换行分隔 JSON 转发，并把 server.py
+    的换行回包重新分帧后返回。
+  - 不修改 server.py 任何代码；server.py 的「通知无回包」语义天然保留
+    （转发通知后不等待回包）。
+
+用法：
+  python mcp_adapter.py
+环境变量（可选）：BEACON_REPO / BEACON_MASK_PHONE / BEACON_SOURCE
+"""
+import json
+import os
+import sys
+import subprocess
+import threading
+import queue
+
+_HERE = os.path.dirname(os.path.abspath(__file__))
+REPO = os.environ.get("BEACON_REPO") or os.path.dirname(_HERE)
+SERVER = os.path.join(_HERE, "server.py")
+MASK = os.environ.get("BEACON_MASK_PHONE", "1")
+
+
+def _spawn_server():
+    return subprocess.Popen(
+        [sys.executable, "-u", SERVER],
+        stdin=subprocess.PIPE,
+        stdout=subprocess.PIPE,
+        env={**os.environ, "BEACON_REPO": REPO, "BEACON_MASK_PHONE": MASK},
+        bufsize=0, text=True, encoding="utf-8",
+    )
+
+
+def _read_stdin_framed():
+    """从父进程(stdio)读取一条标准 MCP 分帧消息，返回 dict；EOF 返回 None。"""
+    buf = b""
+    while True:
+        c = sys.stdin.buffer.read(1)
+        if not c:
+            return None
+        buf += c
+        if buf.endswith(b"\r\n\r\n") or buf.endswith(b"\n\n"):
+            break
+    clen = None
+    for line in buf.split(b"\n"):
+        low = line.strip().lower()
+        if low.startswith(b"content-length:"):
+            try:
+                clen = int(low.split(b":", 1)[1].strip())
+            except ValueError:
+                pass
+    if clen is None:
+        return None
+    body = b""
+    while len(body) < clen:
+        chunk = sys.stdin.buffer.read(clen - len(body))
+        if not chunk:
+            return None
+        body += chunk
+    try:
+        return json.loads(body.decode("utf-8"))
+    except Exception:
+        return None
+
+
+def _write_stdout_framed(obj):
+    """向父进程(stdio)写一条标准 MCP 分帧消息。"""
+    data = json.dumps(obj, ensure_ascii=False).encode("utf-8")
+    sys.stdout.buffer.write(b"Content-Length: %d\r\n\r\n" % len(data))
+    sys.stdout.buffer.write(data)
+    sys.stdout.buffer.flush()
+
+
+def main():
+    child = _spawn_server()
+    responses: "queue.Queue" = queue.Queue()
+    client_protocol = None  # 客户端在 initialize 里请求的协议版本
+
+    def reader():
+        # 子进程用换行分隔 JSON；逐行读、入队。
+        try:
+            for line in child.stdout:
+                line = line.strip()
+                if not line:
+                    continue
+                try:
+                    obj = json.loads(line)
+                except Exception:
+                    continue
+                responses.put(obj)
+        except Exception:
+            pass
+
+    threading.Thread(target=reader, daemon=True).start()
+
+    try:
+        while True:
+            msg = _read_stdin_framed()
+            if msg is None:
+                break
+            method = msg.get("method")
+            mid = msg.get("id")
+            is_notification = (mid is None) or (isinstance(method, str) and method.startswith("notifications/"))
+
+            if method == "initialize":
+                # 记下客户端请求版本，用于回包时回写（兼容严格客户端）
+                client_protocol = (msg.get("params") or {}).get("protocolVersion")
+
+            # 转发给子进程（换行分隔）
+            child.stdin.write(json.dumps(msg, ensure_ascii=False) + "\n")
+            child.stdin.flush()
+
+            if is_notification:
+                continue  # 通知无需回包，server.py 自身也是 continue
+
+            # 请求：等子进程回包（按到达顺序，server.py 严格顺序响应）
+            try:
+                resp = responses.get(timeout=60)
+            except queue.Empty:
+                if mid is not None:
+                    _write_stdout_framed({
+                        "jsonrpc": "2.0", "id": mid,
+                        "error": {"code": -32000, "message": "no response from server.py"},
+                    })
+                continue
+
+            # 兼容严格客户端：server.py 回的是固定 protocolVersion，
+            # 改写回客户端在 initialize 里请求的版本，避免握手因版本不一致卡住。
+            if method == "initialize" and client_protocol and isinstance(resp.get("result"), dict):
+                resp["result"]["protocolVersion"] = client_protocol
+
+            _write_stdout_framed(resp)
+    finally:
+        try:
+            child.terminate()
+        except Exception:
+            pass
+
+
+if __name__ == "__main__":
+    main()

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "beacon-mfg-mcp",
-  "version": "1.3.3",
+  "version": "1.4.0",
   "description": "Read-only MCP server exposing the BeaconMFG灯塔工厂 supplier directory (search vendors, get vendor detail, get capability card, multi-turn sourcing) to any MCP-compatible agent. No API key required.",
   "keywords": [
     "mcp",
@@ -26,6 +26,8 @@
   "files": [
     "server.py",
     "http_server.py",
+    "mcp_adapter.py",
+    "test_protocol.py",
     "bin/",
     "rfq-kernel/",
     "copy-rfk.js",

package/test_protocol.py

@@ -0,0 +1,262 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+BeaconMFG MCP —— 发布前协议回归测试（pre-release smoke test）。
+
+为什么需要它
+------------
+`server.py` 同时服务两条链路：本地 stdio（npm / git 安装）与远程 Streamable HTTP
+（`http_server.py`）。业务逻辑一改，两条链路都可能悄悄坏掉，且**症状隐蔽**
+（能握手但查不出数据、或只有部分工具挂掉）。本脚本把「协议能握手 + 工具齐全 +
+真实检索有结果」三件事固化为一条命令，每次发版前跑一遍。
+
+用法
+----
+    python mcp/test_protocol.py                 # 自动起 HTTP 端点自测
+    python mcp/test_protocol.py --stdio         # 额外测 stdio(server.py)
+    python mcp/test_protocol.py --adapter       # 额外测标准分帧适配层
+    BEACON_REPO=/path/to/repo python mcp/test_protocol.py
+
+退出码：0 全部通过；1 有失败项。
+"""
+import json
+import os
+import subprocess
+import sys
+import threading
+import time
+import urllib.error
+import urllib.request
+
+HERE = os.path.dirname(os.path.abspath(__file__))
+SERVER = os.path.join(HERE, "server.py")
+HTTP_SERVER = os.path.join(HERE, "http_server.py")
+ADAPTER = os.path.join(HERE, "mcp_adapter.py")
+
+EXPECT_TOOLS = {
+    "search_vendors", "get_vendor", "get_capability_card",
+    "start_sourcing", "answer_sourcing", "refine_sourcing",
+}
+
+# 三个城市各取一个高频词，覆盖不同行业与不同分片，避免「只测通一个分片」
+SMOKE_QUERIES = [("修车", "嘉兴"), ("面馆", "苏州"), ("米粉", "贵阳")]
+
+_results = []
+
+
+def _check(name, ok, detail=""):
+    _results.append((name, bool(ok), detail))
+    print("  %s %s%s" % ("PASS" if ok else "FAIL", name, ("  -> " + detail) if detail else ""))
+    return ok
+
+
+def _child_env():
+    env = dict(os.environ)
+    env.setdefault("BEACON_MASK_PHONE", "1")
+    env["PYTHONUTF8"] = "1"
+    env["PYTHONIOENCODING"] = "utf-8"
+    return env
+
+
+# --------------------------------------------------------------------------
+# 公共断言：拿到工具列表 + 跑真实检索
+# --------------------------------------------------------------------------
+def _assert_tools_and_search(call, label):
+    """call(method, params, session_id) -> result dict"""
+    r = call("initialize", {
+        "protocolVersion": "2024-11-05", "capabilities": {},
+        "clientInfo": {"name": "smoke", "version": "1"},
+    })
+    _check("%s initialize" % label, r is not None and "result" in r, "" if r else "无回包")
+
+    # 通知无回包（规范允许），不要等
+    try:
+        call("notifications/initialized", {}, r.get("__sid__") if isinstance(r, dict) else None)
+    except Exception:
+        pass
+
+    r2 = call("tools/list", {})
+    tools = set()
+    if r2 and "result" in r2:
+        tools = {t.get("name") for t in r2["result"].get("tools", [])}
+    missing = EXPECT_TOOLS - tools
+    _check("%s tools/list (6)" % label, not missing, ("缺: " + ",".join(missing)) if missing else "")
+
+    for q, city in SMOKE_QUERIES:
+        r3 = call("tools/call", {"name": "search_vendors",
+                                 "arguments": {"query": q, "city": city, "limit": 3}})
+        n, err = 0, ""
+        if r3 and "result" in r3:
+            try:
+                data = json.loads(r3["result"]["content"][0]["text"])
+                n = len(data.get("results", []))
+            except Exception as e:
+                err = "解析失败 %s" % e
+        else:
+            err = "无回包"
+        _check("%s search %s/%s" % (label, city, q), n > 0, err or "命中 %d" % n)
+
+
+# --------------------------------------------------------------------------
+# 1) Streamable HTTP（http_server.py）
+# --------------------------------------------------------------------------
+def test_http():
+    print("\n[1/3] Streamable HTTP  (mcp/http_server.py)")
+    port = int(os.environ.get("MCP_TEST_PORT", "8899"))
+    env = _child_env()
+    env["MCP_PORT"] = str(port)
+    proc = subprocess.Popen([sys.executable, "-u", HTTP_SERVER],
+                            env=env, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    base = "http://127.0.0.1:%d/mcp" % port
+    try:
+        # 等端口起来
+        for _ in range(40):
+            try:
+                urllib.request.urlopen("http://127.0.0.1:%d/health" % port, timeout=1).read()
+                break
+            except Exception:
+                time.sleep(0.25)
+        else:
+            _check("HTTP 端点启动", False, "40 次探测仍未就绪")
+            return
+
+        def call(method, params, sid=None):
+            payload = {"jsonrpc": "2.0", "id": 1, "method": method, "params": params}
+            if method.startswith("notifications/"):
+                payload.pop("id")
+            data = json.dumps(payload).encode()
+            req = urllib.request.Request(base, data=data, method="POST")
+            req.add_header("Content-Type", "application/json")
+            req.add_header("Accept", "application/json, text/event-stream")
+            if sid:
+                req.add_header("Mcp-Session-Id", sid)
+            with urllib.request.urlopen(req, timeout=90) as resp:
+                new_sid = resp.headers.get("Mcp-Session-Id")
+                body = resp.read().decode()
+            if method.startswith("notifications/"):
+                return None
+            obj = json.loads(body)
+            if new_sid:
+                obj["__sid__"] = new_sid
+            return obj
+
+        _assert_tools_and_search(call, "http")
+    finally:
+        proc.terminate()
+        try:
+            proc.wait(timeout=5)
+        except Exception:
+            proc.kill()
+
+
+# --------------------------------------------------------------------------
+# 2) 换行分隔 stdio（server.py）—— npm/git 安装走的默认链路
+# --------------------------------------------------------------------------
+def test_stdio():
+    print("\n[2/3] stdio 换行分隔  (mcp/server.py)")
+    proc = subprocess.Popen([sys.executable, "-u", SERVER], stdin=subprocess.PIPE,
+                            stdout=subprocess.PIPE, env=_child_env(),
+                            encoding="utf-8", bufsize=1)
+    try:
+        def send(method, params, notify=False):
+            msg = {"jsonrpc": "2.0", "method": method, "params": params}
+            if not notify:
+                msg["id"] = 1
+            proc.stdin.write(json.dumps(msg, ensure_ascii=False) + "\n")
+            proc.stdin.flush()
+            if notify:
+                return None
+            return json.loads(proc.stdout.readline())
+
+        def call(method, params, sid=None):
+            if method.startswith("notifications/"):
+                return send(method, params, notify=True)
+            return send(method, params)
+
+        _assert_tools_and_search(call, "stdio")
+    finally:
+        proc.terminate()
+        try:
+            proc.wait(timeout=5)
+        except Exception:
+            proc.kill()
+
+
+# --------------------------------------------------------------------------
+# 3) 标准分帧适配层（mcp_adapter.py）—— 严格客户端走这条
+# --------------------------------------------------------------------------
+def test_adapter():
+    print("\n[3/3] 标准分帧适配层  (mcp/mcp_adapter.py)")
+    proc = subprocess.Popen([sys.executable, "-u", ADAPTER], stdin=subprocess.PIPE,
+                            stdout=subprocess.PIPE, env=_child_env(), bufsize=0)
+
+    def send(obj):
+        data = json.dumps(obj, ensure_ascii=False).encode()
+        proc.stdin.write(b"Content-Length: %d\r\n\r\n" % len(data) + data)
+        proc.stdin.flush()
+
+    def recv():
+        hdr = b""
+        while not hdr.endswith(b"\r\n\r\n"):
+            c = proc.stdout.read(1)
+            if not c:
+                return None
+            hdr += c
+        n = 0
+        for line in hdr.split(b"\r\n"):
+            if line.lower().startswith(b"content-length:"):
+                n = int(line.split(b":")[1].strip())
+        body = b""
+        while len(body) < n:
+            chunk = proc.stdout.read(n - len(body))
+            if not chunk:
+                return None
+            body += chunk
+        return json.loads(body.decode("utf-8"))
+
+    def call(method, params, sid=None):
+        msg = {"jsonrpc": "2.0", "id": 1, "method": method, "params": params}
+        if method.startswith("notifications/"):
+            msg.pop("id")
+            send(msg)
+            return None
+        send(msg)
+        return recv()
+
+    try:
+        _assert_tools_and_search(call, "adapter")
+    finally:
+        proc.terminate()
+        try:
+            proc.wait(timeout=5)
+        except Exception:
+            proc.kill()
+
+
+def main():
+    print("=" * 60)
+    print("BeaconMFG MCP 协议回归测试")
+    print("数据源: BEACON_REPO=%s" % (os.environ.get("BEACON_REPO") or "(默认/联网)"))
+    print("=" * 60)
+
+    test_http()
+    if "--stdio" in sys.argv:
+        test_stdio()
+    if "--adapter" in sys.argv:
+        test_adapter()
+    else:
+        print("\n[2/3] stdio  (跳过，加 --stdio 启用)")
+        print("[3/3] adapter (跳过，加 --adapter 启用)")
+
+    passed = sum(1 for _, ok, _ in _results if ok)
+    failed = [(n, d) for n, ok, d in _results if not ok]
+    print("\n" + "=" * 60)
+    print("结果: %d 通过, %d 失败" % (passed, len(failed)))
+    for n, d in failed:
+        print("  FAIL %s  %s" % (n, d))
+    print("=" * 60)
+    return 1 if failed else 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())