From 9192d50e72e95ff72ee0b2dd9e9649bb0461e18e Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Feb 2024 18:41:29 -0300 Subject: [PATCH] Remove unnecessary comments and improve result handling in ConversationChainComponent --- src/backend/langflow/components/chains/ConversationChain.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/backend/langflow/components/chains/ConversationChain.py b/src/backend/langflow/components/chains/ConversationChain.py index 46c33447e..dd465c511 100644 --- a/src/backend/langflow/components/chains/ConversationChain.py +++ b/src/backend/langflow/components/chains/ConversationChain.py @@ -32,8 +32,6 @@ class ConversationChainComponent(CustomComponent): else: chain = ConversationChain(llm=llm, memory=memory) result = chain.invoke(inputs) - # result is an AIMessage which is a subclass of BaseMessage - # We need to check if it is a string or a BaseMessage if hasattr(result, "content") and isinstance(result.content, str): result = result.content elif isinstance(result, str):