Follow-up from "Custom decoder for LangChain messages"

The following discussion from !83 (merged) should be addressed:

• @mikolaj_wawrzyniak started a discussion:

  suggestion(non-blocking): Please add test case to cover ToolMessage as it require additional params