AI助手优化-人设与会话绑定

2026-09-29 10:44:39 +08:00
parent f4ad2f8612
commit 900d48697e
5 changed files with 464 additions and 150 deletions
@@ -1,4 +1,5 @@
 import 'package:dio/dio.dart';
+import 'package:flutter/material.dart';
 import 'package:flutter_riverpod/flutter_riverpod.dart';
 import 'package:flutter_test/flutter_test.dart';
 import 'package:shared_preferences/shared_preferences.dart';
@@ -6,6 +7,7 @@ import 'package:shared_preferences/shared_preferences.dart';
 import 'package:qiangbao_accounting/api/ai_api.dart';
 import 'package:qiangbao_accounting/models/ai.dart';
 import 'package:qiangbao_accounting/models/ai_persona.dart';
+import 'package:qiangbao_accounting/pages/assistant/assistant_page.dart';
 import 'package:qiangbao_accounting/providers/ai_provider.dart';
 import 'package:qiangbao_accounting/providers/ai_persona_provider.dart';

@@ -39,7 +41,35 @@ class _CapturingAiApi extends AiApi {
  Future<List<ChatSessionItem>> sessions() async => const [];

  @override
-  Future<List<AiPersonaItem>> personas() async => const [];
+  Future<List<ChatMessageItem>> messages(int sessionId) async => const [
+        ChatMessageItem(id: 1, role: 'user', content: '你好'),
+        ChatMessageItem(id: 2, role: 'assistant', content: '你好呀'),
+      ];
+
+  /// 两个人设，头像区分明显 —— 用于验证气泡上显示的是哪一个
+  @override
+  Future<List<AiPersonaItem>> personas() async => const [
+        AiPersonaItem(
+          id: 7,
+          name: '花花',
+          systemPrompt: '你是花花',
+          promptLength: 4,
+          isDefault: false,
+          editable: true,
+          defaultPrompt: null,
+          avatar: 'persona/huahua.jpg',
+        ),
+        AiPersonaItem(
+          id: 9,
+          name: '简洁',
+          systemPrompt: '只回一句',
+          promptLength: 4,
+          isDefault: false,
+          editable: true,
+          defaultPrompt: null,
+          avatar: 'persona/jianjie.jpg',
+        ),
+      ];
 }

 void main() {
@@ -57,7 +87,7 @@ void main() {
  }

  group('★ 会话内人设固定', () {
-    test('新对话选的人设会一直用下去（中途全局偏好变了也不受影响）', () async {
+    test('新对话选的人设会一直用下去（中途预选偏好变了也不受影响）', () async {
      final api = _CapturingAiApi();
      final c = makeContainer(api);
      final notifier = c.read(assistantProvider.notifier);
@@ -66,14 +96,17 @@ void main() {
      notifier.newChat(personaId: 7);
      expect(c.read(assistantProvider).personaId, 7);

-      // 中途用户在设置页把全局偏好换成了 9
-      c.read(selectedAiPersonaProvider.notifier).select(9);
+      // 之后他又开了一次新对话、选了人设 9（这是偏好，不是「当前会话的人设」）
+      c.read(preferredPersonaProvider.notifier).remember(9);
+
+      // 回到**第一个人设 7 的那条会话**继续说话。
+      // 用 openSession 模拟「切回历史会话」——它会用该会话自己的 personaId
+      await notifier.openSession(1, personaId: 7);

-      // 继续在这个会话里说话
      await notifier.send('第一句');
      await notifier.send('第二句');

-      // 两次都该用 7，不该跟着全局偏好变成 9
+      // 两次都该用 7，不该跟着偏好变成 9
      expect(api.personaIdsSent, [7, 7],
          reason: '会话锁定的人设不该被全局偏好覆盖');
    });
@@ -152,4 +185,143 @@ void main() {
      expect(s.personaName, '默认人设');
    });
  });
+
+  group('★ 气泡头像跟着会话的人设走', () {
+    /// 从渲染出来的 Image 里收集所有网络图片地址。
+    ///
+    /// 要拆一层 ResizeImage：Avatar 传了 cacheWidth，Flutter 会把
+    /// NetworkImage 包成 ResizeImage，直接 whereType<NetworkImage> 会取不到。
+    List<String> imageUrls(WidgetTester tester) {
+      final urls = <String>[];
+      for (final img in tester.widgetList<Image>(find.byType(Image))) {
+        final provider = img.image;
+        if (provider is NetworkImage) {
+          urls.add(provider.url);
+        } else if (provider is ResizeImage &&
+            provider.imageProvider is NetworkImage) {
+          urls.add((provider.imageProvider as NetworkImage).url);
+        }
+      }
+      return urls;
+    }
+
+    testWidgets('会话绑 7 → 气泡显示 7 的头像，不是预选偏好的 9', (tester) async {
+      final api = _CapturingAiApi();
+      await tester.pumpWidget(ProviderScope(
+        overrides: [aiApiProvider.overrideWithValue(api)],
+        child: const MaterialApp(home: AssistantPage()),
+      ));
+      await tester.pump();
+
+      final container = ProviderScope.containerOf(
+        tester.element(find.byType(AssistantPage)),
+      );
+
+      // 预选偏好是 9（比如用户上次新对话选了「简洁」）
+      container.read(preferredPersonaProvider.notifier).remember(9);
+
+      // 但打开的是绑定人设 7 的那条会话
+      await container.read(assistantProvider.notifier).openSession(1, personaId: 7);
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 300));
+
+      final urls = imageUrls(tester);
+      expect(urls.any((u) => u.contains('huahua')), isTrue,
+          reason: '气泡该显示会话绑定的人设（7=花花）的头像');
+      expect(urls.any((u) => u.contains('jianjie')), isFalse,
+          reason: '不该显示预选偏好那个人（9=简洁）的头像 —— 那是另一个会话的人');
+
+      expect(tester.takeException(), isNull);
+    });
+
+    testWidgets('会话没绑人设（内置默认）时不加载任何头像图，退回默认图标', (tester) async {
+      final api = _CapturingAiApi();
+      await tester.pumpWidget(ProviderScope(
+        overrides: [aiApiProvider.overrideWithValue(api)],
+        child: const MaterialApp(home: AssistantPage()),
+      ));
+      await tester.pump();
+
+      final container = ProviderScope.containerOf(
+        tester.element(find.byType(AssistantPage)),
+      );
+      // 预选偏好 9，但会话用的是内置默认（personaId=null）
+      container.read(preferredPersonaProvider.notifier).remember(9);
+      await container.read(assistantProvider.notifier).openSession(1);
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 300));
+
+      final urls = imageUrls(tester);
+      expect(urls.any((u) => u.contains('jianjie')), isFalse,
+          reason: '会话用内置默认人设时，不该借预选偏好那个人来用');
+
+      expect(tester.takeException(), isNull);
+    });
+  });
+
+  group('★ 新建对话：选了才开，关掉就是取消', () {
+    Future<void> openAssistant(WidgetTester tester, _CapturingAiApi api) async {
+      await tester.pumpWidget(ProviderScope(
+        overrides: [aiApiProvider.overrideWithValue(api)],
+        child: const MaterialApp(home: AssistantPage()),
+      ));
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 300));
+    }
+
+    testWidgets('点「新对话」会弹出选人设；选一个 → 该会话锁定它', (tester) async {
+      final api = _CapturingAiApi();
+      await openAssistant(tester, api);
+      final container = ProviderScope.containerOf(
+        tester.element(find.byType(AssistantPage)),
+      );
+
+      await tester.tap(find.byTooltip('新对话'));
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 400));
+
+      // 弹层里应该有两个人设
+      expect(find.text('这次想找谁聊？'), findsOneWidget);
+      expect(find.text('花花'), findsOneWidget);
+      expect(find.text('简洁'), findsOneWidget);
+
+      await tester.tap(find.text('简洁'));
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 400));
+
+      expect(container.read(assistantProvider).personaId, 9,
+          reason: '选了「简洁」就该锁定 9');
+      expect(tester.takeException(), isNull);
+    });
+
+    testWidgets('★ 关掉弹层 = 取消，不会偷偷开一个新对话', (tester) async {
+      final api = _CapturingAiApi();
+      await openAssistant(tester, api);
+      final container = ProviderScope.containerOf(
+        tester.element(find.byType(AssistantPage)),
+      );
+
+      // 先打开一条历史会话，制造「当前有对话」的状态
+      await container.read(assistantProvider.notifier).openSession(1, personaId: 7);
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 300));
+      expect(container.read(assistantProvider).sessionId, 1);
+
+      await tester.tap(find.byTooltip('新对话'));
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 400));
+      expect(find.text('这次想找谁聊？'), findsOneWidget);
+
+      // 点弹层外面（左上角）关掉它
+      await tester.tapAt(const Offset(5, 5));
+      await tester.pump();
+      await tester.pump(const Duration(milliseconds: 400));
+
+      // 当前会话不该被清掉 —— 误触一下不该丢掉正在看的对话
+      expect(container.read(assistantProvider).sessionId, 1,
+          reason: '取消选人设时不该开新对话');
+      expect(container.read(assistantProvider).personaId, 7);
+      expect(tester.takeException(), isNull);
+    });
+  });
 }