fix[visualize-tab]: add support nested object path in color_by para…

…meter for visualiztion tab fixes : #251
qdrant · Oct 26, 2024 · ebac47f · ebac47f
1 parent 681c506
commit ebac47f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/components/VisualizeChart/renderBy.js b/src/components/VisualizeChart/renderBy.js
@@ -66,12 +66,16 @@ export function generateColorBy(points, colorBy = null) {
     colorBy = { payload: colorBy };
   }
 
+  function getNestedValue(obj, path) {
+    return path.split('.').reduce((acc, part) => acc && acc[part], obj);
+  }
+
   if (colorBy.payload) {
     const valuesToColor = {};
 
     return points.map((point) => {
-      const payloadValue = point.payload[colorBy.payload];
-      if (!payloadValue) {
+      const payloadValue = getNestedValue(point.payload, colorBy.payload);
+      if (payloadValue === undefined || payloadValue === null) {
         return BACKGROUND_COLOR;
       }
       return colorByPayload(payloadValue, valuesToColor);