From a1a56758750fba25d070298bee1946646292aeb9 Mon Sep 17 00:00:00 2001
From: "Lucas.Xu" <lucas.xu@appflowy.io>
Date: Wed, 1 Mar 2023 17:38:14 +0800
Subject: [PATCH] fix: openai for summarize doesn't work when mutliple
 selecting (#1904)

---
 .../presentation/plugins/openai/service/openai_client.dart  | 6 ++++--
 .../plugins/openai/widgets/smart_edit_action.dart           | 4 ++--
 .../plugins/openai/widgets/smart_edit_node_widget.dart      | 1 +
 3 files changed, 7 insertions(+), 4 deletions(-)

diff --git a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/service/openai_client.dart b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/service/openai_client.dart
index f8b3b050fb..3bd41e2d52 100644
--- a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/service/openai_client.dart
+++ b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/service/openai_client.dart
@@ -37,7 +37,7 @@ abstract class OpenAIRepository {
   Future<Either<OpenAIError, TextCompletionResponse>> getCompletions({
     required String prompt,
     String? suffix,
-    int maxTokens = 50,
+    int maxTokens = 500,
     double temperature = .3,
   });
 
@@ -72,7 +72,7 @@ class HttpOpenAIRepository implements OpenAIRepository {
   Future<Either<OpenAIError, TextCompletionResponse>> getCompletions({
     required String prompt,
     String? suffix,
-    int maxTokens = 50,
+    int maxTokens = 500,
     double temperature = 0.3,
   }) async {
     final parameters = {
@@ -102,12 +102,14 @@ class HttpOpenAIRepository implements OpenAIRepository {
     required String input,
     required String instruction,
     double temperature = 0.3,
+    int n = 1,
   }) async {
     final parameters = {
       'model': 'text-davinci-edit-001',
       'input': input,
       'instruction': instruction,
       'temperature': temperature,
+      'n': n,
     };
 
     final response = await client.post(
diff --git a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_action.dart b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_action.dart
index a7c2ef2626..5ceb4d7ea6 100644
--- a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_action.dart
+++ b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_action.dart
@@ -10,9 +10,9 @@ enum SmartEditAction {
   String get toInstruction {
     switch (this) {
       case SmartEditAction.summarize:
-        return 'Summarize';
+        return 'Make it shorter';
       case SmartEditAction.fixSpelling:
-        return 'Fix the spelling mistakes';
+        return 'Fix all the spelling mistakes';
     }
   }
 }
diff --git a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_node_widget.dart b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_node_widget.dart
index 2ce7267a7e..d98595e0da 100644
--- a/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_node_widget.dart
+++ b/frontend/appflowy_flutter/lib/plugins/document/presentation/plugins/openai/widgets/smart_edit_node_widget.dart
@@ -254,6 +254,7 @@ class _SmartEditInputState extends State<_SmartEditInput> {
       final edits = await openAIRepository.getEdits(
         input: input,
         instruction: instruction,
+        n: input.split('\n').length,
       );
       return edits.fold((error) async {
         return dartz.Left(