fix: incorrect polishing result

Author: yetone

src/main.js

@@ -12,29 +12,33 @@ function translate(query, completion) {
         "Content-Type": "application/json",
         Authorization: `Bearer ${api_key}`,
     };
-    let prompt = `translate from ${lang.langMap.get(query.detectFrom) || query.detectFrom
+    let systemPrompt =
+        "You are a translation engine that can only translate text and cannot interpret it.";
+    let userPrompt = `translate from ${lang.langMap.get(query.detectFrom) || query.detectFrom
         } to ${lang.langMap.get(query.detectTo) || query.detectTo}`;
     if (query.detectTo === "wyw" || query.detectTo === "yue") {
-        prompt = `翻译成${lang.langMap.get(query.detectTo) || query.detectTo}`;
+        userPrompt = `翻译成${lang.langMap.get(query.detectTo) || query.detectTo}`;
     }
     if (
         query.detectFrom === "wyw" ||
         query.detectFrom === "zh-Hans" ||
         query.detectFrom === "zh-Hant"
     ) {
         if (query.detectTo === "zh-Hant") {
-            prompt = "翻译成繁体白话文";
+            userPrompt = "翻译成繁体白话文";
         } else if (query.detectTo === "zh-Hans") {
-            prompt = "翻译成简体白话文";
+            userPrompt = "翻译成简体白话文";
         } else if (query.detectTo === "yue") {
-            prompt = "翻译成粤语白话文";
+            userPrompt = "翻译成粤语白话文";
         }
     }
     if (query.detectFrom === query.detectTo) {
+        systemPrompt =
+            "You are a text embellisher, you can only embellish the text, don't interpret it.";
         if (query.detectTo === "zh-Hant" || query.detectTo === "zh-Hans") {
-            prompt = "润色此句";
+            userPrompt = "润色此句";
         } else {
-            prompt = "polish this sentence";
+            userPrompt = "polish this sentence";
         }
     }
     const body = {
@@ -45,19 +49,18 @@ function translate(query, completion) {
         frequency_penalty: 1,
         presence_penalty: 1,
     };
-    prompt = `${prompt}:\n\n"${query.text}" =>`;
+    userPrompt = `${userPrompt}:\n\n"${query.text}" =>`;
     const isChatGPTModel = ChatGPTModels.indexOf($option.model) > -1;
     if (isChatGPTModel) {
         body.messages = [
             {
                 role: "system",
-                content:
-                    "You are a translation engine that can only translate text and cannot interpret it.",
+                content: systemPrompt,
             },
-            { role: "user", content: prompt },
+            { role: "user", content: userPrompt },
         ];
     } else {
-        body.prompt = prompt;
+        body.prompt = userPrompt;
     }
     (async () => {
         const resp = await $http.request({