From c77ade5b2f62c5373fc7573e5c45581f003c77a3 Mon Sep 17 00:00:00 2001
From: hunteraraujo <hunter_araujo@msn.com>
Date: Mon, 9 Oct 2023 11:30:58 -0700
Subject: [PATCH] Gate test suites from the task view based off
 isDeveloperModeEnabled flag

---
 frontend/lib/viewmodels/task_viewmodel.dart | 3 +++
 frontend/lib/views/chat/chat_view.dart      | 1 -
 frontend/lib/views/task/task_view.dart      | 6 +++++-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/frontend/lib/viewmodels/task_viewmodel.dart b/frontend/lib/viewmodels/task_viewmodel.dart
index 7bb8f689..b0804740 100644
--- a/frontend/lib/viewmodels/task_viewmodel.dart
+++ b/frontend/lib/viewmodels/task_viewmodel.dart
@@ -14,6 +14,7 @@ class TaskViewModel with ChangeNotifier {
   List<Task> _tasks = [];
   List<TestSuite> _testSuites = [];
   List<dynamic> combinedDataSource = [];
+  List<Task> tasksDataSource = [];
 
   Task? _selectedTask;
   TestSuite? _selectedTestSuite;
@@ -149,6 +150,7 @@ class TaskViewModel with ChangeNotifier {
 
     // Clear the existing combined data source to start fresh.
     combinedDataSource.clear();
+    tasksDataSource.clear();
 
     // Iterate through each task to check if it's contained in any of the test suites.
     for (var task in _tasks) {
@@ -188,6 +190,7 @@ class TaskViewModel with ChangeNotifier {
       // If the task was not found in any test suite, add it to the combined data source.
       if (!found) {
         combinedDataSource.add(task);
+        tasksDataSource.add(task);
       }
     }
 
diff --git a/frontend/lib/views/chat/chat_view.dart b/frontend/lib/views/chat/chat_view.dart
index f54d37c6..5f2b35c2 100644
--- a/frontend/lib/views/chat/chat_view.dart
+++ b/frontend/lib/views/chat/chat_view.dart
@@ -1,6 +1,5 @@
 import 'package:auto_gpt_flutter_client/models/message_type.dart';
 import 'package:auto_gpt_flutter_client/viewmodels/settings_viewmodel.dart';
-import 'package:auto_gpt_flutter_client/viewmodels/skill_tree_viewmodel.dart';
 import 'package:auto_gpt_flutter_client/viewmodels/task_queue_viewmodel.dart';
 import 'package:auto_gpt_flutter_client/viewmodels/task_viewmodel.dart';
 import 'package:auto_gpt_flutter_client/views/chat/agent_message_tile.dart';
diff --git a/frontend/lib/views/task/task_view.dart b/frontend/lib/views/task/task_view.dart
index 78b8e91c..fb426403 100644
--- a/frontend/lib/views/task/task_view.dart
+++ b/frontend/lib/views/task/task_view.dart
@@ -1,5 +1,6 @@
 import 'package:auto_gpt_flutter_client/models/task.dart';
 import 'package:auto_gpt_flutter_client/models/test_suite.dart';
+import 'package:auto_gpt_flutter_client/viewmodels/settings_viewmodel.dart';
 import 'package:auto_gpt_flutter_client/views/task/test_suite_detail_view.dart';
 import 'package:auto_gpt_flutter_client/views/task/test_suite_list_tile.dart';
 import 'package:flutter/material.dart';
@@ -32,7 +33,10 @@ class _TaskViewState extends State<TaskView> {
   @override
   Widget build(BuildContext context) {
     // Combine tasks and test suites into a single list
-    final items = widget.viewModel.combinedDataSource;
+    final items = Provider.of<SettingsViewModel>(context, listen: false)
+            .isDeveloperModeEnabled
+        ? widget.viewModel.combinedDataSource
+        : widget.viewModel.tasksDataSource;
     return Scaffold(
       backgroundColor: Colors.white,
       body: Stack(