feat: Expose ES highlight option (#91)

Signed-off-by: Michele Dolfi <dol@zurich.ibm.com>
DS4SD · Jun 1, 2023 · 8e65cbf · 8e65cbf
1 parent a8cc296
commit 8e65cbf
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