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:
Gabriel Luiz Freitas Almeida 2024-01-29 23:24:45 -03:00
commit 5eb2e7f979
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,8 +1,9 @@
from typing import Callable, Optional, Union from typing import Callable, Optional, Union
from langchain.chains import LLMChain from langchain.chains import LLMChain
from langflow import CustomComponent 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): class LLMChainComponent(CustomComponent):
@ -22,5 +23,5 @@ class LLMChainComponent(CustomComponent):
prompt: BasePromptTemplate, prompt: BasePromptTemplate,
llm: BaseLanguageModel, llm: BaseLanguageModel,
memory: Optional[BaseMemory] = None, memory: Optional[BaseMemory] = None,
) -> Union[Chain, Callable]: ) -> Union[Chain, Callable, Text]:
return LLMChain(prompt=prompt, llm=llm, memory=memory) return LLMChain(prompt=prompt, llm=llm, memory=memory)