jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update LLMChainComponent signature

Browse source
This commit is contained in:
anovazzi1 2024-01-16 18:13:31 -03:00
commit cb5076cba0
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/langflow/components/chains/LLMChain.py
View file

@ -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)