chore: set logger level to error to reduce noise from Elasticsearch and OpenSearch client libraries

argilla-server/src/argilla_server/_app.py

@@ -66,7 +66,7 @@
     for app_configure in [
         configure_app_logging,
         configure_database,
-        ping_search_engine,
+        configure_search_engine,
         configure_telemetry,
         configure_middleware,
         configure_app_security,
@@ -148,18 +148,35 @@
     )
 
 
-def ping_search_engine(app: FastAPI):
+def configure_search_engine(app: FastAPI):
+    @app.on_event("startup")
+    async def configure_elasticsearch():
+        if not settings.search_engine_is_elasticsearch:
+            return
+
+        logging.getLogger("elasticsearch").setLevel(logging.ERROR)
+        logging.getLogger("elastic_transport").setLevel(logging.ERROR)
+
+    @app.on_event("startup")
+    async def configure_opensearch():
+        if not settings.search_engine_is_opensearch:
+            return
+
+        logging.getLogger("opensearch").setLevel(logging.ERROR)
+        logging.getLogger("opensearch_transport").setLevel(logging.ERROR)
+
     @app.on_event("startup")
     @backoff.on_exception(backoff.expo, ConnectionError, max_time=60)
-    async def _ping_search_engine():
+    async def ping_search_engine():
         async for search_engine in get_search_engine():
             if not await search_engine.ping():
                 raise ConnectionError(
-                    f"Your Elasticsearch endpoint at {settings.obfuscated_elasticsearch()} is not available or not responding.\n"
-                    "Please make sure your Elasticsearch instance is launched and correctly running and\n"
+                    f"Your {settings.humanized_search_engine} endpoint at {settings.obfuscated_elasticsearch()} is not available or not responding.\n"
+                    f"Please make sure your {settings.humanized_search_engine} instance is launched and correctly running and\n"
                     "you have the necessary access permissions. Once you have verified this, restart the argilla server.\n"
                 )
 
+
 def configure_app_security(app: FastAPI):
     auth.configure_app(app)
 

argilla-server/src/argilla_server/constants.py

@@ -15,6 +15,9 @@
 API_KEY_HEADER_NAME = "X-Argilla-Api-Key"
 WORKSPACE_HEADER_NAME = "X-Argilla-Workspace"
 
+SEARCH_ENGINE_ELASTICSEARCH = "elasticsearch"
+SEARCH_ENGINE_OPENSEARCH = "opensearch"
+
 DEFAULT_USERNAME = "argilla"
 DEFAULT_PASSWORD = "1234"
 DEFAULT_API_KEY = "argilla.apikey"

argilla-server/src/argilla_server/settings.py

@@ -21,14 +21,16 @@
 import re
 import warnings
 from pathlib import Path
-from typing import List, Optional
+from typing import List, Literal, Optional, Union
 from urllib.parse import urlparse
 
 from argilla_server.constants import (
     DEFAULT_LABEL_SELECTION_OPTIONS_MAX_ITEMS,
     DEFAULT_MAX_KEYWORD_LENGTH,
     DEFAULT_SPAN_OPTIONS_MAX_ITEMS,
     DEFAULT_TELEMETRY_KEY,
+    SEARCH_ENGINE_ELASTICSEARCH,
+    SEARCH_ENGINE_OPENSEARCH,
 )
 from argilla_server.pydantic_v1 import BaseSettings, Field, root_validator, validator
 
@@ -97,7 +99,10 @@
 
     es_mapping_total_fields_limit: int = 2000
 
-    search_engine: str = "elasticsearch"
+    search_engine: Union[
+        Literal[SEARCH_ENGINE_ELASTICSEARCH],
+        Literal[SEARCH_ENGINE_OPENSEARCH],
+    ] = SEARCH_ENGINE_ELASTICSEARCH
 
     vectors_fields_limit: int = Field(
         default=5,
@@ -217,6 +222,21 @@
             return index_name.replace("<NAMESPACE>", "")
         return index_name.replace("<NAMESPACE>", f".{ns}")
 
+    @property
+    def search_engine_is_elasticsearch(self) -> bool:
+        return self.search_engine == SEARCH_ENGINE_ELASTICSEARCH
+
+    @property
+    def search_engine_is_opensearch(self) -> bool:
+        return self.search_engine == SEARCH_ENGINE_OPENSEARCH
+
+    @property
+    def humanized_search_engine(self) -> str:
+        if self.search_engine_is_elasticsearch:
+            return "Elasticsearch"
+        elif self.search_engine_is_opensearch:
+            return "OpenSearch"
+
     def obfuscated_elasticsearch(self) -> str:
         """Returns configured elasticsearch url obfuscating the provided password, if any"""
         parsed = urlparse(self.elasticsearch)