modify egress and ingress

Author: Jyothirmaikottu

vllm/infra/ec2.py

@@ -228,7 +228,7 @@ def setup():
         # create fsx
         sg_fsx = fsx.create_security_group(vpc_id, "vllm-ec2-fsx-sg", "SG for Fsx Mounting")
         ingress_rules = [{"protocol": "tcp", "port": "988-1023"}]
-        fsx.add_security_group_ingress_rules(sg_fsx, ingress_rules)
+        fsx.add_security_group_ingress_and_egress_rules(sg_fsx, ingress_rules)
 
         fsx_config = fsx.create_fsx_filesystem(
             subnet_ids[0], [sg_fsx], 1200, "SCRATCH_2", {"Name": "vllm-fsx-storage"}

vllm/infra/utils/fsx_utils.py

@@ -151,7 +151,7 @@ def delete_security_group(self, group_id: str):
             logger.error(f"Failed to create security group: {e}")
             raise
 
-    def add_security_group_ingress_rules(
+    def add_security_group_ingress_and_egress_rules(
         self, security_group_id: str, ingress_rules: List[Dict[str, Any]]
     ):
         """
@@ -169,6 +169,12 @@ def add_security_group_ingress_rules(
                     cmd += f" --{key} {value}"
                 run(cmd)
 
+            for rule in ingress_rules:
+                cmd = f"aws ec2 authorize-security-group-egress --group-id {security_group_id}"
+                for key, value in rule.items():
+                    cmd += f" --{key} {value}"
+                run(cmd)
+
             logger.info(f"Added ingress rules to security group: {security_group_id}")
 
         except Exception as e: