Refactor manifest ES request creation

DataBiosphere · Mar 21, 2024 · 7a2ef5b · 7a2ef5b
1 parent bd68050
commit 7a2ef5b
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/azul/service/elasticsearch_service.py b/src/azul/service/elasticsearch_service.py
@@ -663,13 +663,14 @@ def create_chain(self,
 
     def create_request(self,
                        catalog: CatalogName,
-                       entity_type: str
+                       entity_type: str,
+                       doc_type: DocumentType = DocumentType.aggregate
                        ) -> Search:
         """
-        Create an Elasticsearch request against the index containing aggregate
-        documents for the given entity type in the given catalog.
+        Create an Elasticsearch request against the index containing documents
+        for the given entity+document type in the given catalog.
         """
         return Search(using=self._es_client,
                       index=str(IndexName.create(catalog=catalog,
                                                  qualifier=entity_type,
-                                                 doc_type=DocumentType.aggregate)))
+                                                 doc_type=doc_type)))