mirror of
https://github.com/microsoft/graphrag.git
synced 2025-07-25 01:41:10 +00:00
11 lines
295 B
Python
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)
|