diff --git a/libs/partners/elasticsearch/tests/integration_tests/test_retrievers.py b/libs/partners/elasticsearch/tests/integration_tests/test_retrievers.py
index 79f9d8ef1d..59c02449b2 100644
--- a/libs/partners/elasticsearch/tests/integration_tests/test_retrievers.py
+++ b/libs/partners/elasticsearch/tests/integration_tests/test_retrievers.py
@@ -1,5 +1,6 @@
 """Test ElasticsearchRetriever functionality."""
 
+import os
 import re
 import uuid
 from typing import Any, Dict
@@ -77,11 +78,19 @@ class TestElasticsearchRetriever:
         def body_func(query: str) -> Dict:
             return {"query": {"match": {text_field: {"query": query}}}}
 
+        es_url = os.environ.get("ES_URL", "http://localhost:9200")
+        cloud_id = os.environ.get("ES_CLOUD_ID")
+        api_key = os.environ.get("ES_API_KEY")
+
+        config = (
+            {"cloud_id": cloud_id, "api_key": api_key} if cloud_id else {"url": es_url}
+        )
+
         retriever = ElasticsearchRetriever.from_es_params(
-            url="http://localhost:9200",
             index_name=index_name,
             body_func=body_func,
             content_field=text_field,
+            **config,  # type: ignore[arg-type]
         )
 
         index_test_data(retriever.es_client, index_name, text_field)