Function formatPlotData

lib/v2/edge-runtime/dataAnalysis.ts:594–642

(plotMessage: {
  type: "plot";
  args: BertieGraphData;
})

}

export function formatPlotData(plotMessage: {
  type: "plot";
  args: BertieGraphData;
}): GraphMessage {
  let graphData = plotMessage.args;
  if (graphData.type !== "table") {
    graphData.data = ensureXandYinData(graphData);

    // Must ensure that the y value is a number
    graphData.data.forEach((item, idx) => {
      if (typeof item.y === "string") {
        const num = Number(item.y);
        if (!isNaN(num)) {
          item.y = num;
        }
      } else if (
        typeof item.y === "object" &&
        !Array.isArray(item.y) &&
        item.y !== null
      ) {
        // Find the first number in the object and use that as y
        const key = Object.keys(item.y).find(
          (k) => !isNaN(item.y[k]) && item.y[k] !== null,
        );
        if (key) {
          item.y.y = item.y[key];
          delete item.y[key];
        }
        graphData.data[idx] = { ...item, ...item.y };
      }
    });
  }
  return {
    role: "graph",
    content: {
      type:
        // Can have tables with only 1 data point
        graphData.data.length === 1 && graphData.type !== "table"
          ? "value"
          : graphData.type,
      // TODO: Fix below typing issues by making the functions type safe
      // @ts-ignore
      data: graphData.data,
      xLabel: graphData.labels?.x ?? "",
      yLabel: graphData.labels?.y ?? "",
      graphTitle: graphData.title,
    },
  };
}

export function ensureXandYinData(
  graphData: BertieGraphData,