Function runDataAnalysis

lib/edge-runtime/dataAnalysis.ts:70–211

(
  instruction: string,
  actions: ActionPlusApiInfo[],
  fullChatHistory: GPTMessageInclSummary[],
  org: Pick<Organization, "id" | "name" | "description">,
  dbData: { conversationId: number; index: number },
  cache: LlmResponseCache,
)

}

export async function runDataAnalysis(
  instruction: string,
  actions: ActionPlusApiInfo[],
  fullChatHistory: GPTMessageInclSummary[],
  org: Pick<Organization, "id" | "name" | "description">,
  dbData: { conversationId: number; index: number },
  cache: LlmResponseCache,
): Promise<GraphData | { error: string } | null> {
  // Step by step, go back through user-response sequences until we find the last function
  // message that's not a data analysis call
  const relevantMessages = getRelevantMessages(fullChatHistory);
  if (!relevantMessages) {
    console.info("No relevant messages found");
    return {
      error: "Cannot perform data analysis: no APIs called",
    };
  }

  // We want to get the output of all API calls since last user message from responseMessages:
  // First, split responseMessages into an array of pairs of assistant message & sequence of
  // function messages that follow it
  const assistantFnMessagePairs = getAssistantFnMessagePairs(relevantMessages);
  console.log("assistantFnMessagePairs", assistantFnMessagePairs);

  // Then convert to 'CalledAction' format
  const calledActions = getCalledActions(assistantFnMessagePairs, actions);
  console.log("calledActions", calledActions);
  const varNames = getVarNames(calledActions);
  console.log(
    "varNames",
    varNames,
    "number of calledActions:",
    calledActions.length,
  );
  const dataAnalysisPrompt = getDataAnalysisPrompt(
    instruction,
    calledActions,
    varNames,
    org,
  );

  let { llmResponse, graphData } = await cache.checkAnalyticsCache(
    dataAnalysisPrompt[1].content,
    org.id,
    fullChatHistory,
    supabase,
  );
  if (graphData) {
    await saveAnalyticsToDB(
      dataAnalysisPrompt[1].content,
      llmResponse,
      org.id,
      dbData,
    );
    return graphData as GraphData;
  }

  if (!llmResponse) {