diff --git a/src/ragas/llms/base.py b/src/ragas/llms/base.py index 3d2e117e2..2f33981cc 100644 --- a/src/ragas/llms/base.py +++ b/src/ragas/llms/base.py @@ -9,12 +9,15 @@ from langchain_core.language_models import BaseLanguageModel from langchain_core.outputs import LLMResult + if t.TYPE_CHECKING: from langchain_core.callbacks import Callbacks from langchain_core.prompts import ChatPromptTemplate - + from ragas.llms.prompt import PromptValue + + MULTIPLE_COMPLETION_SUPPORTED = [ OpenAI, ChatOpenAI, @@ -66,6 +69,9 @@ def generate_text_with_hmpt( stop: t.Optional[t.List[str]] = None, callbacks: Callbacks = [], ) -> LLMResult: + + from ragas.llms.prompt import PromptValue + prompt = PromptValue(prompt_str=prompts[0].format()) return self.generate_text(prompt, n, temperature, stop, callbacks)