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

Remove unnecessary comments and improve result handling in ConversationChainComponent

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

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

@ -32,8 +32,6 @@ class ConversationChainComponent(CustomComponent):
else: else:
chain = ConversationChain(llm=llm, memory=memory) chain = ConversationChain(llm=llm, memory=memory)
result = chain.invoke(inputs) 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): if hasattr(result, "content") and isinstance(result.content, str):
result = result.content result = result.content
elif isinstance(result, str): elif isinstance(result, str):