Catch job cancellation exceptions

trypromptly · Nov 8, 2024 · 77d14c9 · 77d14c9
1 parent c134286
commit 77d14c9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/llmstack/sheets/apis.py b/llmstack/sheets/apis.py
@@ -309,7 +309,11 @@ def cancel_run(self, request, sheet_uuid=None, run_id=None):
 
         if run_id_in_sheet != run_id:
             return DRFResponse(status=status.HTTP_400_BAD_REQUEST)
-        PromptlySheetAppExecuteJob.cancel(job_id)
+
+        try:
+            PromptlySheetAppExecuteJob.cancel(job_id)
+        except Exception as e:
+            logger.error(f"Error cancelling job {job_id}: {e}")
 
         sheet.extra_data["running"] = False
         sheet.extra_data["job_id"] = None