From e555d13dd74a98de90d6f8e3055172c04145cd69 Mon Sep 17 00:00:00 2001
From: David Corbitt <davidlcorbitt@gmail.com>
Date: Mon, 10 Jul 2023 16:33:16 -0600
Subject: [PATCH] Limit prompt tokens to outputs from visible scenarios

---
 src/server/api/routers/promptVariants.router.ts | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/server/api/routers/promptVariants.router.ts b/src/server/api/routers/promptVariants.router.ts
index 7636107..12c384d 100644
--- a/src/server/api/routers/promptVariants.router.ts
+++ b/src/server/api/routers/promptVariants.router.ts
@@ -51,6 +51,7 @@ export const promptVariantsRouter = createTRPCRouter({
     const overallTokens = await prisma.modelOutput.aggregate({
       where: {
         promptVariantId: input.variantId,
+        testScenario: { visible: true },
       },
       _sum: {
         promptTokens: true,