Update chain.invoke() argument to use a dictionary for input value

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-06 14:53:10 -03:00
commit be32d57434

View file

@ -31,7 +31,7 @@ class ConversationChainComponent(CustomComponent):
chain = ConversationChain(llm=llm)
else:
chain = ConversationChain(llm=llm, memory=memory)
result = chain.invoke(inputs)
result = chain.invoke({"input": input_value})
if hasattr(result, "content") and isinstance(result.content, str):
result = result.content
elif isinstance(result, str):