b98eb9b729
* Rename tables, add graphile workers, update types * Add dev:worker command * Update pnpm-lock.yaml * Remove sentry config import from worker.ts * Stop generating new cells in cell router get query * Generate new cells for new scenarios, variants, and experiments * Remove most error throwing from queryLLM.task.ts * Remove promptVariantId and testScenarioId from ModelOutput * Remove duplicate index from ModelOutput * Move inputHash from cell to output * Add TODO * Add todo * Show cost and time for each cell * Always show output stats if there is output * Trigger LLM outputs when scenario variables are updated * Add newlines to ends of files * Add another newline * Cascade ModelOutput deletion * Fix linting and prettier * Return instead of throwing for non-pending cell * Remove pnpm dev:worker from pnpm:dev * Update pnpm-lock.yaml
23 lines
562 B
TypeScript
23 lines
562 B
TypeScript
import { prisma } from "../db";
|
|
import { queryLLM } from "../tasks/queryLLM.task";
|
|
|
|
export const queueLLMRetrievalTask = async (cellId: string) => {
|
|
const updatedCell = await prisma.scenarioVariantCell.update({
|
|
where: {
|
|
id: cellId,
|
|
},
|
|
data: {
|
|
retrievalStatus: "PENDING",
|
|
errorMessage: null,
|
|
},
|
|
include: {
|
|
modelOutput: true,
|
|
},
|
|
});
|
|
|
|
// @ts-expect-error we aren't passing the helpers but that's ok
|
|
void queryLLM.task.handler({ scenarioVariantCellId: cellId }, { logger: console });
|
|
|
|
return updatedCell;
|
|
};
|