@yxw007/translate 0.0.20 → 0.1.1

package/dist/browser/index.umd.js

@@ -1,4 +1,4 @@
-// translate v0.0.20 Copyright (c) 2024 Potter<aa4790139@gmail.com> and contributors
+// translate v0.1.1 Copyright (c) 2024 Potter<aa4790139@gmail.com> and contributors
 (function (global, factory) {
     typeof exports === 'object' && typeof module !== 'undefined' ? factory(exports, require('@aws-sdk/client-translate')) :
     typeof define === 'function' && define.amd ? define(['exports', '@aws-sdk/client-translate'], factory) :
@@ -13,8 +13,34 @@
             Error.captureStackTrace(this, this.constructor);
         }
     }
-
-    function google$1(options) {
+    const OPEN_AI_MODELS = [
+        "o1-preview",
+        "o1-preview-2024-09-12",
+        "o1-mini-2024-09-12",
+        "o1-mini",
+        "dall-e-2",
+        "gpt-3.5-turbo",
+        "gpt-3.5-turbo-0125",
+        "babbage-002",
+        "davinci-002",
+        "dall-e-3",
+        "text-embedding-3-large",
+        "gpt-3.5-turbo-16k",
+        "tts-1-hd-1106",
+        "text-embedding-ada-002",
+        "text-embedding-3-small",
+        "tts-1-hd",
+        "whisper-1",
+        "gpt-3.5-turbo-1106",
+        "gpt-3.5-turbo-instruct",
+        "gpt-4o-mini-2024-07-18",
+        "gpt-4o-mini",
+        "tts-1",
+        "tts-1-1106",
+        "gpt-3.5-turbo-instruct-0914",
+    ];
+
+    function google(options) {
         const base = "https://translate.googleapis.com/translate_a/single";
         return {
             name: "google",
@@ -1349,12 +1375,74 @@
         };
     }
 
+    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: google$1,
+        google,
         azure: azure$1,
         amazon: amazon$1,
         baidu: baidu$1,
         deepl: deepl$2,
+        openai: openai$1,
     };
 
     class Cache {
@@ -1582,7 +1670,7 @@
         Zulu: "zu",
     };
 
-    var google = {
+    var openai = {
         Abkhazian: "ab",
         Acehnese: "ace",
         "Acholi language": "ach",
@@ -2155,10 +2243,11 @@
 
     const originLanguages = {
         azure: azure,
-        google: google,
+        google: openai,
         baidu: baidu,
         deepl: deepl$1,
         amazon: amazon,
+        openai: openai,
     };
 
     var deepl = {
@@ -2202,10 +2291,11 @@
 
     const targetLanguages = {
         azure: azure,
-        google: google,
+        google: openai,
         baidu: baidu,
         deepl: deepl,
         amazon: amazon,
+        openai: openai,
     };
 
     function normalFromLanguage(from, engine) {
@@ -2259,13 +2349,28 @@
             this.engines = new Map();
             this.cache_time = cache_time;
         }
+        /**
+         * This method is obsolete, please use the addEngine method
+         * @param engine {@link Engine}  instance
+         * @deprecated Use {@link addEngine} instead.
+         */
         use(engine) {
+            this.addEngine(engine);
+        }
+        addEngine(engine) {
             if (this.engines.has(engine.name)) {
                 logger.warn("Engine already exists");
                 return;
             }
             this.engines.set(engine.name, engine);
         }
+        removeEngine(engineName) {
+            if (!engineName || !this.engines.has(engineName)) {
+                logger.warn("Engine name is required or not found");
+                return false;
+            }
+            this.engines.delete(engineName);
+        }
         async translate(text, options) {
             const { engine = "google", cache_time = 60 * 1000 } = options;
             let { from = "auto", to } = options;
@@ -2317,6 +2422,7 @@
     };
 
     exports.Cache = Cache;
+    exports.OPEN_AI_MODELS = OPEN_AI_MODELS;
     exports.TranslationError = TranslationError;
     exports.Translator = Translator;
     exports.default = index;