From 1a8825f491b4437da2a1fff132504be0e7fc6932 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 3 Nov 2023 11:11:10 -0300 Subject: [PATCH] Add memory info to ConversationChainComponent --- .../langflow/components/chains/ConversationChain.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/components/chains/ConversationChain.py b/src/backend/langflow/components/chains/ConversationChain.py index 530c6fb01..76e6d8e25 100644 --- a/src/backend/langflow/components/chains/ConversationChain.py +++ b/src/backend/langflow/components/chains/ConversationChain.py @@ -12,7 +12,10 @@ class ConversationChainComponent(CustomComponent): return { "prompt": {"display_name": "Prompt"}, "llm": {"display_name": "LLM"}, - "memory": {"display_name": "Memory"}, + "memory": { + "display_name": "Memory", + "info": "Memory to load context from. If none is provided, a ConversationBufferMemory will be used.", + }, "code": {"show": False}, } @@ -21,4 +24,6 @@ class ConversationChainComponent(CustomComponent): llm: BaseLanguageModel, memory: Optional[BaseMemory] = None, ) -> Union[Chain, Callable]: + if memory is None: + return ConversationChain(llm=llm) return ConversationChain(llm=llm, memory=memory)