Function _make_batch

tests/extraction/llm/test_openai_provider.py:24–45

Build a fake Batch object matching openai.types.Batch shape.

(
    status: str = "in_progress",
    completed: int = 0,
    failed: int = 0,
    total: int = 10,
    input_file_id: str | None = None,
    output_file_id: str | None = None,
    error_file_id: str | None = None,
)

def _make_batch(
    status: str = "in_progress",
    completed: int = 0,
    failed: int = 0,
    total: int = 10,
    input_file_id: str | None = None,
    output_file_id: str | None = None,
    error_file_id: str | None = None,
) -> SimpleNamespace:
    """Build a fake Batch object matching openai.types.Batch shape."""
    return SimpleNamespace(
        status=status,
        request_counts=SimpleNamespace(
            completed=completed,
            failed=failed,
            total=total,
        ),
        input_file_id=input_file_id,
        output_file_id=output_file_id,
        usage=None,
        error_file_id=error_file_id,
    )


@patch("explainshell.extraction.llm.providers.openai.OpenAI")