fix: type hint contexts that are returned when generating

[jakelorocco]

Backends return a context of the same type that was passed in. We should add type hints for this.

[CatOn60Hz]

I’d like to work on this issue.