I added this functionality already some time ago in my Claudine agent:

https://github.com/xemantic/claudine/

It costed roughly 30 lines of code: https://github.com/xemantic/claudine/blob/main/src/commonMai...