minor: but how would you feel about calling this `StreamingContextState`? I found it a bit confusing when looking at it directly. Then you could probably even remove the doc because it would be totally clear.