Add more debug logs

Author: sanjayprabhu

src/isolate/connections/grpc/agent.py

@@ -224,6 +224,7 @@ def abort_with_msg(
 
     def handle_shutdown(self) -> None:
         if self._current_callable is None:
+            print("No current callable, skipping shutdown.")
             return
 
         # Check for teardown on the callable itself or on the wrapped function
@@ -272,7 +273,9 @@ def run_agent(address: str, log_fd: int | None = None) -> int:
 
     # Set up SIGTERM handler
     def sigterm_handler(signum, frame):
+        print("Received SIGTERM, shutting down the agent...")
         servicer.handle_shutdown()
+        priint("Shutdown complete, stopping the agent server.")
         server.stop(grace=0.1)
         executor.shutdown(wait=False, cancel_futures=True)
 

src/isolate/server/server.py

@@ -185,6 +185,7 @@ def cancel(self):
                 self.future.cancel()
 
             if self.agent:
+                print("Terminating the agent...")
                 self.agent.terminate()
 
             try: