From e7378c9a5f353f3d228fc0a7fd9b37be1c310d43 Mon Sep 17 00:00:00 2001
From: wiaam96 <142393189+wiaam96@users.noreply.github.com>
Date: Thu, 22 Aug 2024 15:15:24 +0300
Subject: [PATCH] Update entry.sh

---
 build_system/docker/entry.sh | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/build_system/docker/entry.sh b/build_system/docker/entry.sh
index 9e5a69f..6c8c696 100644
--- a/build_system/docker/entry.sh
+++ b/build_system/docker/entry.sh
@@ -44,8 +44,11 @@ while true; do
 done
 
 if [ -z $var_token ] && [ $var_mode != "--hybrid_mode" ]; then
-    echo "Error: Token was not provided as input argument."
-    exit 1
+    var_token=$(env | grep 'AGENT_TOKEN=' | cut -d'=' -f2-)
+    if  [ -z $var_token ]; then
+        echo "Error: Token was not provided as input argument."
+        exit 1
+    fi
 fi
 
 orchestration_service_installation_flags="--container_mode --skip_registration"