2024-07-01 15:25:30 -06:00

11 lines
295 B
Python

# 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)