Use Callback protocols to type the callbacks module

Replace the usage of mypy_extensions in the callbacks module with
Callback protocols to allow the usage of different type checkers
(pyright, ty, etc).

see: https://typing.python.org/en/latest/spec/callables.html#callback-protocols

Author: MoisesGSalas