@hupan56/wlkj 2.7.12 → 3.1.2

package/templates/qoder/scripts/capability/smoke_test.py

@@ -0,0 +1,211 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+nexus MCP 端到端 smoke test。
+
+用法:
+    python .qoder/scripts/capability/smoke_test.py
+
+验证:
+    1. McpRegistry 能 spawn 每个 MCP server 子进程
+    2. list_tools 能拿到真实工具清单
+    3. call_tool 能端到端调通（以 kg 的 search_code 为例）
+    4. Memory 职责链链头是否可用
+
+输出: 每个 server 的 PASS/FAIL + tools_indexed + latency_ms。
+退出码: 0=全 PASS，1=有 FAIL。
+"""
+from __future__ import annotations
+
+import json
+import os
+import sys
+import time
+
+# 确保 scripts/ 在 sys.path
+_here = os.path.dirname(os.path.abspath(__file__))
+_scripts = os.path.dirname(_here)
+if _scripts not in sys.path:
+    sys.path.insert(0, _scripts)
+
+# UTF-8 (Windows)
+try:
+    sys.stdout.reconfigure(encoding="utf-8", errors="replace")
+except Exception:
+    pass
+
+
+def _section(title: str) -> None:
+    print("\n" + "=" * 60)
+    print(f"  {title}")
+    print("=" * 60)
+
+
+def test_registry() -> dict:
+    """测试 McpRegistry: 每个 server 的 list_tools + call_tool。"""
+    from capability.registry_mcp import McpRegistry
+
+    reg = McpRegistry()
+    results = {}
+    _section("McpRegistry server 探活")
+
+    for server_name in reg._servers:
+        t0 = time.monotonic()
+        try:
+            transport = reg._get_or_create(server_name)
+            available = transport.available
+            tools = transport.list_tools() if available else []
+            latency_ms = (time.monotonic() - t0) * 1000
+            results[server_name] = {
+                "available": available,
+                "tools_indexed": len(tools),
+                "latency_ms": round(latency_ms, 1),
+                "sample_tools": [t.get("name", "") for t in tools[:5]],
+                "spawn_pid": getattr(getattr(transport, "_proc", None), "pid", None),
+                "status": "PASS" if available and len(tools) > 0 else "FAIL",
+            }
+        except Exception as e:
+            latency_ms = (time.monotonic() - t0) * 1000
+            results[server_name] = {
+                "available": False,
+                "tools_indexed": 0,
+                "latency_ms": round(latency_ms, 1),
+                "error": str(e)[:120],
+                "status": "FAIL",
+            }
+
+    for name, r in results.items():
+        icon = "✅" if r["status"] == "PASS" else "❌"
+        pid = f" pid={r['spawn_pid']}" if r.get("spawn_pid") else ""
+        print(f"  {icon} {name}: tools={r['tools_indexed']} "
+              f"latency={r['latency_ms']}ms{pid} sample={r.get('sample_tools', [])[:3]}")
+        if r.get("error"):
+            print(f"     error: {r['error']}")
+
+    return results
+
+
+def test_call_kg_search(reg) -> dict:
+    """端到端调用: kg search_code（验证 call_tool 链路）。"""
+    _section("端到端 call_tool: search_code")
+    t0 = time.monotonic()
+    try:
+        result = reg.call("search_code", {"keyword": "保险"})
+        latency_ms = (time.monotonic() - t0) * 1000
+        ok = not result.is_error and len(result.text) > 0
+        return {
+            "status": "PASS" if ok else "FAIL",
+            "latency_ms": round(latency_ms, 1),
+            "text_preview": result.text[:200] if result.text else "",
+            "is_error": result.is_error,
+        }
+    except Exception as e:
+        latency_ms = (time.monotonic() - t0) * 1000
+        return {
+            "status": "FAIL",
+            "latency_ms": round(latency_ms, 1),
+            "error": str(e)[:120],
+        }
+
+
+def test_memory_chain() -> dict:
+    """测试 memory 职责链链头是否可用。"""
+    _section("Memory 职责链")
+    from capability.registry import resolve
+
+    cap = resolve()
+    mem = cap.memory
+    has_providers = hasattr(mem, "providers_status")
+    available = getattr(mem, "available", False)
+    providers = mem.providers_status() if has_providers else []
+
+    result = {
+        "available": available,
+        "providers": providers,
+        "status": "PASS" if available else "WARN",
+    }
+
+    icon = "✅" if available else "⚠️"
+    print(f"  {icon} memory.available={available}")
+    for p in providers:
+        hit = " ← ACTIVE" if p.get("hit") else ""
+        print(f"     {p['provider']}: available={p['available']}{hit}")
+
+    return result
+
+
+def main() -> int:
+    print("nexus MCP smoke test")
+    print(f"时间: {time.strftime('%Y-%m-%d %H:%M:%S')}")
+    print(f"Python: {sys.executable}")
+    print(f"CWD: {os.getcwd()}")
+
+    # 1. Registry 探活
+    from capability.registry_mcp import McpRegistry
+    reg = McpRegistry()
+    server_results = test_registry()
+
+    # 2. 端到端 call_tool（只在 kg 可用时跑）
+    call_result = {}
+    kg_ok = server_results.get("qoder-knowledge-graph", {}).get("status") == "PASS"
+    if kg_ok:
+        call_result = test_call_kg_search(reg)
+        icon = "✅" if call_result["status"] == "PASS" else "❌"
+        print(f"\n  {icon} search_code: latency={call_result['latency_ms']}ms "
+              f"text_len={len(call_result.get('text_preview', ''))}")
+        if call_result.get("error"):
+            print(f"     error: {call_result['error']}")
+    else:
+        print("\n  ⏭️  kg server 不可用，跳过 call_tool 测试")
+
+    # 3. Memory 链
+    memory_result = test_memory_chain()
+
+    # 4. 汇总
+    _section("汇总")
+    all_pass = all(r.get("status") == "PASS" for r in server_results.values())
+    call_pass = call_result.get("status") == "PASS" or not kg_ok
+    mem_pass = memory_result.get("status") in ("PASS", "WARN")
+    total_pass = all_pass and call_pass and mem_pass
+
+    total_tools = sum(r.get("tools_indexed", 0) for r in server_results.values())
+    active_servers = sum(1 for r in server_results.values() if r.get("status") == "PASS")
+
+    print(f"  Servers: {active_servers}/{len(server_results)} PASS")
+    print(f"  Tools indexed: {total_tools}")
+    print(f"  Call tool: {call_result.get('status', 'SKIP')}")
+    print(f"  Memory chain: {memory_result.get('status', 'SKIP')}")
+    print(f"\n  {'✅ ALL PASS' if total_pass else '❌ SOME FAILED'}")
+
+    # 写 JSON 报告（可选）
+    report = {
+        "timestamp": time.strftime("%Y-%m-%d %H:%M:%S"),
+        "servers": server_results,
+        "call_tool": call_result,
+        "memory": memory_result,
+        "summary": {
+            "all_pass": total_pass,
+            "active_servers": active_servers,
+            "total_servers": len(server_results),
+            "total_tools": total_tools,
+        },
+    }
+    report_path = os.path.join(_here, "smoke_test_report.json")
+    try:
+        with open(report_path, "w", encoding="utf-8") as f:
+            json.dump(report, f, ensure_ascii=False, indent=2)
+        print(f"\n  报告已写入: {report_path}")
+    except Exception:
+        pass
+
+    # 关闭所有 transport
+    try:
+        reg.close_all()
+    except Exception:
+        pass
+
+    return 0 if total_pass else 1
+
+
+if __name__ == "__main__":
+    sys.exit(main())

package/templates/qoder/scripts/capability/smoke_test_report.json

@@ -0,0 +1,94 @@
+{
+  "timestamp": "2026-06-26 21:25:55",
+  "servers": {
+    "qoder-knowledge-graph": {
+      "available": true,
+      "tools_indexed": 17,
+      "latency_ms": 437.0,
+      "sample_tools": [
+        "search_code",
+        "search_api",
+        "search_prd",
+        "get_impact",
+        "context_360"
+      ],
+      "spawn_pid": 38616,
+      "status": "PASS"
+    },
+    "qoder-zentao": {
+      "available": true,
+      "tools_indexed": 66,
+      "latency_ms": 3657.0,
+      "sample_tools": [
+        "check_zentao_status",
+        "get_zentao_url",
+        "list_products",
+        "get_product",
+        "create_product"
+      ],
+      "spawn_pid": 11224,
+      "status": "PASS"
+    },
+    "qoder-mysql": {
+      "available": true,
+      "tools_indexed": 6,
+      "latency_ms": 3484.0,
+      "sample_tools": [
+        "list_envs",
+        "query_schema",
+        "query_data",
+        "query_distinct",
+        "list_databases"
+      ],
+      "spawn_pid": 37868,
+      "status": "PASS"
+    },
+    "lanhu": {
+      "available": true,
+      "tools_indexed": 13,
+      "latency_ms": 3156.0,
+      "sample_tools": [
+        "lanhu_resolve_invite_link",
+        "lanhu_list_product_documents",
+        "lanhu_get_pages",
+        "lanhu_get_ai_analyze_page_result",
+        "lanhu_get_designs"
+      ],
+      "spawn_pid": 36836,
+      "status": "PASS"
+    }
+  },
+  "call_tool": {
+    "status": "PASS",
+    "latency_ms": 3953.0,
+    "text_preview": "Chinese: 保险 -> English: ins insurance vehlife/insurance\n\n[Carmg-H5] 12 files\n  Carmg-H5/src/.../main/List.vue  ->  data/code/Carmg-H5/src/modules/icsAbnormalCenter/module/main/List.vue\n  Carmg-H5/src/",
+    "is_error": false
+  },
+  "memory": {
+    "available": true,
+    "providers": [
+      {
+        "provider": "McpMemoryProvider",
+        "available": true,
+        "hit": true
+      },
+      {
+        "provider": "FileMemoryCap",
+        "available": true,
+        "hit": false
+      },
+      {
+        "provider": "NoOpMemoryCap",
+        "available": false,
+        "hit": false
+      }
+    ],
+    "status": "PASS"
+  },
+  "summary": {
+    "all_pass": true,
+    "active_servers": 4,
+    "total_servers": 4,
+    "total_tools": 102
+  }
+}

package/templates/qoder/scripts/deployment/setup/__init__.py

@@ -0,0 +1,11 @@
+# 子包初始化: 注入 scripts/根 + 所有子包到 sys.path
+# 让跨包 import (旧式不带前缀) 仍可用, 兼容现有代码
+import os as _os, sys as _sys
+_root = _os.path.dirname(_os.path.dirname(_os.path.abspath(__file__)))
+if _root not in _sys.path:
+    _sys.path.insert(0, _root)
+# 注入所有子包目录 (兼容旧式 import xxx)
+for _sub in ('kg', 'build', 'test', 'setup', 'io'):
+    _subdir = _os.path.join(_root, _sub)
+    if _os.path.isdir(_subdir) and _subdir not in _sys.path:
+        _sys.path.insert(0, _subdir)