@yottameta/yotta-dev-mcp-plugin 0.2.0 → 0.2.2

package/skills/yotta-dev-mcp/scripts/dev_model.py

@@ -30,9 +30,14 @@ def _resolve_python_import(source_rel, node):
         if node.level > 1:
             parts = parts[:-(node.level - 1)] if len(parts) >= node.level - 1 else []
         prefix = "/".join(parts)
+        if not node.module:
+            return [
+                (prefix + "/" + alias.name.replace(".", "/")).strip("/")
+                for alias in node.names
+            ]
         module = node.module or ""
         target = (prefix + "/" + module.replace(".", "/")).strip("/")
-        return [target + ".py" if target else source_rel]
+        return [target] if target else [source_rel]
     if node.module:
         return [node.module]
     return []
@@ -47,7 +52,12 @@ def _repo_map_python(path, rel):
     for node in ast.walk(tree):
         if isinstance(node, (ast.Import, ast.ImportFrom)):
             for target in _resolve_python_import(rel, node):
-                imports.append({"source": rel, "target": target, "line": getattr(node, "lineno", 1)})
+                imports.append({
+                    "source": rel,
+                    "target": target,
+                    "line": getattr(node, "lineno", 1),
+                    "relative": isinstance(node, ast.ImportFrom) and bool(node.level),
+                })
     return imports
 
 def _repo_map_js(path, rel):
@@ -77,7 +87,7 @@ def _resolve_relative_module(raw_target, source_rel, module_set, root=None):
         return base, "file"
     return None, None
 
-def _classify_python_import(raw_target, source_rel, module_set):
+def _classify_python_import(raw_target, source_rel, module_set, relative=False):
     if raw_target.endswith(".py"):
         if raw_target in module_set:
             return "internal", raw_target
@@ -91,7 +101,7 @@ def _classify_python_import(raw_target, source_rel, module_set):
         for candidate in (base + ".py", base + "/__init__.py"):
             if candidate in module_set:
                 return "internal", candidate
-    return "external", raw_target
+    return ("unresolved" if relative else "external"), raw_target
 
 def _classify_js_import(raw_target, source_rel, module_set, root=None):
     if raw_target.startswith("."):
@@ -265,19 +275,21 @@ def system_model(path, max_files=2000, contract_file=None):
         suffix = file_path.suffix.lower()
         if suffix == ".py":
             raw_imports = _repo_map_python(file_path, rel)
-            classifier = lambda raw: _classify_python_import(raw, rel, module_set)  # noqa: E731
         elif suffix in JS_EXTS:
             raw_imports = _repo_map_js(file_path, rel)
-            classifier = lambda raw: _classify_js_import(raw, rel, module_set, root)  # noqa: E731
         else:
             raw_imports = []
-            classifier = None
         module_imports = []
         for raw in raw_imports:
             target_raw = raw["target"]
             if target_raw == rel:
                 continue
-            kind, target = classifier(target_raw)
+            if suffix == ".py":
+                kind, target = _classify_python_import(
+                    target_raw, rel, module_set, relative=raw.get("relative", False)
+                )
+            else:
+                kind, target = _classify_js_import(target_raw, rel, module_set, root)
             entry = {
                 "source": rel,
                 "target": target,

package/skills/yotta-dev-mcp/scripts/dev_selftest.py

@@ -15,7 +15,8 @@ from pathlib import Path
 
 from dev_architecture import architecture_review
 from dev_common import (
-    EVIDENCE_LIMIT, VERIFY_REQUIRED_INSTALLED_FILES,
+    EVIDENCE_LIMIT, LEAN_INSTALL_MARKERS, VERIFY_REQUIRED_INSTALLED_ASSETS,
+    VERIFY_REQUIRED_INSTALLED_FILES,
     VERIFY_REQUIRED_SOURCE_FILES, VERIFY_WRITE_GATES, _frontmatter_version,
     _read_text,
 )
@@ -44,26 +45,30 @@ def _function_default(source_text, function_name, parameter_name):
 def _string_constant(node):
     return isinstance(node, ast.Constant) and isinstance(node.value, str)
 
+def _is_lean_install(root):
+    return any((root / marker).is_file() for marker in LEAN_INSTALL_MARKERS)
+
 def _protocol_tool_contracts(source_text):
     tree = ast.parse(source_text)
     function = None
     for node in tree.body:
-        if isinstance(node, ast.FunctionDef) and node.name == "mcp_tools":
+        # v0.2.2 起工具清单字面量在 _all_tools()；mcp_tools(profile) 只做分档过滤。
+        if isinstance(node, ast.FunctionDef) and node.name in ("_all_tools", "mcp_tools"):
             function = node
             break
     if function is None:
-        raise ValueError("mcp_tools() not found")
+        raise ValueError("tool list function not found")
     list_node = None
     for node in ast.walk(function):
         if isinstance(node, ast.Return) and isinstance(node.value, ast.List):
             list_node = node.value
             break
     if list_node is None:
-        raise ValueError("mcp_tools() does not return a literal tool list")
+        raise ValueError("tool list function does not return a literal tool list")
     contracts = []
     for element in list_node.elts:
         if not isinstance(element, ast.Dict):
-            raise ValueError("mcp_tools() contains a non-literal tool entry")
+            raise ValueError("tool list contains a non-literal tool entry")
         values = {}
         for key, value in zip(element.keys, element.values):
             if _string_constant(key):
@@ -380,8 +385,14 @@ def self_test(path, mode="auto", allow_execute=False, timeout=120):
         else:
             mode = "source"
 
-    required = (VERIFY_REQUIRED_SOURCE_FILES if mode == "source"
-                else VERIFY_REQUIRED_INSTALLED_FILES)
+    if mode == "source":
+        required = list(VERIFY_REQUIRED_SOURCE_FILES)
+    else:
+        required = list(VERIFY_REQUIRED_INSTALLED_FILES)
+        # Lean distribution strips assets/ by platform policy; only require
+        # the banner when the copy is not lean or it already ships assets/.
+        if not _is_lean_install(root) or (root / "assets").is_dir():
+            required.extend(VERIFY_REQUIRED_INSTALLED_ASSETS)
     missing = [rel for rel in required if not (root / rel).is_file()]
     files_check = {
         "id": "files",

package/skills/yotta-dev-mcp/scripts/yotta_dev_mcp.py

@@ -25,6 +25,25 @@ MCP_PROTOCOL_MODERN = "2026-07-28"
 MCP_PROTOCOL_LEGACY = "2025-11-25"
 SERVER_INFO = {"name": TOOL_NAME, "version": VERSION}
 
+# 工具分组（v0.2.2）：与元忆 `--tools core|full` 同口径。core 是给「需要在上下文里
+# 常驻工具表」的宿主用小档（5 工具，schema 远低于 8000 字符硬上限）；full 保持 18 工具。
+CORE_TOOL_NAMES = (
+    "repo_map",
+    "find_code",
+    "review_code",
+    "review_diff",
+    "verify_change",
+)
+TOOL_PROFILES = ("core", "full")
+DEFAULT_TOOL_PROFILE = "full"
+
+
+def normalize_tool_profile(value):
+    profile = str(value or "").strip().lower() or DEFAULT_TOOL_PROFILE
+    if profile not in TOOL_PROFILES:
+        raise ValueError("未知工具分组: %s（可用: core / full）" % value)
+    return profile
+
 
 def _text_result(payload):
     return {
@@ -63,7 +82,7 @@ TOOL_HANDLERS = {
 }
 
 
-def mcp_tools():
+def _all_tools():
     return [
         {
             "name": "repo_map",
@@ -386,14 +405,24 @@ def mcp_tools():
         {
             "name": "mcp_doctor",
             "description": (
-                "Inspect installed skills and MCP JSON configuration files for "
-                "versions and obvious configuration issues. Read-only."
+                "Broad, read-only health check for installed skills and MCP client "
+                "configuration. Discovers Codex, Cursor, WorkBuddy, OpenCode, Claude "
+                "Code and other known hosts (plus project-level configs), parses JSON, "
+                "JSONC and a narrow TOML subset, and always returns a coverage report "
+                "with checked / missing / unsupported / unverified hosts. Only server "
+                "names are returned; commands, args and env values are never included. "
+                "A clean result applies only to the covered scope: read coverage and "
+                "coverage_confidence before treating it as all-clear."
             ),
             "inputSchema": {
                 "type": "object",
                 "properties": {
                     "skills_dirs": {"type": "array", "items": {"type": "string"}},
                     "config_paths": {"type": "array", "items": {"type": "string"}},
+                    "include_defaults": {
+                        "type": "boolean",
+                        "description": "Also scan the built-in host registry when config_paths is supplied",
+                    },
                 },
                 "additionalProperties": False,
             },
@@ -511,6 +540,15 @@ def mcp_tools():
     ]
 
 
+def mcp_tools(profile=None):
+    """按工具分档返回 tools/list 内容（core 5 个 / full 18 个）。"""
+    tools = _all_tools()
+    if normalize_tool_profile(profile) == "core":
+        by_name = {tool["name"]: tool for tool in tools}
+        return [by_name[name] for name in CORE_TOOL_NAMES]
+    return tools
+
+
 def _req_version(params):
     meta = (params or {}).get("_meta") or {}
     return meta.get("io.modelcontextprotocol/protocolVersion")
@@ -538,7 +576,7 @@ def _unsupported_version(rid, protocol_version):
     }
 
 
-def handle_message(msg):
+def handle_message(msg, tool_profile=None):
     if not isinstance(msg, dict) or msg.get("jsonrpc") != "2.0":
         rid = msg.get("id") if isinstance(msg, dict) else None
         return {"jsonrpc": "2.0", "id": rid,
@@ -568,7 +606,9 @@ def handle_message(msg):
                         "verify_change for change-centered evidence, self_test for integrity "
                         "checks, and run_adapter to probe or explicitly run optional local "
                         "architecture adapters. Tools are offline and read-only unless an "
-                        "explicit write or execute flag is set."
+                        "explicit write or execute flag is set. Current tool profile: "
+                        + normalize_tool_profile(tool_profile)
+                        + " (start with --tools core|full)."
                     ),
                 }, (3600000, "public")),
             }
@@ -578,11 +618,20 @@ def handle_message(msg):
             return {
                 "jsonrpc": "2.0",
                 "id": rid,
-                "result": _modern_ok({"tools": mcp_tools()}, (300000, "public")),
+                "result": _modern_ok({"tools": mcp_tools(tool_profile)}, (300000, "public")),
             }
         if method == "tools/call":
             name = params.get("name")
             arguments = params.get("arguments") or {}
+            if (normalize_tool_profile(tool_profile) == "core"
+                    and name not in CORE_TOOL_NAMES):
+                return {
+                    "jsonrpc": "2.0",
+                    "id": rid,
+                    "result": _modern_ok(_tool_error(
+                        "工具 %s 属于 full 分组，当前以 --tools core 启动。"
+                        "需要完整工具集时请用 --tools full 重新启动。" % name)),
+                }
             handler = TOOL_HANDLERS.get(name)
             if not handler:
                 return {
@@ -616,10 +665,16 @@ def handle_message(msg):
     if method == "ping":
         return {"jsonrpc": "2.0", "id": rid, "result": {}}
     if method == "tools/list":
-        return {"jsonrpc": "2.0", "id": rid, "result": {"tools": mcp_tools()}}
+        return {"jsonrpc": "2.0", "id": rid,
+                "result": {"tools": mcp_tools(tool_profile)}}
     if method == "tools/call":
         name = params.get("name")
         arguments = params.get("arguments") or {}
+        if (normalize_tool_profile(tool_profile) == "core"
+                and name not in CORE_TOOL_NAMES):
+            return {"jsonrpc": "2.0", "id": rid, "result": _tool_error(
+                "工具 %s 属于 full 分组，当前以 --tools core 启动。"
+                "需要完整工具集时请用 --tools full 重新启动。" % name)}
         handler = TOOL_HANDLERS.get(name)
         if not handler:
             return {
@@ -632,7 +687,42 @@ def handle_message(msg):
             "error": {"code": -32601, "message": "Method not found: " + str(method)}}
 
 
+def parse_args(argv):
+    """解析启动参数：`--tools core|full`（默认 full）、`--version`、`--help`。"""
+    profile = DEFAULT_TOOL_PROFILE
+    index = 0
+    while index < len(argv):
+        arg = argv[index]
+        if arg == "--tools":
+            index += 1
+            if index >= len(argv):
+                sys.stderr.write("用法错误: --tools 需要一个值（core / full）\n")
+                raise SystemExit(2)
+            raw = argv[index]
+        elif arg.startswith("--tools="):
+            raw = arg.split("=", 1)[1]
+        elif arg in ("--version", "-v"):
+            print("%s %s" % (TOOL_NAME, VERSION))
+            raise SystemExit(0)
+        elif arg in ("--help", "-h"):
+            print("用法: yotta-dev-mcp [--tools core|full]")
+            print("  core: 5 工具（%s）" % " / ".join(CORE_TOOL_NAMES))
+            print("  full: 18 工具（默认）")
+            raise SystemExit(0)
+        else:
+            sys.stderr.write("用法错误: 未知参数 %s\n" % arg)
+            raise SystemExit(2)
+        try:
+            profile = normalize_tool_profile(raw)
+        except ValueError as exc:
+            sys.stderr.write("用法错误: %s\n" % exc)
+            raise SystemExit(2)
+        index += 1
+    return profile
+
+
 def main():
+    tool_profile = parse_args(sys.argv[1:])
     try:
         sys.stdin.reconfigure(encoding="utf-8")
         sys.stdout.reconfigure(encoding="utf-8")
@@ -649,7 +739,7 @@ def main():
             response = {"jsonrpc": "2.0", "id": None,
                         "error": {"code": -32700, "message": "parse error"}}
         else:
-            response = handle_message(message)
+            response = handle_message(message, tool_profile)
         if response is not None:
             sys.stdout.write(json.dumps(response, ensure_ascii=False) + "\n")
             sys.stdout.flush()

package/skills/yotta-dev-mcp/server.json

@@ -6,12 +6,12 @@
     "url": "https://github.com/YottaMeta/yotta-dev-mcp",
     "source": "github"
   },
-  "version": "0.2.0",
+  "version": "0.2.2",
   "packages": [
     {
       "registryType": "npm",
       "identifier": "@yottameta/yotta-dev-mcp",
-      "version": "0.2.0",
+      "version": "0.2.2",
       "transport": {
         "type": "stdio"
       }