From c96bc5069c1bb34fab53174f52cdb31111db54a6 Mon Sep 17 00:00:00 2001
From: Anoop Sharma <anoopsharma527@gmail.com>
Date: Mon, 26 Feb 2024 06:51:20 +0530
Subject: [PATCH] MockLLM (#11376)
Update __init__.py
---
llama-index-core/llama_index/core/llms/__init__.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/llama-index-core/llama_index/core/llms/__init__.py b/llama-index-core/llama_index/core/llms/__init__.py
index 5e480086f7..d09954b130 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",
]
--
GitLab