From fe04db828e57c9409321b6941c10a00ec5b0973b Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Mon, 2 Dec 2024 23:56:45 +0000
Subject: [PATCH] Better error handling

---
 pdelfin/beakerpipeline.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/pdelfin/beakerpipeline.py b/pdelfin/beakerpipeline.py
index bc40411..77f23ed 100644
--- a/pdelfin/beakerpipeline.py
+++ b/pdelfin/beakerpipeline.py
@@ -517,9 +517,10 @@ async def sglang_server_host(args, semaphore):
         logger.warning("SGLang server task ended")
         retry += 1
 
-    logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
-    sys.exit(1)
-    
+    if retry >= MAX_RETRIES:
+        logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
+        sys.exit(1)
+
 
 async def sglang_server_ready():
     max_attempts = 300