changes to restart function

- allow setting tag to restart function

Author: DaMandal0rian

scripts/launch-nodes/README.md

@@ -88,15 +88,17 @@ Once the configuration file is ready, make the python script executable and run
 
 ```bash
 chmod +x manage_subspace.py
-python manage_subspace.py --config nodes.toml --release_version gemini-3h-2024-sep-17 --subspace_dir /home/ubuntu/subspace/subspace \
---pot_external_entropy random_value --network gemini-3h --plot_size 10G --cache-percentage 15
+python manage_subspace.py --config nodes.toml --release_version docker-tag --subspace_dir /home/ubuntu/subspace/subspace \
+--pot_external_entropy random_value --network gemini-3h --plot_size 10G --cache_percentage 15
 
 # prune images
-python manage_subspace.py --config nodes.toml --release_version gemini-3h-2024-sep-17 --subspace_dir /home/ubuntu/subspace/subspace --prune
+python manage_subspace.py --config nodes.toml --subspace_dir /home/ubuntu/subspace/subspace --prune
 
 # restart stack
-python manage_subspace.py --config nodes.toml --release_version gemini-3h-2024-sep-17 --subspace_dir /home/ubuntu/subspace/subspace --restart
+python manage_subspace.py --config nodes.toml --subspace_dir /home/ubuntu/subspace/subspace --restart
 
+# restart stack with new docker tag
+python manage_subspace.py --config nodes.toml --release_version docker-tag --subspace_dir /home/ubuntu/subspace/subspace --restart
 ```
 
 ### Command Line Options

scripts/launch-nodes/manage_subspace.py

@@ -89,12 +89,19 @@ def docker_compose_down(client, subspace_dir):
         logger.error(f"Failed to run sudo docker compose down -v: {e}")
         raise
 
-def docker_compose_restart(client, subspace_dir):
+def docker_compose_restart(client, subspace_dir, docker_tag=None):
     """Run sudo docker compose restart in the subspace directory."""
     try:
-        command = f'cd {subspace_dir} && sudo docker compose restart'
+        # Modify .env file if a new DOCKER_TAG is provided
+        if docker_tag:
+            logger.info(f"Updating DOCKER_TAG to {docker_tag} in {subspace_dir}/.env")
+            modify_env_file(client, subspace_dir, release_version=docker_tag)
+
+        # Restart the containers
+        restart_cmd = f'cd {subspace_dir} && sudo docker compose restart'
         logger.info(f"Running sudo docker compose restart in {subspace_dir}")
-        run_command(client, command)
+        run_command(client, restart_cmd)
+
     except Exception as e:
         logger.error(f"Failed to run sudo docker compose restart: {e}")
         raise