Hi @yuga04 . What kind of other attributes do you want to use? By context-1, context-2 do you mean conversation contexts? If yes, then that wouldn’t be possible because you don’t have those during training. However, you may be interested in looking into end to end training scheme.