From 1d7585b40a31bb7433ef1b0acabfb67b6ee0a1b6 Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Wed, 9 Jul 2025 16:32:46 -0700
Subject: [PATCH] update run_task to make it possible to expose ecs task
 publicly (#2916)

---
 skyvern/forge/sdk/api/aws.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/skyvern/forge/sdk/api/aws.py b/skyvern/forge/sdk/api/aws.py
index a23f32c5..a99222ef 100644
--- a/skyvern/forge/sdk/api/aws.py
+++ b/skyvern/forge/sdk/api/aws.py
@@ -268,6 +268,7 @@ class AsyncAWSClient:
         task_definition: str,
         subnets: list[str],
         security_groups: list[str],
+        assign_public_ip: str = "DISABLED",
     ) -> dict:
         # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/ecs/client/run_task.html
         async with self._ecs_client() as client:
@@ -279,7 +280,7 @@ class AsyncAWSClient:
                     "awsvpcConfiguration": {
                         "subnets": subnets,
                         "securityGroups": security_groups,
-                        "assignPublicIp": "DISABLED",
+                        "assignPublicIp": assign_public_ip,
                     }
                 },
             )