fix(assistant): improve middleware dispatch and inject kwargs in middleware

fix(assistant): improve middleware dispatch and inject kwargs in middleware #606