welsberr
/
Didactopus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support GroundRecall query bundles in draft packs

This commit is contained in:
welsberr 2026-04-27 12:41:29 -04:00
parent 35be8a2f39
commit 2260588b70
2 changed files with 38 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/didactopus/pack_emitter.py
View File

@ -71,8 +71,12 @@ def build_draft_pack(
license_name: str,
review_flags: list[str],
conflicts: list[str] | None = None,
groundrecall_query_bundle: dict | None = None,
) -> DraftPack:
pack_name = course.title.lower().replace(" ", "-")
supporting_artifacts = ["source_corpus.json", "knowledge_graph.json"]
if groundrecall_query_bundle is not None:
supporting_artifacts.append("groundrecall_query_bundle.json")
pack = {
"name": pack_name,
"display_name": course.title,
@ -87,7 +91,7 @@ def build_draft_pack(
"overrides": [],
"profile_templates": {},
"cross_pack_links": [],
"supporting_artifacts": ["source_corpus.json", "knowledge_graph.json"],
"supporting_artifacts": supporting_artifacts,
}
concepts_yaml = {
"concepts": [
@ -134,6 +138,8 @@ def build_draft_pack(
for src in course.source_records
],
}
if groundrecall_query_bundle is not None:
attribution["groundrecall_query_bundle"] = groundrecall_query_bundle
return DraftPack(
pack=pack,
concepts=concepts_yaml,
@ -159,6 +165,11 @@ def write_draft_pack(pack: DraftPack, outdir: str | Path) -> None:
conflict_lines = ["# Conflict Report", ""] + [f"- {flag}" for flag in pack.conflicts] if pack.conflicts else ["# Conflict Report", "", "- none"]
(out / "conflict_report.md").write_text("\n".join(conflict_lines), encoding="utf-8")
(out / "license_attribution.json").write_text(json.dumps(pack.attribution, indent=2), encoding="utf-8")
if isinstance(pack.attribution.get("groundrecall_query_bundle"), dict):
(out / "groundrecall_query_bundle.json").write_text(
json.dumps(pack.attribution["groundrecall_query_bundle"], indent=2),
encoding="utf-8",
)
def write_source_corpus(course: NormalizedCourse, outdir: str | Path) -> None:

26
tests/test_pack_emitter.py
View File

@ -27,3 +27,29 @@ def test_emit_pack(tmp_path: Path) -> None:
assert (tmp_path / "review_report.md").exists()
assert (tmp_path / "source_corpus.json").exists()
assert (tmp_path / "knowledge_graph.json").exists()
def test_emit_pack_can_write_groundrecall_query_bundle(tmp_path: Path) -> None:
course = parse_markdown_course(SAMPLE, "Sample Course")
concepts = extract_concept_candidates(course)
ctx = RuleContext(course=course, concepts=concepts)
run_rules(ctx, build_default_rules())
groundrecall_bundle = {
"bundle_kind": "groundrecall_query_bundle",
"concept": {"concept_id": "concept::topic-a", "title": "Topic A"},
"review_candidates": [{"candidate_id": "concept::topic-a", "rationale": "Topic A | lane=knowledge_capture | priority=20"}],
}
draft = build_draft_pack(
course,
ctx.concepts,
"Tester",
"REVIEW",
ctx.review_flags,
groundrecall_query_bundle=groundrecall_bundle,
)
write_draft_pack(draft, tmp_path)
pack_yaml = (tmp_path / "pack.yaml").read_text(encoding="utf-8")
bundle_payload = (tmp_path / "groundrecall_query_bundle.json").read_text(encoding="utf-8")
assert "groundrecall_query_bundle.json" in pack_yaml
assert '"bundle_kind": "groundrecall_query_bundle"' in bundle_payload