fix(chalice): fixed get heatmap with incomplete payload

tahayk · Jul 8, 2024 · 4b6f448 · 4b6f448
1 parent d241b0b
commit 4b6f448
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/api/chalicelib/core/custom_metrics.py b/api/chalicelib/core/custom_metrics.py
@@ -692,9 +692,12 @@ def make_chart_from_card(project_id, user_id, metric_id, data: schemas.CardSessi
                                                     data=data.model_dump())
     elif metric.metric_type == schemas.MetricType.heat_map:
         if raw_metric["data"] and raw_metric["data"].get("sessionId"):
-            return heatmaps.get_selected_session(project_id=project_id, session_id=raw_metric["data"]["sessionId"])
+            return heatmaps.get_selected_session(project_id=project_id,
+                                                 session_id=raw_metric["data"]["sessionId"])
         else:
-            return heatmaps.search_short_session(project_id=project_id, data=metric, user_id=user_id)
+            return heatmaps.search_short_session(project_id=project_id,
+                                                 data=schemas.HeatMapSessionsSearch(**metric.model_dump()),
+                                                 user_id=user_id)
 
     return get_chart(project_id=project_id, data=metric, user_id=user_id)
 

diff --git a/ee/api/chalicelib/core/custom_metrics.py b/ee/api/chalicelib/core/custom_metrics.py
@@ -739,9 +739,12 @@ def make_chart_from_card(project_id, user_id, metric_id, data: schemas.CardSessi
                                                     data=data.model_dump())
     elif metric.metric_type == schemas.MetricType.heat_map:
         if raw_metric["data"] and raw_metric["data"].get("sessionId"):
-            return heatmaps.get_selected_session(project_id=project_id, session_id=raw_metric["data"]["sessionId"])
+            return heatmaps.get_selected_session(project_id=project_id,
+                                                 session_id=raw_metric["data"]["sessionId"])
         else:
-            return heatmaps.search_short_session(project_id=project_id, data=metric, user_id=user_id)
+            return heatmaps.search_short_session(project_id=project_id,
+                                                 data=schemas.HeatMapSessionsSearch(**metric.model_dump()),
+                                                 user_id=user_id)
 
     return get_chart(project_id=project_id, data=metric, user_id=user_id)