Function _make_mock_extractor

tests/extraction/llm/test_extractor.py:29–53

Build an LLMExtractor whose provider is a MagicMock. Patching make_provider sidesteps eager client construction (e.g. OpenAIProvider's OpenAI() call which now fails when credentials are missing — irrelevant for these tests). retryable_exceptions is set so _call_llm's `except retrya

(
    model: str = "openai/test-model",
    run_dir: str | None = None,
    debug: bool = False,
)

def _make_mock_extractor(
    model: str = "openai/test-model",
    run_dir: str | None = None,
    debug: bool = False,
) -> LLMExtractor:
    """Build an LLMExtractor whose provider is a MagicMock.

    Patching make_provider sidesteps eager client construction (e.g.
    OpenAIProvider's OpenAI() call which now fails when credentials are
    missing — irrelevant for these tests).  retryable_exceptions is set
    so _call_llm's `except retryable as e:` clause type-checks.
    """
    cfg = ExtractorConfig(model=model, run_dir=run_dir, debug=debug)
    provider = MagicMock(retryable_exceptions=(ConnectionError,))
    with (
        patch(
            "explainshell.extraction.llm.extractor.make_provider",
            return_value=provider,
        ),
        patch(
            "explainshell.extraction.llm.extractor.make_batch_provider",
            return_value=MagicMock(),
        ),
    ):
        return LLMExtractor(cfg)


class TestExtractIntegration(unittest.TestCase):