graphrag/tests/unit/indexing/verbs/helpers/mock_llm.py

# Copyright (c) 2024 Microsoft Corporation.
# Licensed under the MIT License
from graphrag.llm import CompletionLLM, MockChatLLM


def create_mock_llm(
    responses: list[str],
) -> CompletionLLM:
    """Creates a mock LLM that returns the given responses."""
    return MockChatLLM(responses)