I like your thought process. Small thing - in your agent class, you define a method to populate input variables, which takes a single payload argument. I’m pretty sure you’ll have to decorate the method as @staticmethod to circumvent the self instance argument requirement.