Yes, I am considering using something like https://pkg.go.dev/context for this very purpose, though I need to read a bit more into it first.
Funny that you built a Python wrapper as I originally started this implementation in Python, which was...not a good idea. Claude hallucinated the acknowledgments section though :D