1 files changed, 51 insertions, 0 deletions
diff --git a/frontend/lib/services/benchmark_service.dart b/frontend/lib/services/benchmark_service.dart
new file mode 100644
index 000000000..e58ae40c8
--- /dev/null
+++ b/frontend/lib/services/benchmark_service.dart
@@ -0,0 +1,51 @@
+import 'dart:async';
+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/utils/rest_api_utility.dart';
+import 'package:auto_gpt_flutter_client/models/benchmark/api_type.dart';
+
+class BenchmarkService {
+  final RestApiUtility api;
+
+  BenchmarkService(this.api);
+
+  /// Creates a new benchmark task.
+  ///
+  /// [benchmarkTaskRequestBody] is a Map representing the request body for creating a task.
+  Future<Map<String, dynamic>> createBenchmarkTask(
+      BenchmarkTaskRequestBody benchmarkTaskRequestBody) async {
+    try {
+      return await api.post('agent/tasks', benchmarkTaskRequestBody.toJson(),
+          apiType: ApiType.benchmark);
+    } catch (e) {
+      throw Exception('Failed to create a new task: $e');
+    }
+  }
+
+  /// Executes a step in a specific benchmark task.
+  ///
+  /// [taskId] is the ID of the task.
+  /// [benchmarkStepRequestBody] is a Map representing the request body for executing a step.
+  Future<Map<String, dynamic>> executeBenchmarkStep(
+      String taskId, BenchmarkStepRequestBody benchmarkStepRequestBody) async {
+    try {
+      return await api.post(
+          'agent/tasks/$taskId/steps', benchmarkStepRequestBody.toJson(),
+          apiType: ApiType.benchmark);
+    } catch (e) {
+      throw Exception('Failed to execute step: $e');
+    }
+  }
+
+  /// Triggers an evaluation for a specific benchmark task.
+  ///
+  /// [taskId] is the ID of the task.
+  Future<Map<String, dynamic>> triggerEvaluation(String taskId) async {
+    try {
+      return await api.post('agent/tasks/$taskId/evaluations', {},
+          apiType: ApiType.benchmark);
+    } catch (e) {
+      throw Exception('Failed to trigger evaluation: $e');
+    }
+  }
+}