From 683c52411bf9a407b04192512fe4d5bff0de48a0 Mon Sep 17 00:00:00 2001
From: nkumar-eightfold <nkumar@eightfold.ai>
Date: Tue, 28 Oct 2025 05:41:08 +0000
Subject: [PATCH] adding airflow_sandbox label

---
 entrypoint.sh | 32 ++++++++++++++++++++++++++++++++
 1 file changed, 32 insertions(+)

diff --git a/entrypoint.sh b/entrypoint.sh
index 3054355..8ffc94d 100755
--- a/entrypoint.sh
+++ b/entrypoint.sh
@@ -335,3 +335,35 @@ if [[ $comment_body == "stop_sandbox" ]]; then
     esac
   done
 fi
+
+# Add airflow_sandbox if needs_airflow_sandbox
+already_needs_airflow_sandbox=false
+
+if [[ $comment_body == "needs_airflow_sandbox" ]]; then
+  for label in $labels; do
+    case $label in
+      airflow_sandbox)
+        already_needs_airflow_sandbox=true
+        ;;
+      *)
+        echo "Unknown label $label"
+        ;;
+    esac
+  done
+  if [[ "$already_needs_airflow_sandbox" == false ]]; then
+    add_label "airflow_sandbox"
+  fi
+fi
+
+if [[ $comment_body == "stop_airflow_sandbox" ]]; then
+  for label in $labels; do
+    case $label in
+      airflow_sandbox)
+        remove_label "airflow_sandbox"
+        ;;
+      *)
+        echo "Unknown label $label"
+        ;;
+    esac
+  done
+fi