From 72fee51d3244bf362cb1d639893759a56b40accd Mon Sep 17 00:00:00 2001
From: bennykok <itechbenny@gmail.com>
Date: Thu, 1 Feb 2024 21:04:24 +0800
Subject: [PATCH] fix: making sure the log was sent before setting the status

---
 custom_routes.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/custom_routes.py b/custom_routes.py
index f6698a9..0487a0b 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -259,7 +259,6 @@ def update_run(prompt_id, status: Status):
             "run_id": prompt_id,
             "status": status.value,
         }
-        prompt_metadata[prompt_id]['status'] = status
         print(f"Status: {status.value}")
 
         try:
@@ -297,7 +296,9 @@ def update_run(prompt_id, status: Status):
             error_type = type(e).__name__
             stack_trace = traceback.format_exc().strip()
             print(f"Error occurred while updating run: {e} {stack_trace}")
-
+        finally:
+            prompt_metadata[prompt_id]['status'] = status
+            
 
 async def upload_file(prompt_id, filename, subfolder=None, content_type="image/png", type="output"):
     """