🔧 fix(LLMChain.py): add missing import for Text type from langflow.field_typing module
🚀 feat(LLMChain.py): update return type annotation of LLMChainComponent.__call__ method to include Text type
This commit is contained in:
parent
3079318d6a
commit
23a141cb8d
1 changed files with 2 additions and 1 deletions
|
|
@ -6,6 +6,7 @@ from langflow.field_typing import (
|
|||
BaseLanguageModel,
|
||||
BaseMemory,
|
||||
Chain,
|
||||
Text,
|
||||
)
|
||||
|
||||
|
||||
|
|
@ -26,5 +27,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)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue