refactor: migrate to riverpod 3

2025-09-21 22:31:44 +05:30
parent 37e5633c5c
commit 462bf4cde2
20 changed files with 834 additions and 453 deletions
--- a/lib/core/auth/auth_state_manager.dart
+++ b/lib/core/auth/auth_state_manager.dart
@@ -78,23 +78,28 @@ enum AuthStatus {
 }

 /// Unified auth state manager - single source of truth for all auth operations
-class AuthStateManager extends StateNotifier<AuthState> {
-  AuthStateManager(this._ref)
-    : super(const AuthState(status: AuthStatus.initial)) {
-    _initialize();
-  }
-
-  final Ref _ref;
+class AuthStateManager extends Notifier<AuthState> {
  final AuthCacheManager _cacheManager = AuthCacheManager();
  // Prevent overlapping silent-login attempts from multiple triggers
  Future<bool>? _silentLoginFuture;
+  bool _initialized = false;
+
+  @override
+  AuthState build() {
+    if (!_initialized) {
+      _initialized = true;
+      Future.microtask(_initialize);
+    }
+
+    return const AuthState(status: AuthStatus.initial);
+  }

  /// Initialize auth state from storage
  Future<void> _initialize() async {
    state = state.copyWith(status: AuthStatus.loading, isLoading: true);

    try {
-      final storage = _ref.read(optimizedStorageServiceProvider);
+      final storage = ref.read(optimizedStorageServiceProvider);
      final token = await storage.getAuthToken();

      if (token != null && token.isNotEmpty) {
@@ -171,7 +176,7 @@ class AuthStateManager extends StateNotifier<AuthState> {

      // Ensure API service is available
      await _ensureApiServiceAvailable();
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api == null) {
        throw Exception('No server connection available');
      }
@@ -192,12 +197,12 @@ class AuthStateManager extends StateNotifier<AuthState> {
        await api.getCurrentUser(); // Just validate, don't store user data yet

        // Save token to storage
-        final storage = _ref.read(optimizedStorageServiceProvider);
+        final storage = ref.read(optimizedStorageServiceProvider);
        await storage.saveAuthToken(tokenStr);

        // Save API key if requested (for convenience, though less secure than credentials)
        if (rememberCredentials) {
-          final activeServer = await _ref.read(activeServerProvider.future);
+          final activeServer = await ref.read(activeServerProvider.future);
          if (activeServer != null) {
            // Store API key as a special credential type
            await storage.saveCredentials(
@@ -260,7 +265,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
    try {
      // Ensure API service is available (active server/provider rebuild race)
      await _ensureApiServiceAvailable();
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api == null) {
        throw Exception('No server connection available');
      }
@@ -280,12 +285,12 @@ class AuthStateManager extends StateNotifier<AuthState> {
      }

      // Save token to storage
-      final storage = _ref.read(optimizedStorageServiceProvider);
+      final storage = ref.read(optimizedStorageServiceProvider);
      await storage.saveAuthToken(tokenStr);

      // Save credentials if requested
      if (rememberCredentials) {
-        final activeServer = await _ref.read(activeServerProvider.future);
+        final activeServer = await ref.read(activeServerProvider.future);
        if (activeServer != null) {
          await storage.saveCredentials(
            serverId: activeServer.id,
@@ -332,7 +337,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
  }) async {
    final end = DateTime.now().add(timeout);
    while (DateTime.now().isBefore(end)) {
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api != null) return;
      await Future.delayed(const Duration(milliseconds: 50));
    }
@@ -363,7 +368,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
    );

    try {
-      final storage = _ref.read(optimizedStorageServiceProvider);
+      final storage = ref.read(optimizedStorageServiceProvider);
      final savedCredentials = await storage.getSavedCredentials();

      if (savedCredentials == null) {
@@ -381,10 +386,10 @@ class AuthStateManager extends StateNotifier<AuthState> {

      // Set active server if needed
      await storage.setActiveServerId(serverId);
-      _ref.invalidate(activeServerProvider);
+      ref.invalidate(activeServerProvider);

      // Wait for server connection
-      final activeServer = await _ref.read(activeServerProvider.future);
+      final activeServer = await ref.read(activeServerProvider.future);
      if (activeServer == null) {
        await storage.setActiveServerId(null);
        state = state.copyWith(
@@ -411,7 +416,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
          e.toString().contains('403') ||
          e.toString().contains('authentication') ||
          e.toString().contains('unauthorized')) {
-        final storage = _ref.read(optimizedStorageServiceProvider);
+        final storage = ref.read(optimizedStorageServiceProvider);
        await storage.deleteSavedCredentials();
      }

@@ -434,7 +439,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
    }

    // Clear token from storage
-    final storage = _ref.read(optimizedStorageServiceProvider);
+    final storage = ref.read(optimizedStorageServiceProvider);
    await storage.deleteAuthToken();

    // Update state
@@ -461,7 +466,7 @@ class AuthStateManager extends StateNotifier<AuthState> {

    try {
      // Call server logout if possible
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api != null) {
        try {
          await api.logout();
@@ -471,7 +476,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
      }

      // Clear all local auth data
-      final storage = _ref.read(optimizedStorageServiceProvider);
+      final storage = ref.read(optimizedStorageServiceProvider);
      await storage.clearAuthData();

      // Update state
@@ -524,7 +529,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
  /// Load complete user data from server
  Future<void> _loadServerUserData() async {
    try {
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api != null && state.isAuthenticated) {
        // Check if we already have user data from token validation
        if (state.user != null) {
@@ -546,7 +551,7 @@ class AuthStateManager extends StateNotifier<AuthState> {

  /// Update API service with current token
  void _updateApiServiceToken(String token) {
-    final api = _ref.read(apiServiceProvider);
+    final api = ref.read(apiServiceProvider);
    api?.updateAuthToken(token);
  }

@@ -582,7 +587,7 @@ class AuthStateManager extends StateNotifier<AuthState> {

    // Server validation (async with timeout)
    try {
-      final api = _ref.read(apiServiceProvider);
+      final api = ref.read(apiServiceProvider);
      if (api == null) {
        debugPrint('DEBUG: No API service available for token validation');
        return formatResult.isValid; // Fall back to format validation
@@ -630,7 +635,7 @@ class AuthStateManager extends StateNotifier<AuthState> {
    }

    try {
-      final storage = _ref.read(optimizedStorageServiceProvider);
+      final storage = ref.read(optimizedStorageServiceProvider);
      final hasCredentials = await storage.hasCredentials();

      // Cache the result
@@ -668,10 +673,9 @@ class AuthStateManager extends StateNotifier<AuthState> {
 }

 /// Provider for the unified auth state manager
-final authStateManagerProvider =
-    StateNotifierProvider<AuthStateManager, AuthState>((ref) {
-      return AuthStateManager(ref);
-    });
+final authStateManagerProvider = NotifierProvider<AuthStateManager, AuthState>(
+  AuthStateManager.new,
+);

 /// Computed providers for common auth state queries
 final isAuthenticatedProvider = Provider<bool>((ref) {
--- a/lib/core/providers/app_providers.dart
+++ b/lib/core/providers/app_providers.dart
@@ -63,28 +63,24 @@ final optimizedStorageServiceProvider = Provider<OptimizedStorageService>((
 });

 // Theme provider
-final themeModeProvider = StateNotifierProvider<ThemeModeNotifier, ThemeMode>((
-  ref,
-) {
-  final storage = ref.watch(optimizedStorageServiceProvider);
-  return ThemeModeNotifier(storage);
-});
+final themeModeProvider = NotifierProvider<ThemeModeNotifier, ThemeMode>(
+  ThemeModeNotifier.new,
+);

-class ThemeModeNotifier extends StateNotifier<ThemeMode> {
-  final OptimizedStorageService _storage;
+class ThemeModeNotifier extends Notifier<ThemeMode> {
+  late final OptimizedStorageService _storage;

-  ThemeModeNotifier(this._storage) : super(ThemeMode.system) {
-    _loadTheme();
-  }
-
-  void _loadTheme() {
-    final mode = _storage.getThemeMode();
-    if (mode != null) {
-      state = ThemeMode.values.firstWhere(
-        (e) => e.toString() == mode,
+  @override
+  ThemeMode build() {
+    _storage = ref.watch(optimizedStorageServiceProvider);
+    final storedMode = _storage.getThemeMode();
+    if (storedMode != null) {
+      return ThemeMode.values.firstWhere(
+        (e) => e.toString() == storedMode,
        orElse: () => ThemeMode.system,
      );
    }
+    return ThemeMode.system;
  }

  void setTheme(ThemeMode mode) {
@@ -94,25 +90,21 @@ class ThemeModeNotifier extends StateNotifier<ThemeMode> {
 }

 // Locale provider
-final localeProvider = StateNotifierProvider<LocaleNotifier, Locale?>((ref) {
-  final storage = ref.watch(optimizedStorageServiceProvider);
-  return LocaleNotifier(storage);
-});
+final localeProvider = NotifierProvider<LocaleNotifier, Locale?>(
+  LocaleNotifier.new,
+);

-class LocaleNotifier extends StateNotifier<Locale?> {
-  final OptimizedStorageService _storage;
+class LocaleNotifier extends Notifier<Locale?> {
+  late final OptimizedStorageService _storage;

-  LocaleNotifier(this._storage) : super(null) {
-    _loadLocale();
-  }
-
-  void _loadLocale() {
+  @override
+  Locale? build() {
+    _storage = ref.watch(optimizedStorageServiceProvider);
    final code = _storage.getLocaleCode();
    if (code != null && code.isNotEmpty) {
-      state = Locale(code);
-    } else {
-      state = null; // system
+      return Locale(code);
    }
+    return null; // system default
  }

  Future<void> setLocale(Locale? locale) async {
@@ -325,17 +317,38 @@ final modelsProvider = FutureProvider<List<Model>>((ref) async {
  }
 });

-final selectedModelProvider = StateProvider<Model?>((ref) => null);
+final selectedModelProvider = NotifierProvider<SelectedModelNotifier, Model?>(
+  SelectedModelNotifier.new,
+);

 // Track if the current model selection is manual (user-selected) or automatic (default)
-final isManualModelSelectionProvider = StateProvider<bool>((ref) => false);
+final isManualModelSelectionProvider =
+    NotifierProvider<IsManualModelSelectionNotifier, bool>(
+      IsManualModelSelectionNotifier.new,
+    );
+
+class SelectedModelNotifier extends Notifier<Model?> {
+  @override
+  Model? build() => null;
+
+  void set(Model? model) => state = model;
+
+  void clear() => state = null;
+}
+
+class IsManualModelSelectionNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}

 // Listen for settings changes and reset manual selection when default model changes
 final _settingsWatcherProvider = Provider<void>((ref) {
  ref.listen<AppSettings>(appSettingsProvider, (previous, next) {
    if (previous?.defaultModel != next.defaultModel) {
      // Reset manual selection when default model changes
-      ref.read(isManualModelSelectionProvider.notifier).state = false;
+      ref.read(isManualModelSelectionProvider.notifier).set(false);
    }
  });
 });
@@ -376,7 +389,7 @@ final defaultModelAutoSelectionProvider = Provider<void>((ref) {
            (models.isNotEmpty ? models.first : null);

        if (selected != null) {
-          ref.read(selectedModelProvider.notifier).state = selected;
+          ref.read(selectedModelProvider.notifier).set(selected);
          foundation.debugPrint(
            'DEBUG: Auto-applied default model (by ID): ${selected.name}',
          );
@@ -391,7 +404,17 @@ final defaultModelAutoSelectionProvider = Provider<void>((ref) {
 });

 // Cache timestamp for conversations to prevent rapid re-fetches
-final _conversationsCacheTimestamp = StateProvider<DateTime?>((ref) => null);
+final _conversationsCacheTimestamp =
+    NotifierProvider<_ConversationsCacheTimestampNotifier, DateTime?>(
+      _ConversationsCacheTimestampNotifier.new,
+    );
+
+class _ConversationsCacheTimestampNotifier extends Notifier<DateTime?> {
+  @override
+  DateTime? build() => null;
+
+  void set(DateTime? timestamp) => state = timestamp;
+}

 // Conversation providers - Now using correct OpenWebUI API with caching
 final conversationsProvider = FutureProvider<List<Conversation>>((ref) async {
@@ -585,7 +608,7 @@ final conversationsProvider = FutureProvider<List<Conversation>>((ref) async {
      );

      // Update cache timestamp
-      ref.read(_conversationsCacheTimestamp.notifier).state = DateTime.now();
+      ref.read(_conversationsCacheTimestamp.notifier).set(DateTime.now());

      return sortedConversations;
    } catch (e) {
@@ -597,7 +620,7 @@ final conversationsProvider = FutureProvider<List<Conversation>>((ref) async {
      );

      // Update cache timestamp
-      ref.read(_conversationsCacheTimestamp.notifier).state = DateTime.now();
+      ref.read(_conversationsCacheTimestamp.notifier).set(DateTime.now());

      return conversations; // Return original conversations if folder fetch fails
    }
@@ -618,7 +641,19 @@ final conversationsProvider = FutureProvider<List<Conversation>>((ref) async {
  }
 });

-final activeConversationProvider = StateProvider<Conversation?>((ref) => null);
+final activeConversationProvider =
+    NotifierProvider<ActiveConversationNotifier, Conversation?>(
+      ActiveConversationNotifier.new,
+    );
+
+class ActiveConversationNotifier extends Notifier<Conversation?> {
+  @override
+  Conversation? build() => null;
+
+  void set(Conversation? conversation) => state = conversation;
+
+  void clear() => state = null;
+}

 // Provider to load full conversation with messages
 final loadConversationProvider = FutureProvider.family<Conversation, String>((
@@ -662,7 +697,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
    if (models.isNotEmpty) {
      final defaultModel = models.first;
      if (!ref.read(isManualModelSelectionProvider)) {
-        ref.read(selectedModelProvider.notifier).state = defaultModel;
+        ref.read(selectedModelProvider.notifier).set(defaultModel);
        foundation.debugPrint(
          'DEBUG: Auto-selected demo model: ${defaultModel.name}',
        );
@@ -692,7 +727,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
            name: storedDefaultId,
            supportsStreaming: true,
          );
-          ref.read(selectedModelProvider.notifier).state = placeholder;
+          ref.read(selectedModelProvider.notifier).set(placeholder);
        }
        // Reconcile against real models in background
        Future.microtask(() async {
@@ -709,7 +744,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
            }
            resolved ??= models.isNotEmpty ? models.first : null;
            if (resolved != null && !ref.read(isManualModelSelectionProvider)) {
-              ref.read(selectedModelProvider.notifier).state = resolved;
+              ref.read(selectedModelProvider.notifier).set(resolved);
              foundation.debugPrint(
                'DEBUG: Reconciled default model to ${resolved.name}',
              );
@@ -730,7 +765,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
            name: serverDefault,
            supportsStreaming: true,
          );
-          ref.read(selectedModelProvider.notifier).state = placeholder;
+          ref.read(selectedModelProvider.notifier).set(placeholder);
        }
        // Reconcile against real models in background
        Future.microtask(() async {
@@ -747,7 +782,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
            }
            resolved ??= models.isNotEmpty ? models.first : null;
            if (resolved != null && !ref.read(isManualModelSelectionProvider)) {
-              ref.read(selectedModelProvider.notifier).state = resolved;
+              ref.read(selectedModelProvider.notifier).set(resolved);
              foundation.debugPrint(
                'DEBUG: Reconciled server default to ${resolved.name}',
              );
@@ -766,7 +801,7 @@ final defaultModelProvider = FutureProvider<Model?>((ref) async {
    }
    final selectedModel = models.first;
    if (!ref.read(isManualModelSelectionProvider)) {
-      ref.read(selectedModelProvider.notifier).state = selectedModel;
+      ref.read(selectedModelProvider.notifier).set(selectedModel);
      foundation.debugPrint(
        'DEBUG: Set default model (fallback): ${selectedModel.name}',
      );
@@ -806,7 +841,16 @@ final backgroundModelLoadProvider = Provider<void>((ref) {
 });

 // Search query provider
-final searchQueryProvider = StateProvider<String>((ref) => '');
+final searchQueryProvider = NotifierProvider<SearchQueryNotifier, String>(
+  SearchQueryNotifier.new,
+);
+
+class SearchQueryNotifier extends Notifier<String> {
+  @override
+  String build() => '';
+
+  void set(String query) => state = query;
+}

 // Server-side search provider for chats
 final serverSearchProvider = FutureProvider.family<List<Conversation>, String>((
@@ -1002,17 +1046,29 @@ final archivedConversationsProvider = Provider<List<Conversation>>((ref) {
 });

 // Reviewer mode provider (persisted)
-final reviewerModeProvider = StateNotifierProvider<ReviewerModeNotifier, bool>(
-  (ref) => ReviewerModeNotifier(ref.watch(optimizedStorageServiceProvider)),
+final reviewerModeProvider = NotifierProvider<ReviewerModeNotifier, bool>(
+  ReviewerModeNotifier.new,
 );

-class ReviewerModeNotifier extends StateNotifier<bool> {
-  final OptimizedStorageService _storage;
-  ReviewerModeNotifier(this._storage) : super(false) {
-    _load();
+class ReviewerModeNotifier extends Notifier<bool> {
+  late final OptimizedStorageService _storage;
+  bool _initialized = false;
+
+  @override
+  bool build() {
+    _storage = ref.watch(optimizedStorageServiceProvider);
+    if (!_initialized) {
+      _initialized = true;
+      Future.microtask(_load);
+    }
+    return false;
  }
+
  Future<void> _load() async {
    final enabled = await _storage.getReviewerMode();
+    if (!ref.mounted) {
+      return;
+    }
    state = enabled;
  }

--- a/lib/core/providers/app_startup_providers.dart
+++ b/lib/core/providers/app_startup_providers.dart
@@ -16,19 +16,37 @@ import '../utils/debug_logger.dart';
 enum _ConversationWarmupStatus { idle, warming, complete }

 final _conversationWarmupStatusProvider =
-    StateProvider<_ConversationWarmupStatus>(
-      (ref) => _ConversationWarmupStatus.idle,
+    NotifierProvider<
+      _ConversationWarmupStatusNotifier,
+      _ConversationWarmupStatus
+    >(_ConversationWarmupStatusNotifier.new);
+
+final _conversationWarmupLastAttemptProvider =
+    NotifierProvider<_ConversationWarmupLastAttemptNotifier, DateTime?>(
+      _ConversationWarmupLastAttemptNotifier.new,
    );

-final _conversationWarmupLastAttemptProvider = StateProvider<DateTime?>(
-  (ref) => null,
-);
+class _ConversationWarmupStatusNotifier
+    extends Notifier<_ConversationWarmupStatus> {
+  @override
+  _ConversationWarmupStatus build() => _ConversationWarmupStatus.idle;
+
+  void set(_ConversationWarmupStatus status) => state = status;
+}
+
+class _ConversationWarmupLastAttemptNotifier extends Notifier<DateTime?> {
+  @override
+  DateTime? build() => null;
+
+  void set(DateTime? value) => state = value;
+}

 void _scheduleConversationWarmup(Ref ref, {bool force = false}) {
  final navState = ref.read(authNavigationStateProvider);
  if (navState != AuthNavigationState.authenticated) {
-    ref.read(_conversationWarmupStatusProvider.notifier).state =
-        _ConversationWarmupStatus.idle;
+    ref
+        .read(_conversationWarmupStatusProvider.notifier)
+        .set(_ConversationWarmupStatus.idle);
    return;
  }

@@ -38,7 +56,7 @@ void _scheduleConversationWarmup(Ref ref, {bool force = false}) {
  }

  final statusController = ref.read(_conversationWarmupStatusProvider.notifier);
-  final status = statusController.state;
+  final status = ref.read(_conversationWarmupStatusProvider);

  if (!force) {
    if (status == _ConversationWarmupStatus.warming ||
@@ -56,28 +74,28 @@ void _scheduleConversationWarmup(Ref ref, {bool force = false}) {
      now.difference(lastAttempt) < const Duration(seconds: 30)) {
    return;
  }
-  ref.read(_conversationWarmupLastAttemptProvider.notifier).state = now;
+  ref.read(_conversationWarmupLastAttemptProvider.notifier).set(now);

-  statusController.state = _ConversationWarmupStatus.warming;
+  statusController.set(_ConversationWarmupStatus.warming);

  Future.microtask(() async {
    try {
      final existing = ref.read(conversationsProvider);
      if (existing.hasValue) {
-        statusController.state = _ConversationWarmupStatus.complete;
+        statusController.set(_ConversationWarmupStatus.complete);
        return;
      }
      if (existing.hasError) {
        ref.invalidate(conversationsProvider);
      }
      final conversations = await ref.read(conversationsProvider.future);
-      statusController.state = _ConversationWarmupStatus.complete;
+      statusController.set(_ConversationWarmupStatus.complete);
      DebugLogger.info(
        'Background chats warmup fetched ${conversations.length} conversations',
      );
    } catch (error) {
      DebugLogger.warning('Background chats warmup failed: $error');
-      statusController.state = _ConversationWarmupStatus.idle;
+      statusController.set(_ConversationWarmupStatus.idle);
    }
  });
 }
@@ -148,8 +166,9 @@ final appStartupFlowProvider = Provider<void>((ref) {
      });
    } else {
      // Reset warmup state when leaving authenticated flow
-      ref.read(_conversationWarmupStatusProvider.notifier).state =
-          _ConversationWarmupStatus.idle;
+      ref
+          .read(_conversationWarmupStatusProvider.notifier)
+          .set(_ConversationWarmupStatus.idle);
    }
  });

@@ -167,8 +186,9 @@ final appStartupFlowProvider = Provider<void>((ref) {
  ) {
    final wasReady = previous?.hasValue == true || previous?.hasError == true;
    if (wasReady && next.isLoading) {
-      ref.read(_conversationWarmupStatusProvider.notifier).state =
-          _ConversationWarmupStatus.idle;
+      ref
+          .read(_conversationWarmupStatusProvider.notifier)
+          .set(_ConversationWarmupStatus.idle);
      Future.microtask(() => _scheduleConversationWarmup(ref, force: true));
    }
  });
@@ -195,8 +215,9 @@ class _ForegroundRefreshObserver extends WidgetsBindingObserver {
      Future.microtask(() {
        try {
          _ref.invalidate(conversationsProvider);
-          _ref.read(_conversationWarmupStatusProvider.notifier).state =
-              _ConversationWarmupStatus.idle;
+          _ref
+              .read(_conversationWarmupStatusProvider.notifier)
+              .set(_ConversationWarmupStatus.idle);
        } catch (_) {}
        _scheduleConversationWarmup(_ref, force: true);
      });
--- a/lib/core/services/animation_service.dart
+++ b/lib/core/services/animation_service.dart
@@ -208,12 +208,29 @@ class AnimationService {
 enum PageTransitionType { fade, slide, scale }

 /// Provider for reduced motion preference
-final reducedMotionProvider = StateProvider<bool>((ref) => false);
+final reducedMotionProvider = NotifierProvider<ReducedMotionNotifier, bool>(
+  ReducedMotionNotifier.new,
+);

 /// Provider for animation performance settings
-final animationPerformanceProvider = StateProvider<AnimationPerformance>((ref) {
-  return AnimationPerformance.adaptive;
-});
+final animationPerformanceProvider =
+    NotifierProvider<AnimationPerformanceNotifier, AnimationPerformance>(
+      AnimationPerformanceNotifier.new,
+    );
+
+class ReducedMotionNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class AnimationPerformanceNotifier extends Notifier<AnimationPerformance> {
+  @override
+  AnimationPerformance build() => AnimationPerformance.adaptive;
+
+  void set(AnimationPerformance performance) => state = performance;
+}

 /// Animation performance levels
 enum AnimationPerformance {
@@ -225,8 +242,8 @@ enum AnimationPerformance {

 /// Provider for managing animation settings
 final animationSettingsProvider =
-    StateNotifierProvider<AnimationSettingsNotifier, AnimationSettings>(
-      (ref) => AnimationSettingsNotifier(),
+    NotifierProvider<AnimationSettingsNotifier, AnimationSettings>(
+      AnimationSettingsNotifier.new,
    );

 class AnimationSettings {
@@ -253,8 +270,9 @@ class AnimationSettings {
  }
 }

-class AnimationSettingsNotifier extends StateNotifier<AnimationSettings> {
-  AnimationSettingsNotifier() : super(const AnimationSettings());
+class AnimationSettingsNotifier extends Notifier<AnimationSettings> {
+  @override
+  AnimationSettings build() => const AnimationSettings();

  void setReduceMotion(bool reduce) {
    state = state.copyWith(reduceMotion: reduce);
--- a/lib/core/services/settings_service.dart
+++ b/lib/core/services/settings_service.dart
@@ -20,9 +20,11 @@ class SettingsService {
  static const String _voiceHoldToTalkKey = 'voice_hold_to_talk';
  static const String _voiceAutoSendKey = 'voice_auto_send_final';
  // Realtime transport preference
-  static const String _socketTransportModeKey = 'socket_transport_mode'; // 'auto' or 'ws'
+  static const String _socketTransportModeKey =
+      'socket_transport_mode'; // 'auto' or 'ws'
  // Quick pill visibility selections (max 2)
-  static const String _quickPillsKey = 'quick_pills'; // StringList of identifiers e.g. ['web','image','tools']
+  static const String _quickPillsKey =
+      'quick_pills'; // StringList of identifiers e.g. ['web','image','tools']
  // Chat input behavior
  static const String _sendOnEnterKey = 'send_on_enter';

@@ -335,9 +337,14 @@ class AppSettings {
      highContrast: highContrast ?? this.highContrast,
      largeText: largeText ?? this.largeText,
      darkMode: darkMode ?? this.darkMode,
-      defaultModel: defaultModel is _DefaultValue ? this.defaultModel : defaultModel as String?,
-      omitProviderInModelName: omitProviderInModelName ?? this.omitProviderInModelName,
-      voiceLocaleId: voiceLocaleId is _DefaultValue ? this.voiceLocaleId : voiceLocaleId as String?,
+      defaultModel: defaultModel is _DefaultValue
+          ? this.defaultModel
+          : defaultModel as String?,
+      omitProviderInModelName:
+          omitProviderInModelName ?? this.omitProviderInModelName,
+      voiceLocaleId: voiceLocaleId is _DefaultValue
+          ? this.voiceLocaleId
+          : voiceLocaleId as String?,
      voiceHoldToTalk: voiceHoldToTalk ?? this.voiceHoldToTalk,
      voiceAutoSendFinal: voiceAutoSendFinal ?? this.voiceAutoSendFinal,
      socketTransportMode: socketTransportMode ?? this.socketTransportMode,
@@ -363,7 +370,7 @@ class AppSettings {
        other.voiceAutoSendFinal == voiceAutoSendFinal &&
        other.sendOnEnter == sendOnEnter &&
        _listEquals(other.quickPills, quickPills);
-        // socketTransportMode intentionally not included in == to avoid frequent rebuilds
+    // socketTransportMode intentionally not included in == to avoid frequent rebuilds
  }

  @override
@@ -397,18 +404,27 @@ bool _listEquals(List<String> a, List<String> b) {
 }

 /// Provider for app settings
-final appSettingsProvider =
-    StateNotifierProvider<AppSettingsNotifier, AppSettings>(
-      (ref) => AppSettingsNotifier(),
-    );
+final appSettingsProvider = NotifierProvider<AppSettingsNotifier, AppSettings>(
+  AppSettingsNotifier.new,
+);

-class AppSettingsNotifier extends StateNotifier<AppSettings> {
-  AppSettingsNotifier() : super(const AppSettings()) {
-    _loadSettings();
+class AppSettingsNotifier extends Notifier<AppSettings> {
+  bool _initialized = false;
+
+  @override
+  AppSettings build() {
+    if (!_initialized) {
+      _initialized = true;
+      Future.microtask(_loadSettings);
+    }
+    return const AppSettings();
  }

  Future<void> _loadSettings() async {
    final settings = await SettingsService.loadSettings();
+    if (!ref.mounted) {
+      return;
+    }
    state = settings;
  }

--- a/lib/core/services/share_receiver_service.dart
+++ b/lib/core/services/share_receiver_service.dart
@@ -25,7 +25,17 @@ class SharedPayload {
 }

 /// Holds a pending shared payload until the app is ready (e.g., authed + model loaded)
-final pendingSharedPayloadProvider = StateProvider<SharedPayload?>((_) => null);
+final pendingSharedPayloadProvider =
+    NotifierProvider<PendingSharedPayloadNotifier, SharedPayload?>(
+      PendingSharedPayloadNotifier.new,
+    );
+
+class PendingSharedPayloadNotifier extends Notifier<SharedPayload?> {
+  @override
+  SharedPayload? build() => null;
+
+  void set(SharedPayload? payload) => state = payload;
+}

 /// Initializes listening to OS share intents and handles them
 final shareReceiverInitializerProvider = Provider<void>((ref) {
@@ -45,7 +55,7 @@ final shareReceiverInitializerProvider = Provider<void>((ref) {
        model != null &&
        isOnChatRoute) {
      _processPayload(ref, pending);
-      ref.read(pendingSharedPayloadProvider.notifier).state = null;
+      ref.read(pendingSharedPayloadProvider.notifier).set(null);
    }
  }

@@ -70,7 +80,7 @@ final shareReceiverInitializerProvider = Provider<void>((ref) {
      final dynamic media = await handler.getInitialSharedMedia();
      final payload = _toPayload(media);
      if (payload.hasAnything) {
-        ref.read(pendingSharedPayloadProvider.notifier).state = payload;
+        ref.read(pendingSharedPayloadProvider.notifier).set(payload);
        maybeProcessPending();
      }
    } catch (e) {
@@ -83,7 +93,7 @@ final shareReceiverInitializerProvider = Provider<void>((ref) {
    try {
      final payload = _toPayload(media);
      if (payload.hasAnything) {
-        ref.read(pendingSharedPayloadProvider.notifier).state = payload;
+        ref.read(pendingSharedPayloadProvider.notifier).set(payload);
        maybeProcessPending();
      }
    } catch (e) {
@@ -178,10 +188,10 @@ Future<void> _processPayload(Ref ref, SharedPayload payload) async {
    // Prefill text in the composer (do not auto-send) and request focus
    final text = payload.text?.trim();
    if (text != null && text.isNotEmpty) {
-      ref.read(prefilledInputTextProvider.notifier).state = text;
+      ref.read(prefilledInputTextProvider.notifier).set(text);
      // Bump focus trigger to ensure input focuses after navigation/build
      final current = ref.read(inputFocusTriggerProvider);
-      ref.read(inputFocusTriggerProvider.notifier).state = current + 1;
+      ref.read(inputFocusTriggerProvider.notifier).set(current + 1);
    }
    // Do NOT create a server chat here. The chat is created on first send
    // (with server syncing + title generation) in chat_providers.dart.
--- a/lib/features/chat/providers/chat_providers.dart
+++ b/lib/features/chat/providers/chat_providers.dart
@@ -23,90 +23,155 @@ const bool kSocketVerboseLogging = false;

 // Chat messages for current conversation
 final chatMessagesProvider =
-    StateNotifierProvider<ChatMessagesNotifier, List<ChatMessage>>((ref) {
-      return ChatMessagesNotifier(ref);
-    });
+    NotifierProvider<ChatMessagesNotifier, List<ChatMessage>>(
+      ChatMessagesNotifier.new,
+    );

 // Loading state for conversation (used to show chat skeletons during fetch)
-final isLoadingConversationProvider = StateProvider<bool>((ref) => false);
+final isLoadingConversationProvider =
+    NotifierProvider<IsLoadingConversationNotifier, bool>(
+      IsLoadingConversationNotifier.new,
+    );

 // Prefilled input text (e.g., when sharing text from other apps)
-final prefilledInputTextProvider = StateProvider<String?>((ref) => null);
+final prefilledInputTextProvider =
+    NotifierProvider<PrefilledInputTextNotifier, String?>(
+      PrefilledInputTextNotifier.new,
+    );

 // Trigger to request focus on the chat input (increment to signal)
-final inputFocusTriggerProvider = StateProvider<int>((ref) => 0);
+final inputFocusTriggerProvider =
+    NotifierProvider<InputFocusTriggerNotifier, int>(
+      InputFocusTriggerNotifier.new,
+    );

 // Whether the chat composer currently has focus
-final composerHasFocusProvider = StateProvider<bool>((ref) => false);
+final composerHasFocusProvider = NotifierProvider<ComposerFocusNotifier, bool>(
+  ComposerFocusNotifier.new,
+);

-class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {
-  final Ref _ref;
+class IsLoadingConversationNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class PrefilledInputTextNotifier extends Notifier<String?> {
+  @override
+  String? build() => null;
+
+  void set(String? value) => state = value;
+
+  void clear() => state = null;
+}
+
+class InputFocusTriggerNotifier extends Notifier<int> {
+  @override
+  int build() => 0;
+
+  void set(int value) => state = value;
+
+  int increment() {
+    final next = state + 1;
+    state = next;
+    return next;
+  }
+}
+
+class ComposerFocusNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class ChatMessagesNotifier extends Notifier<List<ChatMessage>> {
  StreamSubscription? _messageStream;
  ProviderSubscription? _conversationListener;
  final List<StreamSubscription> _subscriptions = [];
  // Activity-based watchdog to prevent stuck typing indicator
  InactivityWatchdog? _typingWatchdog;

-  ChatMessagesNotifier(this._ref) : super([]) {
-    // Load messages when conversation changes with proper cleanup
-    _conversationListener = _ref.listen(activeConversationProvider, (
-      previous,
-      next,
-    ) {
-      debugPrint('Conversation changed: ${previous?.id} -> ${next?.id}');
+  bool _initialized = false;

-      // Only react when the conversation actually changes
-      if (previous?.id == next?.id) {
-        // If same conversation but server updated it (e.g., title/content), avoid overwriting
-        // locally streamed assistant content with an outdated server copy.
-        if (previous?.updatedAt != next?.updatedAt) {
-          final serverMessages = next?.messages ?? const [];
-          // Primary rule: adopt server messages when there are strictly more of them.
-          if (serverMessages.length > state.length) {
-            state = serverMessages;
-            return;
-          }
+  @override
+  List<ChatMessage> build() {
+    if (!_initialized) {
+      _initialized = true;
+      _conversationListener = ref.listen(activeConversationProvider, (
+        previous,
+        next,
+      ) {
+        debugPrint('Conversation changed: ${previous?.id} -> ${next?.id}');

-          // Secondary rule: if counts are equal but the last assistant message grew,
-          // adopt the server copy to recover from missed socket events.
-          if (serverMessages.isNotEmpty && state.isNotEmpty) {
-            final serverLast = serverMessages.last;
-            final localLast = state.last;
-            final serverText = serverLast.content.trim();
-            final localText = localLast.content.trim();
-            final sameLastId = serverLast.id == localLast.id;
-            final isAssistant = serverLast.role == 'assistant';
-            final serverHasMore =
-                serverText.isNotEmpty && serverText.length > localText.length;
-            final localEmptyButServerHas =
-                localText.isEmpty && serverText.isNotEmpty;
-            if (sameLastId &&
-                isAssistant &&
-                (serverHasMore || localEmptyButServerHas)) {
+        // Only react when the conversation actually changes
+        if (previous?.id == next?.id) {
+          // If same conversation but server updated it (e.g., title/content), avoid overwriting
+          // locally streamed assistant content with an outdated server copy.
+          if (previous?.updatedAt != next?.updatedAt) {
+            final serverMessages = next?.messages ?? const [];
+            // Primary rule: adopt server messages when there are strictly more of them.
+            if (serverMessages.length > state.length) {
              state = serverMessages;
              return;
            }
+
+            // Secondary rule: if counts are equal but the last assistant message grew,
+            // adopt the server copy to recover from missed socket events.
+            if (serverMessages.isNotEmpty && state.isNotEmpty) {
+              final serverLast = serverMessages.last;
+              final localLast = state.last;
+              final serverText = serverLast.content.trim();
+              final localText = localLast.content.trim();
+              final sameLastId = serverLast.id == localLast.id;
+              final isAssistant = serverLast.role == 'assistant';
+              final serverHasMore =
+                  serverText.isNotEmpty && serverText.length > localText.length;
+              final localEmptyButServerHas =
+                  localText.isEmpty && serverText.isNotEmpty;
+              if (sameLastId &&
+                  isAssistant &&
+                  (serverHasMore || localEmptyButServerHas)) {
+                state = serverMessages;
+                return;
+              }
+            }
          }
+          return;
        }
-        return;
-      }

-      // Cancel any existing message stream when switching conversations
-      _cancelMessageStream();
-      // Also cancel typing guard on conversation switch
-      _cancelTypingGuard();
+        // Cancel any existing message stream when switching conversations
+        _cancelMessageStream();
+        // Also cancel typing guard on conversation switch
+        _cancelTypingGuard();

-      if (next != null) {
-        state = next.messages;
+        if (next != null) {
+          state = next.messages;

-        // Update selected model if conversation has a different model
-        _updateModelForConversation(next);
-      } else {
-        state = [];
-      }
-    });
+          // Update selected model if conversation has a different model
+          _updateModelForConversation(next);
+        } else {
+          state = [];
+        }
+      });

-    // ProviderSubscription will be cleaned up in dispose method
+      ref.onDispose(() {
+        for (final subscription in _subscriptions) {
+          subscription.cancel();
+        }
+        _subscriptions.clear();
+
+        _cancelMessageStream();
+        _cancelTypingGuard();
+
+        _conversationListener?.close();
+        _conversationListener = null;
+      });
+    }
+
+    final activeConversation = ref.read(activeConversationProvider);
+    return activeConversation?.messages ?? const [];
  }

  void _addSubscription(StreamSubscription subscription) {
@@ -137,8 +202,8 @@ class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {
            // Attempt a soft recovery: if content is still empty, try fetching final content from server
            if ((last.content).trim().isEmpty) {
              try {
-                final apiSvc = _ref.read(apiServiceProvider);
-                final activeConv = _ref.read(activeConversationProvider);
+                final apiSvc = ref.read(apiServiceProvider);
+                final activeConv = ref.read(activeConversationProvider);
                final msgId = last.id;
                final chatId = activeConv?.id;
                if (apiSvc != null && chatId != null && chatId.isNotEmpty) {
@@ -228,9 +293,9 @@ class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {
          final isImageGenFlow = (meta['imageGenerationFlow'] == true);

          // Also consult global toggles if metadata not present
-          final globalWebSearch = _ref.read(webSearchEnabledProvider);
-          final webSearchAvailable = _ref.read(webSearchAvailableProvider);
-          final globalImageGen = _ref.read(imageGenerationEnabledProvider);
+          final globalWebSearch = ref.read(webSearchEnabledProvider);
+          final webSearchAvailable = ref.read(webSearchAvailableProvider);
+          final globalImageGen = ref.read(imageGenerationEnabledProvider);

          // Extend guard windows to tolerate long reasoning/tools (> 1 min)
          if (isWebSearchFlow || (globalWebSearch && webSearchAvailable)) {
@@ -262,13 +327,13 @@ class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {
      return;
    }

-    final currentSelectedModel = _ref.read(selectedModelProvider);
+    final currentSelectedModel = ref.read(selectedModelProvider);

    // If the conversation's model is different from the currently selected one
    if (currentSelectedModel?.id != conversation.model) {
      // Get available models to find the matching one
      try {
-        final models = await _ref.read(modelsProvider.future);
+        final models = await ref.read(modelsProvider.future);

        if (models.isEmpty) {
          return;
@@ -281,7 +346,7 @@ class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {

        if (conversationModel != null) {
          // Update the selected model
-          _ref.read(selectedModelProvider.notifier).state = conversationModel;
+          ref.read(selectedModelProvider.notifier).set(conversationModel);
        } else {
          // Model not found in available models - silently continue
        }
@@ -447,33 +512,9 @@ class ChatMessagesNotifier extends StateNotifier<List<ChatMessage>> {
    // can pick up updated titles and ordering once streaming completes.
    // Best-effort: ignore if ref lifecycle/context prevents invalidation.
    try {
-      _ref.invalidate(conversationsProvider);
+      ref.invalidate(conversationsProvider);
    } catch (_) {}
  }
-
-  @override
-  void dispose() {
-    debugPrint(
-      'ChatMessagesNotifier disposing - ${_subscriptions.length} subscriptions',
-    );
-
-    // Cancel all tracked subscriptions
-    for (final subscription in _subscriptions) {
-      subscription.cancel();
-    }
-    _subscriptions.clear();
-
-    // Cancel message stream specifically
-    _cancelMessageStream();
-    // Cancel any active typing guard
-    _cancelTypingGuard();
-
-    // Cancel conversation listener specifically
-    _conversationListener?.close();
-    _conversationListener = null;
-
-    super.dispose();
-  }
 }

 // Pre-seed an assistant skeleton message (with a given id or a new one),
@@ -526,8 +567,8 @@ Future<String> _preseedAssistantAndPersist(
  // Persist the skeleton to the server so the web client sees a correct chain
  try {
    if (api != null && activeConv != null) {
-      final resolvedSystemPrompt = (systemPrompt != null &&
-              systemPrompt.trim().isNotEmpty)
+      final resolvedSystemPrompt =
+          (systemPrompt != null && systemPrompt.trim().isNotEmpty)
          ? systemPrompt.trim()
          : activeConv.systemPrompt;
      final current = ref.read(chatMessagesProvider);
@@ -567,45 +608,92 @@ String? _extractSystemPromptFromSettings(Map<String, dynamic>? settings) {
 // Start a new chat (unified function for both "New Chat" button and home screen)
 void startNewChat(dynamic ref) {
  // Clear active conversation
-  ref.read(activeConversationProvider.notifier).state = null;
+  ref.read(activeConversationProvider.notifier).clear();

  // Clear messages
  ref.read(chatMessagesProvider.notifier).clearMessages();
 }

 // Available tools provider
-final availableToolsProvider = StateProvider<List<String>>((ref) => []);
+final availableToolsProvider =
+    NotifierProvider<AvailableToolsNotifier, List<String>>(
+      AvailableToolsNotifier.new,
+    );

 // Web search enabled state for API-based web search
-final webSearchEnabledProvider = StateProvider<bool>((ref) => false);
+final webSearchEnabledProvider =
+    NotifierProvider<WebSearchEnabledNotifier, bool>(
+      WebSearchEnabledNotifier.new,
+    );

 // Image generation enabled state - behaves like web search
-final imageGenerationEnabledProvider = StateProvider<bool>((ref) => false);
+final imageGenerationEnabledProvider =
+    NotifierProvider<ImageGenerationEnabledNotifier, bool>(
+      ImageGenerationEnabledNotifier.new,
+    );

 // Vision capable models provider
-final visionCapableModelsProvider = StateProvider<List<String>>((ref) {
-  final selectedModel = ref.watch(selectedModelProvider);
-  if (selectedModel == null) return [];
-
-  // Check if the model supports vision (multimodal)
-  if (selectedModel.isMultimodal == true) {
-    return [selectedModel.id];
-  }
-
-  // For now, assume all models support vision unless explicitly marked
-  // This can be enhanced with proper model capability detection
-  return [selectedModel.id];
-});
+final visionCapableModelsProvider =
+    NotifierProvider<VisionCapableModelsNotifier, List<String>>(
+      VisionCapableModelsNotifier.new,
+    );

 // File upload capable models provider
-final fileUploadCapableModelsProvider = StateProvider<List<String>>((ref) {
-  final selectedModel = ref.watch(selectedModelProvider);
-  if (selectedModel == null) return [];
+final fileUploadCapableModelsProvider =
+    NotifierProvider<FileUploadCapableModelsNotifier, List<String>>(
+      FileUploadCapableModelsNotifier.new,
+    );

-  // For now, assume all models support file upload
-  // This can be enhanced with proper model capability detection
-  return [selectedModel.id];
-});
+class AvailableToolsNotifier extends Notifier<List<String>> {
+  @override
+  List<String> build() => [];
+
+  void set(List<String> tools) => state = List<String>.from(tools);
+}
+
+class WebSearchEnabledNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class ImageGenerationEnabledNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class VisionCapableModelsNotifier extends Notifier<List<String>> {
+  @override
+  List<String> build() {
+    final selectedModel = ref.watch(selectedModelProvider);
+    if (selectedModel == null) {
+      return [];
+    }
+
+    if (selectedModel.isMultimodal == true) {
+      return [selectedModel.id];
+    }
+
+    // For now, assume all models support vision unless explicitly marked
+    return [selectedModel.id];
+  }
+}
+
+class FileUploadCapableModelsNotifier extends Notifier<List<String>> {
+  @override
+  List<String> build() {
+    final selectedModel = ref.watch(selectedModelProvider);
+    if (selectedModel == null) {
+      return [];
+    }
+
+    // For now, assume all models support file upload
+    return [selectedModel.id];
+  }
+}

 // Helper function to validate file size
 bool validateFileSize(int fileSize, int? maxSizeMB) {
@@ -790,9 +878,10 @@ Future<void> regenerateMessage(
    if ((activeConversation.systemPrompt == null ||
            activeConversation.systemPrompt!.trim().isEmpty) &&
        (userSystemPrompt?.isNotEmpty ?? false)) {
-      final updated =
-          activeConversation.copyWith(systemPrompt: userSystemPrompt);
-      ref.read(activeConversationProvider.notifier).state = updated;
+      final updated = activeConversation.copyWith(
+        systemPrompt: userSystemPrompt,
+      );
+      ref.read(activeConversationProvider.notifier).set(updated);
      activeConversation = updated;
    }

@@ -831,7 +920,8 @@ Future<void> regenerateMessage(
    }

    final conversationSystemPrompt = activeConversation.systemPrompt?.trim();
-    final effectiveSystemPrompt = (conversationSystemPrompt != null &&
+    final effectiveSystemPrompt =
+        (conversationSystemPrompt != null &&
            conversationSystemPrompt.isNotEmpty)
        ? conversationSystemPrompt
        : userSystemPrompt;
@@ -840,10 +930,10 @@ Future<void> regenerateMessage(
        (m) => (m['role']?.toString().toLowerCase() ?? '') == 'system',
      );
      if (!hasSystemMessage) {
-        conversationMessages.insert(
-          0,
-          {'role': 'system', 'content': effectiveSystemPrompt},
-        );
+        conversationMessages.insert(0, {
+          'role': 'system',
+          'content': effectiveSystemPrompt,
+        });
      }
    }

@@ -974,7 +1064,9 @@ Future<void> regenerateMessage(
    // Resolve tool servers from user settings (if any)
    List<Map<String, dynamic>>? toolServers;
    final uiSettings = userSettingsData?['ui'] as Map<String, dynamic>?;
-    final rawServers = uiSettings != null ? (uiSettings['toolServers'] as List?) : null;
+    final rawServers = uiSettings != null
+        ? (uiSettings['toolServers'] as List?)
+        : null;
    if (rawServers != null && rawServers.isNotEmpty) {
      try {
        toolServers = await _resolveToolServers(rawServers, api);
@@ -1151,7 +1243,7 @@ Future<void> _sendMessageInternal(
    );

    // Set as active conversation locally
-    ref.read(activeConversationProvider.notifier).state = localConversation;
+    ref.read(activeConversationProvider.notifier).set(localConversation);
    activeConversation = localConversation;

    if (!reviewerMode) {
@@ -1170,8 +1262,7 @@ Future<void> _sendMessageInternal(
              ? serverConversation.messages
              : [userMessage],
        );
-        ref.read(activeConversationProvider.notifier).state =
-            updatedConversation;
+        ref.read(activeConversationProvider.notifier).set(updatedConversation);
        activeConversation = updatedConversation;

        // Set messages in the messages provider to keep UI in sync
@@ -1208,7 +1299,7 @@ Future<void> _sendMessageInternal(
          activeConversation.systemPrompt!.trim().isEmpty) &&
      (userSystemPrompt?.isNotEmpty ?? false)) {
    final updated = activeConversation.copyWith(systemPrompt: userSystemPrompt);
-    ref.read(activeConversationProvider.notifier).state = updated;
+    ref.read(activeConversationProvider.notifier).set(updated);
    activeConversation = updated;
  }

@@ -1316,8 +1407,8 @@ Future<void> _sendMessageInternal(
  }

  final conversationSystemPrompt = activeConversation?.systemPrompt?.trim();
-  final effectiveSystemPrompt = (conversationSystemPrompt != null &&
-          conversationSystemPrompt.isNotEmpty)
+  final effectiveSystemPrompt =
+      (conversationSystemPrompt != null && conversationSystemPrompt.isNotEmpty)
      ? conversationSystemPrompt
      : userSystemPrompt;
  if (effectiveSystemPrompt != null && effectiveSystemPrompt.isNotEmpty) {
@@ -1325,10 +1416,10 @@ Future<void> _sendMessageInternal(
      (m) => (m['role']?.toString().toLowerCase() ?? '') == 'system',
    );
    if (!hasSystemMessage) {
-      conversationMessages.insert(
-        0,
-        {'role': 'system', 'content': effectiveSystemPrompt},
-      );
+      conversationMessages.insert(0, {
+        'role': 'system',
+        'content': effectiveSystemPrompt,
+      });
    }
  }

@@ -1478,7 +1569,9 @@ Future<void> _sendMessageInternal(
    // Resolve tool servers from user settings (if any)
    List<Map<String, dynamic>>? toolServers;
    final uiSettings = userSettingsData?['ui'] as Map<String, dynamic>?;
-    final rawServers = uiSettings != null ? (uiSettings['toolServers'] as List?) : null;
+    final rawServers = uiSettings != null
+        ? (uiSettings['toolServers'] as List?)
+        : null;
    if (rawServers != null && rawServers.isNotEmpty) {
      try {
        toolServers = await _resolveToolServers(rawServers, api);
@@ -2377,8 +2470,9 @@ Future<void> _sendMessageInternal(
                    updatedAt: DateTime.now(),
                  );

-                  ref.read(activeConversationProvider.notifier).state =
-                      updatedConversation;
+                  ref
+                      .read(activeConversationProvider.notifier)
+                      .set(updatedConversation);
                } else {
                  // Keep local messages and only refresh conversations list
                  ref.invalidate(conversationsProvider);
@@ -2614,7 +2708,7 @@ Future<void> _checkForTitleInBackground(
              title: updatedConv.title,
              updatedAt: DateTime.now(),
            );
-            ref.read(activeConversationProvider.notifier).state = updated;
+            ref.read(activeConversationProvider.notifier).set(updated);

            // Refresh the conversations list
            ref.invalidate(conversationsProvider);
@@ -2679,7 +2773,7 @@ Future<void> _saveConversationLocally(dynamic ref) async {
    }

    await storage.setString('conversations', jsonEncode(conversations));
-    ref.read(activeConversationProvider.notifier).state = updatedConversation;
+    ref.read(activeConversationProvider.notifier).set(updatedConversation);
    ref.invalidate(conversationsProvider);
  } catch (e) {
    // Handle local storage errors silently
@@ -2723,8 +2817,9 @@ Future<void> pinConversation(
    // Update active conversation if it's the one being pinned
    final activeConversation = ref.read(activeConversationProvider);
    if (activeConversation?.id == conversationId) {
-      ref.read(activeConversationProvider.notifier).state = activeConversation!
-          .copyWith(pinned: pinned);
+      ref
+          .read(activeConversationProvider.notifier)
+          .set(activeConversation!.copyWith(pinned: pinned));
    }
  } catch (e) {
    debugPrint('Error ${pinned ? 'pinning' : 'unpinning'} conversation: $e');
@@ -2743,7 +2838,7 @@ Future<void> archiveConversation(

  // Update local state first
  if (activeConversation?.id == conversationId && archived) {
-    ref.read(activeConversationProvider.notifier).state = null;
+    ref.read(activeConversationProvider.notifier).clear();
    ref.read(chatMessagesProvider.notifier).clearMessages();
  }

@@ -2761,7 +2856,7 @@ Future<void> archiveConversation(

    // If server operation failed and we archived locally, restore the conversation
    if (activeConversation?.id == conversationId && archived) {
-      ref.read(activeConversationProvider.notifier).state = activeConversation;
+      ref.read(activeConversationProvider.notifier).set(activeConversation);
      // Messages will be restored through the listener
    }

@@ -2796,7 +2891,7 @@ Future<void> cloneConversation(WidgetRef ref, String conversationId) async {
    final clonedConversation = await api.cloneConversation(conversationId);

    // Set the cloned conversation as active
-    ref.read(activeConversationProvider.notifier).state = clonedConversation;
+    ref.read(activeConversationProvider.notifier).set(clonedConversation);
    // Load messages through the listener mechanism
    // The ChatMessagesNotifier will automatically load messages when activeConversation changes

@@ -2841,7 +2936,7 @@ final regenerateLastMessageProvider = Provider<Future<void> Function()>((ref) {
      final prev = ref.read(imageGenerationEnabledProvider);
      try {
        // Force image generation enabled during regeneration
-        ref.read(imageGenerationEnabledProvider.notifier).state = true;
+        ref.read(imageGenerationEnabledProvider.notifier).set(true);
        await regenerateMessage(
          ref,
          lastUserMessage.content,
@@ -2849,7 +2944,7 @@ final regenerateLastMessageProvider = Provider<Future<void> Function()>((ref) {
        );
      } finally {
        // restore previous state
-        ref.read(imageGenerationEnabledProvider.notifier).state = prev;
+        ref.read(imageGenerationEnabledProvider.notifier).set(prev);
      }
      return;
    }
--- a/lib/features/chat/providers/text_to_speech_provider.dart
+++ b/lib/features/chat/providers/text_to_speech_provider.dart
@@ -49,21 +49,32 @@ class TextToSpeechState {
  }
 }

-class TextToSpeechController extends StateNotifier<TextToSpeechState> {
-  TextToSpeechController(this._service) : super(const TextToSpeechState()) {
-    _service.bindHandlers(
-      onStart: _handleStart,
-      onComplete: _handleCompletion,
-      onCancel: _handleCancellation,
-      onPause: _handlePause,
-      onContinue: _handleContinue,
-      onError: _handleError,
-    );
-  }
-
-  final TextToSpeechService _service;
+class TextToSpeechController extends Notifier<TextToSpeechState> {
+  late final TextToSpeechService _service;
+  bool _handlersBound = false;
  Future<bool>? _initializationFuture;

+  @override
+  TextToSpeechState build() {
+    _service = ref.watch(textToSpeechServiceProvider);
+    if (!_handlersBound) {
+      _handlersBound = true;
+      _service.bindHandlers(
+        onStart: _handleStart,
+        onComplete: _handleCompletion,
+        onCancel: _handleCancellation,
+        onPause: _handlePause,
+        onContinue: _handleContinue,
+        onError: _handleError,
+      );
+
+      ref.onDispose(() {
+        unawaited(_service.stop());
+      });
+    }
+    return const TextToSpeechState();
+  }
+
  Future<bool> _ensureInitialized() {
    final existing = _initializationFuture;
    if (existing != null) {
@@ -78,7 +89,7 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
    final future = _service
        .initialize()
        .then((available) {
-          if (!mounted) {
+          if (!ref.mounted) {
            return available;
          }

@@ -90,7 +101,7 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
          return available;
        })
        .catchError((error, _) {
-          if (!mounted) {
+          if (!ref.mounted) {
            return false;
          }

@@ -132,7 +143,7 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {

    final available = await _ensureInitialized();
    if (!available) {
-      if (!mounted) {
+      if (!ref.mounted) {
        return;
      }
      state = state.copyWith(
@@ -151,14 +162,14 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {

    try {
      await _service.speak(text);
-      if (!mounted) {
+      if (!ref.mounted) {
        return;
      }
      if (state.status == TtsPlaybackStatus.loading) {
        state = state.copyWith(status: TtsPlaybackStatus.speaking);
      }
    } catch (e) {
-      if (!mounted) {
+      if (!ref.mounted) {
        return;
      }
      state = state.copyWith(
@@ -178,7 +189,7 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {

  Future<void> stop() async {
    await _service.stop();
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(
@@ -189,14 +200,14 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
  }

  void _handleStart() {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(status: TtsPlaybackStatus.speaking);
  }

  void _handleCompletion() {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(
@@ -206,7 +217,7 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
  }

  void _handleCancellation() {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(
@@ -216,21 +227,21 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
  }

  void _handlePause() {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(status: TtsPlaybackStatus.paused);
  }

  void _handleContinue() {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(status: TtsPlaybackStatus.speaking);
  }

  void _handleError(String message) {
-    if (!mounted) {
+    if (!ref.mounted) {
      return;
    }
    state = state.copyWith(
@@ -239,12 +250,6 @@ class TextToSpeechController extends StateNotifier<TextToSpeechState> {
      clearActiveMessageId: true,
    );
  }
-
-  @override
-  void dispose() {
-    unawaited(_service.stop());
-    super.dispose();
-  }
 }

 final textToSpeechServiceProvider = Provider<TextToSpeechService>((ref) {
@@ -256,7 +261,6 @@ final textToSpeechServiceProvider = Provider<TextToSpeechService>((ref) {
 });

 final textToSpeechControllerProvider =
-    StateNotifierProvider<TextToSpeechController, TextToSpeechState>((ref) {
-      final service = ref.watch(textToSpeechServiceProvider);
-      return TextToSpeechController(service);
-    });
+    NotifierProvider<TextToSpeechController, TextToSpeechState>(
+      TextToSpeechController.new,
+    );
--- a/lib/features/chat/services/file_attachment_service.dart
+++ b/lib/features/chat/services/file_attachment_service.dart
@@ -498,8 +498,9 @@ final fileAttachmentServiceProvider = Provider<dynamic>((ref) {
 });

 // State notifier for managing attached files
-class AttachedFilesNotifier extends StateNotifier<List<FileUploadState>> {
-  AttachedFilesNotifier() : super([]);
+class AttachedFilesNotifier extends Notifier<List<FileUploadState>> {
+  @override
+  List<FileUploadState> build() => [];

  void addFiles(List<File> files) {
    final newStates = files
@@ -536,6 +537,6 @@ class AttachedFilesNotifier extends StateNotifier<List<FileUploadState>> {
 }

 final attachedFilesProvider =
-    StateNotifierProvider<AttachedFilesNotifier, List<FileUploadState>>((ref) {
-      return AttachedFilesNotifier();
-    });
+    NotifierProvider<AttachedFilesNotifier, List<FileUploadState>>(
+      AttachedFilesNotifier.new,
+    );
--- a/lib/features/chat/services/message_batch_service.dart
+++ b/lib/features/chat/services/message_batch_service.dart
@@ -523,16 +523,41 @@ final messageBatchServiceProvider = Provider<MessageBatchService>((ref) {
 });

 /// Provider for selected messages (for batch operations)
-final selectedMessagesProvider = StateProvider<Set<String>>((ref) {
-  return <String>{};
-});
+final selectedMessagesProvider =
+    NotifierProvider<SelectedMessagesNotifier, Set<String>>(
+      SelectedMessagesNotifier.new,
+    );

 /// Provider for batch operation mode
-final batchModeProvider = StateProvider<bool>((ref) {
-  return false;
-});
+final batchModeProvider = NotifierProvider<BatchModeNotifier, bool>(
+  BatchModeNotifier.new,
+);

 /// Provider for message filter
-final messageFilterProvider = StateProvider<MessageFilter?>((ref) {
-  return null;
-});
+final messageFilterProvider =
+    NotifierProvider<MessageFilterNotifier, MessageFilter?>(
+      MessageFilterNotifier.new,
+    );
+
+class SelectedMessagesNotifier extends Notifier<Set<String>> {
+  @override
+  Set<String> build() => <String>{};
+
+  void set(Set<String> messages) => state = Set<String>.from(messages);
+
+  void clear() => state = <String>{};
+}
+
+class BatchModeNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class MessageFilterNotifier extends Notifier<MessageFilter?> {
+  @override
+  MessageFilter? build() => null;
+
+  void set(MessageFilter? filter) => state = filter;
+}
--- a/lib/features/chat/views/chat_page.dart
+++ b/lib/features/chat/views/chat_page.dart
@@ -92,7 +92,7 @@ class _ChatPageState extends ConsumerState<ChatPage> {
  void startNewChat() {
    // Clear current conversation
    ref.read(chatMessagesProvider.notifier).clearMessages();
-    ref.read(activeConversationProvider.notifier).state = null;
+    ref.read(activeConversationProvider.notifier).clear();

    // Scroll to top
    if (_scrollController.hasClients) {
@@ -140,7 +140,7 @@ class _ChatPageState extends ConsumerState<ChatPage> {
          'Default provider failed, selecting first model directly',
        );
        // Fallback: select the first available model
-        ref.read(selectedModelProvider.notifier).state = models.first;
+        ref.read(selectedModelProvider.notifier).set(models.first);
        DebugLogger.log('Fallback model selected: ${models.first.name}');
      }
    } catch (e) {
@@ -220,7 +220,7 @@ class _ChatPageState extends ConsumerState<ChatPage> {
        );

        if (!mounted) return;
-        ref.read(activeConversationProvider.notifier).state = welcomeConv;
+        ref.read(activeConversationProvider.notifier).set(welcomeConv);
        debugPrint('Auto-loaded demo conversation');
        return;
      }
@@ -296,7 +296,7 @@ class _ChatPageState extends ConsumerState<ChatPage> {
        }
        if (models.isNotEmpty) {
          selectedModel = models.first;
-          ref.read(selectedModelProvider.notifier).state = selectedModel;
+          ref.read(selectedModelProvider.notifier).set(selectedModel);
        }
      } catch (_) {
        // If models cannot be resolved, bail out without sending
@@ -1007,12 +1007,14 @@ class _ChatPageState extends ConsumerState<ChatPage> {
        if (!mounted) return;
        final current = ref.read(inputFocusTriggerProvider);
        // Immediate focus bump
-        ref.read(inputFocusTriggerProvider.notifier).state = current + 1;
+        ref
+            .read(inputFocusTriggerProvider.notifier)
+            .set(current + 1);
        // Second bump shortly after to overcome route/IME timing
        Future.delayed(const Duration(milliseconds: 120), () {
          if (!mounted) return;
          final cur2 = ref.read(inputFocusTriggerProvider);
-          ref.read(inputFocusTriggerProvider.notifier).state = cur2 + 1;
+          ref.read(inputFocusTriggerProvider.notifier).set(cur2 + 1);
        });
      });
      _didStartupFocus = true;
@@ -1326,9 +1328,8 @@ class _ChatPageState extends ConsumerState<ChatPage> {
                            try {
                              final full = await api.getConversation(active.id);
                              ref
-                                      .read(activeConversationProvider.notifier)
-                                      .state =
-                                  full;
+                                  .read(activeConversationProvider.notifier)
+                                  .set(full);
                            } catch (e) {
                              debugPrint(
                                'DEBUG: Failed to refresh conversation: $e',
@@ -1507,7 +1508,7 @@ class _ChatPageState extends ConsumerState<ChatPage> {
      if (hadFocus) {
        // Bump focus trigger to restore composer focus + IME
        final cur = ref.read(inputFocusTriggerProvider);
-        ref.read(inputFocusTriggerProvider.notifier).state = cur + 1;
+        ref.read(inputFocusTriggerProvider.notifier).set(cur + 1);
      }
    });
  }
@@ -1789,11 +1790,8 @@ class _ModelSelectorSheetState extends ConsumerState<_ModelSelectorSheet> {
                                    onTap: () {
                                      HapticFeedback.selectionClick();
                                      widget.ref
-                                              .read(
-                                                selectedModelProvider.notifier,
-                                              )
-                                              .state =
-                                          model;
+                                          .read(selectedModelProvider.notifier)
+                                          .set(model);
                                      Navigator.pop(context);
                                    },
                                  );
--- a/lib/features/chat/widgets/modern_chat_input.dart
+++ b/lib/features/chat/widgets/modern_chat_input.dart
@@ -115,7 +115,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
        _controller.text = text;
        _controller.selection = TextSelection.collapsed(offset: text.length);
        // Clear after applying so it doesn't re-apply on rebuilds
-        ref.read(prefilledInputTextProvider.notifier).state = null;
+        ref.read(prefilledInputTextProvider.notifier).clear();
      }
    });

@@ -131,7 +131,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
        final hasFocus = _focusNode.hasFocus;
        // Publish composer focus state
        try {
-          ref.read(composerHasFocusProvider.notifier).state = hasFocus;
+          ref.read(composerHasFocusProvider.notifier).set(hasFocus);
        } catch (_) {}
      });
    });
@@ -142,7 +142,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
  @override
  void dispose() {
    try {
-      ref.read(composerHasFocusProvider.notifier).state = false;
+      ref.read(composerHasFocusProvider.notifier).set(false);
    } catch (_) {}
    _controller.removeListener(_handleComposerChanged);
    _controller.dispose();
@@ -583,7 +583,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
          offset: incoming.length,
        );
        try {
-          ref.read(prefilledInputTextProvider.notifier).state = null;
+          ref.read(prefilledInputTextProvider.notifier).clear();
        } catch (_) {}
      });
    });
@@ -658,7 +658,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
            : Icons.search;
        void handleTap() {
          final notifier = ref.read(webSearchEnabledProvider.notifier);
-          notifier.state = !webSearchEnabled;
+          notifier.set(!webSearchEnabled);
        }

        quickPills.add(
@@ -676,7 +676,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
            : Icons.image;
        void handleTap() {
          final notifier = ref.read(imageGenerationEnabledProvider.notifier);
-          notifier.state = !imageGenEnabled;
+          notifier.set(!imageGenEnabled);
        }

        quickPills.add(
@@ -709,7 +709,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
            } else {
              current.add(id);
            }
-            ref.read(selectedToolIdsProvider.notifier).state = current;
+            ref.read(selectedToolIdsProvider.notifier).set(current);
          }

          quickPills.add(
@@ -1459,7 +1459,7 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
                subtitle: l10n.webSearchDescription,
                value: webSearchEnabled,
                onChanged: (next) {
-                  modalRef.read(webSearchEnabledProvider.notifier).state = next;
+                  modalRef.read(webSearchEnabledProvider.notifier).set(next);
                },
              ),
            );
@@ -1479,8 +1479,9 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
                subtitle: l10n.imageGenerationDescription,
                value: imageGenEnabled,
                onChanged: (next) {
-                  modalRef.read(imageGenerationEnabledProvider.notifier).state =
-                      next;
+                  modalRef
+                      .read(imageGenerationEnabledProvider.notifier)
+                      .set(next);
                },
              ),
            );
@@ -1507,8 +1508,9 @@ class _ModernChatInputState extends ConsumerState<ModernChatInput>
                    } else {
                      current.add(tool.id);
                    }
-                    modalRef.read(selectedToolIdsProvider.notifier).state =
-                        current;
+                    modalRef
+                        .read(selectedToolIdsProvider.notifier)
+                        .set(current);
                  },
                );
              }).toList();
--- a/lib/features/navigation/widgets/chats_drawer.dart
+++ b/lib/features/navigation/widgets/chats_drawer.dart
@@ -43,10 +43,12 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
  bool _draggingHasFolder = false;

  // UI state providers for sections
-  static final _showArchivedProvider = StateProvider<bool>((ref) => false);
-  static final _expandedFoldersProvider = StateProvider<Map<String, bool>>(
-    (ref) => {},
-  );
+  static final _showArchivedProvider =
+      NotifierProvider<_ShowArchivedNotifier, bool>(_ShowArchivedNotifier.new);
+  static final _expandedFoldersProvider =
+      NotifierProvider<_ExpandedFoldersNotifier, Map<String, bool>>(
+        _ExpandedFoldersNotifier.new,
+      );

  Future<void> _refreshChats() async {
    try {
@@ -694,7 +696,7 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
            onTap: () {
              final current = {...ref.read(_expandedFoldersProvider)};
              current[folderId] = !isExpanded;
-              ref.read(_expandedFoldersProvider.notifier).state = current;
+              ref.read(_expandedFoldersProvider.notifier).set(current);
            },
            onLongPress: () {
              HapticFeedback.selectionClick();
@@ -1065,7 +1067,7 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
          ),
          child: InkWell(
            borderRadius: BorderRadius.zero,
-            onTap: () => ref.read(_showArchivedProvider.notifier).state = !show,
+            onTap: () => ref.read(_showArchivedProvider.notifier).set(!show),
            overlayColor: WidgetStateProperty.resolveWith((states) {
              if (states.contains(WidgetState.pressed)) {
                return theme.buttonPrimary.withValues(
@@ -1163,11 +1165,11 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
    final container = ProviderScope.containerOf(context, listen: false);
    try {
      // Mark global loading to show skeletons in chat
-      container.read(chat.isLoadingConversationProvider.notifier).state = true;
+      container.read(chat.isLoadingConversationProvider.notifier).set(true);
      _pendingConversationId = id;

      // Immediately clear current chat to show loading skeleton in the chat view
-      container.read(activeConversationProvider.notifier).state = null;
+      container.read(activeConversationProvider.notifier).clear();
      container.read(chat.chatMessagesProvider.notifier).clearMessages();

      // Close the drawer immediately for faster perceived performance
@@ -1185,21 +1187,22 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
      final api = container.read(apiServiceProvider);
      if (api != null) {
        final full = await api.getConversation(id);
-        container.read(activeConversationProvider.notifier).state = full;
+        container.read(activeConversationProvider.notifier).set(full);
      } else {
        // Fallback: use the lightweight item to update the active conversation
-        container
-            .read(activeConversationProvider.notifier)
-            .state = (await container.read(
-          conversationsProvider.future,
-        )).firstWhere((c) => c.id == id);
+        container.read(activeConversationProvider.notifier).set(
+              (await container.read(
+                conversationsProvider.future,
+              ))
+                  .firstWhere((c) => c.id == id),
+            );
      }

      // Clear loading after data is ready
-      container.read(chat.isLoadingConversationProvider.notifier).state = false;
+      container.read(chat.isLoadingConversationProvider.notifier).set(false);
      _pendingConversationId = null;
    } catch (_) {
-      container.read(chat.isLoadingConversationProvider.notifier).state = false;
+      container.read(chat.isLoadingConversationProvider.notifier).set(false);
      _pendingConversationId = null;
    } finally {
      if (mounted) setState(() => _isLoadingConversation = false);
@@ -1311,6 +1314,20 @@ class _ChatsDrawerState extends ConsumerState<ChatsDrawer> {
  }
 }

+class _ShowArchivedNotifier extends Notifier<bool> {
+  @override
+  bool build() => false;
+
+  void set(bool value) => state = value;
+}
+
+class _ExpandedFoldersNotifier extends Notifier<Map<String, bool>> {
+  @override
+  Map<String, bool> build() => {};
+
+  void set(Map<String, bool> value) => state = Map<String, bool>.from(value);
+}
+
 class _DragConversationData {
  final String id;
  final String title;
--- a/lib/features/prompts/providers/prompts_providers.dart
+++ b/lib/features/prompts/providers/prompts_providers.dart
@@ -9,4 +9,14 @@ final promptsListProvider = FutureProvider<List<Prompt>>((ref) async {
  return promptsService.getPrompts();
 });

-final activePromptCommandProvider = StateProvider<String?>((ref) => null);
+final activePromptCommandProvider =
+    NotifierProvider<ActivePromptCommandNotifier, String?>(
+      ActivePromptCommandNotifier.new,
+    );
+
+class ActivePromptCommandNotifier extends Notifier<String?> {
+  @override
+  String? build() => null;
+
+  void set(String? command) => state = command;
+}
--- a/lib/features/tools/providers/tools_providers.dart
+++ b/lib/features/tools/providers/tools_providers.dart
@@ -8,4 +8,14 @@ final toolsListProvider = FutureProvider<List<Tool>>((ref) async {
  return await toolsService.getTools();
 });

-final selectedToolIdsProvider = StateProvider<List<String>>((ref) => []);
+final selectedToolIdsProvider =
+    NotifierProvider<SelectedToolIdsNotifier, List<String>>(
+      SelectedToolIdsNotifier.new,
+    );
+
+class SelectedToolIdsNotifier extends Notifier<List<String>> {
+  @override
+  List<String> build() => [];
+
+  void set(List<String> ids) => state = List<String>.from(ids);
+}
--- a/lib/shared/services/tasks/task_queue.dart
+++ b/lib/shared/services/tasks/task_queue.dart
@@ -10,18 +10,23 @@ import 'outbound_task.dart';
 import 'task_worker.dart';

 final taskQueueProvider =
-    StateNotifierProvider<TaskQueueNotifier, List<OutboundTask>>((ref) {
-  return TaskQueueNotifier(ref);
-});
-
-class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {
-  TaskQueueNotifier(this._ref) : super(const []) {
-    _load();
-  }
+    NotifierProvider<TaskQueueNotifier, List<OutboundTask>>(
+      TaskQueueNotifier.new,
+    );

+class TaskQueueNotifier extends Notifier<List<OutboundTask>> {
  static const _prefsKey = 'outbound_task_queue_v1';
-  final Ref _ref;
  final _uuid = const Uuid();
+  bool _bootstrapScheduled = false;
+
+  @override
+  List<OutboundTask> build() {
+    if (!_bootstrapScheduled) {
+      _bootstrapScheduled = true;
+      Future.microtask(_load);
+    }
+    return const [];
+  }

  bool _processing = false;
  final Set<String> _activeThreads = <String>{};
@@ -29,19 +34,24 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {

  Future<void> _load() async {
    try {
-      final prefs = _ref.read(sharedPreferencesProvider);
+      final prefs = ref.read(sharedPreferencesProvider);
      final jsonStr = prefs.getString(_prefsKey);
      if (jsonStr == null || jsonStr.isEmpty) return;
      final raw = (jsonDecode(jsonStr) as List).cast<Map<String, dynamic>>();
      final tasks = raw.map(OutboundTask.fromJson).toList();
      // Only restore non-completed tasks
      state = tasks
-          .where((t) => t.status == TaskStatus.queued || t.status == TaskStatus.running)
-          .map((t) => t.copyWith(
-                status: TaskStatus.queued,
-                startedAt: null,
-                completedAt: null,
-              ))
+          .where(
+            (t) =>
+                t.status == TaskStatus.queued || t.status == TaskStatus.running,
+          )
+          .map(
+            (t) => t.copyWith(
+              status: TaskStatus.queued,
+              startedAt: null,
+              completedAt: null,
+            ),
+          )
          .toList();
      // Kick processing after load
      _process();
@@ -52,7 +62,7 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {

  Future<void> _save() async {
    try {
-      final prefs = _ref.read(sharedPreferencesProvider);
+      final prefs = ref.read(sharedPreferencesProvider);
      final raw = state.map((t) => t.toJson()).toList(growable: false);
      await prefs.setString(_prefsKey, jsonEncode(raw));
    } catch (e) {
@@ -87,10 +97,7 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {
    state = [
      for (final t in state)
        if (t.id == id)
-          t.copyWith(
-            status: TaskStatus.cancelled,
-            completedAt: DateTime.now(),
-          )
+          t.copyWith(status: TaskStatus.cancelled, completedAt: DateTime.now())
        else
          t,
    ];
@@ -102,10 +109,7 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {
      for (final t in state)
        if ((t.maybeConversationId ?? '') == conversationId &&
            (t.status == TaskStatus.queued || t.status == TaskStatus.running))
-          t.copyWith(
-            status: TaskStatus.cancelled,
-            completedAt: DateTime.now(),
-          )
+          t.copyWith(status: TaskStatus.cancelled, completedAt: DateTime.now())
        else
          t,
    ];
@@ -177,7 +181,9 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {
      // Pump while there is capacity and queued tasks remain
      while (true) {
        // Filter runnable tasks
-        final queued = state.where((t) => t.status == TaskStatus.queued).toList();
+        final queued = state
+            .where((t) => t.status == TaskStatus.queued)
+            .toList();
        if (queued.isEmpty) break;

        // Respect parallelism and one-per-thread
@@ -202,18 +208,23 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {
        state = [
          for (final t in state)
            if (t.id == next.id)
-              next.copyWith(status: TaskStatus.running, startedAt: DateTime.now())
+              next.copyWith(
+                status: TaskStatus.running,
+                startedAt: DateTime.now(),
+              )
            else
              t,
        ];
        await _save();

        // Launch worker
-        unawaited(_run(next).whenComplete(() {
-          _activeThreads.remove(threadKey);
-          // After a task completes, try to schedule more
-          _process();
-        }));
+        unawaited(
+          _run(next).whenComplete(() {
+            _activeThreads.remove(threadKey);
+            // After a task completes, try to schedule more
+            _process();
+          }),
+        );
      }
    } finally {
      _processing = false;
@@ -222,11 +233,14 @@ class TaskQueueNotifier extends StateNotifier<List<OutboundTask>> {

  Future<void> _run(OutboundTask task) async {
    try {
-      await TaskWorker(_ref).perform(task);
+      await TaskWorker(ref).perform(task);
      state = [
        for (final t in state)
          if (t.id == task.id)
-            t.copyWith(status: TaskStatus.succeeded, completedAt: DateTime.now())
+            t.copyWith(
+              status: TaskStatus.succeeded,
+              completedAt: DateTime.now(),
+            )
          else
            t,
      ];
--- a/lib/shared/services/tasks/task_worker.dart
+++ b/lib/shared/services/tasks/task_worker.dart
@@ -49,7 +49,7 @@ class TaskWorker {
          final api = _ref.read(apiServiceProvider);
          if (api != null) {
            final conv = await api.getConversation(task.conversationId!);
-            _ref.read(activeConversationProvider.notifier).state = conv;
+            _ref.read(activeConversationProvider.notifier).set(conv);
          }
        } catch (_) {
          // If loading fails, proceed; send flow can create a new conversation
@@ -167,7 +167,7 @@ class TaskWorker {
          (active == null || active.id != task.conversationId)) {
        try {
          final conv = await api.getConversation(task.conversationId!);
-          _ref.read(activeConversationProvider.notifier).state = conv;
+          _ref.read(activeConversationProvider.notifier).set(conv);
        } catch (_) {}
      }
    } catch (_) {}
@@ -225,7 +225,7 @@ class TaskWorker {
          (active == null || active.id != task.conversationId)) {
        try {
          final conv = await api.getConversation(task.conversationId!);
-          _ref.read(activeConversationProvider.notifier).state = conv;
+          _ref.read(activeConversationProvider.notifier).set(conv);
        } catch (_) {}
      }
    } catch (_) {}
@@ -233,10 +233,10 @@ class TaskWorker {
    // Temporarily enable image-generation background flow for this send
    final prev = _ref.read(chat.imageGenerationEnabledProvider);
    try {
-      _ref.read(chat.imageGenerationEnabledProvider.notifier).state = true;
+      _ref.read(chat.imageGenerationEnabledProvider.notifier).set(true);
      await chat.sendMessageFromService(_ref, task.prompt, null, null);
    } finally {
-      _ref.read(chat.imageGenerationEnabledProvider.notifier).state = prev;
+      _ref.read(chat.imageGenerationEnabledProvider.notifier).set(prev);
    }
  }

@@ -368,7 +368,7 @@ class TaskWorker {
            title: title.length > 100 ? '${title.substring(0, 100)}...' : title,
            updatedAt: DateTime.now(),
          );
-          _ref.read(activeConversationProvider.notifier).state = updated;
+          _ref.read(activeConversationProvider.notifier).set(updated);
          // Do not push full messages to server; skip remote update.
          // Optionally refresh list to reflect server-side title when it’s generated there.
          _ref.invalidate(conversationsProvider);
--- a/lib/shared/utils/conversation_context_menu.dart
+++ b/lib/shared/utils/conversation_context_menu.dart
@@ -224,9 +224,9 @@ Future<void> _renameConversation(
    ref.invalidate(conversationsProvider);
    final active = ref.read(activeConversationProvider);
    if (active?.id == conversationId) {
-      ref.read(activeConversationProvider.notifier).state = active!.copyWith(
-        title: newName,
-      );
+      ref
+          .read(activeConversationProvider.notifier)
+          .set(active!.copyWith(title: newName));
    }
  } catch (_) {
    if (!context.mounted) return;
@@ -259,7 +259,7 @@ Future<void> _confirmAndDeleteConversation(
    HapticFeedback.mediumImpact();
    final active = ref.read(activeConversationProvider);
    if (active?.id == conversationId) {
-      ref.read(activeConversationProvider.notifier).state = null;
+      ref.read(activeConversationProvider.notifier).clear();
      ref.read(chat.chatMessagesProvider.notifier).clearMessages();
    }
    ref.invalidate(conversationsProvider);