fix: no translation for some sentences (#16)

openai-translator · Mar 3, 2023 · 449ae6c · 449ae6c
1 parent 0fcd520
commit 449ae6c
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/main.js b/src/main.js
@@ -49,7 +49,7 @@ function translate(query, completion) {
     if (isChatGPTModel) {
         body.messages = [
             { role: "system", content: prompt },
-            { role: "user", content: query.text },
+            { role: "user", content: `"${query.text}"` },
         ];
     } else {
         body.prompt = `${prompt}:\n\n"${query.text}" =>`;
@@ -95,14 +95,14 @@ function translate(query, completion) {
             } else {
                 targetTxt = choices[0].text.trim();
             }
-            if (!isChatGPTModel) {
-                if (targetTxt.startsWith('"')) {
-                    targetTxt = targetTxt.slice(1);
-                }
-                if (targetTxt.endsWith('"')) {
-                    targetTxt = targetTxt.slice(0, -1);
-                }
+
+            if (targetTxt.startsWith('"')) {
+                targetTxt = targetTxt.slice(1);
             }
+            if (targetTxt.endsWith('"')) {
+                targetTxt = targetTxt.slice(0, -1);
+            }
+
             completion({
                 result: {
                     from: query.detectFrom,