The LangGraph store (NullDriftLangGraphStore in nulldrift_agents.py) uses httpx.AsyncClient(), but it never explicitly closes the client when the script finishes, which can lead to unclosed socket warnings. Implement __aenter__ and __aexit__ asynchronous context managers so the store can be safely used inside an async with block, automatically calling await self.client.aclose().
The LangGraph store (
NullDriftLangGraphStoreinnulldrift_agents.py) useshttpx.AsyncClient(), but it never explicitly closes the client when the script finishes, which can lead to unclosed socket warnings. Implement__aenter__and__aexit__asynchronous context managers so the store can be safely used inside anasync withblock, automatically callingawait self.client.aclose().