Function runCodeGen

lib/v2/edge-runtime/ai.ts:898–962

(
  userRequest: string,
  actions: ActionPlusApiInfo[],
  org: Pick<
    Organization,
    | "id"
    | "name"
    | "description"
    | "chatbot_instructions"
    | "fun_loading_messages"
  >,
  dbData: { conversationId: number; index: number },
  userDescription: string,
  chatMessageCache: LlmResponseCache,
  streamInfo: (step: StreamingStepInput) => void,
  userApiKey: string | undefined,
)

}

async function runCodeGen(
  userRequest: string,
  actions: ActionPlusApiInfo[],
  org: Pick<
    Organization,
    | "id"
    | "name"
    | "description"
    | "chatbot_instructions"
    | "fun_loading_messages"
  >,
  dbData: { conversationId: number; index: number },
  userDescription: string,
  chatMessageCache: LlmResponseCache,
  streamInfo: (step: StreamingStepInput) => void,
  userApiKey: string | undefined,
): Promise<FunctionMessage[]> {
  console.log("Running data analysis!");
  const graphData = await runDataAnalysis(
    userRequest,
    actions,
    org,
    dbData,
    userDescription,
    chatMessageCache,
    streamInfo,
    userApiKey,
  );
  // Used in if statements further down
  const fnMsg: FunctionMessage = {
    role: "function",
    name: dataAnalysisActionName,
    content: "",
  };
  // Return graph data to the user & add message to chat history
  if (graphData === null) {
    fnMsg.content = "Failed to run data analysis";
    streamInfo(fnMsg);
    return [fnMsg];
  } else if ("error" in graphData) {
    // Handle error - add function message
    fnMsg.content = graphData.error;
    streamInfo(fnMsg);
    return [fnMsg];
  } else {
    // Stream API calls
    graphData
      .filter((m) => m.type === "call-human-format")
      .map((m) =>
        streamInfo({
          role: "debug",
          // @ts-ignore
          content: m.args.message,
        }),
      );
    const graphDataArr = convertToGraphData(graphData);
    graphDataArr.map(streamInfo);
    return graphDataArr.map((m) => ({