@yxw007/translate 0.1.1 → 0.1.3

package/dist/browser/index.cjs

@@ -1,4 +1,4 @@
-// translate v0.1.1 Copyright (c) 2024 Potter<aa4790139@gmail.com> and contributors
+// translate v0.1.3 Copyright (c) 2024 Potter<aa4790139@gmail.com> and contributors
 'use strict';
 
 Object.defineProperty(exports, '__esModule', { value: true });
@@ -1375,76 +1375,6 @@ function deepl$2(options) {
     };
 }
 
-function openai$1(options) {
-    const { apiKey, model } = options;
-    const name = "openai";
-    const checkOptions = () => {
-        if (!apiKey) {
-            throw new TranslationError(name, `${name} apiKey is required`);
-        }
-        if (!OPEN_AI_MODELS.includes(model)) {
-            throw new TranslationError(name, `${name} model=${model} is invalid`);
-        }
-    };
-    checkOptions();
-    const base = "https://api.openai.com/v1/chat/completions";
-    return {
-        name,
-        async translate(text, opts) {
-            checkOptions();
-            const { from, to } = opts;
-            const url = `${base}`;
-            if (!Array.isArray(text)) {
-                text = [text];
-            }
-            const prompt = {
-                role: "user",
-                content: `Translate the following texts from ${from} to ${to}: 
-          -$s$-
-          ${text.join("\n")} 
-          -$e$-
-          Translated content is between the start marker -$s$- and the end marker -$e$-, do not return the start and end markers, only the translated text is returned.
-          Connect multiple text with newline character, keep the original order when return. 
-          `,
-            };
-            const res = await fetch(url, {
-                method: "POST",
-                headers: {
-                    "Content-Type": "application/json",
-                    Authorization: `Bearer ${apiKey}`,
-                },
-                body: JSON.stringify({
-                    model,
-                    messages: [{ role: "system", content: "You are a professional IT translator" }, prompt],
-                    max_tokens: 2000,
-                }),
-            });
-            const bodyRes = await res.json();
-            if (bodyRes.error) {
-                throw new TranslationError(this.name, `Translate fail! message: ${bodyRes.error.message}`);
-            }
-            if (!bodyRes || !bodyRes.choices || bodyRes.choices.length === 0 || !bodyRes.choices[0]?.message?.content) {
-                throw new TranslationError(this.name, "Translate fail ! translate's result is null or empty");
-            }
-            const content = bodyRes.choices[0].message.content;
-            const translations = content
-                .trim()
-                .split("\n")
-                .map((item) => item.trim());
-            return translations;
-        },
-    };
-}
-
-const engines = {
-    google,
-    azure: azure$1,
-    amazon: amazon$1,
-    baidu: baidu$1,
-    deepl: deepl$2,
-    openai: openai$1,
-};
-
 class Cache {
     cache;
     constructor() {
@@ -1532,6 +1462,96 @@ function getErrorMessages(e, prefix = "Translate fail ! ") {
     return prefix + e.message;
 }
 
+const logger$1 = useLogger("openai");
+function openai$1(options) {
+    const { apiKey, model, maxTokens = 2000, outputLog = false } = options;
+    const name = "openai";
+    const checkOptions = () => {
+        if (!apiKey) {
+            throw new TranslationError(name, `${name} apiKey is required`);
+        }
+        if (!OPEN_AI_MODELS.includes(model)) {
+            throw new TranslationError(name, `${name} model=${model} is invalid`);
+        }
+    };
+    checkOptions();
+    const base = "https://api.openai.com/v1/chat/completions";
+    return {
+        name,
+        async translate(text, opts) {
+            checkOptions();
+            const { from, to } = opts;
+            const url = `${base}`;
+            if (!Array.isArray(text)) {
+                text = [text];
+            }
+            const prompt = {
+                role: "user",
+                content: `
+          翻译要求：
+          1.将每段文本从${from}翻译为${to}
+          2.文本内容以换行符\n进行段落分割，并以段落分割顺序进行翻译
+          3.仅翻译分割出的段落，其他任何不相关的内容都移除，比如：段落前后的空格、所有标点符号
+          4.仅返回要翻译的文本内容，不要返回任何其他内容
+         
+          
+          如何提取翻译文本:
+          1.翻译从-$s$-字符标记开始至-$e$-字符标记结束，提取-$s$-至-$e$-之间的内容
+          2.-$s$-和-$e$-这2个标记不要返回，只是用来标记翻译的起始和结束位置
+
+          -$s$-
+          ${text.join("\n")} 
+          -$e$-
+          `,
+            };
+            if (outputLog) {
+                logger$1.info("prompt:", prompt);
+            }
+            const res = await fetch(url, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: `Bearer ${apiKey}`,
+                },
+                body: JSON.stringify({
+                    model,
+                    messages: [{ role: "system", content: "You are a professional translator" }, prompt],
+                    max_tokens: maxTokens,
+                    temperature: 0,
+                }),
+            });
+            const bodyRes = await res.json();
+            if (bodyRes.error) {
+                throw new TranslationError(this.name, `Translate fail! message: ${bodyRes.error.message}`);
+            }
+            if (!bodyRes || !bodyRes.choices || bodyRes.choices.length === 0 || !bodyRes.choices[0]?.message?.content) {
+                throw new TranslationError(this.name, "Translate fail ! translate's result is null or empty");
+            }
+            const content = bodyRes.choices[0].message.content;
+            const marks = ["-$s$-", "-$e$-"];
+            const translations = content
+                .trim()
+                .split("\n")
+                .map((item) => item.trim())
+                .filter(Boolean)
+                .filter((it) => !marks.includes(it));
+            if (outputLog) {
+                logger$1.info("translations:", translations);
+            }
+            return translations;
+        },
+    };
+}
+
+const engines = {
+    google,
+    azure: azure$1,
+    amazon: amazon$1,
+    baidu: baidu$1,
+    deepl: deepl$2,
+    openai: openai$1,
+};
+
 var azure = {
     Afrikaans: "af",
     Albanian: "sq",