From 4a5d6c785551ef573e5eaed7f444c7b3d26738fa Mon Sep 17 00:00:00 2001
From: Donato Capitella <donato.capitella@reversec.com>
Date: Thu, 19 Feb 2026 20:29:28 +0000
Subject: [PATCH] fix broken stuff

---
 scripts/cluster_manager.py    |  2 +-
 scripts/start_vllm_cluster.py | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/scripts/cluster_manager.py b/scripts/cluster_manager.py
index 45a4b9e..11d796b 100644
--- a/scripts/cluster_manager.py
+++ b/scripts/cluster_manager.py
@@ -59,7 +59,7 @@ def setup_worker_node(worker_ip, head_ip):
     export NCCL_IB_TIMEOUT=23
     export NCCL_IB_RETRY_CNT=7
     echo "Starting Ray Worker on {worker_ip} connecting to {head_ip}..."
-    ray start --address='{head_ip}:6379' --num-gpus=1 --num-cpus=8 --disable-usage-stats --include-dashboard=false
+    ray start --address='{head_ip}:6379' --num-gpus=1 --num-cpus=8 --disable-usage-stats
     """
     
     print(f"Setting up Worker Node ({worker_ip})...")
diff --git a/scripts/start_vllm_cluster.py b/scripts/start_vllm_cluster.py
index 18509ba..e31c72e 100755
--- a/scripts/start_vllm_cluster.py
+++ b/scripts/start_vllm_cluster.py
@@ -303,6 +303,27 @@ def configure_and_launch_vllm(model_idx, head_ip):
     # Exec
     os.execvpe("vllm", cmd, env)
 
+def setup_ips_dialog(current_head, current_worker):
+    # Using a form to edit both IPs
+    # label y x item y x flen ilen
+    form_args = [
+        "--title", "Cluster Configuration",
+        "--form", "Enter IP addresses for Head and Worker nodes:",
+        "10", "60", "2",
+        "Head Node IP:", "1", "1",  current_head, "1", "20", "20", "0",
+        "Worker Node IP:", "2", "1", current_worker, "2", "20", "20", "0"
+    ]
+    
+    result = run_dialog(form_args)
+    if not result:
+        return None
+        
+    lines = result.splitlines()
+    if len(lines) >= 2:
+        return lines[0].strip(), lines[1].strip()
+    return None
+
+
 def main():
     check_dependencies()