From 5eb2e7f9794782c2205422dfe6048c2e3540ab79 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 29 Jan 2024 23:24:45 -0300 Subject: [PATCH] Update LLMChainComponent signature --- src/backend/langflow/components/chains/LLMChain.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/chains/LLMChain.py b/src/backend/langflow/components/chains/LLMChain.py index 309f69215..8a62a8c55 100644 --- a/src/backend/langflow/components/chains/LLMChain.py +++ b/src/backend/langflow/components/chains/LLMChain.py @@ -1,8 +1,9 @@ from typing import Callable, Optional, Union from langchain.chains import LLMChain + from langflow import CustomComponent -from langflow.field_typing import BaseLanguageModel, BaseMemory, BasePromptTemplate, Chain +from langflow.field_typing import BaseLanguageModel, BaseMemory, BasePromptTemplate, Chain, Text class LLMChainComponent(CustomComponent): @@ -22,5 +23,5 @@ class LLMChainComponent(CustomComponent): prompt: BasePromptTemplate, llm: BaseLanguageModel, memory: Optional[BaseMemory] = None, - ) -> Union[Chain, Callable]: + ) -> Union[Chain, Callable, Text]: return LLMChain(prompt=prompt, llm=llm, memory=memory)