Complete local hybrid search and improve agent reliability

SovereignAI-Starter/flutter_app/test/widget_test.dart

@@ -5,6 +5,7 @@ import 'package:record/record.dart';
 import 'dart:async';
 
 import 'package:flutter_app/core/network/api_repository.dart';
+import 'package:flutter_app/core/settings/local_settings_store.dart';
 import 'package:flutter_app/features/chat/presentation/cubit/chat_cubit.dart';
 import 'package:flutter_app/features/chat/presentation/cubit/chat_state.dart';
 import 'package:flutter_app/features/audio/presentation/cubit/voice_cubit.dart';
@@ -17,7 +18,37 @@ class _TestApiRepository extends ApiRepository {
   Future<String> getModelName() async => 'gemma4:e2b';
 
   @override
-  Future<List<String>> getAvailableModels() async => ['gemma4:e2b'];
+  Future<List<ModelDescriptor>> getAvailableModels() async => const [
+    ModelDescriptor(
+      id: 'gemma4:e2b',
+      capabilities: ['completion', 'vision', 'audio', 'tools'],
+      capabilitiesVerified: true,
+    ),
+  ];
+
+  @override
+  Future<List<AgentSkillDescriptor>> getAgentSkills() async => const [
+    AgentSkillDescriptor(
+      id: 'code_explain',
+      name: 'شرح الكود',
+      description: 'شرح المصدر.',
+    ),
+    AgentSkillDescriptor(
+      id: 'code_review',
+      name: 'مراجعة الكود',
+      description: 'مراجعة المصدر.',
+    ),
+    AgentSkillDescriptor(
+      id: 'test_plan',
+      name: 'خطة اختبارات',
+      description: 'اقتراح الاختبارات.',
+    ),
+  ];
+
+  @override
+  Future<List<String>> listWorkspaceFiles(String workspacePath) async => [
+    'lib/main.dart',
+  ];
 
   @override
   Future<List<SavedConversation>> listConversations() async => [];
@@ -39,6 +70,63 @@ class _CancellableWidgetApi extends _TestApiRepository {
   }
 }
 
+class _MarkdownWidgetApi extends _TestApiRepository {
+  @override
+  Future<void> saveConversation(SavedConversation conversation) async {}
+
+  @override
+  Stream<String> completeStream(List<ChatMessage> messages, {String? model}) =>
+      Stream.value('''| Feature | Status |
+| --- | --- |
+| Markdown | **ready** |
+
+1. Parent item
+   - Nested item
+- [x] GFM task''');
+}
+
+class _ProposalWidgetApi extends _TestApiRepository {
+  int proposalCount = 0;
+  int applyCount = 0;
+
+  @override
+  Future<void> saveConversation(SavedConversation conversation) async {}
+
+  @override
+  Future<List<String>> listWorkspaceFiles(String workspacePath) async => [
+    'src/hello.py',
+  ];
+
+  @override
+  Future<AgentRunResult> runAgent(
+    String task, {
+    String? model,
+    String? workspacePath,
+    List<String> workspaceFiles = const [],
+    String? skillId,
+    void Function(String message)? onProgress,
+  }) async {
+    onProgress?.call('يبني معاينة diff دون الكتابة إلى الملف.');
+    proposalCount++;
+    return AgentRunResult(
+      answer: 'حضّرت معاينة، ولم أكتب الملف.',
+      proposal: FileChangeProposal(
+        token: 'proposal-$proposalCount',
+        path: 'src/hello.py',
+        operation: 'create',
+        diff: '--- /dev/null\n+++ b/src/hello.py\n+print("hello")',
+        expiresInSeconds: 600,
+      ),
+    );
+  }
+
+  @override
+  Future<String> applyFileChange(FileChangeProposal proposal) async {
+    applyCount++;
+    return 'أُنشئ الملف ${proposal.path} بعد موافقتك.';
+  }
+}
+
 void main() {
   testWidgets('chat page fits a compact Windows window', (tester) async {
     tester.view.physicalSize = const Size(800, 600);
@@ -47,12 +135,16 @@ void main() {
     addTearDown(tester.view.resetDevicePixelRatio);
 
     final api = _TestApiRepository();
+    final chatCubit = ChatCubit(
+      api,
+      settingsStore: LocalSettingsStore.inMemory(),
+    );
     await tester.pumpWidget(
       MultiRepositoryProvider(
         providers: [RepositoryProvider<ApiRepository>.value(value: api)],
         child: MultiBlocProvider(
           providers: [
-            BlocProvider(create: (_) => ChatCubit(api)),
+            BlocProvider.value(value: chatCubit),
             BlocProvider(create: (_) => VoiceCubit(api, AudioRecorder())),
           ],
           child: const SovereignApp(),
@@ -63,6 +155,57 @@ void main() {
 
     expect(find.text('مساعدك المحلي'), findsOneWidget);
     expect(tester.takeException(), isNull);
+
+    await chatCubit.setDarkMode(true);
+    await tester.pumpAndSettle();
+    expect(
+      tester.widget<MaterialApp>(find.byType(MaterialApp)).themeMode,
+      ThemeMode.dark,
+    );
+    expect(tester.takeException(), isNull);
+    await chatCubit.close();
+  });
+
+  testWidgets('model selector displays capabilities reported by Ollama', (
+    tester,
+  ) async {
+    tester.view.physicalSize = const Size(1000, 700);
+    tester.view.devicePixelRatio = 1;
+    addTearDown(tester.view.resetPhysicalSize);
+    addTearDown(tester.view.resetDevicePixelRatio);
+
+    final api = _TestApiRepository();
+    final chatCubit = ChatCubit(
+      api,
+      settingsStore: LocalSettingsStore.inMemory(),
+    );
+    await tester.pumpWidget(
+      MultiRepositoryProvider(
+        providers: [RepositoryProvider<ApiRepository>.value(value: api)],
+        child: MultiBlocProvider(
+          providers: [
+            BlocProvider.value(value: chatCubit),
+            BlocProvider(create: (_) => VoiceCubit(api, AudioRecorder())),
+          ],
+          child: const SovereignApp(),
+        ),
+      ),
+    );
+    await tester.pumpAndSettle();
+    await chatCubit.selectWorkspace('C:/sample-project');
+    await tester.pumpAndSettle();
+    await tester.tap(find.byTooltip('اختيار مهارة للوكيل'));
+    await tester.pumpAndSettle();
+    expect(find.text('خطة اختبارات'), findsOneWidget);
+    await tester.tap(find.text('مراجعة الكود'));
+    await tester.pumpAndSettle();
+    expect(chatCubit.state.selectedSkillId, 'code_review');
+    await tester.tap(find.byType(DropdownButton<String>));
+    await tester.pumpAndSettle();
+
+    expect(find.text('نص · صور · صوت · أدوات'), findsOneWidget);
+    expect(tester.takeException(), isNull);
+    await chatCubit.close();
   });
 
   testWidgets('stop button cancels an active answer', (tester) async {
@@ -72,7 +215,7 @@ void main() {
     addTearDown(tester.view.resetDevicePixelRatio);
 
     final api = _CancellableWidgetApi();
-    final cubit = ChatCubit(api);
+    final cubit = ChatCubit(api, settingsStore: LocalSettingsStore.inMemory());
     await tester.pumpWidget(
       MultiRepositoryProvider(
         providers: [RepositoryProvider<ApiRepository>.value(value: api)],
@@ -103,4 +246,101 @@ void main() {
     await cubit.close();
     await api.controller.close();
   });
+
+  testWidgets('assistant renders tables and nested GFM task lists', (
+    tester,
+  ) async {
+    tester.view.physicalSize = const Size(1000, 700);
+    tester.view.devicePixelRatio = 1;
+    addTearDown(tester.view.resetPhysicalSize);
+    addTearDown(tester.view.resetDevicePixelRatio);
+
+    final api = _MarkdownWidgetApi();
+    await tester.pumpWidget(
+      MultiRepositoryProvider(
+        providers: [RepositoryProvider<ApiRepository>.value(value: api)],
+        child: MultiBlocProvider(
+          providers: [
+            BlocProvider(
+              create:
+                  (_) => ChatCubit(
+                    api,
+                    settingsStore: LocalSettingsStore.inMemory(),
+                  ),
+            ),
+            BlocProvider(create: (_) => VoiceCubit(api, AudioRecorder())),
+          ],
+          child: const SovereignApp(),
+        ),
+      ),
+    );
+    await tester.pumpAndSettle();
+
+    await tester.enterText(find.byType(TextField), 'اختبار Markdown');
+    await tester.tap(find.byTooltip('إرسال'));
+    await tester.pumpAndSettle();
+
+    expect(find.byType(Table), findsOneWidget);
+    expect(find.text('Feature'), findsOneWidget);
+    expect(
+      find.byWidgetPredicate(
+        (widget) =>
+            widget is RichText &&
+            widget.text.toPlainText().contains('Nested item'),
+      ),
+      findsOneWidget,
+    );
+    expect(find.text('☑'), findsOneWidget);
+    expect(tester.takeException(), isNull);
+  });
+
+  testWidgets('file changes require explicit review before apply', (
+    tester,
+  ) async {
+    tester.view.physicalSize = const Size(1100, 800);
+    tester.view.devicePixelRatio = 1;
+    addTearDown(tester.view.resetPhysicalSize);
+    addTearDown(tester.view.resetDevicePixelRatio);
+
+    final api = _ProposalWidgetApi();
+    final cubit = ChatCubit(api, settingsStore: LocalSettingsStore.inMemory());
+    await cubit.selectWorkspace(r'C:\Projects\sample');
+    cubit.selectWorkspaceFiles(['src/hello.py']);
+    await tester.pumpWidget(
+      MultiRepositoryProvider(
+        providers: [RepositoryProvider<ApiRepository>.value(value: api)],
+        child: MultiBlocProvider(
+          providers: [
+            BlocProvider<ChatCubit>.value(value: cubit),
+            BlocProvider(create: (_) => VoiceCubit(api, AudioRecorder())),
+          ],
+          child: const SovereignApp(),
+        ),
+      ),
+    );
+    await tester.pumpAndSettle();
+
+    Future<void> requestProposal() async {
+      await tester.enterText(find.byType(TextField), 'أنشئ ملف Python');
+      await tester.tap(find.byTooltip('إرسال'));
+      await tester.pumpAndSettle();
+    }
+
+    await requestProposal();
+    expect(find.text('مراجعة إنشاء الملف'), findsOneWidget);
+    expect(find.text('src/hello.py'), findsOneWidget);
+    expect(api.applyCount, 0);
+
+    await tester.tap(find.text('إلغاء'));
+    await tester.pumpAndSettle();
+    expect(api.applyCount, 0);
+
+    await requestProposal();
+    expect(find.text('مراجعة إنشاء الملف'), findsOneWidget);
+    await tester.tap(find.text('إنشاء الملف'));
+    await tester.pumpAndSettle();
+    expect(api.applyCount, 1);
+    expect(tester.takeException(), isNull);
+    await cubit.close();
+  });
 }