codebee 0.1.71 → 0.1.73
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.
- package/CHANGELOG.md +601 -589
- package/LICENSE +21 -21
- package/README.md +508 -509
- package/app/core/acceptance.py +100 -100
- package/app/core/aiflavor.py +197 -197
- package/app/core/artifacts.py +148 -148
- package/app/core/assets.py +110 -110
- package/app/core/attachments.py +623 -623
- package/app/core/audit.py +30 -30
- package/app/core/backup.py +470 -470
- package/app/core/beekeeper.py +183 -183
- package/app/core/benchstore.py +95 -95
- package/app/core/board.py +209 -209
- package/app/core/bookmeta.py +935 -935
- package/app/core/bookmeta_catalog.py +180 -180
- package/app/core/branching.py +137 -137
- package/app/core/builtin_agent.py +1680 -1673
- package/app/core/capability.py +144 -144
- package/app/core/chaptersafe.py +32 -32
- package/app/core/cleanup.py +303 -303
- package/app/core/compaction.py +201 -201
- package/app/core/covergen.py +356 -356
- package/app/core/defectretro.py +43 -43
- package/app/core/diagnostics.py +114 -114
- package/app/core/dispatch.py +190 -190
- package/app/core/dispatch_log.py +128 -128
- package/app/core/env_scrub.py +85 -85
- package/app/core/errorlog.py +137 -137
- package/app/core/evalbench.py +836 -720
- package/app/core/evaluation.py +174 -174
- package/app/core/failure_notify.py +290 -290
- package/app/core/flows.py +8 -8
- package/app/core/goal_service.py +167 -167
- package/app/core/health.py +374 -374
- package/app/core/history.py +38 -38
- package/app/core/hooks.py +162 -162
- package/app/core/jobs.py +25 -1
- package/app/core/manager.py +2208 -2143
- package/app/core/market.py +90 -4
- package/app/core/market_remote.py +16 -0
- package/app/core/mocks.py +64 -64
- package/app/core/modelhub.py +3 -2
- package/app/core/notify.py +201 -201
- package/app/core/novel_quality.py +189 -189
- package/app/core/operations.py +228 -228
- package/app/core/paihang.py +145 -138
- package/app/core/pipeline.py +102 -30
- package/app/core/planner.py +2 -0
- package/app/core/plugins.py +34 -0
- package/app/core/portguard.py +111 -111
- package/app/core/preview.py +293 -293
- package/app/core/prompt_guard.py +78 -0
- package/app/core/publish/__init__.py +7 -7
- package/app/core/publish/auto.py +396 -367
- package/app/core/publish/browser.py +655 -655
- package/app/core/publish/fanqie.py +165 -165
- package/app/core/publish/flow.py +640 -589
- package/app/core/publish/flows-fanqie-calibrated.json +336 -336
- package/app/core/publish/ledger.py +307 -229
- package/app/core/publish/manager.py +1118 -824
- package/app/core/publish/qimao.py +176 -146
- package/app/core/publish/ws.py +139 -139
- package/app/core/quality_gate.py +194 -0
- package/app/core/redact.py +38 -38
- package/app/core/repeat_guard.py +129 -129
- package/app/core/router.py +398 -398
- package/app/core/runner.py +2142 -2142
- package/app/core/selfupdate.py +384 -384
- package/app/core/session_log.py +161 -161
- package/app/core/sessions.py +312 -312
- package/app/core/settings_schema.py +270 -270
- package/app/core/share_page.py +114 -114
- package/app/core/skill_scan.py +139 -122
- package/app/core/skillpacks/fanqie-novel.md +98 -80
- package/app/core/skillpacks/market/character-bible.md +66 -66
- package/app/core/skillpacks/market/code-risk-checklist.md +58 -58
- package/app/core/skillpacks/market/git-workflow.md +57 -57
- package/app/core/skillpacks/market/release-notes.md +72 -72
- package/app/core/skillpacks/market/weekly-report.md +71 -71
- package/app/core/skillpacks/market/worldview-consistency.md +70 -70
- package/app/core/skillpacks/qimao-signing.md +70 -59
- package/app/core/skills.py +58 -8
- package/app/core/step_runner.py +132 -132
- package/app/core/store.py +51 -0
- package/app/core/task_compile.py +199 -199
- package/app/core/telemetry.py +292 -292
- package/app/core/tlsctx.py +64 -64
- package/app/core/token_meter.py +147 -147
- package/app/core/tracing.py +144 -144
- package/app/core/upstream.py +41 -41
- package/app/core/usage.py +1035 -1035
- package/app/core/volumes.py +389 -389
- package/app/core/wxdigest.py +858 -858
- package/app/core/zentao.py +2692 -2614
- package/app/main.py +110 -39
- package/app/mcp_server.py +175 -175
- package/app/pick_dialog.py +110 -110
- package/app/ui/app.js +509 -114
- package/app/ui/artdiff.js +74 -74
- package/app/ui/board.html +162 -162
- package/app/ui/board.js +305 -305
- package/app/ui/i18n.js +41 -2
- package/app/ui/icons/bee.svg +79 -79
- package/app/ui/index.html +8 -3
- package/app/ui/manifest.json +16 -16
- package/app/ui/qrcode.js +2297 -2297
- package/app/ui/style.css +74 -26
- package/bin/tutti.js +147 -147
- package/package.json +48 -47
- package/tools/wx_reader.py +261 -261
- package/app/.playwright-cli/console-2026-09-23T08-26-03-635Z.log +0 -88
- package/app/.playwright-cli/page-2026-09-23T08-26-03-845Z.yml +0 -126
- package/app/.playwright-cli/page-2026-09-23T08-29-44-203Z.png +0 -0
- package/app/.playwright-cli/page-2026-09-23T08-35-42-046Z.yml +0 -210
- package/app/.playwright-cli/page-2026-09-23T08-36-52-858Z.png +0 -0
- package/app/.playwright-cli/page-2026-09-23T08-37-33-938Z.png +0 -0
- package/app/.playwright-cli/run-detail-after.png +0 -0
- package/app//344/275/234/345/223/201/344/277/241/346/201/257-/344/270/203/347/214/253.md +0 -18
|
@@ -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
|
package/app/core/flows.py
CHANGED
|
@@ -40,19 +40,19 @@ BUILTIN_FLOWS = [
|
|
|
40
40
|
{"id": "code", "name": "代码", "icon": "i-code", "engine": "code", "builtin": True,
|
|
41
41
|
"goal_hint": "要实现/修复什么(一句话)",
|
|
42
42
|
"note": "按难度动态编排:简单任务走实现+验证短链,复杂任务增加规划、评审、修复与换将"},
|
|
43
|
-
{"id": "novel", "name": "小说", "icon": "i-book-open", "engine": "review", "builtin": True,
|
|
44
|
-
"manuscript": "manuscript.md",
|
|
45
|
-
"rubric": ["情节", "人物", "文笔", "节奏", "吸引力"],
|
|
43
|
+
{"id": "novel", "name": "小说", "icon": "i-book-open", "engine": "review", "builtin": True,
|
|
44
|
+
"manuscript": "manuscript.md",
|
|
45
|
+
"rubric": ["情节", "人物", "文笔", "节奏", "吸引力"],
|
|
46
46
|
"threshold": 7.0, "rounds": 2,
|
|
47
47
|
"goal_hint": "写什么(题材 / 篇幅 / 风格)",
|
|
48
48
|
"note": "起草 → 多维评审 → 修订循环 → 发布门禁"},
|
|
49
|
-
{"id": "serial_novel", "name": "连载小说", "icon": "i-library", "engine": "review", "builtin": True,
|
|
50
|
-
"manuscript": "manuscript.md",
|
|
51
|
-
"rubric": ["情节", "人物", "文笔", "节奏", "吸引力"],
|
|
49
|
+
{"id": "serial_novel", "name": "连载小说", "icon": "i-library", "engine": "review", "builtin": True,
|
|
50
|
+
"manuscript": "manuscript.md",
|
|
51
|
+
"rubric": ["情节", "人物", "文笔", "节奏", "吸引力"],
|
|
52
52
|
"threshold": 7.0, "rounds": 2,
|
|
53
53
|
"serial": {"chapters": 8, "words_per_chapter": 2500},
|
|
54
54
|
"goal_hint": "题材/受众/卖点 + 总字数(例:2 万字都市女频,可签约平台)",
|
|
55
|
-
"note": "大纲 → 逐章起草 → 签约质量门禁(开篇/推进/文风/情感/节奏)→ 全局一致性评审 → 合并(可断点续跑)"},
|
|
55
|
+
"note": "大纲 → 逐章起草 → 签约质量门禁(开篇/推进/文风/情感/节奏)→ 全局一致性评审 → 合并(可断点续跑)"},
|
|
56
56
|
{"id": "article", "name": "自媒体文章", "icon": "i-news", "engine": "review", "builtin": True,
|
|
57
57
|
"manuscript": "article.md",
|
|
58
58
|
"rubric": ["选题与标题", "开头吸引力", "结构节奏", "平台适配", "传播性"],
|
|
@@ -79,7 +79,7 @@ BUILTIN_FLOWS = [
|
|
|
79
79
|
"note": "起草 → 多维评审 → 修订循环 → 发布门禁"},
|
|
80
80
|
{"id": "rank_scan", "name": "扫榜选材", "icon": "i-chart", "engine": "direct", "builtin": True,
|
|
81
81
|
"goal_hint": "想写哪个方向(一句话,可留空默认分析总榜热门题材)",
|
|
82
|
-
"note": "抓取七猫/番茄/起点/纵横四平台排行榜公开数据 → AI
|
|
82
|
+
"note": "抓取七猫/番茄/起点/纵横四平台排行榜公开数据 → AI 提炼跨平台热门题材/人设/差异化切入点,每条附 A-E 跨平台证据分级(快档直出报告)"},
|
|
83
83
|
{"id": "defect_retro", "name": "缺陷复盘", "icon": "i-history", "engine": "direct", "builtin": True,
|
|
84
84
|
"goal_hint": "复盘哪个版本/时间段的缺陷(建议附带禅道或 Jira 导出 CSV)",
|
|
85
85
|
"note": "读取附带的缺陷导出数据 → AI 输出产品/开发/测试三视角复盘报告:整体画像、根因聚类、漏测分析、改进动作(快档直出报告)"},
|