Function getRelevantMessages

lib/edge-runtime/dataAnalysis.ts:273–302

(
  chatHistory: GPTMessageInclSummary[],
)

}

export function getRelevantMessages(
  chatHistory: GPTMessageInclSummary[],
): NonUserMessage[] | null {
  /** A semi-common error the LLM makes is not including the API calls required for
   * data analysis in the same output as the data analysis call. So if there are no
   * API calls we check previous query-response sections in the chat history. **/
  let lastUserIdx = -1;

  for (let i = 0; i < chatHistory.length; i++) {
    lastUserIdx = findLastIndex(
      chatHistory.slice(0, lastUserIdx),
      (m) => m.role === "user",
    );
    const relevantMessages = chatHistory
      .slice(lastUserIdx)
      .filter(
        (m) =>
          m.role !== "user" &&
          (m.role !== "function" || m.name !== dataAnalysisActionName),
      ) as NonUserMessage[];
    if (relevantMessages.filter((m) => m.role === "function").length > 0) {
      console.log("Found non-data analysis function call");
      return relevantMessages;
    } else if (lastUserIdx === 0) {
      // Terminate early if you reach the start of the chat history
      return null;
    }
  }
  return null;
}