Mirrored_Repos/AppFlowy
1
0
Fork 0
mirror of https://github.com/AppFlowy-IO/AppFlowy.git synced 2024-08-30 18:12:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Revert "feat: support regenerating ai results #2012 (#2528)" (#2622)" (#2636)

Browse Source 
This reverts commit 2b5ecf5609.
This commit is contained in:
Alex Wallen 2023-05-26 14:00:20 -10:00 committed by GitHub
parent 2b5ecf5609
commit 6fbd88fe76
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 127 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
frontend/appflowy_flutter/assets/translations/en.json
View File

@ -360,6 +360,7 @@
"autoGeneratorGenerate": "Generate",
"autoGeneratorHintText": "Ask OpenAI ...",
"autoGeneratorCantGetOpenAIKey": "Can't get OpenAI key",
"autoGeneratorRewrite": "Rewrite",
"smartEdit": "AI Assistants",
"openAI": "OpenAI",
"smartEditFixSpelling": "Fix spelling",

104
frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/auto_completion_node_widget.dart
View File

@ -1,5 +1,4 @@
import 'dart:convert';
import 'package:appflowy/plugins/document/presentation/plugins/openai/service/openai_client.dart';
import 'package:appflowy/plugins/document/presentation/plugins/openai/util/learn_more_action.dart';
import 'package:appflowy/plugins/document/presentation/plugins/openai/widgets/discard_dialog.dart';
@ -21,6 +20,8 @@ import '../util/editor_extension.dart';
const String kAutoCompletionInputType = 'auto_completion_input';
const String kAutoCompletionInputString = 'auto_completion_input_string';
const String kAutoCompletionGenerationCount =
'auto_completion_generation_count';
const String kAutoCompletionInputStartSelection =
'auto_completion_input_start_selection';
@ -124,7 +125,8 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
}
Widget _buildAutoGeneratorPanel(BuildContext context) {
if (text.isEmpty) {
if (text.isEmpty &&
widget.node.attributes[kAutoCompletionGenerationCount] < 1) {
return Column(
mainAxisSize: MainAxisSize.min,
children: [
@ -204,6 +206,15 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
);
}
Future<void> _updateGenerationCount() async {
final transaction = widget.editorState.transaction;
transaction.updateNode(widget.node, {
kAutoCompletionGenerationCount:
widget.node.attributes[kAutoCompletionGenerationCount] + 1
});
await widget.editorState.apply(transaction);
}
Widget _buildFooterWidget(BuildContext context) {
return Row(
children: [
@ -212,6 +223,11 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
onPressed: () => _onExit(),
),
const Space(10, 0),
SecondaryTextButton(
LocaleKeys.document_plugins_autoGeneratorRewrite.tr(),
onPressed: () => _onRewrite(),
),
const Space(10, 0),
SecondaryTextButton(
LocaleKeys.button_discard.tr(),
onPressed: () => _onDiscard(),
@ -272,6 +288,7 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
await _showError(error.message);
},
);
await _updateGenerationCount();
}, (error) async {
loading.stop();
await _showError(
@ -280,6 +297,88 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
});
}
Future<void> _onRewrite() async {
String previousOutput = _getPreviousOutput()!;
final loading = Loading(context);
loading.start();
// clear previous response
final selection =
widget.node.attributes[kAutoCompletionInputStartSelection];
if (selection != null) {
final start = Selection.fromJson(json.decode(selection)).start.path;
final end = widget.node.previous?.path;
if (end != null) {
final transaction = widget.editorState.transaction;
transaction.deleteNodesAtPath(
start,
end.last - start.last + 1,
);
await widget.editorState.apply(transaction);
}
}
// generate new response
final result = await UserBackendService.getCurrentUserProfile();
result.fold((userProfile) async {
final openAIRepository = HttpOpenAIRepository(
client: http.Client(),
apiKey: userProfile.openaiKey,
);
await openAIRepository.getStreamedCompletions(
prompt: _rewritePrompt(previousOutput),
onStart: () async {
loading.stop();
await _makeSurePreviousNodeIsEmptyTextNode();
},
onProcess: (response) async {
if (response.choices.isNotEmpty) {
final text = response.choices.first.text;
await widget.editorState.autoInsertText(
text,
inputType: TextRobotInputType.word,
delay: Duration.zero,
);
}
},
onEnd: () async {},
onError: (error) async {
loading.stop();
await _showError(error.message);
},
);
await _updateGenerationCount();
}, (error) async {
loading.stop();
await _showError(
LocaleKeys.document_plugins_autoGeneratorCantGetOpenAIKey.tr(),
);
});
}
String? _getPreviousOutput() {
final selection =
widget.node.attributes[kAutoCompletionInputStartSelection];
if (selection != null) {
final start = Selection.fromJson(json.decode(selection)).start.path;
final end = widget.node.previous?.path;
if (end != null) {
String lastOutput = "";
for (var i = start.last; i < end.last - start.last + 2; i++) {
TextNode? textNode =
widget.editorState.document.nodeAtPath([i]) as TextNode?;
lastOutput = "$lastOutput ${textNode!.toPlainText()}";
}
return lastOutput.trim();
}
}
return null;
}
String _rewritePrompt(String previousOutput) {
String prompt =
'I am not satisfied with your previous response($previousOutput) to the query ($text) please write another one';
return prompt;
}
Future<void> _onDiscard() async {
final selection =
widget.node.attributes[kAutoCompletionInputStartSelection];
@ -293,6 +392,7 @@ class _AutoCompletionInputState extends State<_AutoCompletionInput> {
end.last - start.last + 1,
);
await widget.editorState.apply(transaction);
await _makeSurePreviousNodeIsEmptyTextNode();
}
}
_onExit();

1
frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/auto_completion_plugins.dart
View File

@ -14,6 +14,7 @@ SelectionMenuItem autoGeneratorMenuItem = SelectionMenuItem.node(
type: kAutoCompletionInputType,
attributes: {
kAutoCompletionInputString: '',
kAutoCompletionGenerationCount: 0,
},
);
return node;

28
frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_node_widget.dart
View File

@ -211,15 +211,15 @@ class _SmartEditInputState extends State<_SmartEditInput> {
}
Widget _buildResultWidget(BuildContext context) {
final loading = Padding(
final loadingWidget = Padding(
padding: const EdgeInsets.symmetric(horizontal: 4.0),
child: SizedBox.fromSize(
size: const Size.square(14),
child: const CircularProgressIndicator(),
),
);
if (result.isEmpty) {
return loading;
if (result.isEmpty || loading) {
return loadingWidget;
}
return Flexible(
child: Text(
@ -231,6 +231,18 @@ class _SmartEditInputState extends State<_SmartEditInput> {
Widget _buildInputFooterWidget(BuildContext context) {
return Row(
children: [
FlowyRichTextButton(
TextSpan(
children: [
TextSpan(
text: LocaleKeys.document_plugins_autoGeneratorRewrite.tr(),
style: Theme.of(context).textTheme.bodyMedium,
),
],
),
onPressed: () => _requestCompletions(rewrite: true),
),
const Space(10, 0),
FlowyRichTextButton(
TextSpan(
children: [
@ -272,7 +284,7 @@ class _SmartEditInputState extends State<_SmartEditInput> {
),
onPressed: () async => await _onExit(),
),
const Spacer(flex: 2),
const Spacer(flex: 1),
Expanded(
child: FlowyText.regular(
overflow: TextOverflow.ellipsis,
@ -359,7 +371,13 @@ class _SmartEditInputState extends State<_SmartEditInput> {
);
}
Future<void> _requestCompletions() async {
Future<void> _requestCompletions({bool rewrite = false}) async {
if (rewrite) {
setState(() {
loading = true;
result = "";
});
}
final openAIRepository = await getIt.getAsync<OpenAIRepository>();
var lines = input.split('\n\n');