Function startTestJobsForModel

app/src/server/utils/nodes/startTestJobs.ts:157–193

({
  modelId,
  datasetId,
}: {
  modelId: string;
  datasetId: string;
})

// Used for both fine-tuned and comparison models
export const startTestJobsForModel = async ({
  modelId,
  datasetId,
}: {
  modelId: string;
  datasetId: string;
}) => {
  const dataset = await prisma.dataset.findUniqueOrThrow({
    where: {
      id: datasetId,
    },
  });

  const nodeEntriesToRun = await kysely
    .selectFrom("NodeEntry as ne")
    .innerJoin("DataChannel as dc", (join) =>
      join.onRef("dc.id", "=", "ne.dataChannelId").on("dc.destinationId", "=", dataset.nodeId),
    )
    .where("ne.status", "=", "PROCESSED")
    .where("ne.split", "=", "TEST")
    .leftJoin("FineTuneTestingEntry as ftte", (join) =>
      join.onRef("ftte.inputHash", "=", "ne.inputHash").on("ftte.modelId", "=", modelId),
    )
    .where("ftte.id", "is", null)
    .select(["ne.id as nodeEntryId"])
    .execute();

  await Promise.all(
    nodeEntriesToRun.map((nodeEntry) =>
      generateTestSetEntry.enqueue({
        modelId,
        nodeEntryId: nodeEntry.nodeEntryId,
        numPreviousTries: 0,
      }),
    ),
  );
};