v0.20 alpha skill-driven python core

tests/test_search_grounded_packets.py

@@ -0,0 +1,115 @@
+from __future__ import annotations
+
+import json
+import sys
+from pathlib import Path
+
+REPO_ROOT = Path(__file__).resolve().parents[1]
+if str(REPO_ROOT) not in sys.path:
+    sys.path.insert(0, str(REPO_ROOT))
+
+from scripts.runtime.roles import resolve_runtime_profile
+from scripts.runtime.sources import append_packet_sources, rebuild_sources_from_packets
+from scripts.runtime.tasks import TaskCard
+from scripts.runtime.workers import PacketWorker, build_search_context
+
+
+class FakeSearchProvider:
+    def search(self, *, query: str, route: str, num_results: int):
+        return [
+            {
+                "title": f"{route} result for {query}",
+                "url": f"https://example.com/{route}",
+                "snippet": "候选证据摘要",
+                "route": route,
+            }
+        ][:num_results]
+
+
+class FakeClient:
+    def __init__(self, response: dict) -> None:
+        self.response = response
+        self.calls: list[dict[str, object]] = []
+
+    def chat_complete(self, **kwargs) -> str:
+        self.calls.append(kwargs)
+        return json.dumps(self.response, ensure_ascii=False)
+
+
+def sample_card() -> TaskCard:
+    return TaskCard(
+        task_id="ch01-literature",
+        chapter_ids=["ch01"],
+        topic_axis="literature",
+        questions=["围绕临床证据提炼结论。"],
+        search_routes=["scholar", "general"],
+        output_packet="phase2/packets/ch01-literature.json",
+    )
+
+
+def test_build_search_context_assigns_stable_source_ids() -> None:
+    context = build_search_context(sample_card(), FakeSearchProvider(), num_results_per_route=1)
+
+    assert [source["id"] for source in context["candidate_sources"]] == [
+        "src_ch01_literature_001",
+        "src_ch01_literature_002",
+    ]
+    assert context["routes_used"] == ["scholar", "general"]
+
+
+def test_packet_worker_includes_search_context_in_prompt() -> None:
+    context = build_search_context(sample_card(), FakeSearchProvider(), num_results_per_route=1)
+    response = {
+        "task_id": "ch01-literature",
+        "claims": [{"claim": "候选证据支持判断", "source_ids": ["src_ch01_literature_001"]}],
+        "evidence_items": [{"source_id": "src_ch01_literature_001", "summary": "摘要"}],
+        "counter_evidence": [{"claim": "仍需更多数据", "source_ids": ["src_ch01_literature_002"]}],
+        "source_ids": ["src_ch01_literature_001", "src_ch01_literature_002"],
+        "sources": context["candidate_sources"],
+        "source_quality_notes": ["候选来源需要后续评级"],
+        "open_questions": [],
+        "raw_quotes_or_notes": [],
+    }
+    role = resolve_runtime_profile(profile="medium").role_for_task("evidence_packet")
+    fake = FakeClient(response)
+
+    packet = PacketWorker(role=role, client=fake, search_provider=FakeSearchProvider()).run(sample_card())
+
+    assert packet["sources"][0]["url"] == "https://example.com/scholar"
+    assert "candidate_sources" in fake.calls[0]["user"]
+
+
+def test_append_packet_sources_dedupes_by_url(tmp_path: Path) -> None:
+    packet = {
+        "sources": [
+            {"id": "src_a", "title": "A", "url": "https://example.com/a", "tier": "Tier 2", "score": 7},
+            {"id": "src_b", "title": "B", "url": "https://example.com/a", "tier": "Tier 2", "score": 7},
+        ]
+    }
+
+    written = append_packet_sources(tmp_path / "sources.jsonl", packet)
+
+    assert written == 1
+    assert len((tmp_path / "sources.jsonl").read_text(encoding="utf-8").splitlines()) == 1
+
+
+def test_rebuild_sources_from_packets_dedupes_manual_packets(tmp_path: Path) -> None:
+    project = tmp_path / "project"
+    packets = project / "phase2" / "packets"
+    packets.mkdir(parents=True)
+    packet = {
+        "sources": [
+            {"id": "src_001", "title": "A", "url": "https://example.com/a"},
+            {"id": "src_002", "title": "A duplicate", "url": "https://example.com/a"},
+            {"id": "src_003", "title": "Local", "url": "phase0/extracted/local.md"},
+        ]
+    }
+    (packets / "ch01-a.json").write_text(json.dumps(packet, ensure_ascii=False), encoding="utf-8")
+
+    count = rebuild_sources_from_packets(project)
+
+    lines = (project / "phase2" / "sources.jsonl").read_text(encoding="utf-8").splitlines()
+    assert count == 2
+    assert len(lines) == 2
+    assert "src_001" in lines[0]
+    assert "src_003" in lines[1]