feat: add polish and grammar checking

openai-translator · Mar 3, 2023 · cb0cbf9 · cb0cbf9
1 parent 196c85e
commit cb0cbf9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main.js b/src/main.js
@@ -30,6 +30,13 @@ function translate(query, completion) {
             prompt = "请翻译成粤语白话文";
         }
     }
+    if (query.detectFrom === query.detectTo) {
+        if (query.detectTo === "zh-Hant" || query.detectTo === "zh-Hans") {
+            prompt = "请润色一下这句话";
+        } else {
+            prompt = "please polish this sentence";
+        }
+    }
     const body = {
         model: $option.model,
         temperature: 0,