Function handler

pages/api/write-action-descriptions.ts:55–132

(req: NextRequest)

type WriteActionDescriptionType = z.infer<typeof WriteActionDescriptionZod>;

export default async function handler(req: NextRequest): Promise<Response> {
  if (req.method !== "POST") {
    return new Response("Only POST requests allowed", { status: 405, headers });
  }
  const reqBody = await req.json();
  if (
    !isValidBody<WriteActionDescriptionType>(reqBody, WriteActionDescriptionZod)
  ) {
    return new Response("Invalid request body", { status: 400, headers });
  }
  // Check key is service level key
  const authHeader = req.headers.get("Authorization");
  if (
    !authHeader ||
    !authHeader.includes(process.env.SERVICE_LEVEL_KEY_SUPABASE!)
  ) {
    return new Response("Unauthorized", { status: 401, headers });
  }
  await new Promise((resolve) => setTimeout(resolve, 1000));

  // Get the actions from the database
  const orgResponse = await supabase
    .from("actions")
    .select("*, organizations(name, description)")
    .eq("org_id", reqBody.org_id)
    .in("id", reqBody.action_ids);
  if (orgResponse.error) {
    console.error(orgResponse.error);
    return new Response(orgResponse.error.message, { status: 500, headers });
  }
  console.log(
    "Total actions found:",
    orgResponse.data.map((a) => a.name),
  );
  let actions = orgResponse.data.filter((a) => !a.filtering_description);
  // Limit to 40 actions max
  if (actions.length > 40) {
    actions = actions.filter((a) => a.active).slice(0, 40);
  }
  console.log(
    "Chosen action names:",
    actions.map((a) => a.name),
  );

  // Generate action filtering_descriptions
  await Promise.all(
    actions.map(async (action) => {
      const prompt = writeActionDescriptionPrompt({
        action,
        org: action.organizations!,
      });
      console.log("WRITE ACTIONS PROMPT:", prompt[0].content);
      const llmOut = await exponentialRetryWrapper(
        getLLMResponse,
        [
          prompt,
          writeActionDescriptionLLMParams,
          process.env.WRITE_ACTION_FILTERING_DESCRIPTION_MODEL!,