feat(mcp): queue long-running calculations

2026-10-29 18:25:15 +08:00
parent 2865811f1b
commit 63af998045
32 changed files with 1588 additions and 72 deletions
@@ -8,15 +8,14 @@ import sys
 import pandas as pd
 import wntr

-from app.algorithms._utils import _cleanup_temp_files
+from app.algorithms._utils import _cleanup_temp_files, calculation_temp_directory

 _PIPE2LEAKNODE = None
 _SIGNATURE_WORKER_DATA = {}


 def _make_temp_prefix(tag):
-    temp_dir = os.path.abspath(os.path.join("temp", "burst_location"))
-    os.makedirs(temp_dir, exist_ok=True)
+    temp_dir = calculation_temp_directory("burst_location")
    safe_tag = str(tag).replace(os.sep, "_").replace(" ", "_")
    return os.path.join(temp_dir, f"{safe_tag}_{os.getpid()}")