From 3547223edb300fff5bee8a512d5d2b35afee4689 Mon Sep 17 00:00:00 2001
From: Akhil <akhil@athina.ai>
Date: Fri, 31 Jan 2025 23:18:12 +0530
Subject: [PATCH] Fixed query inside highlights

---
 athina/steps/search.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/athina/steps/search.py b/athina/steps/search.py
index f8267f1..4fcf866 100644
--- a/athina/steps/search.py
+++ b/athina/steps/search.py
@@ -87,7 +87,10 @@ def execute(self, input_data: Any) -> Union[Dict[str, Any], None]:
             "excludeText": self.excludeText,
             "includeText": self.includeText,
             "contents": {
-                "highlights": self.highlights,
+                "highlights": {
+                    "query": self.query,
+                    **(self.highlights or {})  # Merging self.highlights if it exists, otherwise an empty dict
+                },
                 "summary": {"query": self.query},
             },
             "startPublishedDate": self.startPublishedDate,