From 8927bb0c903fcf2c1299a2127ca95302e93f5f15 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Damir=20Jeli=C4=87?= Date: Mon, 10 Jun 2019 15:53:19 +0200 Subject: [PATCH] client: Fix the search response format. --- pantalaimon/client.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/pantalaimon/client.py b/pantalaimon/client.py index facf8f5..793803c 100644 --- a/pantalaimon/client.py +++ b/pantalaimon/client.py @@ -644,7 +644,13 @@ class PanClient(AsyncClient): result_dict["results"].append(event_dict) result_dict["count"] = len(result_dict["results"]) - result_dict["highlight"] = [] - result_dict["state"] = state_cache + result_dict["highlights"] = [] - return result_dict + if include_state: + result_dict["state"] = state_cache + + return { + "search_categories": { + "room_events": result_dict + } + }