Function pullInMessageData

components/approval/verifyAnswerScreen.tsx:893–959

(
  answer_id: string,
  supabase: SupabaseClient<Database>,
)

}

async function pullInMessageData(
  answer_id: string,
  supabase: SupabaseClient<Database>,
): Promise<{
  answer: UIAnswerType;
  messagesData: UIMessageData[];
  messagesToViz: StreamingStepInput[];
}> {
  const { data, error } = await supabase
    .from("approval_answers")
    .select(
      "id,group_id,org_id,is_generating,generation_failed,approved, approval_questions(text, primary_question), approval_answer_messages(id, raw_text, message_idx, message_type, generated_output)",
    )
    .match({ id: answer_id })
    .single();
  if (error) throw new Error(error.message);
  const primaryQuestion = data.approval_questions.find(
    (q) => q.primary_question,
  );
  if (!primaryQuestion) throw new Error("No primary question found");

  const messagesToViz = data.approval_answer_messages
    .sort((a, b) => a.message_idx - b.message_idx)
    .map((m) => {
      if (m.message_type === "user") {
        const userText = getUserMessageText(
          primaryQuestion.text,
          JSON.parse(m.raw_text),
        );
        return { role: "user", content: userText } as UserMessage;
      } else if (m.message_type === "text")
        return { role: "assistant", content: m.raw_text } as AssistantMessage;
      else if (m.message_type === "code") {
        return m.generated_output.map((item) =>
          item &&
          typeof item === "object" &&
          "name" in item &&
          "content" in item &&
          item.name === "plot"
            ? {
                role: "graph",
                // @ts-ignore
                content: JSON.parse(item.content),
              }
            : item,
        );
      } else if (m.message_type === "error") {
        return { role: "error", content: m.raw_text };
      } else if (m.message_type === "function") {
        return m.generated_output.map(
          (item) =>
            item &&
            typeof item === "object" &&
            "name" in item &&
            "content" in item &&
            item,
        );
      }