From 8338ca63940a14e120b40bb591c8c6b2f05d70bd Mon Sep 17 00:00:00 2001
From: Shuchang Zheng <wintonzheng0325@gmail.com>
Date: Fri, 27 Jun 2025 09:41:38 +0900
Subject: [PATCH] Add RESIDENTIAL_AU proxy support for Australia (#2803)

---
 skyvern/client/types/proxy_location.py       | 1 +
 skyvern/client/types/workflow_run_request.py | 1 +
 skyvern/schemas/docs/doc_strings.py          | 1 +
 skyvern/schemas/runs.py                      | 7 +++++++
 4 files changed, 10 insertions(+)

diff --git a/skyvern/client/types/proxy_location.py b/skyvern/client/types/proxy_location.py
index a5ed452b..b5807702 100644
--- a/skyvern/client/types/proxy_location.py
+++ b/skyvern/client/types/proxy_location.py
@@ -20,6 +20,7 @@ ProxyLocation = typing.Union[
         "RESIDENTIAL_NZ",
         "RESIDENTIAL_ZA",
         "RESIDENTIAL_AR",
+        "RESIDENTIAL_AU",
         "RESIDENTIAL_ISP",
         "NONE",
     ],
diff --git a/skyvern/client/types/workflow_run_request.py b/skyvern/client/types/workflow_run_request.py
index a9907407..b5b9f0cd 100644
--- a/skyvern/client/types/workflow_run_request.py
+++ b/skyvern/client/types/workflow_run_request.py
@@ -40,6 +40,7 @@ class WorkflowRunRequest(UniversalBaseModel):
     - RESIDENTIAL_NZ: New Zealand
     - RESIDENTIAL_ZA: South Africa
     - RESIDENTIAL_AR: Argentina
+    - RESIDENTIAL_AU: Australia
     - RESIDENTIAL_ISP: ISP proxy
     - US-CA: California
     - US-NY: New York
diff --git a/skyvern/schemas/docs/doc_strings.py b/skyvern/schemas/docs/doc_strings.py
index 3bab0bf0..7d26d40c 100644
--- a/skyvern/schemas/docs/doc_strings.py
+++ b/skyvern/schemas/docs/doc_strings.py
@@ -25,6 +25,7 @@ Available geotargeting options:
 - RESIDENTIAL_NZ: New Zealand
 - RESIDENTIAL_ZA: South Africa
 - RESIDENTIAL_AR: Argentina
+- RESIDENTIAL_AU: Australia
 - RESIDENTIAL_ISP: ISP proxy
 - US-CA: California
 - US-NY: New York
diff --git a/skyvern/schemas/runs.py b/skyvern/schemas/runs.py
index 4aaec6ee..663de818 100644
--- a/skyvern/schemas/runs.py
+++ b/skyvern/schemas/runs.py
@@ -50,6 +50,7 @@ class ProxyLocation(StrEnum):
     RESIDENTIAL_NZ = "RESIDENTIAL_NZ"
     RESIDENTIAL_ZA = "RESIDENTIAL_ZA"
     RESIDENTIAL_AR = "RESIDENTIAL_AR"
+    RESIDENTIAL_AU = "RESIDENTIAL_AU"
     RESIDENTIAL_ISP = "RESIDENTIAL_ISP"
     NONE = "NONE"
 
@@ -81,6 +82,7 @@ class ProxyLocation(StrEnum):
             cls.RESIDENTIAL_NZ,
             cls.RESIDENTIAL_ZA,
             cls.RESIDENTIAL_AR,
+            cls.RESIDENTIAL_AU,
         }
 
     @staticmethod
@@ -97,6 +99,7 @@ class ProxyLocation(StrEnum):
             ProxyLocation.RESIDENTIAL_NZ: 2000,
             ProxyLocation.RESIDENTIAL_ZA: 2000,
             ProxyLocation.RESIDENTIAL_AR: 2000,
+            ProxyLocation.RESIDENTIAL_AU: 2000,
         }
         return counts.get(proxy_location, 10000)
 
@@ -114,6 +117,7 @@ class ProxyLocation(StrEnum):
             ProxyLocation.RESIDENTIAL_NZ: "NZ",
             ProxyLocation.RESIDENTIAL_ZA: "ZA",
             ProxyLocation.RESIDENTIAL_AR: "AR",
+            ProxyLocation.RESIDENTIAL_AU: "AU",
         }
         return mapping.get(proxy_location, "US")
 
@@ -170,6 +174,9 @@ def get_tzinfo_from_proxy(proxy_location: ProxyLocation) -> ZoneInfo | None:
     if proxy_location == ProxyLocation.RESIDENTIAL_AR:
         return ZoneInfo("America/Argentina/Buenos_Aires")
 
+    if proxy_location == ProxyLocation.RESIDENTIAL_AU:
+        return ZoneInfo("Australia/Sydney")
+
     if proxy_location == ProxyLocation.RESIDENTIAL_ISP:
         return ZoneInfo("America/New_York")