eduvia/workavia-mail-front
11
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

feat(ai-scribe): Improve data & domain layer for ai scribe

Browse Source
This commit is contained in:
dab246
2025-12-18 04:31:18 +07:00
committed by Dat H. Pham
parent 1a1b8f489f
commit 860393fc90
16 changed files with 94 additions and 117 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/lib/data/network/dio_client.dart
+5 -2
View File
@@ -40,9 +40,12 @@ class DioClient {
CancelToken? cancelToken,
ProgressCallback? onSendProgress,
ProgressCallback? onReceiveProgress,
bool useJMAPHeader = true,
}) async {
final newOptions = options?.appendHeaders({HttpHeaders.acceptHeader : jmapHeader})
?? Options(headers: {HttpHeaders.acceptHeader : jmapHeader}) ;
Map<String, dynamic> defaultHeaders =
useJMAPHeader ? {HttpHeaders.acceptHeader: jmapHeader} : {};
final newOptions = options?.appendHeaders(defaultHeaders) ??
Options(headers: defaultHeaders);
return await _dio.post(path,
data: data,
lib/features/base/mixin/ai_scribe_mixin.dart
+26 -4
View File
@@ -1,13 +1,35 @@
import 'package:core/utils/app_logger.dart';
import 'package:core/utils/platform_info.dart';
import 'package:jmap_dart_client/jmap/account_id.dart';
import 'package:jmap_dart_client/jmap/core/session/session.dart';
import 'package:scribe/scribe.dart';
import 'package:scribe/scribe/ai/presentation/bindings/ai_scribe_bindings.dart';
import 'package:tmail_ui_user/features/home/domain/extensions/session_extensions.dart';
mixin AiScribeMixin {
bool isAICapabilitySupported({Session? session, AccountId? accountId}) {
AICapability? getAICapability({Session? session, AccountId? accountId}) {
if (PlatformInfo.isMobile) return null;
if (accountId == null || session == null) {
return false;
return null;
}
return session.getAICapability(accountId);
}
void injectAIScribeBindings(Session? session, AccountId? accountId) {
try {
final aiCapability = getAICapability(
session: session,
accountId: accountId,
);
final scribeEndpoint = aiCapability?.scribeEndpoint;
if (scribeEndpoint == null || scribeEndpoint.isEmpty) return;
AIScribeBindings(scribeEndpoint).dependencies();
} catch (e) {
logError('AiScribeMixin::injectAIScribeBindings(): $e');
}
final aiCapability = session.getAICapability(accountId);
return aiCapability != null;
}
}
lib/features/composer/presentation/composer_controller.dart
-2
View File
@@ -263,8 +263,6 @@ class ComposerController extends BaseController
TransformHtmlEmailContentInteractor get transformHtmlEmailContentInteractor => _transformHtmlEmailContentInteractor;
GenerateAITextInteractor get generateAITextInteractor => Get.find<GenerateAITextInteractor>();
String get ownEmailAddress =>
mailboxDashBoardController.ownEmailAddress.value;
lib/features/composer/presentation/extensions/ai_scribe/handle_ai_scribe_in_composer_extension.dart
+2 -1
View File
@@ -10,10 +10,11 @@ import 'package:tmail_ui_user/features/composer/presentation/mixin/text_selectio
extension HandleAiScribeInComposerExtension on ComposerController {
bool get isAIScribeAvailable {
return isAICapabilitySupported(
final aiCapability = getAICapability(
session: mailboxDashBoardController.sessionCurrent,
accountId: mailboxDashBoardController.accountId.value,
);
return aiCapability?.isScribeEndpointAvailable == true;
}
Future<String> _getTextOnlyContentInEditor() async {
lib/features/mailbox_dashboard/presentation/controller/mailbox_dashboard_controller.dart
+3 -43
View File
@@ -31,6 +31,7 @@ import 'package:receive_sharing_intent/receive_sharing_intent.dart';
import 'package:rxdart/transformers.dart';
import 'package:server_settings/server_settings/tmail_server_settings_extension.dart';
import 'package:tmail_ui_user/features/base/action/ui_action.dart';
import 'package:tmail_ui_user/features/base/mixin/ai_scribe_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/contact_support_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/message_dialog_action_manager.dart';
import 'package:tmail_ui_user/features/base/mixin/own_email_address_mixin.dart';
@@ -78,15 +79,8 @@ import 'package:tmail_ui_user/features/email/presentation/model/composer_argumen
import 'package:tmail_ui_user/features/email/presentation/utils/email_utils.dart';
import 'package:tmail_ui_user/features/email_recovery/presentation/model/email_recovery_arguments.dart';
import 'package:tmail_ui_user/features/home/data/exceptions/session_exceptions.dart';
import 'package:tmail_ui_user/features/home/domain/extensions/session_extensions.dart';
import 'package:tmail_ui_user/features/home/domain/state/auto_sign_in_via_deep_link_state.dart';
import 'package:tmail_ui_user/features/home/domain/usecases/store_session_interactor.dart';
import 'package:scribe/scribe/ai/data/datasource/ai_datasource.dart';
import 'package:scribe/scribe/ai/data/datasource_impl/ai_datasource_impl.dart';
import 'package:scribe/scribe/ai/data/repository/ai_repository_impl.dart';
import 'package:scribe/scribe/ai/domain/repository/ai_scribe_repository.dart';
import 'package:scribe/scribe/ai/domain/usecases/generate_ai_text_interactor.dart';
import 'package:scribe/scribe/ai/presentation/bindings/ai_scribe_bindings.dart';
import 'package:tmail_ui_user/features/identity_creator/domain/state/get_identity_cache_on_web_state.dart';
import 'package:tmail_ui_user/features/identity_creator/domain/usecase/get_identity_cache_on_web_interactor.dart';
import 'package:tmail_ui_user/features/login/domain/exceptions/logout_exception.dart';
@@ -223,7 +217,8 @@ import 'package:uuid/uuid.dart';
class MailboxDashBoardController extends ReloadableController
with ContactSupportMixin,
OwnEmailAddressMixin,
SaaSPremiumMixin {
SaaSPremiumMixin,
AiScribeMixin {
final RemoveEmailDraftsInteractor _removeEmailDraftsInteractor = Get.find<RemoveEmailDraftsInteractor>();
final EmailReceiveManager _emailReceiveManager = Get.find<EmailReceiveManager>();
@@ -825,41 +820,6 @@ class MailboxDashBoardController extends ReloadableController
}
}
void injectAIScribeBindings(Session? session, AccountId? accountId) {
try {
if (!PlatformInfo.isWeb) return;
if (session == null || accountId == null) return;
final aiCapability = session.getAICapability(accountId);
final scribeEndpoint = aiCapability?.scribeEndpoint;
if (scribeEndpoint == null || scribeEndpoint.isEmpty) return;
// Delete existing AI bindings if they exist
if (Get.isRegistered<AIDataSourceImpl>()) {
Get.delete<AIDataSourceImpl>(force: true);
}
if (Get.isRegistered<AIDataSource>()) {
Get.delete<AIDataSource>(force: true);
}
if (Get.isRegistered<AIScribeRepositoryImpl>()) {
Get.delete<AIScribeRepositoryImpl>(force: true);
}
if (Get.isRegistered<AIScribeRepository>()) {
Get.delete<AIScribeRepository>(force: true);
}
if (Get.isRegistered<GenerateAITextInteractor>()) {
Get.delete<GenerateAITextInteractor>(force: true);
}
// Reinitialize with the correct endpoint
AIScribeBindings(scribeEndpoint: scribeEndpoint).dependencies();
} catch (e) {
logError('MailboxDashBoardController::injectAIScribeBindings(): $e');
}
}
@override
Future<void> injectFCMBindings(Session? session, AccountId? accountId) async {
try {
lib/features/manage_account/presentation/manage_account_dashboard_controller.dart
+3 -1
View File
@@ -14,6 +14,7 @@ import 'package:model/model.dart';
import 'package:rule_filter/rule_filter/capability_rule_filter.dart';
import 'package:server_settings/server_settings/capability_server_settings.dart';
import 'package:tmail_ui_user/features/base/action/ui_action.dart';
import 'package:tmail_ui_user/features/base/mixin/ai_scribe_mixin.dart';
import 'package:tmail_ui_user/features/base/mixin/own_email_address_mixin.dart';
import 'package:tmail_ui_user/features/base/reloadable/reloadable_controller.dart';
import 'package:tmail_ui_user/features/base/widget/dialog_picker/color_dialog_picker.dart';
@@ -55,7 +56,7 @@ import 'package:tmail_ui_user/main/routes/route_utils.dart';
import 'package:tmail_ui_user/main/utils/app_config.dart';
class ManageAccountDashBoardController extends ReloadableController
with OwnEmailAddressMixin {
with OwnEmailAddressMixin, AiScribeMixin {
GetAllVacationInteractor? _getAllVacationInteractor;
UpdateVacationInteractor? _updateVacationInteractor;
@@ -152,6 +153,7 @@ class ManageAccountDashBoardController extends ReloadableController
_setUpMinInputLengthAutocomplete();
_bindingInteractorForMenuItemView(sessionCurrent, accountId.value);
_getVacationResponse();
injectAIScribeBindings(sessionCurrent, accountId.value);
paywallController = PaywallController(
ownEmailAddress: ownEmailAddress.value,
);
scribe/lib/scribe/ai/data/datasource/ai_datasource.dart
+1 -1
View File
@@ -1,5 +1,5 @@
import 'package:scribe/scribe/ai/domain/model/ai_response.dart';
abstract class AIDataSource {
Future<AIResponse> request(String prompt);
Future<AIResponse> generateMessage(String prompt);
}
scribe/lib/scribe/ai/data/datasource_impl/ai_datasource_impl.dart
+3 -20
View File
@@ -1,34 +1,17 @@
import 'package:dio/dio.dart';
import 'package:scribe/scribe/ai/data/datasource/ai_datasource.dart';
import 'package:scribe/scribe/ai/data/model/ai_message.dart';
import 'package:scribe/scribe/ai/data/model/ai_api_request.dart';
import 'package:scribe/scribe/ai/data/network/ai_api.dart';
import 'package:scribe/scribe/ai/domain/model/ai_response.dart';
class AIDataSourceImpl implements AIDataSource {
final AIApi _aiApi;
AIDataSourceImpl({
required Dio dio,
String? endpoint,
}) : _aiApi = AIApi(
dio: dio,
endpoint: endpoint,
);
AIDataSourceImpl(this._aiApi);
@override
Future<AIResponse> request(String prompt) async {
Future<AIResponse> generateMessage(String prompt) async {
try {
final aiRequest = AIAPIRequest(
messages: [
AIMessage(
role: 'user',
content: prompt,
),
],
);
final apiResponse = await _aiApi.chatCompletion(aiRequest);
final apiResponse = await _aiApi.generateMessage(prompt);
return AIResponse(result: apiResponse.content);
} on DioError catch (e) {
throw Exception('Failed to generate AI text: ${e.message}');
scribe/lib/scribe/ai/data/model/ai_message.dart
+7 -1
View File
@@ -12,7 +12,13 @@ class AIMessage {
required this.content,
});
factory AIMessage.fromJson(Map<String, dynamic> json) => _$AIMessageFromJson(json);
factory AIMessage.fromJson(Map<String, dynamic> json) =>
_$AIMessageFromJson(json);
Map<String, dynamic> toJson() => _$AIMessageToJson(this);
factory AIMessage.ofUser(String content) => AIMessage(
role: 'user',
content: content,
);
}
scribe/lib/scribe/ai/data/network/ai_api.dart
+16 -20
View File
@@ -1,30 +1,22 @@
import 'package:dio/dio.dart';
import 'package:scribe/scribe/ai/data/model/ai_api_response.dart';
import 'package:core/data/network/dio_client.dart';
import 'package:scribe/scribe/ai/data/model/ai_api_request.dart';
import 'package:scribe/scribe/ai/data/model/ai_api_response.dart';
import 'package:scribe/scribe/ai/data/model/ai_message.dart';
import 'package:scribe/scribe/ai/data/network/ai_api_exception.dart';
class AIApi {
final Dio _dio;
final String? _endpoint;
final DioClient _dioClient;
final String aiEndpoint;
AIApi({
required Dio dio,
String? endpoint,
}) : _dio = dio,
_endpoint = endpoint;
AIApi(this._dioClient, this.aiEndpoint);
Future<AIApiResponse> chatCompletion(
AIAPIRequest request
) async {
final url = _endpoint;
Future<AIApiResponse> generateMessage(String prompt) async {
final aiRequest = _generateRequest(prompt);
if (url == null) {
throw AIApiNotAvailableException();
}
final response = await _dio.post(
url,
data: request.toJson(),
final response = await _dioClient.post(
aiEndpoint,
data: aiRequest.toJson(),
useJMAPHeader: false,
);
if (response.statusCode == 200) {
@@ -39,4 +31,8 @@ class AIApi {
);
}
}
AIAPIRequest _generateRequest(String prompt) {
return AIAPIRequest(messages: [AIMessage.ofUser(prompt)]);
}
}
scribe/lib/scribe/ai/data/repository/ai_repository_impl.dart
+4 -4
View File
@@ -3,12 +3,12 @@ import 'package:scribe/scribe/ai/domain/model/ai_response.dart';
import 'package:scribe/scribe/ai/domain/repository/ai_scribe_repository.dart';
class AIScribeRepositoryImpl implements AIScribeRepository {
final AIDataSource _dataSource;
final AIDataSource _aiDataSource;
AIScribeRepositoryImpl(this._dataSource);
AIScribeRepositoryImpl(this._aiDataSource);
@override
Future<AIResponse> generateText(String prompt) async {
return _dataSource.request(prompt);
Future<AIResponse> generateMessage(String prompt) {
return _aiDataSource.generateMessage(prompt);
}
}
scribe/lib/scribe/ai/domain/repository/ai_scribe_repository.dart
+1 -1
View File
@@ -1,5 +1,5 @@
import 'package:scribe/scribe/ai/domain/model/ai_response.dart';
abstract class AIScribeRepository {
Future<AIResponse> generateText(String prompt);
Future<AIResponse> generateMessage(String prompt);
}
scribe/lib/scribe/ai/domain/usecases/generate_ai_text_interactor.dart
+3 -3
View File
@@ -1,5 +1,6 @@
import 'package:core/presentation/state/failure.dart';
import 'package:core/presentation/state/success.dart';
import 'package:core/utils/app_logger.dart';
import 'package:dartz/dartz.dart';
import 'package:scribe/scribe/ai/domain/constants/ai_prompts.dart';
import 'package:scribe/scribe/ai/domain/repository/ai_scribe_repository.dart';
@@ -17,9 +18,8 @@ class GenerateAITextInteractor {
) async {
try {
final prompt = AIPrompts.buildPrompt(action, selectedText);
final response = await _repository.generateText(prompt);
log('$runtimeType::execute():prompt: $prompt');
final response = await _repository.generateMessage(prompt);
return Right(GenerateAITextSuccess(response));
} catch (e) {
return Left(GenerateAITextFailure(e));
scribe/lib/scribe/ai/presentation/bindings/ai_scribe_bindings.dart
+17 -13
View File
@@ -1,18 +1,21 @@
import 'package:dio/dio.dart';
import 'package:core/core.dart';
import 'package:get/get.dart';
import 'package:scribe/scribe/ai/data/datasource/ai_datasource.dart';
import 'package:scribe/scribe/ai/data/datasource_impl/ai_datasource_impl.dart';
import 'package:scribe/scribe/ai/data/network/ai_api.dart';
import 'package:scribe/scribe/ai/data/repository/ai_repository_impl.dart';
import 'package:scribe/scribe/ai/domain/repository/ai_scribe_repository.dart';
import 'package:scribe/scribe/ai/domain/usecases/generate_ai_text_interactor.dart';
class AIScribeBindings extends Bindings {
final String? scribeEndpoint;
AIScribeBindings({this.scribeEndpoint});
final String aiEndpoint;
AIScribeBindings(this.aiEndpoint);
@override
void dependencies() {
_bindingsAPI();
_bindingsDataSourceImpl();
_bindingsDataSource();
_bindingsRepositoryImpl();
@@ -20,11 +23,12 @@ class AIScribeBindings extends Bindings {
_bindingsInteractor();
}
void _bindingsAPI() {
Get.lazyPut<AIApi>(() => AIApi(Get.find<DioClient>(), aiEndpoint));
}
void _bindingsDataSourceImpl() {
Get.lazyPut<AIDataSourceImpl>(() => AIDataSourceImpl(
dio: Get.find<Dio>(),
endpoint: scribeEndpoint,
));
Get.lazyPut<AIDataSourceImpl>(() => AIDataSourceImpl(Get.find<AIApi>()));
}
void _bindingsDataSource() {
@@ -32,9 +36,9 @@ class AIScribeBindings extends Bindings {
}
void _bindingsRepositoryImpl() {
Get.lazyPut<AIScribeRepositoryImpl>(() => AIScribeRepositoryImpl(
Get.find<AIDataSource>(),
));
Get.lazyPut<AIScribeRepositoryImpl>(
() => AIScribeRepositoryImpl(Get.find<AIDataSource>()),
);
}
void _bindingsRepository() {
@@ -42,8 +46,8 @@ class AIScribeBindings extends Bindings {
}
void _bindingsInteractor() {
Get.lazyPut<GenerateAITextInteractor>(() => GenerateAITextInteractor(
Get.find<AIScribeRepository>(),
));
Get.lazyPut<GenerateAITextInteractor>(
() => GenerateAITextInteractor(Get.find<AIScribeRepository>()),
);
}
}
scribe/lib/scribe/ai/presentation/model/ai_capability.dart
+2
View File
@@ -15,6 +15,8 @@ class AICapability extends CapabilityProperties {
factory AICapability.deserialize(Map<String, dynamic> json) => _$AICapabilityFromJson(json);
bool get isScribeEndpointAvailable => scribeEndpoint?.trim().isNotEmpty == true;
@override
List<Object?> get props => [scribeEndpoint];
}
scribe/lib/scribe/ai/presentation/widgets/modal/ai_scribe_suggestion_widget.dart
+1 -1
View File
@@ -66,7 +66,7 @@ class _AiScribeSuggestionWidgetModalState
}
}
void _getAiSuggestion(
Future<void> _getAiSuggestion(
GenerateAITextInteractor interactor,
AIAction action,
String? content,