diff --git a/llama-index-core/llama_index/core/llms/__init__.py b/llama-index-core/llama_index/core/llms/__init__.py index 5e480086f76e9dd6bd97713603bb0e09bda5c4a0..d09954b1306c1442032bcf9802f083486c7ad514 100644 --- a/llama-index-core/llama_index/core/llms/__init__.py +++ b/llama-index-core/llama_index/core/llms/__init__.py @@ -11,6 +11,7 @@ from llama_index.core.base.llms.types import ( ) from llama_index.core.llms.custom import CustomLLM from llama_index.core.llms.llm import LLM +from llama_index.core.llms.mock import MockLLM __all__ = [ "CustomLLM", @@ -24,4 +25,5 @@ __all__ = [ "CompletionResponseGen", "LLMMetadata", "MessageRole", + "MockLLM", ]