chore: expose search engine ping max time as a new environment variable

argilla-server/CHANGELOG.md

@@ -16,6 +16,10 @@ These are the section headers that we use:
 
 ## [Unreleased]()
 
+### Added
+
+- Added `ARGILLA_SEARCH_ENGINE_PING_TIMEOUT` new environment variable to specify a number of seconds that Argilla will wait for the search engine to be ready.
+
 ### Removed
 
 - Removed all API v0 endpoints. ([#4852](https://github.com/argilla-io/argilla/pull/4852))

argilla-server/src/argilla_server/_app.py

@@ -166,7 +166,7 @@ async def configure_opensearch():
         logging.getLogger("opensearch_transport").setLevel(logging.ERROR)
 
     @app.on_event("startup")
-    @backoff.on_exception(backoff.expo, ConnectionError, max_time=60)
+    @backoff.on_exception(backoff.expo, ConnectionError, max_time=settings.search_engine_ping_timeout)
     async def ping_search_engine():
         async for search_engine in get_search_engine():
             if not await search_engine.ping():

argilla-server/src/argilla_server/constants.py

@@ -18,6 +18,8 @@
 SEARCH_ENGINE_ELASTICSEARCH = "elasticsearch"
 SEARCH_ENGINE_OPENSEARCH = "opensearch"
 
+DEFAULT_SEARCH_ENGINE_PING_TIMEOUT = 60
+
 DEFAULT_USERNAME = "argilla"
 DEFAULT_PASSWORD = "1234"
 DEFAULT_API_KEY = "argilla.apikey"

argilla-server/src/argilla_server/settings.py

@@ -27,6 +27,7 @@
 from argilla_server.constants import (
     DEFAULT_LABEL_SELECTION_OPTIONS_MAX_ITEMS,
     DEFAULT_MAX_KEYWORD_LENGTH,
+    DEFAULT_SEARCH_ENGINE_PING_TIMEOUT,
     DEFAULT_SPAN_OPTIONS_MAX_ITEMS,
     DEFAULT_TELEMETRY_KEY,
     SEARCH_ENGINE_ELASTICSEARCH,
@@ -100,6 +101,10 @@ class Settings(BaseSettings):
     es_mapping_total_fields_limit: int = 2000
 
     search_engine: str = SEARCH_ENGINE_ELASTICSEARCH
+    search_engine_ping_timeout: int = Field(
+        default=DEFAULT_SEARCH_ENGINE_PING_TIMEOUT,
+        description="Maximum time in seconds that Argilla will wait for the search engine to respond when starting",
+    )
 
     vectors_fields_limit: int = Field(
         default=5,

docs/_source/getting_started/installation/configurations/server_configuration.md

@@ -73,6 +73,8 @@ You can set the following environment variables to further configure your server
 
 - `ARGILLA_SEARCH_ENGINE`: (Only for Feedback datasets) Search engine to use. Valid values are "elasticsearch" and "opensearch" (Default: "elasticsearch").
 
+- `ARGILLA_SEARCH_ENGINE_PING_TIMEOUT`: The maximum time in seconds that Argilla will wait for the search engine to be up and running (Default: `60`).
+
 - `ARGILLA_ELASTICSEARCH_SSL_VERIFY`: If "False", disables SSL certificate verification when connecting to the Elasticsearch backend.
 
 - `ARGILLA_ELASTICSEARCH_CA_PATH`: Path to CA cert for ES host. For example: `/full/path/to/root-ca.pem` (Optional)