From 67eb5e573467ef2e9221652518712152801914c1 Mon Sep 17 00:00:00 2001
From: Dotta <bippadotta@protonmail.com>
Date: Thu, 5 Mar 2026 07:39:30 -0600
Subject: [PATCH] Limit session reset to assignment wakes only

---
 server/src/__tests__/heartbeat-workspace-session.test.ts | 8 ++++----
 server/src/services/heartbeat.ts                         | 6 +-----
 2 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/server/src/__tests__/heartbeat-workspace-session.test.ts b/server/src/__tests__/heartbeat-workspace-session.test.ts
index 67536fe9..9a34402a 100644
--- a/server/src/__tests__/heartbeat-workspace-session.test.ts
+++ b/server/src/__tests__/heartbeat-workspace-session.test.ts
@@ -93,22 +93,22 @@ describe("shouldResetTaskSessionForWake", () => {
     expect(shouldResetTaskSessionForWake({ wakeReason: "issue_assigned" })).toBe(true);
   });
 
-  it("resets session context on mention wake comment", () => {
+  it("does not reset session context on mention wake comment", () => {
     expect(
       shouldResetTaskSessionForWake({
         wakeReason: "issue_comment_mentioned",
         wakeCommentId: "comment-1",
       }),
-    ).toBe(true);
+    ).toBe(false);
   });
 
-  it("resets session context when commentId is present", () => {
+  it("does not reset session context when commentId is present", () => {
     expect(
       shouldResetTaskSessionForWake({
         wakeReason: "issue_commented",
         commentId: "comment-2",
       }),
-    ).toBe(true);
+    ).toBe(false);
   });
 
   it("does not reset for comment wakes", () => {
diff --git a/server/src/services/heartbeat.ts b/server/src/services/heartbeat.ts
index 3ea9421f..88155bbc 100644
--- a/server/src/services/heartbeat.ts
+++ b/server/src/services/heartbeat.ts
@@ -199,11 +199,7 @@ export function shouldResetTaskSessionForWake(
   contextSnapshot: Record<string, unknown> | null | undefined,
 ) {
   const wakeReason = readNonEmptyString(contextSnapshot?.wakeReason);
-  if (wakeReason === "issue_assigned") return true;
-  const wakeCommentId =
-    readNonEmptyString(contextSnapshot?.wakeCommentId) ??
-    readNonEmptyString(contextSnapshot?.commentId);
-  return wakeCommentId != null;
+  return wakeReason === "issue_assigned";
 }
 
 function deriveCommentId(