feat: add an option to customize prompt (#18)

openai-translator · Mar 28, 2023 · e075a48 · e075a48
1 parent 2b5b6fd
commit e075a48
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/src/info.json b/src/info.json
@@ -47,6 +47,13 @@
                 }
             ]
         },
+        {
+            "identifier": "prompt",
+            "type": "text",
+            "title": "Prompt",
+            "defaultValue": "Revise the following sentences to make them more clear, concise, and coherent",
+            "desc": "通过自定义 prompt 获取不同的润色结果"
+        },
         {
             "identifier": "polishing_mode",
             "type": "menu",

diff --git a/src/main.js b/src/main.js
@@ -13,8 +13,7 @@ function translate(query, completion) {
         Authorization: `Bearer ${api_key}`,
     };
     const detailedPolishingMode = $option.polishing_mode === "detailed";
-    let prompt =
-        "Revise the following sentences to make them more clear, concise, and coherent";
+    let prompt = $option.prompt;
     if (detailedPolishingMode) {
         prompt = `${prompt}. Please note that you need to list the changes and briefly explain why`;
     }