PYTHONPATH=. python - <<'PY'
import asyncio
import json
import types
from unittest.mock import patch

from deeptutor.agents.research.pipeline import ResearchPipeline, ResearchedBlock, SubTopicItem
from deeptutor.core.context import UnifiedContext
from deeptutor.core.stream_bus import StreamBus


class FakeLLM:
    binding = "openai"
    model = "gpt-x"
    api_key = "k"
    base_url = "u"
    api_version = None
    extra_headers = {}
    reasoning_effort = None


class FakeRegistry:
    def build_openai_schemas(self, _names):
        return []

    def build_prompt_text(self, _names, **_kwargs):
        return "- none"

    def get(self, _name):
        return None

    def get_enabled(self, _names):
        return []


async def main() -> None:
    with patch("deeptutor.agents.research.pipeline.get_llm_config", lambda: FakeLLM()), patch(
        "deeptutor.agents.research.pipeline.get_tool_registry", lambda: FakeRegistry()
    ):
        pipeline = ResearchPipeline(language="en", runtime_config={"queue": {"max_length": 5}})

    captured = {}

    async def fake_research_block(self, *, block, queue, citations, topic, context, stream, client):
        queue.mark_researching(block.block_id)
        if block.block_id == "block_1":
            raise RuntimeError("synthetic block failure")
        queue.mark_completed(block.block_id)
        return ResearchedBlock(block=block, knowledge=f"knowledge for {block.block_id}")

    async def fake_write_report(self, *, topic, blocks, citations, stream, client):
        captured["blocks"] = [
            {
                "id": rb.block.block_id,
                "status": getattr(rb.block.status, "value", str(rb.block.status)),
                "knowledge": rb.knowledge,
            }
            for rb in blocks
        ]
        return "REPORT_OK"

    async def fake_emit(*args, **kwargs):
        return None

    pipeline._research_block = types.MethodType(fake_research_block, pipeline)
    pipeline._write_report = types.MethodType(fake_write_report, pipeline)

    with patch("deeptutor.agents.research.pipeline.emit_capability_result", fake_emit):
        result = await pipeline._run_inner(
            context=UnifiedContext(session_id="s1", user_message="research this"),
            topic="Research topic",
            image_attachments=[],
            confirmed_outline=[SubTopicItem(title="A"), SubTopicItem(title="B")],
            stream=StreamBus(),
            client=None,
        )

    print(json.dumps({"result": result, "captured_blocks": captured["blocks"]}, indent=2, default=str))


asyncio.run(main())
PY