From cb5076cba0c7f2d030dbbec099ab19e7e3a182d8 Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Tue, 16 Jan 2024 18:13:31 -0300 Subject: [PATCH] Update LLMChainComponent signature --- src/backend/langflow/components/chains/LLMChain.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/langflow/components/chains/LLMChain.py b/src/backend/langflow/components/chains/LLMChain.py index ec88e128a..85f7ae96d 100644 --- a/src/backend/langflow/components/chains/LLMChain.py +++ b/src/backend/langflow/components/chains/LLMChain.py @@ -28,5 +28,5 @@ class LLMChainComponent(CustomComponent): prompt: BasePromptTemplate, llm: BaseLanguageModel, memory: Optional[BaseMemory] = None, - ) -> Union[Chain, Callable]: + ) -> Union[Chain, Callable,LLMChain]: return LLMChain(prompt=prompt, llm=llm, memory=memory)