@hupan56/wlkj 3.4.9 → 3.5.1

package/templates/qoder/scripts/protocol/mcp/kg_mcp_server.py

@@ -1,351 +1,351 @@
-#!/usr/bin/env python
-# -*- coding: utf-8 -*-
-# v3.0 路径自举: 引导到 common/bootstrap, 统一 sys.path 逻辑
-import os as _o, sys as _s
-_f = _o.path.abspath(__file__)
-for _ in range(10):
-    _f = _o.path.dirname(_f)
-    _cp = _o.path.join(_f, 'foundation')
-    if _o.path.isfile(_o.path.join(_cp, 'bootstrap.py')):
-        break
-if _cp not in _s.path: _s.path.insert(0, _cp)
-from bootstrap import setup; setup()
-from foundation.core.paths import get_repo_root
-
-"""
-Knowledge Graph MCP Server for QoderWork (pure Python, no SDK dependency).
-
-Exposes the QODER knowledge graph as MCP tools that QoderWork's AI agent
-can call directly. Uses the MCP stdio protocol (JSON-RPC 2.0 over stdin/stdout)
-implemented from scratch — no external dependencies beyond the standard library.
-
-Registered in ~/.qoderwork/mcp.json as "qoder-knowledge-graph".
-Tools surface as mcp__qoder-knowledge-graph__<tool> to the AI.
-
-v3.1 重构: 用 @tool 装饰器 + ToolRegistryMeta 元类, 消除:
-  - 手写 TOOLS 列表 (78-253 行)
-  - _execute_tool 的 17 个 if/elif 分支 (386-576 行)
-两处合一, 新增工具只需加一个 @tool 方法。
-"""
-import io
-import os
-import sys
-import json
-import subprocess
-
-# Ensure UTF-8 output
-try:
-    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
-    sys.stderr.reconfigure(encoding='utf-8', errors='replace')
-except Exception:
-    pass
-
-# Locate the scripts dir
-SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  # scripts/ 根
-BASE_DIR = os.path.dirname(os.path.dirname(SCRIPTS_DIR))  # repo 根
-sys.path.insert(0, SCRIPTS_DIR)
-_KG_DIR = os.path.dirname(os.path.abspath(__file__))
-if _KG_DIR not in sys.path: sys.path.insert(0, _KG_DIR)
-
-
-# ── stdout 捕获辅助 ───────────────────────────────────────────────────
-# 底层脚本 (search_index/context_pack) 用 print 输出, MCP 需要捕获为字符串返回。
-
-def _capture(func, *args, **kwargs):
-    """Run a function, capture its stdout.
-
-    v3.0: sys.stdout 在 run() 后已是 stderr (基类 _setup_streams 重定向),
-    这里捕获时临时换回 buf, finally 还原。
-    """
-    old = sys.stdout
-    buf = io.StringIO()
-    sys.stdout = buf
-    try:
-        func(*args, **kwargs)
-    except Exception as e:
-        buf.write('\n[error] %s' % str(e))
-    finally:
-        sys.stdout = old
-    return buf.getvalue().strip()
-
-
-# Lazy imports (首次调用工具时才 import, 加快启动)
-_search = None
-_context = None
-
-def _ensure_imports():
-    global _search, _context
-    if _search is None:
-        from domain.kg.search import search_index as _search
-    if _context is None:
-        from domain.kg.search import context_pack as _context
-
-
-# ── MCP Server (继承 BaseMCPServer, @tool 装饰器自动分发) ──────────────
-
-from foundation.protocol.mcp_base import BaseMCPServer, tool
-
-
-def _require_kgdb():
-    """检查 kg.duckdb 是否存在。存在返回 None; 不存在返回友好提示字符串。
-
-    ★ 共识：本地引擎全走平台。kg MCP 走云 SSE url，不依赖本地 kg.duckdb。
-    但本文件是"本地 launcher 模式"的 server——只在 mcp.json 没配 SSE url 时被调用。
-    配了 SSE url 时 QoderWork 直连云平台，本文件根本不会启动。
-    所以这里只需检查本地 duckdb，报错了也只影响 launcher 模式（非共识路径）。
-    """
-    try:
-        from foundation.core.paths import DATA_INDEX_DIR
-        db_path = DATA_INDEX_DIR / 'kg.duckdb'
-        if not db_path.is_file():
-            return ("知识图谱走云平台（mcp.json kg → SSE url），本地 kg.duckdb 不需要。\n"
-                    "如果云平台不通，可本地构建: python .qoder/scripts/orchestration/wlkj.py kg-build")
-    except Exception:
-        pass
-    return None
-
-
-class KnowledgeGraphMCPServer(BaseMCPServer):
-    """知识图谱 MCP Server。
-
-    v3.1: 每个工具用 @tool 装饰, 元类自动收集 TOOLS + 自动 handle_tool 分发。
-    新增工具 = 加一个 @tool 方法, 无需改 TOOLS 列表或 if 链。
-    """
-
-    SERVER_NAME = "qoder-knowledge-graph"
-    SERVER_VERSION = "2.0.0"
-
-    # ★ v2.0 共识改造：kg 走云平台。本 server 的所有工具不再直连本地 duckdb，
-    # 而是转发到云平台 HTTP MCP@10010（http://10.89.7.120:10010/mcp）。
-    # 工具名映射：本地工具名 → 云平台等价 MCP 工具
-    PLATFORM_MCP_URL = os.environ.get("WLKJ_PLATFORM_MCP", "http://10.89.7.120:10010/mcp")
-    # 本地工具 → 云平台工具映射（名称不同或参数不同）
-    _TOOL_MAP = {
-        "search_code": "search_code",
-        "search_api": "search_api",
-        "search_prd": "search_prd_semantic",
-        "get_impact": "get_impact",
-        "context_360": "context_360",
-        "context_pack": "context_pack",
-        "get_workflow": "business_trace",
-        "multi_hop": "business_trace",
-        "search_wiki": "rag_search",
-        "search_field": "search_field",
-        "list_modules": "get_modules",
-        "coverage_matrix": "impact_coverage",
-        "feature_overview": "feature_profile",
-        "table_impact": "table_impact",
-        "fill_prototype": None,
-        "get_design_system": "get_design_system",
-        "semantic_search": "rag_search",
-    }
-
-    def _call_platform(self, tool_name, args):
-        """转发工具调用到云平台。
-
-        主路径：mcp.json kg→SSE url，QoderWork直连云平台（不经本server）。
-        备选路径：launcher模式（本server被调），走SSE协议转发。
-        """
-        mapped = self._TOOL_MAP.get(tool_name)
-        if not mapped:
-            return json.dumps({"error": f"工具 {tool_name} 无云平台等价（本地功能）"}, ensure_ascii=False)
-        # 尝试HTTP POST（简单协议，部分端点支持）
-        try:
-            import urllib.request, urllib.error
-            body = json.dumps({"tool": mapped, "args": args or {}}).encode("utf-8")
-            req = urllib.request.Request(
-                self.PLATFORM_MCP_URL,
-                data=body,
-                headers={"Content-Type": "application/json"},
-                method="POST",
-            )
-            with urllib.request.urlopen(req, timeout=30) as resp:
-                return resp.read().decode("utf-8")
-        except urllib.error.HTTPError as e:
-            if e.code == 401:
-                return json.dumps({"error": "云平台需鉴权。请在 mcp.json 配 kg 为 SSE url（npx @hupan56/wlkj update 自动配置）。"}, ensure_ascii=False)
-            return json.dumps({"error": f"云平台 HTTP {e.code}: {e.reason}"}, ensure_ascii=False)
-        except Exception as e:
-            return json.dumps({"error": f"云平台不可达: {type(e).__name__}: {str(e)[:120]}"}, ensure_ascii=False)
-
-    # ── 工具实现（全部转发到云平台）──────────────────────────────────
-
-    @tool("search_code",
-          "搜索代码: 按关键词查找代码文件在哪。例: search_code(keyword='考勤') 或 search_code(keyword='车辆', platform='web')",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "搜索关键词(中文或英文)"},
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web=PC管理端, app=移动端。可选"},
-          }, "required": ["keyword"]})
-    def _t_search_code(self, args):
-        return self._call_platform("search_code", args)
-
-    @tool("search_code",
-          "搜索代码: 按关键词查找代码文件在哪。例: search_code(keyword='考勤') 或 search_code(keyword='车辆', platform='web')",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "搜索关键词(中文或英文)"},
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web=PC管理端, app=移动端。可选"},
-          }, "required": ["keyword"]})
-    def _t_search_code(self, args):
-        return self._call_platform("search_code", args)
-
-    @tool("search_api",
-          "搜索API端点: 按关键词查找后端接口。例: search_api(keyword='salary')",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "API关键词"},
-          }, "required": ["keyword"]})
-    def _t_search_api(self, args):
-        return self._call_platform("search_api", args)
-
-    @tool("search_prd",
-          "搜索PRD: 查已有需求文档,防止重复造轮子。例: search_prd(keyword='考勤')",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "PRD关键词"},
-          }, "required": ["keyword"]})
-    def _t_search_prd(self, args):
-        return self._call_platform("search_prd", args)
-
-    @tool("get_impact",
-          "影响分析: 改某接口/路径会影响哪些前端页面和按钮?评估改动风险。例: get_impact(endpoint='/asset') 或 get_impact(endpoint='export')",
-          {"type": "object", "properties": {
-              "endpoint": {"type": "string", "description": "API端点路径或关键词, 如 /asset/list 或 export"},
-          }, "required": ["endpoint"]})
-    def _t_get_impact(self, args):
-        return self._call_platform("get_impact", args)
-
-    @tool("context_360",
-          "360度视图: 查看符号(端点/函数/处理器)的全部关联信息。例: context_360(symbol='handleExport') 或 context_360(symbol='/system/role')",
-          {"type": "object", "properties": {
-              "symbol": {"type": "string", "description": "符号名: 端点/函数名/处理器名"},
-          }, "required": ["symbol"]})
-    def _t_context_360(self, args):
-        return self._call_platform("context_360", args)
-
-    @tool("context_pack",
-          "上下文打包: 一次返回关键词相关的全部上下文(代码+页面+字段+PRD+API+布局指纹)。避免来回查。例: context_pack(keyword='车辆', platform='web')",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "业务关键词"},
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web或app。可选"},
-              "page_type": {"type": "string", "description": "页面类型: table-page/form-page/detail-page。可选"},
-              "role": {"type": "string", "enum": ["pm", "design", "dev", "test", "admin"],
-                       "description": "角色裁剪: pm=字段+PRD+API, design=页面+布局+风格, dev=代码+API, test=代码+PRD+API。可选"},
-          }, "required": ["keyword"]})
-    def _t_context_pack(self, args):
-        return self._call_platform("context_pack", args)
-
-    @tool("get_workflow",
-          "业务流程: 查看一个模块的完整操作链。",
-          {"type": "object", "properties": {
-              "module": {"type": "string", "description": "模块名(英文优先)"},
-          }, "required": ["module"]})
-    def _t_get_workflow(self, args):
-        return self._call_platform("get_workflow", args)
-
-    @tool("multi_hop",
-          "知识图谱多跳遍历: 从一个符号出发, 沿edges递归N跳。",
-          {"type": "object", "properties": {
-              "symbol": {"type": "string", "description": "起点符号"},
-              "max_depth": {"type": "integer", "default": 3, "description": "最大跳数(1-5)"},
-          }, "required": ["symbol"]})
-    def _t_multi_hop(self, args):
-        return self._call_platform("multi_hop", args)
-
-    @tool("search_wiki",
-          "搜索Repo Wiki: 查模块级语义文档。",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "Wiki关键词"},
-          }, "required": ["keyword"]})
-    def _t_search_wiki(self, args):
-        return self._call_platform("search_wiki", args)
-
-    @tool("search_field",
-          "搜索字段: 查某个字段在哪些页面用过。",
-          {"type": "object", "properties": {
-              "field": {"type": "string", "description": "字段名(英文field或中文标题)"},
-          }, "required": ["field"]})
-    def _t_search_field(self, args):
-        return self._call_platform("search_field", args)
-
-    @tool("list_modules",
-          "模块清单: 列出项目所有业务模块。",
-          {"type": "object", "properties": {
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台过滤"},
-          }})
-    def _t_list_modules(self, args):
-        return self._call_platform("list_modules", args)
-
-    @tool("coverage_matrix",
-          "功能×测试覆盖矩阵。",
-          {"type": "object", "properties": {}})
-    def _t_coverage_matrix(self, args):
-        return self._call_platform("coverage_matrix", args)
-
-    @tool("feature_overview",
-          "功能画像: 一个功能的完整画像。",
-          {"type": "object", "properties": {
-              "feature": {"type": "string", "description": "功能名(中/英文)"},
-          }, "required": ["feature"]})
-    def _t_feature_overview(self, args):
-        return self._call_platform("feature_overview", args)
-
-    @tool("table_impact",
-          "改表影响分析: 改某数据库表会影响哪些 Entity 类和接口。",
-          {"type": "object", "properties": {
-              "table": {"type": "string", "description": "数据库表名"},
-          }, "required": ["table"]})
-    def _t_table_impact(self, args):
-        return self._call_platform("table_impact", args)
-
-    # ── 原型/设计/语义 (全部走云平台) ────────────────────────────────
-
-    @tool("fill_prototype",
-          "原型预填: 用真实代码数据自动填模板。",
-          {"type": "object", "properties": {
-              "keyword": {"type": "string", "description": "业务关键词"},
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台"},
-          }, "required": ["keyword"]})
-    def _t_fill_prototype(self, args):
-        return self._call_platform("fill_prototype", args)
-
-    @tool("get_design_system",
-          "设计系统查询: 返回项目完整设计规范。",
-          {"type": "object", "properties": {
-              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台过滤"},
-          }})
-    def _t_get_design_system(self, args):
-        return self._call_platform("get_design_system", args)
-
-    @tool("semantic_search",
-          "语义搜索: 用自然语言找代码/API/功能。",
-          {"type": "object", "properties": {
-              "query": {"type": "string", "description": "自然语言查询"},
-              "top_k": {"type": "integer", "default": 10, "description": "返回前N条"},
-          }, "required": ["query"]})
-    def _t_semantic_search(self, args):
-        return self._call_platform("semantic_search", args)
-        from domain.kg.graph.kg_semantic import semantic_search, embedding_status
-        query = (args.get("query") or '').strip()
-        if not query:
-            return "请提供查询关键词"
-        try:
-            top_k = int(args.get("top_k", 10))
-        except (TypeError, ValueError):
-            top_k = 10
-        results = semantic_search(query, top_k=top_k)
-        if results is None:
-            st = embedding_status()
-            if not st['available']:
-                return ("语义检索未启用: sentence-transformers 未安装。\n"
-                        "启用方法: pip install sentence-transformers\n"
-                        "(95MB 模型首次自动下载, 之后完全离线)")
-            elif st['embeddings_count'] == 0:
-                return ("语义检索未构建向量。请管理员运行:\n"
-                        "  python .qoder/scripts/domain/kg/graph/kg_semantic.py build-embeddings")
-            return None
-        lines = ["语义搜索: '%s' (top %d)" % (query, len(results))]
-        for eid, text, score in results:
-            lines.append("  %.3f  %s" % (score, text[:70]))
-        return '\n'.join(lines)
-
-
-if __name__ == "__main__":
-    KnowledgeGraphMCPServer().run()
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# v3.0 路径自举: 引导到 common/bootstrap, 统一 sys.path 逻辑
+import os as _o, sys as _s
+_f = _o.path.abspath(__file__)
+for _ in range(10):
+    _f = _o.path.dirname(_f)
+    _cp = _o.path.join(_f, 'foundation')
+    if _o.path.isfile(_o.path.join(_cp, 'bootstrap.py')):
+        break
+if _cp not in _s.path: _s.path.insert(0, _cp)
+from bootstrap import setup; setup()
+from foundation.core.paths import get_repo_root
+
+"""
+Knowledge Graph MCP Server for QoderWork (pure Python, no SDK dependency).
+
+Exposes the QODER knowledge graph as MCP tools that QoderWork's AI agent
+can call directly. Uses the MCP stdio protocol (JSON-RPC 2.0 over stdin/stdout)
+implemented from scratch — no external dependencies beyond the standard library.
+
+Registered in ~/.qoderwork/mcp.json as "qoder-knowledge-graph".
+Tools surface as mcp__qoder-knowledge-graph__<tool> to the AI.
+
+v3.1 重构: 用 @tool 装饰器 + ToolRegistryMeta 元类, 消除:
+  - 手写 TOOLS 列表 (78-253 行)
+  - _execute_tool 的 17 个 if/elif 分支 (386-576 行)
+两处合一, 新增工具只需加一个 @tool 方法。
+"""
+import io
+import os
+import sys
+import json
+import subprocess
+
+# Ensure UTF-8 output
+try:
+    sys.stdout.reconfigure(encoding='utf-8', errors='replace')
+    sys.stderr.reconfigure(encoding='utf-8', errors='replace')
+except Exception:
+    pass
+
+# Locate the scripts dir
+SCRIPTS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  # scripts/ 根
+BASE_DIR = os.path.dirname(os.path.dirname(SCRIPTS_DIR))  # repo 根
+sys.path.insert(0, SCRIPTS_DIR)
+_KG_DIR = os.path.dirname(os.path.abspath(__file__))
+if _KG_DIR not in sys.path: sys.path.insert(0, _KG_DIR)
+
+
+# ── stdout 捕获辅助 ───────────────────────────────────────────────────
+# 底层脚本 (search_index/context_pack) 用 print 输出, MCP 需要捕获为字符串返回。
+
+def _capture(func, *args, **kwargs):
+    """Run a function, capture its stdout.
+
+    v3.0: sys.stdout 在 run() 后已是 stderr (基类 _setup_streams 重定向),
+    这里捕获时临时换回 buf, finally 还原。
+    """
+    old = sys.stdout
+    buf = io.StringIO()
+    sys.stdout = buf
+    try:
+        func(*args, **kwargs)
+    except Exception as e:
+        buf.write('\n[error] %s' % str(e))
+    finally:
+        sys.stdout = old
+    return buf.getvalue().strip()
+
+
+# Lazy imports (首次调用工具时才 import, 加快启动)
+_search = None
+_context = None
+
+def _ensure_imports():
+    global _search, _context
+    if _search is None:
+        from domain.kg.search import search_index as _search
+    if _context is None:
+        from domain.kg.search import context_pack as _context
+
+
+# ── MCP Server (继承 BaseMCPServer, @tool 装饰器自动分发) ──────────────
+
+from foundation.protocol.mcp_base import BaseMCPServer, tool
+
+
+def _require_kgdb():
+    """检查 kg.duckdb 是否存在。存在返回 None; 不存在返回友好提示字符串。
+
+    ★ 共识：本地引擎全走平台。kg MCP 走云 SSE url，不依赖本地 kg.duckdb。
+    但本文件是"本地 launcher 模式"的 server——只在 mcp.json 没配 SSE url 时被调用。
+    配了 SSE url 时 QoderWork 直连云平台，本文件根本不会启动。
+    所以这里只需检查本地 duckdb，报错了也只影响 launcher 模式（非共识路径）。
+    """
+    try:
+        from foundation.core.paths import DATA_INDEX_DIR
+        db_path = DATA_INDEX_DIR / 'kg.duckdb'
+        if not db_path.is_file():
+            return ("知识图谱走云平台（mcp.json kg → SSE url），本地 kg.duckdb 不需要。\n"
+                    "如果云平台不通，可本地构建: python .qoder/scripts/orchestration/wlkj.py kg-build")
+    except Exception:
+        pass
+    return None
+
+
+class KnowledgeGraphMCPServer(BaseMCPServer):
+    """知识图谱 MCP Server。
+
+    v3.1: 每个工具用 @tool 装饰, 元类自动收集 TOOLS + 自动 handle_tool 分发。
+    新增工具 = 加一个 @tool 方法, 无需改 TOOLS 列表或 if 链。
+    """
+
+    SERVER_NAME = "qoder-knowledge-graph"
+    SERVER_VERSION = "2.0.0"
+
+    # ★ v2.0 共识改造：kg 走云平台。本 server 的所有工具不再直连本地 duckdb，
+    # 而是转发到云平台 HTTP MCP@10010（http://10.87.106.252/mcp）。
+    # 工具名映射：本地工具名 → 云平台等价 MCP 工具
+    PLATFORM_MCP_URL = os.environ.get("WLKJ_PLATFORM_MCP", "http://10.87.106.252/mcp")
+    # 本地工具 → 云平台工具映射（名称不同或参数不同）
+    _TOOL_MAP = {
+        "search_code": "search_code",
+        "search_api": "search_api",
+        "search_prd": "search_prd_semantic",
+        "get_impact": "get_impact",
+        "context_360": "context_360",
+        "context_pack": "context_pack",
+        "get_workflow": "business_trace",
+        "multi_hop": "business_trace",
+        "search_wiki": "rag_search",
+        "search_field": "search_field",
+        "list_modules": "get_modules",
+        "coverage_matrix": "impact_coverage",
+        "feature_overview": "feature_profile",
+        "table_impact": "table_impact",
+        "fill_prototype": None,
+        "get_design_system": "get_design_system",
+        "semantic_search": "rag_search",
+    }
+
+    def _call_platform(self, tool_name, args):
+        """转发工具调用到云平台。
+
+        主路径：mcp.json kg→SSE url，QoderWork直连云平台（不经本server）。
+        备选路径：launcher模式（本server被调），走SSE协议转发。
+        """
+        mapped = self._TOOL_MAP.get(tool_name)
+        if not mapped:
+            return json.dumps({"error": f"工具 {tool_name} 无云平台等价（本地功能）"}, ensure_ascii=False)
+        # 尝试HTTP POST（简单协议，部分端点支持）
+        try:
+            import urllib.request, urllib.error
+            body = json.dumps({"tool": mapped, "args": args or {}}).encode("utf-8")
+            req = urllib.request.Request(
+                self.PLATFORM_MCP_URL,
+                data=body,
+                headers={"Content-Type": "application/json"},
+                method="POST",
+            )
+            with urllib.request.urlopen(req, timeout=30) as resp:
+                return resp.read().decode("utf-8")
+        except urllib.error.HTTPError as e:
+            if e.code == 401:
+                return json.dumps({"error": "云平台需鉴权。请在 mcp.json 配 kg 为 SSE url（npx @hupan56/wlkj update 自动配置）。"}, ensure_ascii=False)
+            return json.dumps({"error": f"云平台 HTTP {e.code}: {e.reason}"}, ensure_ascii=False)
+        except Exception as e:
+            return json.dumps({"error": f"云平台不可达: {type(e).__name__}: {str(e)[:120]}"}, ensure_ascii=False)
+
+    # ── 工具实现（全部转发到云平台）──────────────────────────────────
+
+    @tool("search_code",
+          "搜索代码: 按关键词查找代码文件在哪。例: search_code(keyword='考勤') 或 search_code(keyword='车辆', platform='web')",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "搜索关键词(中文或英文)"},
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web=PC管理端, app=移动端。可选"},
+          }, "required": ["keyword"]})
+    def _t_search_code(self, args):
+        return self._call_platform("search_code", args)
+
+    @tool("search_code",
+          "搜索代码: 按关键词查找代码文件在哪。例: search_code(keyword='考勤') 或 search_code(keyword='车辆', platform='web')",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "搜索关键词(中文或英文)"},
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web=PC管理端, app=移动端。可选"},
+          }, "required": ["keyword"]})
+    def _t_search_code(self, args):
+        return self._call_platform("search_code", args)
+
+    @tool("search_api",
+          "搜索API端点: 按关键词查找后端接口。例: search_api(keyword='salary')",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "API关键词"},
+          }, "required": ["keyword"]})
+    def _t_search_api(self, args):
+        return self._call_platform("search_api", args)
+
+    @tool("search_prd",
+          "搜索PRD: 查已有需求文档,防止重复造轮子。例: search_prd(keyword='考勤')",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "PRD关键词"},
+          }, "required": ["keyword"]})
+    def _t_search_prd(self, args):
+        return self._call_platform("search_prd", args)
+
+    @tool("get_impact",
+          "影响分析: 改某接口/路径会影响哪些前端页面和按钮?评估改动风险。例: get_impact(endpoint='/asset') 或 get_impact(endpoint='export')",
+          {"type": "object", "properties": {
+              "endpoint": {"type": "string", "description": "API端点路径或关键词, 如 /asset/list 或 export"},
+          }, "required": ["endpoint"]})
+    def _t_get_impact(self, args):
+        return self._call_platform("get_impact", args)
+
+    @tool("context_360",
+          "360度视图: 查看符号(端点/函数/处理器)的全部关联信息。例: context_360(symbol='handleExport') 或 context_360(symbol='/system/role')",
+          {"type": "object", "properties": {
+              "symbol": {"type": "string", "description": "符号名: 端点/函数名/处理器名"},
+          }, "required": ["symbol"]})
+    def _t_context_360(self, args):
+        return self._call_platform("context_360", args)
+
+    @tool("context_pack",
+          "上下文打包: 一次返回关键词相关的全部上下文(代码+页面+字段+PRD+API+布局指纹)。避免来回查。例: context_pack(keyword='车辆', platform='web')",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "业务关键词"},
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台: web或app。可选"},
+              "page_type": {"type": "string", "description": "页面类型: table-page/form-page/detail-page。可选"},
+              "role": {"type": "string", "enum": ["pm", "design", "dev", "test", "admin"],
+                       "description": "角色裁剪: pm=字段+PRD+API, design=页面+布局+风格, dev=代码+API, test=代码+PRD+API。可选"},
+          }, "required": ["keyword"]})
+    def _t_context_pack(self, args):
+        return self._call_platform("context_pack", args)
+
+    @tool("get_workflow",
+          "业务流程: 查看一个模块的完整操作链。",
+          {"type": "object", "properties": {
+              "module": {"type": "string", "description": "模块名(英文优先)"},
+          }, "required": ["module"]})
+    def _t_get_workflow(self, args):
+        return self._call_platform("get_workflow", args)
+
+    @tool("multi_hop",
+          "知识图谱多跳遍历: 从一个符号出发, 沿edges递归N跳。",
+          {"type": "object", "properties": {
+              "symbol": {"type": "string", "description": "起点符号"},
+              "max_depth": {"type": "integer", "default": 3, "description": "最大跳数(1-5)"},
+          }, "required": ["symbol"]})
+    def _t_multi_hop(self, args):
+        return self._call_platform("multi_hop", args)
+
+    @tool("search_wiki",
+          "搜索Repo Wiki: 查模块级语义文档。",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "Wiki关键词"},
+          }, "required": ["keyword"]})
+    def _t_search_wiki(self, args):
+        return self._call_platform("search_wiki", args)
+
+    @tool("search_field",
+          "搜索字段: 查某个字段在哪些页面用过。",
+          {"type": "object", "properties": {
+              "field": {"type": "string", "description": "字段名(英文field或中文标题)"},
+          }, "required": ["field"]})
+    def _t_search_field(self, args):
+        return self._call_platform("search_field", args)
+
+    @tool("list_modules",
+          "模块清单: 列出项目所有业务模块。",
+          {"type": "object", "properties": {
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台过滤"},
+          }})
+    def _t_list_modules(self, args):
+        return self._call_platform("list_modules", args)
+
+    @tool("coverage_matrix",
+          "功能×测试覆盖矩阵。",
+          {"type": "object", "properties": {}})
+    def _t_coverage_matrix(self, args):
+        return self._call_platform("coverage_matrix", args)
+
+    @tool("feature_overview",
+          "功能画像: 一个功能的完整画像。",
+          {"type": "object", "properties": {
+              "feature": {"type": "string", "description": "功能名(中/英文)"},
+          }, "required": ["feature"]})
+    def _t_feature_overview(self, args):
+        return self._call_platform("feature_overview", args)
+
+    @tool("table_impact",
+          "改表影响分析: 改某数据库表会影响哪些 Entity 类和接口。",
+          {"type": "object", "properties": {
+              "table": {"type": "string", "description": "数据库表名"},
+          }, "required": ["table"]})
+    def _t_table_impact(self, args):
+        return self._call_platform("table_impact", args)
+
+    # ── 原型/设计/语义 (全部走云平台) ────────────────────────────────
+
+    @tool("fill_prototype",
+          "原型预填: 用真实代码数据自动填模板。",
+          {"type": "object", "properties": {
+              "keyword": {"type": "string", "description": "业务关键词"},
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台"},
+          }, "required": ["keyword"]})
+    def _t_fill_prototype(self, args):
+        return self._call_platform("fill_prototype", args)
+
+    @tool("get_design_system",
+          "设计系统查询: 返回项目完整设计规范。",
+          {"type": "object", "properties": {
+              "platform": {"type": "string", "enum": ["web", "app"], "description": "平台过滤"},
+          }})
+    def _t_get_design_system(self, args):
+        return self._call_platform("get_design_system", args)
+
+    @tool("semantic_search",
+          "语义搜索: 用自然语言找代码/API/功能。",
+          {"type": "object", "properties": {
+              "query": {"type": "string", "description": "自然语言查询"},
+              "top_k": {"type": "integer", "default": 10, "description": "返回前N条"},
+          }, "required": ["query"]})
+    def _t_semantic_search(self, args):
+        return self._call_platform("semantic_search", args)
+        from domain.kg.graph.kg_semantic import semantic_search, embedding_status
+        query = (args.get("query") or '').strip()
+        if not query:
+            return "请提供查询关键词"
+        try:
+            top_k = int(args.get("top_k", 10))
+        except (TypeError, ValueError):
+            top_k = 10
+        results = semantic_search(query, top_k=top_k)
+        if results is None:
+            st = embedding_status()
+            if not st['available']:
+                return ("语义检索未启用: sentence-transformers 未安装。\n"
+                        "启用方法: pip install sentence-transformers\n"
+                        "(95MB 模型首次自动下载, 之后完全离线)")
+            elif st['embeddings_count'] == 0:
+                return ("语义检索未构建向量。请管理员运行:\n"
+                        "  python .qoder/scripts/domain/kg/graph/kg_semantic.py build-embeddings")
+            return None
+        lines = ["语义搜索: '%s' (top %d)" % (query, len(results))]
+        for eid, text, score in results:
+            lines.append("  %.3f  %s" % (score, text[:70]))
+        return '\n'.join(lines)
+
+
+if __name__ == "__main__":
+    KnowledgeGraphMCPServer().run()