From 176f6b16409e1de3f2bed035b587328dbad9b8ef Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 22 Feb 2024 18:41:18 -0300 Subject: [PATCH] Refactor ConversationChainComponent class to handle different types of result objects --- src/backend/langflow/components/chains/ConversationChain.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/src/backend/langflow/components/chains/ConversationChain.py b/src/backend/langflow/components/chains/ConversationChain.py index 43f71e67b..46c33447e 100644 --- a/src/backend/langflow/components/chains/ConversationChain.py +++ b/src/backend/langflow/components/chains/ConversationChain.py @@ -35,13 +35,10 @@ class ConversationChainComponent(CustomComponent): # 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): - self.status = "is message" result = result.content elif isinstance(result, str): - self.status = "is_string" result = result else: - # is dict result = result.get("response") self.status = result return result