From 97f44ea5c42b89a86b527f3b58c7daa2d2a15bd3 Mon Sep 17 00:00:00 2001
From: Kyle Corbitt <kyle@corbt.com>
Date: Thu, 6 Jul 2023 17:56:02 -0700
Subject: [PATCH] handle issues with OutputCell mutation

---
 src/components/OutputsTable/OutputCell.tsx | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/components/OutputsTable/OutputCell.tsx b/src/components/OutputsTable/OutputCell.tsx
index 5594466..892c092 100644
--- a/src/components/OutputsTable/OutputCell.tsx
+++ b/src/components/OutputsTable/OutputCell.tsx
@@ -59,6 +59,7 @@ export default function OutputCell({
   const [output, setOutput] = useState<RouterOutputs["outputs"]["get"]>(null);
 
   const [fetchOutput, fetchingOutput] = useHandledAsyncCallback(async () => {
+    setOutput(null);
     const output = await outputMutation.mutateAsync({
       scenarioId: scenario.id,
       variantId: variant.id,
@@ -68,7 +69,7 @@ export default function OutputCell({
     await utils.promptVariants.stats.invalidate();
   }, [outputMutation, scenario.id, variant.id, channel]);
 
-  useEffect(fetchOutput, []);
+  useEffect(fetchOutput, [scenario.id, variant.id, channel]);
 
   // Disconnect from socket if we're not streaming anymore
   const streamedMessage = useSocket(fetchingOutput ? channel : undefined);