chore: implement local ai config

Author: appflowy

frontend/appflowy_flutter/lib/ai/service/ai_input_control.dart

@@ -14,7 +14,7 @@ class AIModelStateNotifier {
       : _isDesktop = UniversalPlatform.isDesktop,
         _localAIListener =
             UniversalPlatform.isDesktop ? LocalAIStateListener() : null,
-        _aiModelSwitchListener = AIModelSwitchListener(chatId: objectId);
+        _aiModelSwitchListener = AIModelSwitchListener(objectId: objectId);
 
   final String objectId;
   final bool _isDesktop;

frontend/appflowy_flutter/lib/mobile/presentation/home/mobile_home_setting_page.dart

@@ -100,7 +100,6 @@ class _MobileHomeSettingPageState extends State<MobileHomeSettingPage> {
                       key: ValueKey(currentWorkspaceId),
                       userProfile: userProfile,
                       workspaceId: currentWorkspaceId,
-                      currentWorkspaceMemberRole: state.currentWorkspace?.role,
                     ),
                   const SupportSettingGroup(),
                   const AboutSettingGroup(),

frontend/appflowy_flutter/lib/mobile/presentation/setting/ai/ai_settings_group.dart

@@ -5,8 +5,6 @@ import 'package:appflowy/mobile/presentation/setting/widgets/mobile_setting_item
 import 'package:appflowy/mobile/presentation/widgets/flowy_option_tile.dart';
 import 'package:appflowy/workspace/application/settings/ai/settings_ai_bloc.dart';
 import 'package:appflowy_backend/protobuf/flowy-user/user_profile.pb.dart';
-import 'package:appflowy_backend/protobuf/flowy-user/workspace.pbenum.dart';
-import 'package:collection/collection.dart';
 import 'package:easy_localization/easy_localization.dart';
 import 'package:flowy_infra_ui/style_widget/text.dart';
 import 'package:flutter/material.dart';
@@ -18,12 +16,10 @@ class AiSettingsGroup extends StatelessWidget {
     super.key,
     required this.userProfile,
     required this.workspaceId,
-    this.currentWorkspaceMemberRole,
   });
 
   final UserProfilePB userProfile;
   final String workspaceId;
-  final AFRolePB? currentWorkspaceMemberRole;
 
   @override
   Widget build(BuildContext context) {
@@ -32,7 +28,6 @@ class AiSettingsGroup extends StatelessWidget {
       create: (context) => SettingsAIBloc(
         userProfile,
         workspaceId,
-        currentWorkspaceMemberRole,
       )..add(const SettingsAIEvent.started()),
       child: BlocBuilder<SettingsAIBloc, SettingsAIState>(
         builder: (context, state) {
@@ -48,7 +43,7 @@ class AiSettingsGroup extends StatelessWidget {
                     children: [
                       Flexible(
                         child: FlowyText(
-                          state.selectedAIModel,
+                          state.availableModels?.selectedModel.name ?? "",
                           color: theme.colorScheme.onSurface,
                           overflow: TextOverflow.ellipsis,
                         ),
@@ -84,16 +79,19 @@ class AiSettingsGroup extends StatelessWidget {
       title: LocaleKeys.settings_aiPage_keys_llmModelType.tr(),
       builder: (_) {
         return Column(
-          children: availableModels
-              .mapIndexed(
-                (index, model) => FlowyOptionTile.checkbox(
-                  text: model.name,
-                  showTopBorder: index == 0,
-                  isSelected: state.selectedAIModel == model.name,
+          children: (availableModels?.models ?? [])
+              .asMap()
+              .entries
+              .map(
+                (entry) => FlowyOptionTile.checkbox(
+                  text: entry.value.name,
+                  showTopBorder: entry.key == 0,
+                  isSelected:
+                      availableModels?.selectedModel.name == entry.value.name,
                   onTap: () {
                     context
                         .read<SettingsAIBloc>()
-                        .add(SettingsAIEvent.selectModel(model.name));
+                        .add(SettingsAIEvent.selectModel(entry.value));
                     context.pop();
                   },
                 ),

frontend/appflowy_flutter/lib/plugins/ai_chat/application/ai_model_switch_listener.dart

@@ -12,14 +12,14 @@ import 'package:appflowy_result/appflowy_result.dart';
 typedef OnUpdateSelectedModel = void Function(AIModelPB model);
 
 class AIModelSwitchListener {
-  AIModelSwitchListener({required this.chatId}) {
-    _parser = ChatNotificationParser(id: chatId, callback: _callback);
+  AIModelSwitchListener({required this.objectId}) {
+    _parser = ChatNotificationParser(id: objectId, callback: _callback);
     _subscription = RustStreamReceiver.listen(
       (observable) => _parser?.parse(observable),
     );
   }
 
-  final String chatId;
+  final String objectId;
   StreamSubscription<SubscribeObject>? _subscription;
   ChatNotificationParser? _parser;
 

frontend/appflowy_flutter/lib/workspace/application/settings/ai/settings_ai_bloc.dart

@@ -1,5 +1,5 @@
+import 'package:appflowy/plugins/ai_chat/application/ai_model_switch_listener.dart';
 import 'package:appflowy/user/application/user_listener.dart';
-import 'package:appflowy/user/application/user_service.dart';
 import 'package:appflowy_backend/dispatch/dispatch.dart';
 import 'package:appflowy_backend/log.dart';
 import 'package:appflowy_backend/protobuf/flowy-ai/entities.pb.dart';
@@ -10,48 +10,38 @@ import 'package:bloc/bloc.dart';
 import 'package:freezed_annotation/freezed_annotation.dart';
 
 part 'settings_ai_bloc.freezed.dart';
-part 'settings_ai_bloc.g.dart';
 
 class SettingsAIBloc extends Bloc<SettingsAIEvent, SettingsAIState> {
   SettingsAIBloc(
     this.userProfile,
     this.workspaceId,
-    AFRolePB? currentWorkspaceMemberRole,
   )   : _userListener = UserListener(userProfile: userProfile),
-        _userService = UserBackendService(userId: userProfile.id),
+        _aiModelSwitchListener =
+            AIModelSwitchListener(objectId: "ai_models_global_active_model"),
         super(
           SettingsAIState(
-            selectedAIModel: userProfile.aiModel,
             userProfile: userProfile,
-            currentWorkspaceMemberRole: currentWorkspaceMemberRole,
           ),
         ) {
+    _aiModelSwitchListener.start(
+      onUpdateSelectedModel: (model) {
+        if (!isClosed) {
+          _loadModelList();
+        }
+      },
+    );
     _dispatch();
-
-    if (currentWorkspaceMemberRole == null) {
-      _userService.getWorkspaceMember().then((result) {
-        result.fold(
-          (member) {
-            if (!isClosed) {
-              add(SettingsAIEvent.refreshMember(member));
-            }
-          },
-          (err) {
-            Log.error(err);
-          },
-        );
-      });
-    }
   }
 
   final UserListener _userListener;
   final UserProfilePB userProfile;
-  final UserBackendService _userService;
   final String workspaceId;
+  final AIModelSwitchListener _aiModelSwitchListener;
 
   @override
   Future<void> close() async {
     await _userListener.stop();
+    await _aiModelSwitchListener.stop();
     return super.close();
   }
 
@@ -67,7 +57,6 @@ class SettingsAIBloc extends Bloc<SettingsAIEvent, SettingsAIState> {
               }
             },
           );
-          _loadUserWorkspaceSetting();
           _loadModelList();
         },
         didReceiveUserProfile: (userProfile) {
@@ -82,38 +71,25 @@ class SettingsAIBloc extends Bloc<SettingsAIEvent, SettingsAIState> {
                 !(state.aiSettings?.disableSearchIndexing ?? false),
           );
         },
-        selectModel: (String model) async {
-          await _updateUserWorkspaceSetting(model: model);
+        selectModel: (AIModelPB model) async {
+          if (!model.isLocal) {
+            await _updateUserWorkspaceSetting(model: model.name);
+          }
         },
         didLoadAISetting: (UseAISettingPB settings) {
           emit(
             state.copyWith(
               aiSettings: settings,
-              selectedAIModel: settings.aiModel,
               enableSearchIndexing: !settings.disableSearchIndexing,
             ),
           );
         },
-        didLoadAvailableModels: (List<AvailableModelPB> models) {
-          if (state.selectedAIModel.isEmpty) {
-            final m = models.firstWhere((model) => model.isDefault);
-            _updateUserWorkspaceSetting(model: m.name);
-            emit(
-              state.copyWith(
-                availableModels: models,
-                selectedAIModel: m.name,
-              ),
-            );
-          } else {
-            emit(
-              state.copyWith(
-                availableModels: models,
-              ),
-            );
-          }
-        },
-        refreshMember: (member) {
-          emit(state.copyWith(currentWorkspaceMemberRole: member.role));
+        didLoadAvailableModels: (AvailableModelsPB models) {
+          emit(
+            state.copyWith(
+              availableModels: models,
+            ),
+          );
         },
       );
     });
@@ -148,24 +124,11 @@ class SettingsAIBloc extends Bloc<SettingsAIEvent, SettingsAIState> {
         (err) => Log.error(err),
       );
 
-  void _loadUserWorkspaceSetting() {
-    final payload = UserWorkspaceIdPB(workspaceId: workspaceId);
-    UserEventGetWorkspaceSetting(payload).send().then((result) {
-      result.fold((settings) {
-        if (!isClosed) {
-          add(SettingsAIEvent.didLoadAISetting(settings));
-        }
-      }, (err) {
-        Log.error(err);
-      });
-    });
-  }
-
   void _loadModelList() {
     AIEventGetServerAvailableModels().send().then((result) {
-      result.fold((config) {
+      result.fold((models) {
         if (!isClosed) {
-          add(SettingsAIEvent.didLoadAvailableModels(config.models));
+          add(SettingsAIEvent.didLoadAvailableModels(models));
         }
       }, (err) {
         Log.error(err);
@@ -182,17 +145,15 @@ class SettingsAIEvent with _$SettingsAIEvent {
   ) = _DidLoadWorkspaceSetting;
 
   const factory SettingsAIEvent.toggleAISearch() = _toggleAISearch;
-  const factory SettingsAIEvent.refreshMember(WorkspaceMemberPB member) =
-      _RefreshMember;
 
-  const factory SettingsAIEvent.selectModel(String model) = _SelectAIModel;
+  const factory SettingsAIEvent.selectModel(AIModelPB model) = _SelectAIModel;
 
   const factory SettingsAIEvent.didReceiveUserProfile(
     UserProfilePB newUserProfile,
   ) = _DidReceiveUserProfile;
 
   const factory SettingsAIEvent.didLoadAvailableModels(
-    List<AvailableModelPB> models,
+    AvailableModelsPB models,
   ) = _DidLoadAvailableModels;
 }
 
@@ -201,23 +162,7 @@ class SettingsAIState with _$SettingsAIState {
   const factory SettingsAIState({
     required UserProfilePB userProfile,
     UseAISettingPB? aiSettings,
-    @Default("Default") String selectedAIModel,
-    AFRolePB? currentWorkspaceMemberRole,
-    @Default([]) List<AvailableModelPB> availableModels,
+    AvailableModelsPB? availableModels,
     @Default(true) bool enableSearchIndexing,
   }) = _SettingsAIState;
 }
-
-@JsonSerializable()
-class ModelList {
-  ModelList({
-    required this.models,
-  });
-
-  factory ModelList.fromJson(Map<String, dynamic> json) =>
-      _$ModelListFromJson(json);
-
-  final List<String> models;
-
-  Map<String, dynamic> toJson() => _$ModelListToJson(this);
-}

frontend/appflowy_flutter/lib/workspace/presentation/settings/pages/setting_ai_view/model_selection.dart

@@ -1,3 +1,4 @@
+import 'package:appflowy_backend/protobuf/flowy-ai/entities.pb.dart';
 import 'package:flutter/material.dart';
 
 import 'package:appflowy/generated/locale_keys.g.dart';
@@ -15,6 +16,10 @@ class AIModelSelection extends StatelessWidget {
   Widget build(BuildContext context) {
     return BlocBuilder<SettingsAIBloc, SettingsAIState>(
       builder: (context, state) {
+        if (state.availableModels == null) {
+          return const SizedBox.shrink();
+        }
+
         return Padding(
           padding: const EdgeInsets.symmetric(vertical: 6),
           child: Row(
@@ -28,17 +33,17 @@ class AIModelSelection extends StatelessWidget {
               ),
               const Spacer(),
               Flexible(
-                child: SettingsDropdown<String>(
+                child: SettingsDropdown<AIModelPB>(
                   key: const Key('_AIModelSelection'),
                   onChanged: (model) => context
                       .read<SettingsAIBloc>()
                       .add(SettingsAIEvent.selectModel(model)),
-                  selectedOption: state.selectedAIModel,
-                  options: state.availableModels
+                  selectedOption: state.availableModels!.selectedModel,
+                  options: state.availableModels!.models
                       .map(
-                        (model) => buildDropdownMenuEntry<String>(
+                        (model) => buildDropdownMenuEntry<AIModelPB>(
                           context,
-                          value: model.name,
+                          value: model,
                           label: model.name,
                         ),
                       )

frontend/appflowy_flutter/lib/workspace/presentation/settings/pages/setting_ai_view/settings_ai_view.dart

@@ -42,9 +42,8 @@ class SettingsAIView extends StatelessWidget {
   @override
   Widget build(BuildContext context) {
     return BlocProvider<SettingsAIBloc>(
-      create: (_) =>
-          SettingsAIBloc(userProfile, workspaceId, currentWorkspaceMemberRole)
-            ..add(const SettingsAIEvent.started()),
+      create: (_) => SettingsAIBloc(userProfile, workspaceId)
+        ..add(const SettingsAIEvent.started()),
       child: BlocBuilder<SettingsAIBloc, SettingsAIState>(
         builder: (context, state) {
           final children = <Widget>[

frontend/rust-lib/Cargo.lock

@@ -152,5 +152,5 @@ collab-importer = { version = "0.1", git = "https://github.com/AppFlowy-IO/AppFl
 # To update the commit ID, run:
 # scripts/tool/update_local_ai_rev.sh new_rev_id
 # ⚠️⚠️⚠️️
-appflowy-local-ai = { version = "0.1", git = "https://github.com/AppFlowy-IO/AppFlowy-LocalAI", rev = "cd991507870d225e76df41164deee356cd9921a7" }
-appflowy-plugin = { version = "0.1", git = "https://github.com/AppFlowy-IO/AppFlowy-LocalAI", rev = "cd991507870d225e76df41164deee356cd9921a7" }
+appflowy-local-ai = { version = "0.1", git = "https://github.com/AppFlowy-IO/AppFlowy-LocalAI", rev = "52ad76f21f8f3a7b510dae029836b5fe86479e5a" }
+appflowy-plugin = { version = "0.1", git = "https://github.com/AppFlowy-IO/AppFlowy-LocalAI", rev = "52ad76f21f8f3a7b510dae029836b5fe86479e5a" }

frontend/rust-lib/Cargo.toml

@@ -153,7 +153,6 @@ pub fn parse_event_crate(event_crate: &TsEventCrate) -> Vec<EventASTContext> {
             attrs
               .iter()
               .filter(|attr| !attr.attrs.event_attrs.ignore)
-              .into_iter()
               .map(|variant| EventASTContext::from(&variant.attrs))
               .collect::<Vec<_>>()
           },