From 47de6805e1e42041f04f2124c6bdab58123d5119 Mon Sep 17 00:00:00 2001
From: tellet-q <elena.dubrovina@qdrant.com>
Date: Thu, 18 Apr 2024 11:45:16 +0200
Subject: [PATCH] Fix wait for ES and OS

---
 .github/workflows/actions/run-engine-benchmark/action.yaml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/.github/workflows/actions/run-engine-benchmark/action.yaml b/.github/workflows/actions/run-engine-benchmark/action.yaml
index 8339aa33..b8dea2de 100644
--- a/.github/workflows/actions/run-engine-benchmark/action.yaml
+++ b/.github/workflows/actions/run-engine-benchmark/action.yaml
@@ -30,5 +30,8 @@ runs:
     - name: Execution
       shell: bash
       run: |
+        flag="${{ inputs.engine }}"
+        if [[ "$flag" == *"elasticsearch"* || "$flag" == *"opensearch"* ]]; then
+          ./tools/wait_for_green_status.sh
         source $(poetry env info -p)/bin/activate
         poetry run python3 run.py --engines "${{ inputs.engine }}" --datasets "${{ inputs.dataset }}"
\ No newline at end of file