mirror of
https://github.com/microsoft/graphrag.git
synced 2025-07-23 00:42:56 +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)
|