feat: Expose ES highlight option (DS4SD#91)

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
HolyMichael · Jun 1, 2023 · db99f95 · db99f95
1 parent cc2b11b
commit db99f95
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/deepsearch/cps/queries/__init__.py b/deepsearch/cps/queries/__init__.py
@@ -33,6 +33,7 @@ def DataQuery(
     *,
     source: Optional[List[str]] = None,
     aggregations: Optional[dict] = None,
+    highlight: Optional[dict] = None,
     sort: Optional[List[Dict[str, Any]]] = None,
     limit: int = 20,
     search_after: Optional[List[str]] = None,
@@ -57,6 +58,9 @@ def DataQuery(
     if aggregations is not None:
         payload["aggregations"] = aggregations
 
+    if highlight is not None:
+        payload["highlight"] = highlight
+
     if search_after is not None:
         payload["search_after"] = search_after