@hupan56/wlkj 3.3.12 → 3.3.14

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (103) hide show
  1. package/bin/cli.js +1 -23
  2. package/package.json +29 -29
  3. package/templates/qoder/agents/design-agent.md +20 -0
  4. package/templates/qoder/agents/spec-generator.md +21 -0
  5. package/templates/qoder/commands/optional/wl-spec.md +2 -12
  6. package/templates/qoder/commands/optional/wl-status.md +0 -8
  7. package/templates/qoder/commands/wl-code.md +0 -8
  8. package/templates/qoder/commands/wl-commit.md +1 -11
  9. package/templates/qoder/commands/wl-init.md +129 -129
  10. package/templates/qoder/commands/wl-prd.md +0 -24
  11. package/templates/qoder/commands/wl-search.md +11 -26
  12. package/templates/qoder/commands/wl-task.md +1 -11
  13. package/templates/qoder/commands/wl-test.md +0 -8
  14. package/templates/qoder/config.yaml +8 -3
  15. package/templates/qoder/hooks/session-start.py +365 -384
  16. package/templates/qoder/hooks/stop-eval.py +254 -383
  17. package/templates/qoder/scripts/capability/adapters/mcp.py +314 -322
  18. package/templates/qoder/scripts/capability/registry.py +248 -245
  19. package/templates/qoder/scripts/capability/registry_mcp.py +4 -10
  20. package/templates/qoder/scripts/capability/smoke_test_report.json.new +94 -0
  21. package/templates/qoder/scripts/deployment/setup/init_doctor.py +2 -6
  22. package/templates/qoder/scripts/deployment/setup/install_qoderwork.py +14 -0
  23. package/templates/qoder/scripts/deployment/setup/setup.py +2 -16
  24. package/templates/qoder/scripts/domain/integration/return_to_platform.py +2 -4
  25. package/templates/qoder/scripts/domain/integration/spec_upload.py +4 -4
  26. package/templates/qoder/scripts/domain/kg/build/build_entity_registry.py +12 -12
  27. package/templates/qoder/scripts/domain/kg/build/build_relations.py +12 -12
  28. package/templates/qoder/scripts/domain/kg/extract/asset/__init__.py +10 -0
  29. package/templates/qoder/scripts/domain/kg/extract/asset/asset_tree.py +57 -0
  30. package/templates/qoder/scripts/domain/kg/extract/asset/discussion_importer.py +62 -0
  31. package/templates/qoder/scripts/domain/kg/extract/asset/prd_importer.py +146 -0
  32. package/templates/qoder/scripts/domain/kg/extract/asset/prototype_importer.py +64 -0
  33. package/templates/qoder/scripts/domain/kg/extract/asset/returns_importer.py +52 -0
  34. package/templates/qoder/scripts/domain/kg/extract/build_goal3.py +104 -0
  35. package/templates/qoder/scripts/domain/kg/extract/build_goal4.py +55 -0
  36. package/templates/qoder/scripts/domain/kg/extract/build_goal5.py +95 -0
  37. package/templates/qoder/scripts/domain/kg/extract/db/__init__.py +8 -0
  38. package/templates/qoder/scripts/domain/kg/extract/db/data_profile.py +22 -0
  39. package/templates/qoder/scripts/domain/kg/extract/db/fk_extractor.py +55 -0
  40. package/templates/qoder/scripts/domain/kg/extract/db/schema_extractor.py +90 -0
  41. package/templates/qoder/scripts/domain/kg/extract/inference/__init__.py +9 -0
  42. package/templates/qoder/scripts/domain/kg/extract/inference/community_summarizer.py +206 -0
  43. package/templates/qoder/scripts/domain/kg/extract/inference/embed_builder.py +132 -0
  44. package/templates/qoder/scripts/domain/kg/extract/inference/naming_matcher.py +80 -0
  45. package/templates/qoder/scripts/domain/kg/extract/inference/promote.py +59 -0
  46. package/templates/qoder/scripts/domain/kg/extract/inference/recompute.py +93 -0
  47. package/templates/qoder/scripts/domain/kg/extract/inference/weak_link.py +421 -0
  48. package/templates/qoder/scripts/domain/kg/extract/mybatis/__init__.py +9 -0
  49. package/templates/qoder/scripts/domain/kg/extract/mybatis/all.py +79 -0
  50. package/templates/qoder/scripts/domain/kg/extract/mybatis/mapper_parser.py +99 -0
  51. package/templates/qoder/scripts/domain/kg/extract/mybatis/relation_builder.py +69 -0
  52. package/templates/qoder/scripts/domain/kg/extract/mybatis/sql_extractor.py +78 -0
  53. package/templates/qoder/scripts/domain/kg/extract/prd/__init__.py +8 -0
  54. package/templates/qoder/scripts/domain/kg/extract/prd/prd_chunk_embed.py +105 -0
  55. package/templates/qoder/scripts/domain/kg/extract/prd/prd_llm_extract.py +153 -0
  56. package/templates/qoder/scripts/domain/kg/extract/prd/req_anchor.py +120 -0
  57. package/templates/qoder/scripts/domain/kg/graph/kg_link_db.py +235 -0
  58. package/templates/qoder/scripts/domain/kg/graph/kg_semantic.py +5 -41
  59. package/templates/qoder/scripts/domain/kg/kg.py +0 -14
  60. package/templates/qoder/scripts/domain/kg/search/enrich_prompt.py +238 -0
  61. package/templates/qoder/scripts/domain/kg/server/perf_bench.py +197 -0
  62. package/templates/qoder/scripts/domain/kg/switch_project.py +2 -2
  63. package/templates/qoder/scripts/domain/kg/sync_repowiki.py +109 -0
  64. package/templates/qoder/scripts/domain/task/wlkj_panel.py +1348 -1503
  65. package/templates/qoder/scripts/engine/poller.py +219 -0
  66. package/templates/qoder/scripts/foundation/bootstrap.py +145 -145
  67. package/templates/qoder/scripts/orchestration/wlkj.py +189 -306
  68. package/templates/qoder/settings.json +0 -8
  69. package/templates/qoder/skills/wl-spec/SKILL.md +1 -1
  70. package/templates/qoder/commands/wl-data.md +0 -46
  71. package/templates/qoder/commands/wl-fix.md +0 -47
  72. package/templates/qoder/commands/wl-knowledge.md +0 -41
  73. package/templates/qoder/commands/wl-review.md +0 -49
  74. package/templates/qoder/contracts/insight.md +0 -55
  75. package/templates/qoder/hooks/pre-tool-use-commit.py +0 -124
  76. package/templates/qoder/scripts/foundation/io/context_cache.py +0 -94
  77. package/templates/qoder/scripts/tool_guide.md +0 -70
  78. package/templates/qoder/scripts/validation/eval/alignment_matrix.py +0 -176
  79. package/templates/qoder/scripts/validation/eval/bf2_content_fidelity.py +0 -110
  80. package/templates/qoder/scripts/validation/eval/bf2_llmjudge.py +0 -104
  81. package/templates/qoder/scripts/validation/eval/bf_score.py +0 -218
  82. package/templates/qoder/scripts/validation/eval/code_flywheel.py +0 -150
  83. package/templates/qoder/scripts/validation/eval/dispatcher_ab.py +0 -156
  84. package/templates/qoder/scripts/validation/eval/dispatcher_ab_2026-07-21.json +0 -23
  85. package/templates/qoder/scripts/validation/eval/feature_fidelity_flywheel.py +0 -143
  86. package/templates/qoder/scripts/validation/eval/gradient_matrix.py +0 -261
  87. package/templates/qoder/scripts/validation/eval/gradient_matrix_baseline_2026-07-21.json +0 -33
  88. package/templates/qoder/scripts/validation/eval/metrics_dashboard.py +0 -105
  89. package/templates/qoder/scripts/validation/eval/multi_turn_flywheel.py +0 -118
  90. package/templates/qoder/scripts/validation/eval/prd_fidelity_flywheel.py +0 -128
  91. package/templates/qoder/scripts/validation/eval/prd_flywheel.py +0 -148
  92. package/templates/qoder/scripts/validation/eval/prototype_fidelity_flywheel.py +0 -166
  93. package/templates/qoder/scripts/validation/eval/recall_flywheel.py +0 -148
  94. package/templates/qoder/scripts/validation/eval/robustness_flywheel.py +0 -139
  95. package/templates/qoder/scripts/validation/eval/speed_accuracy_flywheel.py +0 -188
  96. package/templates/qoder/scripts/validation/eval/task_flywheel.py +0 -124
  97. package/templates/qoder/scripts/validation/eval/token_flywheel.py +0 -88
  98. package/templates/qoder/scripts/validation/metrics/eval_code_ac.py +0 -177
  99. package/templates/qoder/scripts/validation/metrics/lint_cases.py +0 -170
  100. package/templates/qoder/scripts/validation/test/test_context_cache.py +0 -78
  101. package/templates/qoder/scripts/validation/test/test_lint_cases.py +0 -60
  102. package/templates/qoder/scripts/validation/test/test_pre_tool_use_commit.py +0 -70
  103. package/templates/workspace/specs/REQ-TEST-CTX/.context-cache.json +0 -1
@@ -1,322 +1,314 @@
1
- # -*- coding: utf-8 -*-
2
- """
3
- adapters.mcp —— L1 MCP 能力统一入口(已接通 transport 层)。
4
-
5
- 设计模式: Adapter + Facade
6
-
7
- 职责:
8
- 把 McpRegistry(tool→server→transport 路由)适配成 McpCap 的
9
- has_tool/call/list_tools 接口,供 Nexus Facade 统一暴露。
10
-
11
- L1 能力定位:禅道/知识图谱/MySQL/蓝湖/playwright 已 MCP 化,
12
- 通过本入口跨宿主统一调用,复用已有 MCP 资产,绝不重写它们。
13
-
14
- 跨宿主行为:
15
- - QoderWork / 任意宿主:通过 mcp_launcher spawn 本地 MCP server(stdio transport)
16
- - 未来服务器:改 McpRegistry.SERVERS 配置 kind=http + url,工作流零改动
17
- - 无可用 server:available=False,has_tool 返回 False(静默回退)
18
-
19
- McpMemoryProvider:memory 职责链的 MCP provider,接通 kg.duckdb learning_events 表。
20
- 当 mem0/Zep MCP server 更成熟时可替换为 MCP 版本,接口不变。
21
- """
22
- from __future__ import annotations
23
-
24
- import json
25
- from typing import Any, Optional
26
-
27
- from ..registry_mcp import McpRegistry, _load_servers
28
- from protocol.transports import ToolResult
29
-
30
-
31
- # 单例 Registry(同一进程复用进程池,避免重复 spawn)
32
- _REGISTRY: Optional[McpRegistry] = None
33
-
34
-
35
- def _get_registry() -> McpRegistry:
36
- global _REGISTRY
37
- if _REGISTRY is None:
38
- _REGISTRY = McpRegistry()
39
- return _REGISTRY
40
-
41
-
42
- class McpCap:
43
- """L1 MCP 能力统一入口 —— 复用已有 MCP server(禅道/知识图谱/MySQL/蓝湖)。
44
-
45
- 工作流调用:
46
- cap = resolve()
47
- result = cap.mcp.call("search_code", {"keyword": "保险"})
48
- # 自动路由:MCP 可用走 MCP,不可用走 wlkj.py(CLI 降级)
49
- """
50
-
51
- def __init__(self, mcp_servers=None):
52
- # mcp_servers (= [] None) 视为"不限制 = 用全部本地 server"。
53
- # 原版: mcp_servers=[] (空列表, is not None) filtered 成空 → _servers={} → 0 工具,
54
- # 整个 cap.mcp 自废武功。但本地 stdio transport (StdioTransport) 通过 mcp_launcher
55
- # spawn 子进程, 不依赖宿主是否原生支持 MCP —— CLI / Codex / Claude 同样能起 server
56
- # 故: 宿主未声明 mcp_servers 或声明为空 = "没有宿主原生限制, 用全部本地可起的"。
57
- if mcp_servers: # 非空列表才过滤
58
- # _load_servers() 而非硬编码 SERVERS —— 这样 mcp_config.json 改配置才生效
59
- all_servers = _load_servers()
60
- filtered = {k: v for k, v in all_servers.items() if k in mcp_servers}
61
- self._registry = McpRegistry(servers=filtered)
62
- else:
63
- self._registry = _get_registry()
64
-
65
- @property
66
- def available(self) -> bool:
67
- """是否有任一 MCP server 可用。"""
68
- try:
69
- return self._registry.available()
70
- except Exception:
71
- return False
72
-
73
- # ------------------------------------------------------------------
74
- # McpCap 接口
75
- # ------------------------------------------------------------------
76
-
77
- def has_tool(self, name: str) -> bool:
78
- """某 MCP 工具是否可用(跨所有已配置 server 查找)。"""
79
- try:
80
- return self._registry.has_tool(name)
81
- except Exception:
82
- return False
83
-
84
- def call(self, name: str, args: Optional[dict] = None,
85
- server: Optional[str] = None) -> ToolResult:
86
- """调用 MCP 工具(自动 CLI 降级)。
87
-
88
- 优先走 MCP;MCP 不可用时自动降级到 wlkj.py CLI 脚本。
89
- 工作流永远写 cap.mcp.call("tool_name", args),零宿主感知。
90
- """
91
- # 1. MCP 可用 → 走 MCP
92
- result = self._registry.call(name, args, server=server)
93
- if not result.is_error:
94
- return result
95
- # 2. CLI 降级(MCP 不可用或调用失败)
96
- cli_text = self._cli_fallback(name, args or {})
97
- if cli_text is not None:
98
- return ToolResult(text=cli_text)
99
- return result # 无降级映射,返回原始错误
100
-
101
- # ------------------------------------------------------------------
102
- # CLI 降级(MCP 不可用时走 wlkj.py)
103
- # ------------------------------------------------------------------
104
-
105
- # MCP 工具名 → (wlkj 命令, 基础参数) 映射
106
- # 只映射有 wlkj.py 对应的知识图谱/任务工具。
107
- # MySQL/禅道/蓝湖等 MCP 专属工具不映射(MCP 不可用时返回错误提示)。
108
- _TOOL_TO_WLKJ = {
109
- "search_code": ("kg", ["search"]),
110
- "search_api": ("kg", ["api"]),
111
- "search_prd": ("kg", ["prd"]),
112
- "context_pack": ("kg", ["context"]),
113
- "get_impact": ("kg", ["impact"]),
114
- "context_360": ("kg", ["context360"]),
115
- "coverage_matrix": ("kg", ["coverage"]),
116
- "feature_overview": ("kg", ["feature"]),
117
- "get_workflow": ("kg", ["workflow"]),
118
- "multi_hop": ("kg", ["hop"]),
119
- "search_wiki": ("kg", ["wiki"]),
120
- "fill_prototype": ("kg", ["fill-prototype"]),
121
- "get_design_system": ("kg", ["design-system"]),
122
- # RAG/GraphRAG 降级映射(MCP 挂时走本地 wlkj.py kg search,语义降级为关键词)
123
- "rag_search": ("kg", ["search"]),
124
- "ask_corpus": ("kg", ["search"]),
125
- # 学习工具降级(MCP 挂时本地无 learning 表,返回提示)
126
- "learn": ("noop", []),
127
- "get_learnings": ("noop", []),
128
- "list_tasks": ("task", ["list"]),
129
- "create_task": ("task", ["create"]),
130
- "start_task": ("task", ["start"]),
131
- "finish_task": ("task", ["finish"]),
132
- }
133
-
134
- def _cli_fallback(self, tool_name: str, args: dict) -> Optional[str]:
135
- """MCP 工具名 → wlkj.py 执行。无映射返回 None。"""
136
- # 递归守卫:search_code→kg search→search_index→call_remote→call→_cli_fallback 会无限套娃。
137
- # 已在降级子进程内(_run_wlkj 设了 _WLKJ_IN_CLI_FALLBACK=1)则直接放弃降级,返回原始错误。
138
- import os as _os
139
- if _os.environ.get("_WLKJ_IN_CLI_FALLBACK"):
140
- return None
141
- entry = self._TOOL_TO_WLKJ.get(tool_name)
142
- if entry is None:
143
- return None
144
- cmd, base_args = entry
145
- # noop 类工具(learn/get_learnings 等):本地无等价 CLI,返回提示不崩
146
- if cmd == "noop":
147
- return json.dumps({"degraded": True, "message": "MCP不可用,该工具(learning)降级跳过"}, ensure_ascii=False)
148
- cli_args = list(base_args)
149
- # MCP args 构建命令行参数。
150
- # 参数名对齐: 平台 MCP search 工具入参是 query, 引擎命令文档用 keyword。
151
- # 这里两边都取值, 确保无论上游传哪个都能落到 CLI 查询词上。
152
- kw = args.get("keyword") or args.get("query")
153
- if kw is not None:
154
- cli_args.append(str(kw))
155
- if "platform" in args and args["platform"]:
156
- cli_args.extend(["--platform", str(args["platform"])])
157
- if "role" in args and args["role"]:
158
- cli_args.extend(["--role", str(args["role"])])
159
- if "endpoint" in args:
160
- cli_args.append(str(args["endpoint"]))
161
- if "symbol" in args:
162
- cli_args.append(str(args["symbol"]))
163
- if "depth" in args:
164
- cli_args.extend(["--depth", str(args["depth"])])
165
- if "feature" in args:
166
- cli_args.append(str(args["feature"]))
167
- if "module" in args:
168
- cli_args.append(str(args["module"]))
169
- if "title" in args:
170
- cli_args.append(str(args["title"]))
171
- if "table" in args:
172
- cli_args.append(str(args["table"]))
173
- if "column" in args:
174
- cli_args.append(str(args["column"]))
175
- if "db" in args and args["db"]:
176
- cli_args.extend(["--db", str(args["db"])])
177
- if "env" in args and args["env"]:
178
- cli_args.extend(["--env", str(args["env"])])
179
- return self._run_wlkj(cmd, cli_args)
180
-
181
- @staticmethod
182
- def _run_wlkj(cmd: str, args: list) -> str:
183
- """调 wlkj.py 执行,返回 stdout。"""
184
- import subprocess, sys, os
185
- try:
186
- from foundation.core.paths import get_repo_root
187
- repo = str(get_repo_root())
188
- except Exception:
189
- repo = os.getcwd()
190
- wlkj = os.path.join(repo, ".qoder", "scripts", "orchestration", "wlkj.py")
191
- try:
192
- # 递归守卫:子进程内 _WLKJ_IN_CLI_FALLBACK=1,其 McpCap 不再降级(否则套娃)
193
- env = dict(os.environ, _WLKJ_IN_CLI_FALLBACK="1")
194
- r = subprocess.run(
195
- [sys.executable, wlkj, cmd] + args,
196
- capture_output=True, text=True, timeout=60,
197
- encoding="utf-8", errors="replace", env=env,
198
- )
199
- if r.returncode == 0:
200
- return r.stdout.strip()
201
- return "[cli-fallback error] %s" % (r.stderr or r.stdout)[:200]
202
- except subprocess.TimeoutExpired:
203
- return "[cli-fallback error] timeout (60s)"
204
- except Exception as e:
205
- return "[cli-fallback error] %s" % str(e)[:100]
206
-
207
- def list_tools(self, keyword: str = "") -> list[dict]:
208
- """枚举可用 MCP 工具(可选按 keyword 过滤)。"""
209
- try:
210
- self._registry._ensure_index()
211
- tools = []
212
- for server_name in self._registry._servers:
213
- t = self._registry._pool.get(server_name)
214
- if t is None:
215
- continue
216
- for tool in (t._tools_cache or []):
217
- if keyword and keyword not in tool.get("name", ""):
218
- continue
219
- tool_copy = dict(tool)
220
- tool_copy["_server"] = server_name
221
- tools.append(tool_copy)
222
- return tools
223
- except Exception:
224
- return []
225
-
226
- def describe(self) -> dict:
227
- """诊断:各 server 状态。"""
228
- try:
229
- return self._registry.describe()
230
- except Exception as e:
231
- return {"error": str(e)[:120]}
232
-
233
-
234
- class McpMemoryProvider:
235
- """memory 职责链的 MCP provider —— 接通平台 PG.learning_patterns。
236
-
237
- 设计模式: Adapter(把平台 MCP get_learnings/learn 适配成 MemoryCap 契约)
238
-
239
- 职责链位置: 链头优先。平台 MCP 可用(has_tool get_learnings)时抢占,
240
- 复用项目级学习表(PG.learning_patterns,按 token 解析 pid 隔离,跨会话积累)。
241
-
242
- 数据源: 平台 MCP 工具 get_learnings / learn(后端 services.knowledge.learn)。
243
- """
244
-
245
- def __init__(self):
246
- self._registry = None # 懒取 MCP 单例(避免 import 时即连后端)
247
-
248
- def _reg(self):
249
- """复用 McpCap 同款模块级单例(_get_registry),零重复 spawn。"""
250
- if self._registry is None:
251
- self._registry = _get_registry()
252
- return self._registry
253
-
254
- @property
255
- def available(self) -> bool:
256
- """平台 MCP 可用且 get_learnings 工具在索引中即为可用。
257
- 全迁 PG 后记忆走平台 get_learnings(PG.learning_patterns,按项目隔离)。
258
- """
259
- try:
260
- return self._reg().has_tool("get_learnings")
261
- except Exception:
262
- return False
263
-
264
- def recall(self, query: str) -> list[dict]:
265
- """检索记忆: 取平台学习记录,按 query 模糊过滤 pattern/category。"""
266
- if not self.available:
267
- return []
268
- try:
269
- import json as _json
270
- result = self._reg().call("get_learnings", {"limit": 50})
271
- text = getattr(result, "text", "") or ""
272
- data = _json.loads(text) if text else {}
273
- items = data.get("items", []) if isinstance(data, dict) else []
274
- q = (query or "").lower()
275
- out = []
276
- for it in items:
277
- hay = "%s %s" % (it.get("category", ""), it.get("pattern", ""))
278
- if not q or q in hay.lower():
279
- out.append({
280
- "event": it.get("category", ""),
281
- "data": it.get("pattern", ""),
282
- "ts": it.get("last_seen", ""),
283
- "dev": ",".join(it.get("source_devs", []) or []),
284
- })
285
- return out[:20]
286
- except Exception:
287
- return []
288
-
289
- def remember(self, key: str, content: str) -> None:
290
- """写入一条记忆: 调平台 learn(PG.learning_patterns 累加)。"""
291
- if not self.available:
292
- return
293
- try:
294
- self._reg().call("learn", {
295
- "category": "rule",
296
- "pattern": "%s: %s" % (key, content[:500]),
297
- "source": "engine/memory",
298
- })
299
- except Exception:
300
- pass # 写入失败不阻塞
301
-
302
- def get_fact(self, key: str) -> str | None:
303
- """按 key 取一条事实,无则 None。"""
304
- results = self.recall(key)
305
- return results[0].get("data") if results else None
306
-
307
- def list_recent(self, limit: int = 10) -> list[dict]:
308
- """最近 N 条记忆(平台按 frequency 倒序返回)。"""
309
- if not self.available:
310
- return []
311
- try:
312
- import json as _json
313
- result = self._reg().call("get_learnings", {"limit": limit})
314
- text = getattr(result, "text", "") or ""
315
- data = _json.loads(text) if text else {}
316
- items = data.get("items", []) if isinstance(data, dict) else []
317
- return [{"event": it.get("category", ""), "data": it.get("pattern", ""),
318
- "ts": it.get("last_seen", ""), "dev": ",".join(it.get("source_devs", []) or [])}
319
- for it in items[:limit]]
320
- except Exception:
321
- return []
322
-
1
+ # -*- coding: utf-8 -*-
2
+ """
3
+ adapters.mcp —— L1 MCP 能力统一入口(已接通 transport 层)。
4
+
5
+ 设计模式: Adapter + Facade
6
+
7
+ 职责:
8
+ 把 McpRegistry(tool→server→transport 路由)适配成 McpCap 的
9
+ has_tool/call/list_tools 接口,供 Nexus Facade 统一暴露。
10
+
11
+ L1 能力定位:禅道/知识图谱/MySQL/蓝湖/playwright 已 MCP 化,
12
+ 通过本入口跨宿主统一调用,复用已有 MCP 资产,绝不重写它们。
13
+
14
+ 跨宿主行为:
15
+ - QoderWork / 任意宿主:通过 mcp_launcher spawn 本地 MCP server(stdio transport)
16
+ - 未来服务器:改 McpRegistry.SERVERS 配置 kind=http + url,工作流零改动
17
+ - 无可用 server:available=False,has_tool 返回 False(静默回退)
18
+
19
+ McpMemoryProvider:memory 职责链的 MCP provider,接通 kg.duckdb learning_events 表。
20
+ 当 mem0/Zep MCP server 更成熟时可替换为 MCP 版本,接口不变。
21
+ """
22
+ from __future__ import annotations
23
+
24
+ from typing import Any, Optional
25
+
26
+ from ..registry_mcp import McpRegistry, _load_servers
27
+ from protocol.transports import ToolResult
28
+
29
+
30
+ # 单例 Registry(同一进程复用进程池,避免重复 spawn)
31
+ _REGISTRY: Optional[McpRegistry] = None
32
+
33
+
34
+ def _get_registry() -> McpRegistry:
35
+ global _REGISTRY
36
+ if _REGISTRY is None:
37
+ _REGISTRY = McpRegistry()
38
+ return _REGISTRY
39
+
40
+
41
+ class McpCap:
42
+ """L1 MCP 能力统一入口 —— 复用已有 MCP server(禅道/知识图谱/MySQL/蓝湖)。
43
+
44
+ 工作流调用:
45
+ cap = resolve()
46
+ result = cap.mcp.call("search_code", {"keyword": "保险"})
47
+ # 自动路由:MCP 可用走 MCP,不可用走 wlkj.py(CLI 降级)
48
+ """
49
+
50
+ def __init__(self, mcp_servers=None):
51
+ # mcp_servers (= [] 或 None) 视为"不限制 = 用全部本地 server"。
52
+ # 原版: mcp_servers=[] (空列表, is not None) filtered 成空 → _servers={} 0 工具,
53
+ # 整个 cap.mcp 自废武功。但本地 stdio transport (StdioTransport) 通过 mcp_launcher
54
+ # spawn 子进程, 不依赖宿主是否原生支持 MCP —— CLI / Codex / Claude 同样能起 server。
55
+ # 故: 宿主未声明 mcp_servers 或声明为空 = "没有宿主原生限制, 用全部本地可起的"
56
+ if mcp_servers: # 非空列表才过滤
57
+ # 用 _load_servers() 而非硬编码 SERVERS —— 这样 mcp_config.json 改配置才生效
58
+ all_servers = _load_servers()
59
+ filtered = {k: v for k, v in all_servers.items() if k in mcp_servers}
60
+ self._registry = McpRegistry(servers=filtered)
61
+ else:
62
+ self._registry = _get_registry()
63
+
64
+ @property
65
+ def available(self) -> bool:
66
+ """是否有任一 MCP server 可用。"""
67
+ try:
68
+ return self._registry.available()
69
+ except Exception:
70
+ return False
71
+
72
+ # ------------------------------------------------------------------
73
+ # McpCap 接口
74
+ # ------------------------------------------------------------------
75
+
76
+ def has_tool(self, name: str) -> bool:
77
+ """某 MCP 工具是否可用(跨所有已配置 server 查找)。"""
78
+ try:
79
+ return self._registry.has_tool(name)
80
+ except Exception:
81
+ return False
82
+
83
+ def call(self, name: str, args: Optional[dict] = None,
84
+ server: Optional[str] = None) -> ToolResult:
85
+ """调用 MCP 工具(自动 CLI 降级)。
86
+
87
+ 优先走 MCP;MCP 不可用时自动降级到 wlkj.py CLI 脚本。
88
+ 工作流永远写 cap.mcp.call("tool_name", args),零宿主感知。
89
+ """
90
+ # 1. MCP 可用 → 走 MCP
91
+ result = self._registry.call(name, args, server=server)
92
+ if not result.is_error:
93
+ return result
94
+ # 2. CLI 降级(MCP 不可用或调用失败)
95
+ cli_text = self._cli_fallback(name, args or {})
96
+ if cli_text is not None:
97
+ return ToolResult(text=cli_text)
98
+ return result # 无降级映射,返回原始错误
99
+
100
+ # ------------------------------------------------------------------
101
+ # CLI 降级(MCP 不可用时走 wlkj.py)
102
+ # ------------------------------------------------------------------
103
+
104
+ # MCP 工具名 → (wlkj 命令, 基础参数) 映射
105
+ # 只映射有 wlkj.py 对应的知识图谱/任务工具。
106
+ # MySQL/禅道/蓝湖等 MCP 专属工具不映射(MCP 不可用时返回错误提示)。
107
+ _TOOL_TO_WLKJ = {
108
+ "search_code": ("kg", ["search"]),
109
+ "search_api": ("kg", ["api"]),
110
+ "search_prd": ("kg", ["prd"]),
111
+ "context_pack": ("kg", ["context"]),
112
+ "get_impact": ("kg", ["impact"]),
113
+ "context_360": ("kg", ["context360"]),
114
+ "coverage_matrix": ("kg", ["coverage"]),
115
+ "feature_overview": ("kg", ["feature"]),
116
+ "get_workflow": ("kg", ["workflow"]),
117
+ "multi_hop": ("kg", ["hop"]),
118
+ "search_wiki": ("kg", ["wiki"]),
119
+ "fill_prototype": ("kg", ["fill-prototype"]),
120
+ "get_design_system": ("kg", ["design-system"]),
121
+ # RAG/GraphRAG 降级映射(MCP 挂时走本地 wlkj.py kg search,语义降级为关键词)
122
+ "rag_search": ("kg", ["search"]),
123
+ "ask_corpus": ("kg", ["search"]),
124
+ # 学习工具降级(MCP 挂时本地无 learning 表,返回提示)
125
+ "learn": ("noop", []),
126
+ "get_learnings": ("noop", []),
127
+ "list_tasks": ("task", ["list"]),
128
+ "create_task": ("task", ["create"]),
129
+ "start_task": ("task", ["start"]),
130
+ "finish_task": ("task", ["finish"]),
131
+ }
132
+
133
+ def _cli_fallback(self, tool_name: str, args: dict) -> Optional[str]:
134
+ """MCP 工具名 wlkj.py 执行。无映射返回 None。"""
135
+ entry = self._TOOL_TO_WLKJ.get(tool_name)
136
+ if entry is None:
137
+ return None
138
+ cmd, base_args = entry
139
+ # noop 类工具(learn/get_learnings 等):本地无等价 CLI,返回提示不崩
140
+ if cmd == "noop":
141
+ return json.dumps({"degraded": True, "message": "MCP不可用,该工具(learning)降级跳过"}, ensure_ascii=False)
142
+ cli_args = list(base_args)
143
+ # 从 MCP args 构建命令行参数。
144
+ # 参数名对齐: 平台 MCP search 工具入参是 query, 引擎命令文档用 keyword。
145
+ # 这里两边都取值, 确保无论上游传哪个都能落到 CLI 查询词上。
146
+ kw = args.get("keyword") or args.get("query")
147
+ if kw is not None:
148
+ cli_args.append(str(kw))
149
+ if "platform" in args and args["platform"]:
150
+ cli_args.extend(["--platform", str(args["platform"])])
151
+ if "role" in args and args["role"]:
152
+ cli_args.extend(["--role", str(args["role"])])
153
+ if "endpoint" in args:
154
+ cli_args.append(str(args["endpoint"]))
155
+ if "symbol" in args:
156
+ cli_args.append(str(args["symbol"]))
157
+ if "depth" in args:
158
+ cli_args.extend(["--depth", str(args["depth"])])
159
+ if "feature" in args:
160
+ cli_args.append(str(args["feature"]))
161
+ if "module" in args:
162
+ cli_args.append(str(args["module"]))
163
+ if "title" in args:
164
+ cli_args.append(str(args["title"]))
165
+ if "table" in args:
166
+ cli_args.append(str(args["table"]))
167
+ if "column" in args:
168
+ cli_args.append(str(args["column"]))
169
+ if "db" in args and args["db"]:
170
+ cli_args.extend(["--db", str(args["db"])])
171
+ if "env" in args and args["env"]:
172
+ cli_args.extend(["--env", str(args["env"])])
173
+ return self._run_wlkj(cmd, cli_args)
174
+
175
+ @staticmethod
176
+ def _run_wlkj(cmd: str, args: list) -> str:
177
+ """调 wlkj.py 执行,返回 stdout。"""
178
+ import subprocess, sys, os
179
+ try:
180
+ from foundation.core.paths import get_repo_root
181
+ repo = str(get_repo_root())
182
+ except Exception:
183
+ repo = os.getcwd()
184
+ wlkj = os.path.join(repo, ".qoder", "scripts", "orchestration", "wlkj.py")
185
+ try:
186
+ r = subprocess.run(
187
+ [sys.executable, wlkj, cmd] + args,
188
+ capture_output=True, text=True, timeout=60,
189
+ encoding="utf-8", errors="replace",
190
+ )
191
+ if r.returncode == 0:
192
+ return r.stdout.strip()
193
+ return "[cli-fallback error] %s" % (r.stderr or r.stdout)[:200]
194
+ except subprocess.TimeoutExpired:
195
+ return "[cli-fallback error] timeout (60s)"
196
+ except Exception as e:
197
+ return "[cli-fallback error] %s" % str(e)[:100]
198
+
199
+ def list_tools(self, keyword: str = "") -> list[dict]:
200
+ """枚举可用 MCP 工具(可选按 keyword 过滤)。"""
201
+ try:
202
+ self._registry._ensure_index()
203
+ tools = []
204
+ for server_name in self._registry._servers:
205
+ t = self._registry._pool.get(server_name)
206
+ if t is None:
207
+ continue
208
+ for tool in (t._tools_cache or []):
209
+ if keyword and keyword not in tool.get("name", ""):
210
+ continue
211
+ tool_copy = dict(tool)
212
+ tool_copy["_server"] = server_name
213
+ tools.append(tool_copy)
214
+ return tools
215
+ except Exception:
216
+ return []
217
+
218
+ def describe(self) -> dict:
219
+ """诊断:各 server 状态。"""
220
+ try:
221
+ return self._registry.describe()
222
+ except Exception as e:
223
+ return {"error": str(e)[:120]}
224
+
225
+
226
+ class McpMemoryProvider:
227
+ """memory 职责链的 MCP provider —— 接通平台 PG.learning_patterns。
228
+
229
+ 设计模式: Adapter(把平台 MCP get_learnings/learn 适配成 MemoryCap 契约)
230
+
231
+ 职责链位置: 链头优先。平台 MCP 可用(has_tool get_learnings)时抢占,
232
+ 复用项目级学习表(PG.learning_patterns,按 token 解析 pid 隔离,跨会话积累)。
233
+
234
+ 数据源: 平台 MCP 工具 get_learnings / learn(后端 services.knowledge.learn)。
235
+ """
236
+
237
+ def __init__(self):
238
+ self._registry = None # 懒取 MCP 单例(避免 import 时即连后端)
239
+
240
+ def _reg(self):
241
+ """复用 McpCap 同款模块级单例(_get_registry),零重复 spawn。"""
242
+ if self._registry is None:
243
+ self._registry = _get_registry()
244
+ return self._registry
245
+
246
+ @property
247
+ def available(self) -> bool:
248
+ """平台 MCP 可用且 get_learnings 工具在索引中即为可用。
249
+ 全迁 PG 后记忆走平台 get_learnings(PG.learning_patterns,按项目隔离)。
250
+ """
251
+ try:
252
+ return self._reg().has_tool("get_learnings")
253
+ except Exception:
254
+ return False
255
+
256
+ def recall(self, query: str) -> list[dict]:
257
+ """检索记忆: 取平台学习记录,按 query 模糊过滤 pattern/category。"""
258
+ if not self.available:
259
+ return []
260
+ try:
261
+ import json as _json
262
+ result = self._reg().call("get_learnings", {"limit": 50})
263
+ text = getattr(result, "text", "") or ""
264
+ data = _json.loads(text) if text else {}
265
+ items = data.get("items", []) if isinstance(data, dict) else []
266
+ q = (query or "").lower()
267
+ out = []
268
+ for it in items:
269
+ hay = "%s %s" % (it.get("category", ""), it.get("pattern", ""))
270
+ if not q or q in hay.lower():
271
+ out.append({
272
+ "event": it.get("category", ""),
273
+ "data": it.get("pattern", ""),
274
+ "ts": it.get("last_seen", ""),
275
+ "dev": ",".join(it.get("source_devs", []) or []),
276
+ })
277
+ return out[:20]
278
+ except Exception:
279
+ return []
280
+
281
+ def remember(self, key: str, content: str) -> None:
282
+ """写入一条记忆: 调平台 learn(PG.learning_patterns 累加)。"""
283
+ if not self.available:
284
+ return
285
+ try:
286
+ self._reg().call("learn", {
287
+ "category": "rule",
288
+ "pattern": "%s: %s" % (key, content[:500]),
289
+ "source": "engine/memory",
290
+ })
291
+ except Exception:
292
+ pass # 写入失败不阻塞
293
+
294
+ def get_fact(self, key: str) -> str | None:
295
+ """按 key 取一条事实,无则 None。"""
296
+ results = self.recall(key)
297
+ return results[0].get("data") if results else None
298
+
299
+ def list_recent(self, limit: int = 10) -> list[dict]:
300
+ """最近 N 条记忆(平台按 frequency 倒序返回)。"""
301
+ if not self.available:
302
+ return []
303
+ try:
304
+ import json as _json
305
+ result = self._reg().call("get_learnings", {"limit": limit})
306
+ text = getattr(result, "text", "") or ""
307
+ data = _json.loads(text) if text else {}
308
+ items = data.get("items", []) if isinstance(data, dict) else []
309
+ return [{"event": it.get("category", ""), "data": it.get("pattern", ""),
310
+ "ts": it.get("last_seen", ""), "dev": ",".join(it.get("source_devs", []) or [])}
311
+ for it in items[:limit]]
312
+ except Exception:
313
+ return []
314
+