fix: type hint contexts that are returned when generating #307
Description
Backends return a context of the same type that was passed in. We should add type hints for this.
{{ message }}
Backends return a context of the same type that was passed in. We should add type hints for this.