codebee 0.1.101 → 0.1.103

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 (134) hide show
  1. package/CHANGELOG.md +14 -0
  2. package/LICENSE +21 -21
  3. package/README.md +3 -6
  4. package/app/.playwright-cli/console-2026-09-23T08-26-03-635Z.log +88 -0
  5. package/app/.playwright-cli/page-2026-09-23T08-26-03-845Z.yml +126 -0
  6. package/app/.playwright-cli/page-2026-09-23T08-29-44-203Z.png +0 -0
  7. package/app/.playwright-cli/page-2026-09-23T08-35-42-046Z.yml +210 -0
  8. package/app/.playwright-cli/page-2026-09-23T08-36-52-858Z.png +0 -0
  9. package/app/.playwright-cli/page-2026-09-23T08-37-33-938Z.png +0 -0
  10. package/app/.playwright-cli/run-detail-after.png +0 -0
  11. package/app/core/acceptance.py +100 -100
  12. package/app/core/acp_client.py +457 -426
  13. package/app/core/aiflavor.py +197 -197
  14. package/app/core/analytics.py +38 -38
  15. package/app/core/artifacts.py +148 -148
  16. package/app/core/assets.py +110 -110
  17. package/app/core/attachments.py +623 -623
  18. package/app/core/audit.py +30 -30
  19. package/app/core/backend_profiles.py +188 -188
  20. package/app/core/backup.py +470 -470
  21. package/app/core/beekeeper.py +183 -183
  22. package/app/core/benchstore.py +102 -102
  23. package/app/core/board.py +21 -17
  24. package/app/core/bookmeta.py +1036 -1036
  25. package/app/core/bookmeta_catalog.py +180 -180
  26. package/app/core/branching.py +137 -137
  27. package/app/core/builtin_agent.py +13 -4
  28. package/app/core/capability.py +144 -144
  29. package/app/core/catalog.py +28 -6
  30. package/app/core/chaptersafe.py +32 -32
  31. package/app/core/checkpoints.py +117 -31
  32. package/app/core/cleanup.py +303 -303
  33. package/app/core/command_auth.py +302 -302
  34. package/app/core/compaction.py +201 -201
  35. package/app/core/contracts.py +255 -255
  36. package/app/core/covergen.py +356 -356
  37. package/app/core/defectretro.py +43 -43
  38. package/app/core/diagnostics.py +114 -114
  39. package/app/core/dispatch.py +1 -1
  40. package/app/core/dispatch_log.py +128 -128
  41. package/app/core/env_scrub.py +85 -85
  42. package/app/core/errorlog.py +137 -137
  43. package/app/core/eval_matrix.py +91 -82
  44. package/app/core/evaluation.py +174 -174
  45. package/app/core/failure_notify.py +290 -290
  46. package/app/core/flow_graph.py +33 -33
  47. package/app/core/goal_service.py +167 -167
  48. package/app/core/health.py +389 -389
  49. package/app/core/history.py +38 -38
  50. package/app/core/hooks.py +162 -162
  51. package/app/core/knowledge_pipeline.py +86 -86
  52. package/app/core/manager.py +2385 -2385
  53. package/app/core/mcp_client.py +7 -2
  54. package/app/core/mocks.py +64 -64
  55. package/app/core/notify.py +201 -201
  56. package/app/core/novel_quality.py +181 -181
  57. package/app/core/openapi.py +217 -209
  58. package/app/core/operations.py +228 -228
  59. package/app/core/paihang.py +145 -145
  60. package/app/core/pipeline.py +6 -3
  61. package/app/core/policy.py +9 -0
  62. package/app/core/portguard.py +111 -111
  63. package/app/core/presence.py +77 -77
  64. package/app/core/preview.py +293 -293
  65. package/app/core/project_memory.py +206 -206
  66. package/app/core/prompt_guard.py +78 -78
  67. package/app/core/publish/__init__.py +7 -7
  68. package/app/core/publish/auto.py +659 -562
  69. package/app/core/publish/chaprows.py +138 -138
  70. package/app/core/publish/fanqie.py +16 -10
  71. package/app/core/publish/flow.py +29 -1
  72. package/app/core/publish/flows-fanqie-calibrated.json +442 -442
  73. package/app/core/publish/manager.py +1185 -1177
  74. package/app/core/publish/qimao.py +193 -193
  75. package/app/core/publish/volumes.py +108 -108
  76. package/app/core/publish/ws.py +139 -139
  77. package/app/core/quality_gate.py +201 -201
  78. package/app/core/redact.py +38 -38
  79. package/app/core/repeat_guard.py +129 -129
  80. package/app/core/retrieval.py +137 -137
  81. package/app/core/sandbox_runtime.py +95 -0
  82. package/app/core/selfupdate.py +384 -384
  83. package/app/core/session_log.py +161 -161
  84. package/app/core/sessions.py +312 -312
  85. package/app/core/settings.py +215 -215
  86. package/app/core/settings_schema.py +282 -282
  87. package/app/core/share_page.py +114 -114
  88. package/app/core/skill_scan.py +139 -139
  89. package/app/core/skillpacks/market/character-bible.md +66 -66
  90. package/app/core/skillpacks/market/code-risk-checklist.md +58 -58
  91. package/app/core/skillpacks/market/git-workflow.md +57 -57
  92. package/app/core/skillpacks/market/release-notes.md +72 -72
  93. package/app/core/skillpacks/market/weekly-report.md +71 -71
  94. package/app/core/skillpacks/market/worldview-consistency.md +70 -70
  95. package/app/core/step_runner.py +132 -132
  96. package/app/core/store.py +13 -3
  97. package/app/core/story_tracking.py +275 -275
  98. package/app/core/task_compile.py +199 -199
  99. package/app/core/telemetry.py +292 -292
  100. package/app/core/tlsctx.py +64 -64
  101. package/app/core/token_meter.py +147 -147
  102. package/app/core/tool_outputs.py +107 -107
  103. package/app/core/tracing.py +207 -207
  104. package/app/core/upstream.py +41 -41
  105. package/app/core/usage.py +1038 -1038
  106. package/app/core/volumes.py +389 -389
  107. package/app/core/webhooks.py +89 -89
  108. package/app/core/wxdigest.py +865 -865
  109. package/app/core/zentao.py +2692 -2692
  110. package/app/main.py +1 -0
  111. package/app/mcp_server.py +244 -244
  112. package/app/output/_createbook_body.json +1 -0
  113. package/app/output/_probe_fq_click.py +50 -0
  114. package/app/output/_probe_fq_form.py +42 -0
  115. package/app/output/_probe_fq_modal.py +55 -0
  116. package/app/output/_probe_fq_points.py +71 -0
  117. package/app/output/_probe_fq_submit.py +34 -0
  118. package/app/output/_probe_fq_watch.py +62 -0
  119. package/app/output/workbench-clean-preview.jpg +0 -0
  120. package/app/output/workbench-reference-render.png +0 -0
  121. package/app/output/workbench-shell.png +0 -0
  122. package/app/pick_dialog.py +110 -110
  123. package/app/ui/app.js +165 -29
  124. package/app/ui/artdiff.js +74 -74
  125. package/app/ui/hive3d.js +491 -223
  126. package/app/ui/i18n.js +18 -1
  127. package/app/ui/icons/bee.svg +79 -79
  128. package/app/ui/index.html +24 -4
  129. package/app/ui/manifest.json +16 -16
  130. package/app/ui/qrcode.js +2297 -2297
  131. package/app/ui/style.css +548 -18
  132. package/bin/tutti.js +147 -147
  133. package/package.json +1 -1
  134. package/tools/wx_reader.py +261 -261
@@ -1,290 +1,290 @@
1
- # -*- coding: utf-8 -*-
2
- """Human notification for repeated exhaustion of a model candidate chain.
3
-
4
- This module is intentionally separate from ``health``. A provider can be
5
- down while another candidate still completes the task; the alarm here means a
6
- whole candidate set has failed repeatedly and a person should intervene.
7
-
8
- The default Windows channel uses the native PowerShell toast API when running
9
- on Windows. SMTP is available through environment configuration for hosts
10
- where mail is the preferred on-call channel. Sending is best effort: a
11
- notification failure is recorded but never changes the task result.
12
- """
13
- from __future__ import annotations
14
-
15
- import base64
16
- from email.message import EmailMessage
17
- import json
18
- import logging
19
- import os
20
- from pathlib import Path
21
- import re
22
- import smtplib
23
- import subprocess
24
- import threading
25
- import time
26
- import uuid
27
-
28
- log = logging.getLogger(__name__)
29
-
30
- DEFAULT_THRESHOLD = 3
31
- DEFAULT_COOLDOWN_SECONDS = 3600
32
- MAX_SEEN_RUNS = 64
33
- _LOCK = threading.RLock()
34
-
35
- _SECRET_ASSIGNMENT = re.compile(
36
- r"(?i)\b(api[_ -]?key|access[_ -]?token|authorization|bearer|"
37
- r"password|secret|token)\b\s*[:=]\s*[^\s,;]+")
38
- _BEARER_VALUE = re.compile(r"(?i)\bbearer\s+[^\s,;]+")
39
- _SECRET_NAME = re.compile(
40
- r"(?i)\b(?:[A-Za-z0-9_-]*(?:api[_-]?key|secret|password)"
41
- r"[A-Za-z0-9_-]*|[A-Za-z0-9_-]+[_-]token[A-Za-z0-9_-]*)\b")
42
- _TOKEN_PREFIX = re.compile(
43
- r"\b(?:sk|pk|rk|ghp|github_pat|xox[baprs])[-_][A-Za-z0-9_-]{8,}\b")
44
-
45
-
46
- def _state_path() -> Path:
47
- from . import paths
48
- return Path(paths.DATA_DIR) / "failure_notifications.json"
49
-
50
-
51
- def _threshold() -> int:
52
- try:
53
- return max(1, min(20, int(os.environ.get(
54
- "TUTTI_FAILURE_NOTIFY_THRESHOLD", DEFAULT_THRESHOLD))))
55
- except (TypeError, ValueError):
56
- return DEFAULT_THRESHOLD
57
-
58
-
59
- def _cooldown() -> float:
60
- try:
61
- return max(0.0, float(os.environ.get(
62
- "TUTTI_FAILURE_NOTIFY_COOLDOWN_S", DEFAULT_COOLDOWN_SECONDS)))
63
- except (TypeError, ValueError):
64
- return float(DEFAULT_COOLDOWN_SECONDS)
65
-
66
-
67
- def _read() -> dict:
68
- try:
69
- value = json.loads(_state_path().read_text(encoding="utf-8"))
70
- return value if isinstance(value, dict) else {}
71
- except (OSError, json.JSONDecodeError):
72
- return {}
73
-
74
-
75
- def _write(value: dict) -> None:
76
- path = _state_path()
77
- try:
78
- path.parent.mkdir(parents=True, exist_ok=True)
79
- tmp = path.with_suffix(".tmp")
80
- tmp.write_text(json.dumps(value, ensure_ascii=False, indent=2),
81
- encoding="utf-8")
82
- tmp.replace(path)
83
- except OSError as exc:
84
- log.warning("[failure-notify] 无法保存通知状态:%s", exc)
85
-
86
-
87
- def _scope(role: str = "") -> str:
88
- role = " ".join(str(role or "").split())[:80]
89
- return "all-candidates:%s" % (role or "default")
90
-
91
-
92
- def _safe_text(value, limit: int = 120) -> str:
93
- # Metadata is normally safe, but provider/model labels can originate from
94
- # local configuration. Redact common credential forms before placing them
95
- # in a toast, e-mail, or durable state; control characters are removed too.
96
- text = " ".join(str(value or "").replace("\x00", "").split())
97
- text = _SECRET_ASSIGNMENT.sub("[REDACTED]", text)
98
- text = _BEARER_VALUE.sub("Bearer [REDACTED]", text)
99
- text = _SECRET_NAME.sub("[REDACTED]", text)
100
- text = _TOKEN_PREFIX.sub("[REDACTED]", text)
101
- return text[:limit]
102
-
103
-
104
- def _event_from_attempts(*, run_id: str, task_id: str, role: str,
105
- attempts, error_code: str = "", provider: str = "",
106
- model: str = "") -> dict:
107
- attempts = [a for a in (attempts or ()) if isinstance(a, dict)]
108
- first = next((a for a in reversed(attempts) if a), {})
109
- return {
110
- "event_id": uuid.uuid4().hex,
111
- "kind": "all_candidates_failed",
112
- "scope": _scope(role),
113
- "run_id": _safe_text(run_id, 100),
114
- "task_id": _safe_text(task_id, 100),
115
- "role": _safe_text(role, 80),
116
- "consecutive_failures": 0,
117
- "candidate_count": len(attempts),
118
- "error_code": _safe_text(error_code, 48),
119
- "provider": _safe_text(provider or first.get("provider_id") or
120
- first.get("provider"), 100),
121
- "model": _safe_text(model or first.get("model"), 120),
122
- "at": time.time(),
123
- }
124
-
125
-
126
- def _toast_script(title: str, message: str) -> str:
127
- # Base64 keeps user-controlled metadata out of PowerShell source code.
128
- def enc(value: str) -> str:
129
- return base64.b64encode(value.encode("utf-8", "replace")).decode("ascii")
130
- t64, m64 = enc(title), enc(message)
131
- return ("$ErrorActionPreference='Stop';"
132
- "Add-Type -AssemblyName System.Runtime.WindowsRuntime;"
133
- "[Windows.UI.Notifications.ToastNotificationManager,Windows.UI.Notifications,ContentType=WindowsRuntime]>$null;"
134
- "$e=[Text.Encoding]::UTF8;"
135
- "$t=$e.GetString([Convert]::FromBase64String('%s'));"
136
- "$m=$e.GetString([Convert]::FromBase64String('%s'));"
137
- "$x=New-Object Windows.Data.Xml.Dom.XmlDocument;"
138
- "$et=[System.Security.SecurityElement]::Escape($t);"
139
- "$em=[System.Security.SecurityElement]::Escape($m);"
140
- "$x.LoadXml(\"<toast><visual><binding template='ToastGeneric'>"
141
- "<text>$et</text><text>$em</text></binding></visual></toast>\");"
142
- "$n=[Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier('CodeBee');"
143
- "$n.Show((New-Object Windows.UI.Notifications.ToastNotification $x));" %
144
- (t64, m64))
145
-
146
-
147
- def _send_windows_toast(event: dict) -> tuple[bool, str]:
148
- if os.name != "nt":
149
- return False, "windows_unavailable"
150
- if os.environ.get("TUTTI_NOTIFY_WINDOWS_TOAST", "1").strip().lower() in (
151
- "0", "false", "off", "no"):
152
- return False, "windows_disabled"
153
- title = "CodeBee:候选链连续失败"
154
- message = ("连续 %d 轮全候选失败(%s),请检查余额、异上游和配置。"
155
- % (event.get("consecutive_failures") or 0,
156
- event.get("error_code") or "未分类"))
157
- try:
158
- result = subprocess.run(
159
- ["powershell.exe", "-NoProfile", "-NonInteractive", "-Command",
160
- _toast_script(title, message)],
161
- check=False, capture_output=True, text=True, timeout=15)
162
- if result.returncode:
163
- return False, "powershell_exit_%s" % result.returncode
164
- return True, "windows_toast"
165
- except (OSError, subprocess.SubprocessError) as exc:
166
- return False, "windows_error:%s" % type(exc).__name__
167
-
168
-
169
- def _send_email(event: dict) -> tuple[bool, str]:
170
- try:
171
- host = os.environ.get("TUTTI_NOTIFY_SMTP_HOST", "").strip()
172
- recipient = os.environ.get("TUTTI_NOTIFY_EMAIL_TO", "").strip()
173
- if not host or not recipient:
174
- return False, "email_unconfigured"
175
- sender = os.environ.get("TUTTI_NOTIFY_EMAIL_FROM", "codebee@localhost").strip()
176
- port = int(os.environ.get("TUTTI_NOTIFY_SMTP_PORT", "587"))
177
- except ValueError:
178
- port = 587
179
- try:
180
- msg = EmailMessage()
181
- msg["Subject"] = "CodeBee 候选链连续失败告警"
182
- msg["From"] = sender
183
- msg["To"] = recipient
184
- msg.set_content(
185
- "连续 %d 轮全候选失败。\nrun=%s task=%s role=%s code=%s provider=%s model=%s\n"
186
- % (event.get("consecutive_failures") or 0, event.get("run_id") or "",
187
- event.get("task_id") or "", event.get("role") or "",
188
- event.get("error_code") or "", event.get("provider") or "",
189
- event.get("model") or ""))
190
- with smtplib.SMTP(host, port, timeout=15) as smtp:
191
- if os.environ.get("TUTTI_NOTIFY_SMTP_TLS", "1").lower() not in (
192
- "0", "false", "off", "no"):
193
- smtp.starttls()
194
- user = os.environ.get("TUTTI_NOTIFY_SMTP_USER", "")
195
- password = os.environ.get("TUTTI_NOTIFY_SMTP_PASSWORD", "")
196
- if user:
197
- smtp.login(user, password)
198
- smtp.send_message(msg)
199
- return True, "email"
200
- except (OSError, smtplib.SMTPException, ValueError, TypeError) as exc:
201
- return False, "email_error:%s" % type(exc).__name__
202
-
203
-
204
- def send(event: dict) -> dict:
205
- """Send configured channels and return per-channel outcomes."""
206
- outcomes = {}
207
- if os.name == "nt":
208
- outcomes["windows"] = _send_windows_toast(event)
209
- if os.environ.get("TUTTI_NOTIFY_SMTP_HOST") or os.environ.get("TUTTI_NOTIFY_EMAIL_TO"):
210
- outcomes["email"] = _send_email(event)
211
- if not outcomes:
212
- outcomes["none"] = (False, "no_channel_configured")
213
- return {name: {"sent": bool(result[0]), "detail": result[1]}
214
- for name, result in outcomes.items()}
215
-
216
-
217
- def record_success(*, role: str = "") -> None:
218
- """Reset the consecutive-failure episode after a successful candidate call."""
219
- key = _scope(role)
220
- with _LOCK:
221
- state = _read()
222
- item = state.get(key) or {}
223
- if item.get("consecutive_failures") or item.get("episode_notified"):
224
- item.update({"consecutive_failures": 0, "episode_notified": False,
225
- "seen_runs": []})
226
- state[key] = item
227
- _write(state)
228
-
229
-
230
- def record_all_candidates_failed(*, run_id: str, task_id: str = "", role: str = "",
231
- attempts=(), error_code: str = "", provider: str = "",
232
- model: str = "") -> dict:
233
- """Record one exhausted chain and notify after the configured threshold.
234
-
235
- Duplicate observations for the same run are ignored. The return value is
236
- useful to tests and diagnostics and contains no prompt or secret material.
237
- """
238
- attempts = [a for a in (attempts or ()) if isinstance(a, dict)]
239
- if not attempts or any(a.get("ok") for a in attempts):
240
- return {"recorded": False, "reason": "chain_not_exhausted"}
241
- key = _scope(role)
242
- now = time.time()
243
- event = _event_from_attempts(run_id=run_id, task_id=task_id, role=role,
244
- attempts=attempts, error_code=error_code,
245
- provider=provider, model=model)
246
- with _LOCK:
247
- state = _read()
248
- item = state.get(key) or {"consecutive_failures": 0, "seen_runs": []}
249
- seen = list(item.get("seen_runs") or [])
250
- if run_id and run_id in seen:
251
- return {"recorded": False, "reason": "duplicate_run",
252
- "consecutive_failures": int(item.get("consecutive_failures") or 0)}
253
- item["consecutive_failures"] = int(item.get("consecutive_failures") or 0) + 1
254
- if run_id:
255
- seen.append(run_id)
256
- item["seen_runs"] = seen[-MAX_SEEN_RUNS:]
257
- event["consecutive_failures"] = item["consecutive_failures"]
258
- should_notify = (item["consecutive_failures"] >= _threshold()
259
- and not item.get("episode_notified")
260
- and now - float(item.get("last_notified_at") or 0) >= _cooldown())
261
- if should_notify:
262
- item["episode_notified"] = True
263
- item["last_notified_at"] = now
264
- state[key] = item
265
- _write(state)
266
- outcomes = send(event) if should_notify else {}
267
- if should_notify:
268
- with _LOCK:
269
- state = _read()
270
- item = state.get(key) or {}
271
- item["last_outcomes"] = outcomes
272
- # A configured channel can fail transiently. Only close the
273
- # episode after at least one channel confirms delivery; otherwise
274
- # the cooldown permits a later exhausted run to retry the alert.
275
- item["episode_notified"] = any(
276
- bool(value.get("sent")) for value in outcomes.values()
277
- if isinstance(value, dict))
278
- state[key] = item
279
- _write(state)
280
- return {"recorded": True, "notified": should_notify,
281
- "consecutive_failures": event["consecutive_failures"],
282
- "outcomes": outcomes}
283
-
284
-
285
- def reset_for_tests() -> None:
286
- with _LOCK:
287
- try:
288
- _state_path().unlink()
289
- except OSError:
290
- pass
1
+ # -*- coding: utf-8 -*-
2
+ """Human notification for repeated exhaustion of a model candidate chain.
3
+
4
+ This module is intentionally separate from ``health``. A provider can be
5
+ down while another candidate still completes the task; the alarm here means a
6
+ whole candidate set has failed repeatedly and a person should intervene.
7
+
8
+ The default Windows channel uses the native PowerShell toast API when running
9
+ on Windows. SMTP is available through environment configuration for hosts
10
+ where mail is the preferred on-call channel. Sending is best effort: a
11
+ notification failure is recorded but never changes the task result.
12
+ """
13
+ from __future__ import annotations
14
+
15
+ import base64
16
+ from email.message import EmailMessage
17
+ import json
18
+ import logging
19
+ import os
20
+ from pathlib import Path
21
+ import re
22
+ import smtplib
23
+ import subprocess
24
+ import threading
25
+ import time
26
+ import uuid
27
+
28
+ log = logging.getLogger(__name__)
29
+
30
+ DEFAULT_THRESHOLD = 3
31
+ DEFAULT_COOLDOWN_SECONDS = 3600
32
+ MAX_SEEN_RUNS = 64
33
+ _LOCK = threading.RLock()
34
+
35
+ _SECRET_ASSIGNMENT = re.compile(
36
+ r"(?i)\b(api[_ -]?key|access[_ -]?token|authorization|bearer|"
37
+ r"password|secret|token)\b\s*[:=]\s*[^\s,;]+")
38
+ _BEARER_VALUE = re.compile(r"(?i)\bbearer\s+[^\s,;]+")
39
+ _SECRET_NAME = re.compile(
40
+ r"(?i)\b(?:[A-Za-z0-9_-]*(?:api[_-]?key|secret|password)"
41
+ r"[A-Za-z0-9_-]*|[A-Za-z0-9_-]+[_-]token[A-Za-z0-9_-]*)\b")
42
+ _TOKEN_PREFIX = re.compile(
43
+ r"\b(?:sk|pk|rk|ghp|github_pat|xox[baprs])[-_][A-Za-z0-9_-]{8,}\b")
44
+
45
+
46
+ def _state_path() -> Path:
47
+ from . import paths
48
+ return Path(paths.DATA_DIR) / "failure_notifications.json"
49
+
50
+
51
+ def _threshold() -> int:
52
+ try:
53
+ return max(1, min(20, int(os.environ.get(
54
+ "TUTTI_FAILURE_NOTIFY_THRESHOLD", DEFAULT_THRESHOLD))))
55
+ except (TypeError, ValueError):
56
+ return DEFAULT_THRESHOLD
57
+
58
+
59
+ def _cooldown() -> float:
60
+ try:
61
+ return max(0.0, float(os.environ.get(
62
+ "TUTTI_FAILURE_NOTIFY_COOLDOWN_S", DEFAULT_COOLDOWN_SECONDS)))
63
+ except (TypeError, ValueError):
64
+ return float(DEFAULT_COOLDOWN_SECONDS)
65
+
66
+
67
+ def _read() -> dict:
68
+ try:
69
+ value = json.loads(_state_path().read_text(encoding="utf-8"))
70
+ return value if isinstance(value, dict) else {}
71
+ except (OSError, json.JSONDecodeError):
72
+ return {}
73
+
74
+
75
+ def _write(value: dict) -> None:
76
+ path = _state_path()
77
+ try:
78
+ path.parent.mkdir(parents=True, exist_ok=True)
79
+ tmp = path.with_suffix(".tmp")
80
+ tmp.write_text(json.dumps(value, ensure_ascii=False, indent=2),
81
+ encoding="utf-8")
82
+ tmp.replace(path)
83
+ except OSError as exc:
84
+ log.warning("[failure-notify] 无法保存通知状态:%s", exc)
85
+
86
+
87
+ def _scope(role: str = "") -> str:
88
+ role = " ".join(str(role or "").split())[:80]
89
+ return "all-candidates:%s" % (role or "default")
90
+
91
+
92
+ def _safe_text(value, limit: int = 120) -> str:
93
+ # Metadata is normally safe, but provider/model labels can originate from
94
+ # local configuration. Redact common credential forms before placing them
95
+ # in a toast, e-mail, or durable state; control characters are removed too.
96
+ text = " ".join(str(value or "").replace("\x00", "").split())
97
+ text = _SECRET_ASSIGNMENT.sub("[REDACTED]", text)
98
+ text = _BEARER_VALUE.sub("Bearer [REDACTED]", text)
99
+ text = _SECRET_NAME.sub("[REDACTED]", text)
100
+ text = _TOKEN_PREFIX.sub("[REDACTED]", text)
101
+ return text[:limit]
102
+
103
+
104
+ def _event_from_attempts(*, run_id: str, task_id: str, role: str,
105
+ attempts, error_code: str = "", provider: str = "",
106
+ model: str = "") -> dict:
107
+ attempts = [a for a in (attempts or ()) if isinstance(a, dict)]
108
+ first = next((a for a in reversed(attempts) if a), {})
109
+ return {
110
+ "event_id": uuid.uuid4().hex,
111
+ "kind": "all_candidates_failed",
112
+ "scope": _scope(role),
113
+ "run_id": _safe_text(run_id, 100),
114
+ "task_id": _safe_text(task_id, 100),
115
+ "role": _safe_text(role, 80),
116
+ "consecutive_failures": 0,
117
+ "candidate_count": len(attempts),
118
+ "error_code": _safe_text(error_code, 48),
119
+ "provider": _safe_text(provider or first.get("provider_id") or
120
+ first.get("provider"), 100),
121
+ "model": _safe_text(model or first.get("model"), 120),
122
+ "at": time.time(),
123
+ }
124
+
125
+
126
+ def _toast_script(title: str, message: str) -> str:
127
+ # Base64 keeps user-controlled metadata out of PowerShell source code.
128
+ def enc(value: str) -> str:
129
+ return base64.b64encode(value.encode("utf-8", "replace")).decode("ascii")
130
+ t64, m64 = enc(title), enc(message)
131
+ return ("$ErrorActionPreference='Stop';"
132
+ "Add-Type -AssemblyName System.Runtime.WindowsRuntime;"
133
+ "[Windows.UI.Notifications.ToastNotificationManager,Windows.UI.Notifications,ContentType=WindowsRuntime]>$null;"
134
+ "$e=[Text.Encoding]::UTF8;"
135
+ "$t=$e.GetString([Convert]::FromBase64String('%s'));"
136
+ "$m=$e.GetString([Convert]::FromBase64String('%s'));"
137
+ "$x=New-Object Windows.Data.Xml.Dom.XmlDocument;"
138
+ "$et=[System.Security.SecurityElement]::Escape($t);"
139
+ "$em=[System.Security.SecurityElement]::Escape($m);"
140
+ "$x.LoadXml(\"<toast><visual><binding template='ToastGeneric'>"
141
+ "<text>$et</text><text>$em</text></binding></visual></toast>\");"
142
+ "$n=[Windows.UI.Notifications.ToastNotificationManager]::CreateToastNotifier('CodeBee');"
143
+ "$n.Show((New-Object Windows.UI.Notifications.ToastNotification $x));" %
144
+ (t64, m64))
145
+
146
+
147
+ def _send_windows_toast(event: dict) -> tuple[bool, str]:
148
+ if os.name != "nt":
149
+ return False, "windows_unavailable"
150
+ if os.environ.get("TUTTI_NOTIFY_WINDOWS_TOAST", "1").strip().lower() in (
151
+ "0", "false", "off", "no"):
152
+ return False, "windows_disabled"
153
+ title = "CodeBee:候选链连续失败"
154
+ message = ("连续 %d 轮全候选失败(%s),请检查余额、异上游和配置。"
155
+ % (event.get("consecutive_failures") or 0,
156
+ event.get("error_code") or "未分类"))
157
+ try:
158
+ result = subprocess.run(
159
+ ["powershell.exe", "-NoProfile", "-NonInteractive", "-Command",
160
+ _toast_script(title, message)],
161
+ check=False, capture_output=True, text=True, timeout=15)
162
+ if result.returncode:
163
+ return False, "powershell_exit_%s" % result.returncode
164
+ return True, "windows_toast"
165
+ except (OSError, subprocess.SubprocessError) as exc:
166
+ return False, "windows_error:%s" % type(exc).__name__
167
+
168
+
169
+ def _send_email(event: dict) -> tuple[bool, str]:
170
+ try:
171
+ host = os.environ.get("TUTTI_NOTIFY_SMTP_HOST", "").strip()
172
+ recipient = os.environ.get("TUTTI_NOTIFY_EMAIL_TO", "").strip()
173
+ if not host or not recipient:
174
+ return False, "email_unconfigured"
175
+ sender = os.environ.get("TUTTI_NOTIFY_EMAIL_FROM", "codebee@localhost").strip()
176
+ port = int(os.environ.get("TUTTI_NOTIFY_SMTP_PORT", "587"))
177
+ except ValueError:
178
+ port = 587
179
+ try:
180
+ msg = EmailMessage()
181
+ msg["Subject"] = "CodeBee 候选链连续失败告警"
182
+ msg["From"] = sender
183
+ msg["To"] = recipient
184
+ msg.set_content(
185
+ "连续 %d 轮全候选失败。\nrun=%s task=%s role=%s code=%s provider=%s model=%s\n"
186
+ % (event.get("consecutive_failures") or 0, event.get("run_id") or "",
187
+ event.get("task_id") or "", event.get("role") or "",
188
+ event.get("error_code") or "", event.get("provider") or "",
189
+ event.get("model") or ""))
190
+ with smtplib.SMTP(host, port, timeout=15) as smtp:
191
+ if os.environ.get("TUTTI_NOTIFY_SMTP_TLS", "1").lower() not in (
192
+ "0", "false", "off", "no"):
193
+ smtp.starttls()
194
+ user = os.environ.get("TUTTI_NOTIFY_SMTP_USER", "")
195
+ password = os.environ.get("TUTTI_NOTIFY_SMTP_PASSWORD", "")
196
+ if user:
197
+ smtp.login(user, password)
198
+ smtp.send_message(msg)
199
+ return True, "email"
200
+ except (OSError, smtplib.SMTPException, ValueError, TypeError) as exc:
201
+ return False, "email_error:%s" % type(exc).__name__
202
+
203
+
204
+ def send(event: dict) -> dict:
205
+ """Send configured channels and return per-channel outcomes."""
206
+ outcomes = {}
207
+ if os.name == "nt":
208
+ outcomes["windows"] = _send_windows_toast(event)
209
+ if os.environ.get("TUTTI_NOTIFY_SMTP_HOST") or os.environ.get("TUTTI_NOTIFY_EMAIL_TO"):
210
+ outcomes["email"] = _send_email(event)
211
+ if not outcomes:
212
+ outcomes["none"] = (False, "no_channel_configured")
213
+ return {name: {"sent": bool(result[0]), "detail": result[1]}
214
+ for name, result in outcomes.items()}
215
+
216
+
217
+ def record_success(*, role: str = "") -> None:
218
+ """Reset the consecutive-failure episode after a successful candidate call."""
219
+ key = _scope(role)
220
+ with _LOCK:
221
+ state = _read()
222
+ item = state.get(key) or {}
223
+ if item.get("consecutive_failures") or item.get("episode_notified"):
224
+ item.update({"consecutive_failures": 0, "episode_notified": False,
225
+ "seen_runs": []})
226
+ state[key] = item
227
+ _write(state)
228
+
229
+
230
+ def record_all_candidates_failed(*, run_id: str, task_id: str = "", role: str = "",
231
+ attempts=(), error_code: str = "", provider: str = "",
232
+ model: str = "") -> dict:
233
+ """Record one exhausted chain and notify after the configured threshold.
234
+
235
+ Duplicate observations for the same run are ignored. The return value is
236
+ useful to tests and diagnostics and contains no prompt or secret material.
237
+ """
238
+ attempts = [a for a in (attempts or ()) if isinstance(a, dict)]
239
+ if not attempts or any(a.get("ok") for a in attempts):
240
+ return {"recorded": False, "reason": "chain_not_exhausted"}
241
+ key = _scope(role)
242
+ now = time.time()
243
+ event = _event_from_attempts(run_id=run_id, task_id=task_id, role=role,
244
+ attempts=attempts, error_code=error_code,
245
+ provider=provider, model=model)
246
+ with _LOCK:
247
+ state = _read()
248
+ item = state.get(key) or {"consecutive_failures": 0, "seen_runs": []}
249
+ seen = list(item.get("seen_runs") or [])
250
+ if run_id and run_id in seen:
251
+ return {"recorded": False, "reason": "duplicate_run",
252
+ "consecutive_failures": int(item.get("consecutive_failures") or 0)}
253
+ item["consecutive_failures"] = int(item.get("consecutive_failures") or 0) + 1
254
+ if run_id:
255
+ seen.append(run_id)
256
+ item["seen_runs"] = seen[-MAX_SEEN_RUNS:]
257
+ event["consecutive_failures"] = item["consecutive_failures"]
258
+ should_notify = (item["consecutive_failures"] >= _threshold()
259
+ and not item.get("episode_notified")
260
+ and now - float(item.get("last_notified_at") or 0) >= _cooldown())
261
+ if should_notify:
262
+ item["episode_notified"] = True
263
+ item["last_notified_at"] = now
264
+ state[key] = item
265
+ _write(state)
266
+ outcomes = send(event) if should_notify else {}
267
+ if should_notify:
268
+ with _LOCK:
269
+ state = _read()
270
+ item = state.get(key) or {}
271
+ item["last_outcomes"] = outcomes
272
+ # A configured channel can fail transiently. Only close the
273
+ # episode after at least one channel confirms delivery; otherwise
274
+ # the cooldown permits a later exhausted run to retry the alert.
275
+ item["episode_notified"] = any(
276
+ bool(value.get("sent")) for value in outcomes.values()
277
+ if isinstance(value, dict))
278
+ state[key] = item
279
+ _write(state)
280
+ return {"recorded": True, "notified": should_notify,
281
+ "consecutive_failures": event["consecutive_failures"],
282
+ "outcomes": outcomes}
283
+
284
+
285
+ def reset_for_tests() -> None:
286
+ with _LOCK:
287
+ try:
288
+ _state_path().unlink()
289
+ except OSError:
290
+ pass
@@ -1,33 +1,33 @@
1
- # -*- coding: utf-8 -*-
2
- """Stable graph projection for the existing flow/version registry."""
3
- from __future__ import annotations
4
-
5
- import hashlib
6
- import json
7
-
8
- from . import flows
9
-
10
-
11
- def build(flow_id):
12
- flow = flows.get_flow(str(flow_id or ""))
13
- if not flow:
14
- raise ValueError("流程不存在")
15
- engine = flow.get("engine") or "review"
16
- if engine == "code":
17
- names = [("plan", "规划"), ("implement", "实现"), ("verify", "验证"),
18
- ("review", "评审"), ("repair", "修复")]
19
- elif engine == "direct":
20
- names = [("input", "输入"), ("execute", "执行"), ("output", "产出")]
21
- else:
22
- names = [("outline", "大纲"), ("draft", "起草"), ("review", "评审"),
23
- ("revise", "修订"), ("publish", "发布门禁")]
24
- nodes = [{"id": key, "label": label, "kind": "step", "order": i}
25
- for i, (key, label) in enumerate(names)]
26
- edges = [{"from": names[i][0], "to": names[i + 1][0]} for i in range(len(names) - 1)]
27
- digest = hashlib.sha256(json.dumps({"flow": flow, "nodes": nodes, "edges": edges},
28
- ensure_ascii=False, sort_keys=True, default=str).encode("utf-8")).hexdigest()
29
- return {"flow_id": flow.get("id"), "name": flow.get("name") or flow.get("id"),
30
- "engine": engine, "version_id": "flowgraph-" + digest[:16],
31
- "nodes": nodes, "edges": edges,
32
- "summary": {"threshold": flow.get("threshold"), "rounds": flow.get("rounds"),
33
- "builtin": bool(flow.get("builtin")), "edited": bool(flow.get("edited"))}}
1
+ # -*- coding: utf-8 -*-
2
+ """Stable graph projection for the existing flow/version registry."""
3
+ from __future__ import annotations
4
+
5
+ import hashlib
6
+ import json
7
+
8
+ from . import flows
9
+
10
+
11
+ def build(flow_id):
12
+ flow = flows.get_flow(str(flow_id or ""))
13
+ if not flow:
14
+ raise ValueError("流程不存在")
15
+ engine = flow.get("engine") or "review"
16
+ if engine == "code":
17
+ names = [("plan", "规划"), ("implement", "实现"), ("verify", "验证"),
18
+ ("review", "评审"), ("repair", "修复")]
19
+ elif engine == "direct":
20
+ names = [("input", "输入"), ("execute", "执行"), ("output", "产出")]
21
+ else:
22
+ names = [("outline", "大纲"), ("draft", "起草"), ("review", "评审"),
23
+ ("revise", "修订"), ("publish", "发布门禁")]
24
+ nodes = [{"id": key, "label": label, "kind": "step", "order": i}
25
+ for i, (key, label) in enumerate(names)]
26
+ edges = [{"from": names[i][0], "to": names[i + 1][0]} for i in range(len(names) - 1)]
27
+ digest = hashlib.sha256(json.dumps({"flow": flow, "nodes": nodes, "edges": edges},
28
+ ensure_ascii=False, sort_keys=True, default=str).encode("utf-8")).hexdigest()
29
+ return {"flow_id": flow.get("id"), "name": flow.get("name") or flow.get("id"),
30
+ "engine": engine, "version_id": "flowgraph-" + digest[:16],
31
+ "nodes": nodes, "edges": edges,
32
+ "summary": {"threshold": flow.get("threshold"), "rounds": flow.get("rounds"),
33
+ "builtin": bool(flow.get("builtin")), "edited": bool(flow.get("edited"))}}