Merge pull request #4387 from freelawproject/4385-tweak-sweep-indexer…

…-supported-models 4385 Moved sweep indexer supported_models to settings.
freelawproject · Aug 30, 2024 · b641780 · b641780
2 parents b351cde + fb99c28
commit b641780
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/cl/search/management/commands/sweep_indexer.py b/cl/search/management/commands/sweep_indexer.py
@@ -27,14 +27,7 @@
 )
 from cl.search.types import ESDocumentClassType
 
-supported_models = [
-    "audio.Audio",
-    "people_db.Person",
-    "search.OpinionCluster",
-    "search.Opinion",
-    "search.Docket",
-    "search.RECAPDocument",
-]
+supported_models = settings.ELASTICSEARCH_SWEEP_INDEXER_MODELS  # type: ignore
 r = get_redis_interface("CACHE")
 
 

diff --git a/cl/settings/third_party/elasticsearch.py b/cl/settings/third_party/elasticsearch.py
@@ -248,6 +248,18 @@
 ELASTICSEARCH_SWEEP_INDEXER_HEADS_RATE = env(
     "ELASTICSEARCH_SWEEP_INDEXER_HEADS_RATE", default=60
 )
+ELASTICSEARCH_SWEEP_INDEXER_MODELS = env(
+    "ELASTICSEARCH_SWEEP_INDEXER_MODELS",
+    default=[
+        "audio.Audio",
+        "people_db.Person",
+        "search.OpinionCluster",
+        "search.Opinion",
+        "search.Docket",
+        "search.RECAPDocument",
+    ],
+)
+
 
 ELASTICSEARCH_MAX_RESULT_COUNT = 10_000
 ELASTICSEARCH_CARDINALITY_PRECISION = 2000