jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor ConversationChainComponent class to handle different types of result objects

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-22 18:41:18 -03:00
commit 176f6b1640
1 changed files with 0 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/backend/langflow/components/chains/ConversationChain.py
View file

@ -35,13 +35,10 @@ class ConversationChainComponent(CustomComponent):
# result is an AIMessage which is a subclass of BaseMessage # result is an AIMessage which is a subclass of BaseMessage
# We need to check if it is a string or a BaseMessage # We need to check if it is a string or a BaseMessage
if hasattr(result, "content") and isinstance(result.content, str): if hasattr(result, "content") and isinstance(result.content, str):
self.status = "is message"
result = result.content result = result.content
elif isinstance(result, str): elif isinstance(result, str):
self.status = "is_string"
result = result result = result
else: else:
# is dict
result = result.get("response") result = result.get("response")
self.status = result self.status = result
return result return result