aljaz/Auto-GPT
1
0
Fork 0
mirror of https://github.com/aljazceru/Auto-GPT.git synced 2025-12-18 14:34:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Refactor SkillTreeViewModel and Update TaskQueueView UI for Task Status (#5269)

Browse Source 
* Refactor SkillTreeViewModel and Update TaskQueueView UI for Task Status

* Notify UI when updating benchmark status
This commit is contained in:
hunteraraujo
2023-09-19 23:30:22 -07:00
committed by GitHub
parent 99035103e0
commit 377d0af228
3 changed files with 68 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
frontend/lib/viewmodels/skill_tree_viewmodel.dart
View File

@@ -1,6 +1,7 @@
import 'dart:convert';
import 'package:auto_gpt_flutter_client/models/benchmark/benchmark_step_request_body.dart';
import 'package:auto_gpt_flutter_client/models/benchmark/benchmark_task_request_body.dart';
import 'package:auto_gpt_flutter_client/models/benchmark/benchmark_task_status.dart';
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_edge.dart';
import 'package:auto_gpt_flutter_client/models/skill_tree/skill_tree_node.dart';
import 'package:auto_gpt_flutter_client/models/step.dart';
@@ -20,7 +21,7 @@ class SkillTreeViewModel extends ChangeNotifier {
// TODO: Potentially move to task queue view model when we create one
bool isBenchmarkRunning = false;
// TODO: Potentially move to task queue view model when we create one
List<Map<SkillTreeNode, bool>> benchmarkStatusList = [];
Map<SkillTreeNode, BenchmarkTaskStatus> benchmarkStatusMap = {};
List<SkillTreeNode> _skillTreeNodes = [];
List<SkillTreeEdge> _skillTreeEdges = [];
@@ -144,7 +145,7 @@ class SkillTreeViewModel extends ChangeNotifier {
Future<void> runBenchmark(
ChatViewModel chatViewModel, TaskViewModel taskViewModel) async {
// Clear the benchmarkStatusList
benchmarkStatusList.clear();
benchmarkStatusMap.clear();
// Create a new TestSuite object with the current timestamp
final testSuite =
@@ -159,12 +160,15 @@ class SkillTreeViewModel extends ChangeNotifier {
final reversedSelectedNodeHierarchy =
List.from(_selectedNodeHierarchy!.reversed);
for (var node in reversedSelectedNodeHierarchy) {
benchmarkStatusList.add({node: false});
benchmarkStatusMap[node] = BenchmarkTaskStatus.notStarted;
}
try {
// Loop through the nodes in the hierarchy
for (var node in reversedSelectedNodeHierarchy) {
benchmarkStatusMap[node] = BenchmarkTaskStatus.inProgress;
notifyListeners();
// Create a BenchmarkTaskRequestBody
final benchmarkTaskRequestBody = BenchmarkTaskRequestBody(
input: node.data.task, evalId: node.data.evalId);
@@ -204,10 +208,10 @@ class SkillTreeViewModel extends ChangeNotifier {
// Update the benchmarkStatusList based on the evaluation response
bool successStatus = evaluationResponse['metrics']['success'];
var nodeStatus = benchmarkStatusList.firstWhere(
(element) => element.keys.first.id == node.id,
);
nodeStatus[node] = successStatus;
benchmarkStatusMap[node] = successStatus
? BenchmarkTaskStatus.success
: BenchmarkTaskStatus.failure;
notifyListeners();
// If successStatus is false, break out of the loop
if (!successStatus) {