From 282df9a642e248f71503ebfe149364bfe272ba1d Mon Sep 17 00:00:00 2001
From: Daishan Peng <daishan@acorn.io>
Date: Fri, 28 Mar 2025 00:00:33 -0700
Subject: [PATCH] Fix: fix close bug when run is aborted

Signed-off-by: Daishan Peng <daishan@acorn.io>
---
 pkg/sdkserver/routes.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/pkg/sdkserver/routes.go b/pkg/sdkserver/routes.go
index 66449749..1a4e28ea 100644
--- a/pkg/sdkserver/routes.go
+++ b/pkg/sdkserver/routes.go
@@ -176,9 +176,13 @@ func (s *server) execHandler(w http.ResponseWriter, r *http.Request) {
 
 	defer func() {
 		s.runningLock.Lock()
+		// Need to check if the cancel is still in map. In case when user abort, the channel will be deleted from map and closed already, and closing it again will panic
+		_, ok := s.running[runID]
+		if ok {
+			close(cancel)
+		}
 		delete(s.running, runID)
 		s.runningLock.Unlock()
-		close(cancel)
 	}()
 
 	// Ensure chat state is not empty.