feat: remove pagination from queries that return one field

weni-ai · Jun 13, 2024 · 8cfdfa0 · 8cfdfa0
1 parent 3267645
commit 8cfdfa0
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/insights/sources/rooms/usecases/query_execute.py b/insights/sources/rooms/usecases/query_execute.py
@@ -31,5 +31,12 @@ def execute(
         with get_cursor(db_name="chats") as cur:
             query_exec = cur.execute(query, params)
             query_results = dictfetchall(query_exec)
-        paginated_results = {"next": None, "previous": None, "results": query_results}
+        if operation in ["count", "avg"]:
+            paginated_results = query_results
+        else:
+            paginated_results = {
+                "next": None,
+                "previous": None,
+                "results": query_results,
+            }
         return paginated_results  # parser(paginated_results)