Compare commits
17 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a24408521c | |||
| c091b6c256 | |||
| cbafd30ec7 | |||
| f84908aa36 | |||
| 500152510a | |||
| 0d5bfa9276 | |||
| eb2af77c90 | |||
| eccaa28b1d | |||
| 2101a43b68 | |||
| 9f0872c36a | |||
| d73da7cda9 | |||
| 268520d453 | |||
| 1b8baed542 | |||
| f2b9301fa1 | |||
| a8ba8d4b4a | |||
| 1477dbdd18 | |||
| 6d0a5284e7 |
+58
-30
@@ -56,6 +56,27 @@ def _req(method, path, data=None):
|
||||
sys.exit(1)
|
||||
|
||||
|
||||
def _req_safe(method, path, data=None):
|
||||
"""Like _req but returns None on HTTPError instead of crashing.
|
||||
Used for probing issue/PR existence where the caller can handle absence.
|
||||
"""
|
||||
url = f"{BASE}{path}"
|
||||
payload = json.dumps(data).encode("utf-8") if data else None
|
||||
req = urllib.request.Request(url, data=payload, method=method)
|
||||
req.add_header("Authorization", f"token {GITEA_TOKEN}")
|
||||
req.add_header("Content-Type", "application/json")
|
||||
try:
|
||||
with urllib.request.urlopen(req) as resp:
|
||||
raw = resp.read()
|
||||
if not raw:
|
||||
return {}
|
||||
return json.loads(raw)
|
||||
except urllib.error.HTTPError as e:
|
||||
body = e.read().decode()
|
||||
print(f"API Error {e.code}: {body}", file=sys.stderr)
|
||||
return None
|
||||
|
||||
|
||||
# ── Issue operations ─────────────────────────────────────────────────────────
|
||||
|
||||
def list_issues(labels: list[str] | None = None):
|
||||
@@ -82,17 +103,17 @@ def _get_blocking_refs(issue_num: int) -> set[int]:
|
||||
"""
|
||||
refs: set[int] = set()
|
||||
# Body
|
||||
issue = _req("GET", f"/issues/{issue_num}")
|
||||
issue = _req_safe("GET", f"/issues/{issue_num}")
|
||||
if issue is None:
|
||||
return refs # API error → return empty set, keep blocked
|
||||
body = issue.get("body", "") or ""
|
||||
refs.update(int(m.group(1)) for m in re.finditer(r'#(\d+)', body))
|
||||
# Comments
|
||||
try:
|
||||
comments = _req("GET", f"/issues/{issue_num}/comments")
|
||||
comments = _req_safe("GET", f"/issues/{issue_num}/comments")
|
||||
if comments:
|
||||
for c in comments:
|
||||
cbody = c.get("body", "") or ""
|
||||
refs.update(int(m.group(1)) for m in re.finditer(r'#(\d+)', cbody))
|
||||
except SystemExit:
|
||||
pass
|
||||
return refs
|
||||
|
||||
|
||||
@@ -103,12 +124,7 @@ def blocked_check():
|
||||
If no references found or all referenced issues are closed,
|
||||
removes the 'blocked' label.
|
||||
"""
|
||||
try:
|
||||
all_blocked = _req("GET", "/issues?state=open&labels=blocked")
|
||||
except SystemExit:
|
||||
print("No blocked issues found.")
|
||||
return
|
||||
|
||||
all_blocked = _req_safe("GET", "/issues?state=open&labels=blocked")
|
||||
if not all_blocked:
|
||||
print("No blocked issues found.")
|
||||
return
|
||||
@@ -119,13 +135,13 @@ def blocked_check():
|
||||
|
||||
all_resolved = True
|
||||
for blk in blocking_nums:
|
||||
try:
|
||||
blk_issue = _req("GET", f"/issues/{blk}")
|
||||
if blk_issue.get("state") != "closed":
|
||||
all_resolved = False
|
||||
break
|
||||
except SystemExit:
|
||||
pass
|
||||
blk_issue = _req_safe("GET", f"/issues/{blk}")
|
||||
if blk_issue is None:
|
||||
all_resolved = False # API error → keep blocked
|
||||
break
|
||||
if blk_issue.get("state") != "closed":
|
||||
all_resolved = False
|
||||
break
|
||||
|
||||
if all_resolved:
|
||||
current_label_names = [l["name"] for l in issue.get("labels", [])]
|
||||
@@ -172,6 +188,15 @@ def close_issue(num, body=None):
|
||||
return i
|
||||
|
||||
|
||||
def reopen_issue(num, body=None):
|
||||
"""Reopen a closed issue, optionally with a reason comment."""
|
||||
if body:
|
||||
comment_issue(num, f"## REOPEN\n\n{body}")
|
||||
i = _req("PATCH", f"/issues/{num}", {"state": "open"})
|
||||
print(f"Issue #{num} reopened")
|
||||
return i
|
||||
|
||||
|
||||
def _unblock_issues_blocked_by(closed_num):
|
||||
"""Check issues blocked by *closed_num* and unblock if all blockers resolved.
|
||||
|
||||
@@ -179,10 +204,7 @@ def _unblock_issues_blocked_by(closed_num):
|
||||
in any blocked issue and all referenced issues are now closed,
|
||||
removes the 'blocked' label and comments on the unblocked issue.
|
||||
"""
|
||||
try:
|
||||
all_blocked = _req("GET", "/issues?state=open&labels=blocked")
|
||||
except SystemExit:
|
||||
return
|
||||
all_blocked = _req_safe("GET", "/issues?state=open&labels=blocked")
|
||||
if not all_blocked:
|
||||
return
|
||||
|
||||
@@ -196,13 +218,13 @@ def _unblock_issues_blocked_by(closed_num):
|
||||
for blk in blocking_nums:
|
||||
if blk == closed_num:
|
||||
continue
|
||||
try:
|
||||
blk_issue = _req("GET", f"/issues/{blk}")
|
||||
if blk_issue.get("state") != "closed":
|
||||
all_resolved = False
|
||||
break
|
||||
except SystemExit:
|
||||
pass # Inaccessible → treat as resolved
|
||||
blk_issue = _req_safe("GET", f"/issues/{blk}")
|
||||
if blk_issue is None:
|
||||
all_resolved = False # API error → keep blocked
|
||||
break
|
||||
if blk_issue.get("state") != "closed":
|
||||
all_resolved = False
|
||||
break
|
||||
|
||||
if all_resolved:
|
||||
current_label_names = [l["name"] for l in issue.get("labels", [])]
|
||||
@@ -369,7 +391,8 @@ def main():
|
||||
parser = argparse.ArgumentParser(description="Dev agent Gitea helper")
|
||||
parser.add_argument("--action", required=True,
|
||||
choices=["list", "get", "comment", "close-issue",
|
||||
"create-issue", "create-pr", "pr-status", "merge-pr", "lifecycle",
|
||||
"create-issue", "reopen-issue",
|
||||
"create-pr", "pr-status", "merge-pr", "lifecycle",
|
||||
"blocked-check"])
|
||||
parser.add_argument("--issue", type=int)
|
||||
parser.add_argument("--pr", type=int)
|
||||
@@ -407,6 +430,11 @@ def main():
|
||||
print("--title is required for 'create-issue' action", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
create_issue(args.title, args.body, args.labels)
|
||||
elif args.action == "reopen-issue":
|
||||
if not args.issue:
|
||||
print("--issue is required for 'reopen-issue' action", file=sys.stderr)
|
||||
sys.exit(1)
|
||||
reopen_issue(args.issue, args.body)
|
||||
elif args.action == "create-pr":
|
||||
if not args.issue or not args.branch:
|
||||
print("--issue and --branch are required for 'create-pr' action", file=sys.stderr)
|
||||
|
||||
@@ -880,11 +880,19 @@ def run_ensemble_semantic_index(doc: dict) -> dict:
|
||||
if v:
|
||||
print(f" {k}: {len(v)} 个问题")
|
||||
|
||||
# Feedback retry: re-run with coverage feedback (one retry)
|
||||
feedback = _build_coverage_feedback(gaps)
|
||||
if feedback:
|
||||
print(f"\n 覆盖反馈重试 (feedback长度={len(feedback)}字符)...", flush=True)
|
||||
# Feedback retry: re-run with coverage feedback (up to 2 retries, quality-gated)
|
||||
retry_count = 0
|
||||
while retry_count < 2:
|
||||
feedback = _build_coverage_feedback(gaps)
|
||||
if not feedback:
|
||||
break
|
||||
retry_count += 1
|
||||
print(f"\n 覆盖反馈重试 #{retry_count} (feedback长度={len(feedback)}字符)...", flush=True)
|
||||
try:
|
||||
# record pre-retry coverage to gate quality
|
||||
pre_warnings = len(gaps.get("coverage_warnings", []))
|
||||
pre_missing_rows = len(gaps.get("missing_table_rows", []))
|
||||
|
||||
retry_prompt = build_prompt(doc, feedback, all_paths)
|
||||
print(f" 重试 prompt 长度: {len(retry_prompt)} 字符", flush=True)
|
||||
retry_result = call_llm(retry_prompt, max_retries=1, temperature=0.3)
|
||||
@@ -892,27 +900,39 @@ def run_ensemble_semantic_index(doc: dict) -> dict:
|
||||
n_retry_concepts = len(retry_result.get("concepts", []))
|
||||
print(f" 重试返回: {n_retry_concepts} 概念, {n_retry_units} 功能单元", flush=True)
|
||||
if n_retry_units > 0:
|
||||
# Check which new sections were covered
|
||||
retry_sections = set()
|
||||
for fu in retry_result.get("function_units", []):
|
||||
for src in fu.get("sources", []):
|
||||
if src.get("section"):
|
||||
retry_sections.add(src["section"])
|
||||
print(f" 重试新增 sections: {sorted(retry_sections)}", flush=True)
|
||||
# Merge retry into results and re-validate
|
||||
semantic_indices.append(retry_result)
|
||||
merged = ensemble_merge(semantic_indices)
|
||||
merged["ensemble_temperatures"] = list(temperatures) + ["feedback_retry"]
|
||||
passed, gaps = _quick_validate(merged, doc, all_paths)
|
||||
merged["validation_passed"] = passed
|
||||
merged["validation_gaps"] = {
|
||||
k: v for k, v in gaps.items() if v
|
||||
}
|
||||
print(f" 重试后验证: {'PASS' if passed else 'GAPS FOUND'}", flush=True)
|
||||
# Quality gate: only include retry if it improves coverage
|
||||
trial_indices = semantic_indices + [retry_result]
|
||||
trial_merged = ensemble_merge(trial_indices)
|
||||
trial_passed, trial_gaps = _quick_validate(trial_merged, doc, all_paths)
|
||||
trial_warnings = len(trial_gaps.get("coverage_warnings", []))
|
||||
trial_missing = len(trial_gaps.get("missing_table_rows", []))
|
||||
if trial_warnings < pre_warnings or trial_missing < pre_missing_rows:
|
||||
semantic_indices.append(retry_result)
|
||||
merged = trial_merged
|
||||
passed, gaps = trial_passed, trial_gaps
|
||||
merged["ensemble_temperatures"] = list(temperatures) + [f"feedback_retry_{retry_count}"]
|
||||
merged["validation_passed"] = passed
|
||||
merged["validation_gaps"] = {
|
||||
k: v for k, v in gaps.items() if v
|
||||
}
|
||||
print(f" 重试后验证 (已采纳): {'PASS' if passed else 'GAPS FOUND'} "
|
||||
f"(warnings {pre_warnings}→{trial_warnings}, "
|
||||
f"missing_rows {pre_missing_rows}→{trial_missing})", flush=True)
|
||||
else:
|
||||
print(f" 重试结果未提升覆盖率,丢弃 "
|
||||
f"(warnings {pre_warnings}→{trial_warnings}, "
|
||||
f"missing_rows {pre_missing_rows}→{trial_missing})", flush=True)
|
||||
except Exception as e:
|
||||
print(f" 覆盖反馈重试失败: {e}", flush=True)
|
||||
import traceback
|
||||
traceback.print_exc()
|
||||
break
|
||||
|
||||
return merged
|
||||
|
||||
|
||||
@@ -169,6 +169,40 @@ def _normalize_rule(rule: dict) -> dict:
|
||||
"value": "active"
|
||||
}]
|
||||
|
||||
# Ensure table/text sources have a section field (defensive against LLM omission)
|
||||
# Also normalize invalid source types (LLM hallucinations like function_unit_description)
|
||||
sources = rule.get("sources", [])
|
||||
valid_types = {"table", "text", "logic_tree"}
|
||||
|
||||
# try to infer a default section from the rule path
|
||||
default_section = ""
|
||||
for s in sources:
|
||||
sec = s.get("section", "")
|
||||
if sec and sec.strip():
|
||||
default_section = sec.strip()
|
||||
break
|
||||
if not default_section:
|
||||
path = rule.get("path", "")
|
||||
if path:
|
||||
default_section = path.split(" > ")[0] if " > " in path else path
|
||||
|
||||
if sources:
|
||||
for src in sources:
|
||||
stype = src.get("type", "")
|
||||
if stype and stype not in valid_types:
|
||||
src["type"] = "text"
|
||||
stype = "text"
|
||||
if stype in ("table", "text"):
|
||||
if not src.get("section"):
|
||||
src["section"] = default_section
|
||||
else:
|
||||
# Empty sources list — add a minimal text source (defensive against schema failure)
|
||||
src = {"type": "text", "text_snippet": "inferred from rule context"}
|
||||
if default_section:
|
||||
src["section"] = default_section
|
||||
sources.append(src)
|
||||
rule["sources"] = sources
|
||||
|
||||
return rule
|
||||
|
||||
|
||||
|
||||
@@ -465,3 +465,76 @@ class TestNormalizeRule:
|
||||
normalized = _normalize_rule(rule)
|
||||
assert normalized["trigger"]["operator"] == "AND"
|
||||
assert normalized["trigger"]["conditions"][0]["operator"] == ">="
|
||||
|
||||
def test_normalize_source_missing_section_from_sibling(self):
|
||||
"""Table/text sources without section get it from sibling sources."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"sources": [
|
||||
{"type": "table", "section": "3.1.1 系统限制", "row": 1},
|
||||
{"type": "text", "text_snippet": "missing section"},
|
||||
],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert normalized["sources"][1]["section"] == "3.1.1 系统限制"
|
||||
|
||||
def test_normalize_source_missing_section_from_path(self):
|
||||
"""Table/text sources without section and no sibling fall back to rule path."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"path": "4.2 关闭流程 > decision_speed > action_disable",
|
||||
"sources": [
|
||||
{"type": "table", "row": 3, "text_snippet": "no section anywhere"},
|
||||
],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert normalized["sources"][0]["section"] == "4.2 关闭流程"
|
||||
|
||||
def test_normalize_source_keeps_existing_section(self):
|
||||
"""Sources that already have section are not modified."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"sources": [
|
||||
{"type": "table", "section": "1.0 概述", "row": 1},
|
||||
],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert normalized["sources"][0]["section"] == "1.0 概述"
|
||||
|
||||
def test_normalize_source_skips_logic_tree(self):
|
||||
"""Logic tree sources are not touched (don't need section)."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"sources": [
|
||||
{"type": "logic_tree", "image_id": "img1", "node_ids": ["n1"]},
|
||||
],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert "section" not in normalized["sources"][0]
|
||||
|
||||
def test_normalize_source_invalid_type(self):
|
||||
"""Invalid source types (LLM hallucinations) are normalized to text."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"sources": [
|
||||
{"type": "function_unit_description", "text_snippet": "desc",
|
||||
"section": "3.1 功能"},
|
||||
{"type": "unknown_type", "text_snippet": "also invalid"},
|
||||
],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert normalized["sources"][0]["type"] == "text"
|
||||
assert normalized["sources"][1]["type"] == "text"
|
||||
assert normalized["sources"][0]["section"] == "3.1 功能"
|
||||
|
||||
def test_normalize_empty_sources(self):
|
||||
"""Rules with empty sources get a minimal text source (defensive)."""
|
||||
rule = {
|
||||
"trigger": {"conditions": [{"signal": "x", "operator": "==", "value": "1"}]},
|
||||
"path": "3.1 策略 > decision_speed",
|
||||
"sources": [],
|
||||
}
|
||||
normalized = _normalize_rule(rule)
|
||||
assert len(normalized["sources"]) == 1
|
||||
assert normalized["sources"][0]["type"] == "text"
|
||||
assert normalized["sources"][0]["section"] == "3.1 策略"
|
||||
|
||||
@@ -140,9 +140,19 @@ def ir_path(request) -> str:
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def ir_data(ir_path: str) -> dict:
|
||||
"""Load the IR JSON data."""
|
||||
"""Load the IR JSON data, normalizing each rule for defensive schema fixes."""
|
||||
with open(ir_path, "r", encoding="utf-8") as f:
|
||||
return json.load(f)
|
||||
data = json.load(f)
|
||||
|
||||
# Apply normalize to every rule so old IR files benefit from latest fixes
|
||||
# (invalid source types, missing section fields, trigger nulls, etc.)
|
||||
sys.path.insert(0, str(_PROJECT_ROOT / "skills" / "ir_generation_skill"))
|
||||
from step3_merge_and_audit import _normalize_rule
|
||||
rules = data.get("rules", [])
|
||||
if rules:
|
||||
data["rules"] = [_normalize_rule(r) for r in rules]
|
||||
|
||||
return data
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
|
||||
Reference in New Issue
Block a user