From 734bf3c09d292012b45b93a4813ccdd0cf26d08a Mon Sep 17 00:00:00 2001
From: Ogulcan Celik <ogulcancelik@gmail.com>
Date: Sun, 21 Jun 2026 17:26:45 +0300
Subject: fix: release pi and omp agents on shutdown

---
 src/integration/assets/omp/herdr-agent-state.ts | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

(limited to 'src/integration/assets/omp')

diff --git a/src/integration/assets/omp/herdr-agent-state.ts b/src/integration/assets/omp/herdr-agent-state.ts
index a933e867..ba97b1c7 100644
--- a/src/integration/assets/omp/herdr-agent-state.ts
+++ b/src/integration/assets/omp/herdr-agent-state.ts
@@ -144,6 +144,19 @@ function sendState(state: AgentState, message?: string, seq = nextReportSeq()):
   });
 }
 
+function releaseAgent(): Promise<void> {
+  return sendRequest({
+    id: `${source}:release:${Date.now()}:${Math.random().toString(36).slice(2)}`,
+    method: "pane.release_agent",
+    params: {
+      pane_id: paneId,
+      source,
+      agent: "omp",
+      seq: nextReportSeq(),
+    },
+  });
+}
+
 let sendInFlight = false;
 let queuedState: QueuedState | undefined;
 
@@ -343,4 +356,12 @@ export default function (pi) {
 
     scheduleIdle();
   });
+
+  pi.on("session_shutdown", async () => {
+    if (!rootSession) {
+      return;
+    }
+    clearPendingTimers();
+    await releaseAgent();
+  });
 }
-- 
cgit v1.2.3