From 1dae1a6be59c6d63151b931056fa3bbd303e33d3 Mon Sep 17 00:00:00 2001
From: Egor fureunoir Gorbunov <contact@fureunoir.com>
Date: Fri, 20 Jun 2025 04:10:55 +0300
Subject: [PATCH] Features: 1) Add `chunk_size` attribute to `_BaseDoc` class;
 2) Enable additional Elasticsearch configurations (`timeout`, `max_retries`,
 `retry_on_timeout`); 3) Set `ELASTICSEARCH_DSL_PARALLEL` to `True`;

Fixes: None;

Extra: None;
---
 core/elasticsearch/documents.py  | 2 ++
 evibes/settings/elasticsearch.py | 5 ++++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/core/elasticsearch/documents.py b/core/elasticsearch/documents.py
index 908ea9c9..8b55f4fb 100644
--- a/core/elasticsearch/documents.py
+++ b/core/elasticsearch/documents.py
@@ -6,6 +6,8 @@ from core.models import Brand, Category, Product
 
 
 class _BaseDoc(ActiveOnlyMixin, Document):
+    chunk_size = 300
+
     name = fields.TextField(
         attr="name",
         analyzer="standard",
diff --git a/evibes/settings/elasticsearch.py b/evibes/settings/elasticsearch.py
index 9cfeb456..b3cac42e 100644
--- a/evibes/settings/elasticsearch.py
+++ b/evibes/settings/elasticsearch.py
@@ -6,9 +6,12 @@ ELASTICSEARCH_DSL = {
         "basic_auth": ("elastic", getenv("ELASTIC_PASSWORD")),  # noqa: F405
         "verify_certs": False,
         "ssl_show_warn": False,
+        "timeout": 30,
+        "max_retries": 3,
+        "retry_on_timeout": True,
     },
 }
 
 ELASTICSEARCH_DSL_AUTOSYNC = True
-ELASTICSEARCH_DSL_PARALLEL = False
+ELASTICSEARCH_DSL_PARALLEL = True
 ELASTICSEARCH_DSL_SIGNAL_PROCESSOR = "evibes.signal_processors.SelectiveSignalProcessor"