Update execution.py

execution.py

@@ -44,23 +44,26 @@ def safe_execute(
             stdout=subprocess.PIPE,
             stderr=subprocess.PIPE,
         )
-        outputs = execution_process.communicate(timeout=timeout)
-    
-        stdout, stderr = outputs
-        stdout = stdout.decode('utf-8')
-        stderr = stderr.decode('utf-8')
-        runtime = (datetime.datetime.now() - start_time).total_seconds()
-        return_code = execution_process.returncode
-    except subprocess.TimeoutExpired:
-        timed_out = True
-        runtime = timeout
-    
+        try:
+            outputs = execution_process.communicate(timeout=timeout)
+        
+            stdout, stderr = outputs
+            stdout = stdout.decode('utf-8')
+            stderr = stderr.decode('utf-8')
+            runtime = (datetime.datetime.now() - start_time).total_seconds()
+            return_code = execution_process.returncode
+        except subprocess.TimeoutExpired:
+            timed_out = True
+            runtime = timeout
+        
+        finally:
+            execution_process.kill()
     except Exception as e:
         stderr = str(e)
         stdout = ""
         return_code = -1
-    finally:
-        execution_process.kill()
+        runtime = -1
+        timed_out = False
     
     return CommandResult(
         return_code=return_code,