From d91236dedabc60d10a62e448fbdc015c3a1735ab Mon Sep 17 00:00:00 2001
From: hunteraraujo <hunter_araujo@msn.com>
Date: Thu, 28 Sep 2023 23:59:14 -0700
Subject: [PATCH] Fix bug where benchmarks are not showing chat convo

---
 frontend/lib/viewmodels/skill_tree_viewmodel.dart | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/frontend/lib/viewmodels/skill_tree_viewmodel.dart b/frontend/lib/viewmodels/skill_tree_viewmodel.dart
index c1d52376..9358e27e 100644
--- a/frontend/lib/viewmodels/skill_tree_viewmodel.dart
+++ b/frontend/lib/viewmodels/skill_tree_viewmodel.dart
@@ -203,16 +203,17 @@ class SkillTreeViewModel extends ChangeNotifier {
             await benchmarkService.executeBenchmarkStep(
                 task.id, BenchmarkStepRequestBody(input: node.data.task));
         Step step = Step.fromMap(stepResponse);
+        chatViewModel.fetchChatsForTask();
 
         // Check if it's the last step
         while (!step.isLast) {
-          // Fetch chats for the task
-          chatViewModel.fetchChatsForTask();
-
           // Execute next step and update the Step object
           stepResponse = await benchmarkService.executeBenchmarkStep(
               task.id, BenchmarkStepRequestBody(input: null));
           step = Step.fromMap(stepResponse);
+
+          // Fetch chats for the task
+          chatViewModel.fetchChatsForTask();
         }
 
         // Trigger the evaluation