feat: 支持 agent 运行取消功能

2026-03-25 18:33:25 +08:00
parent 599c597e69
commit 96fc4a1e77
21 changed files with 778 additions and 85 deletions
@@ -16,6 +16,7 @@ class _FakeApiClient implements IApiClient {
  final List<String> sseLines;
  final Stream<String> Function()? sseLineStreamFactory;
  final String Function()? runIdFactory;
+  final List<String> postPaths = <String>[];

  @override
  Future<Response<T>> delete<T>(String path, {data, Options? options}) {
@@ -51,6 +52,19 @@ class _FakeApiClient implements IApiClient {

  @override
  Future<Response<T>> post<T>(String path, {data, Options? options}) async {
+    postPaths.add(path);
+    if (path.contains('/cancel?runId=')) {
+      final payload = <String, dynamic>{
+        'threadId': 'thread-1',
+        'runId': 'run-new',
+        'accepted': true,
+      };
+      return Response<T>(
+        requestOptions: RequestOptions(path: path),
+        data: payload as T,
+        statusCode: 202,
+      );
+    }
    final runIdFactory = this.runIdFactory;
    final payload = <String, dynamic>{
      'taskId': 'task-1',
@@ -192,6 +206,39 @@ void main() {
    await streamController.close();
  });

+  test(
+    'cancelCurrentRun calls backend cancel endpoint for active run',
+    () async {
+      final streamController = StreamController<String>();
+      final fakeApi = _FakeApiClient(
+        sseLines: const <String>[],
+        sseLineStreamFactory: () => streamController.stream,
+      );
+      final service = AgUiService(apiClient: fakeApi);
+
+      final sendFuture = service.sendMessage('hello');
+      await Future<void>.delayed(Duration.zero);
+      for (final line in _buildSseEvent(
+        id: '51',
+        type: AgUiEventTypeWire.runStarted,
+        payload:
+            '{"type":"RUN_STARTED","threadId":"thread-1","runId":"run-new"}',
+      )) {
+        streamController.add(line);
+      }
+      await Future<void>.delayed(Duration.zero);
+
+      await service.cancelCurrentRun();
+      await sendFuture;
+
+      expect(
+        fakeApi.postPaths,
+        contains('/api/v1/agent/runs/thread-1/cancel?runId=run-new'),
+      );
+      await streamController.close();
+    },
+  );
+
  test(
    'new sendMessage cancels previous SSE subscription explicitly',
    () async {
@@ -206,6 +206,28 @@ void main() {
      expect(bloc.state.error, 'runtime execution failed');
    });

+    test('run canceled error clears error and marks tool as canceled', () {
+      service.emitEvent(
+        ToolCallStartEvent(
+          toolCallId: 'tool-cancel',
+          toolCallName: 'ocr_image',
+        ),
+      );
+      service.emitEvent(ToolCallEndEvent(toolCallId: 'tool-cancel'));
+
+      service.emitEvent(
+        RunErrorEvent(message: 'run canceled by user', code: 'RUN_CANCELED'),
+      );
+
+      final toolItem = bloc.state.items.whereType<ToolCallItem>().single;
+      expect(toolItem.status, ToolCallStatus.error);
+      expect(toolItem.errorMessage, '本次运行已取消');
+      expect(bloc.state.error, isNull);
+      expect(bloc.state.isWaitingFirstToken, isFalse);
+      expect(bloc.state.isStreaming, isFalse);
+      expect(bloc.state.isCancelling, isFalse);
+    });
+
    test('text event with ui schema is rendered into chat items', () {
      service.emitEvent(RunStartedEvent(threadId: 'thread-1', runId: 'run-1'));