From b754daf1af1fa5c121f8edc49fed38500fb2b7ed Mon Sep 17 00:00:00 2001
From: alpha nerd <alpha-nerd@nomyo.ai>
Date: Thu, 28 May 2026 10:16:54 +0200
Subject: [PATCH] feat: after closing the probe session, reset

---
 router.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/router.py b/router.py
index d1a1e38..e294555 100644
--- a/router.py
+++ b/router.py
@@ -396,6 +396,8 @@ async def shutdown_event() -> None:
     await app_state["session"].close()
     if app_state.get("probe_session") is not None:
         await app_state["probe_session"].close()
+        app_state["probe_session"] = None
+        app_state["probe_connector"] = None
 
     # Close Unix socket sessions
     for ep, sess in list(app_state.get("socket_sessions", {}).items()):