generaltranslation 1.1.21 → 1.1.22

package/index.js

@@ -5,7 +5,7 @@
 
 const { _getLanguageName, _getLanguageCode } = require('./codes/codes.js');
 const { _getModelList, _getModelInfo, _getAllModelInfo, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.js');
-const { _translate } = require('./translate/translate.js');
+const { _translatePrompt } = require('./prompts/translate.js');
 
 // ----- CORE CLASS ----- // 
 
@@ -13,10 +13,12 @@ class GT {
 
     constructor({
         apiKey = '', 
-        defaultLanguage = 'en'
+        defaultLanguage = 'en',
+        baseURL = 'https://translate.gtx.dev'
     } = {}) {
         this.apiKey = apiKey;
         this.defaultLanguage = defaultLanguage;
+        this.baseURL = baseURL;
     }
 
     // Language code functions
@@ -32,13 +34,11 @@ class GT {
     getModelsByLanguage = _getModelsByLanguage; // returns array of model names
     getModelsByDeveloper = _getModelsByDeveloper; // returns array of model names
 
-    // Prompt internationalization
+    // Prompt I18N
     getPrompt = async (prompt, language) => {
-        return await _translate(prompt, language, this.defaultLanguage, this.apiKey);
-    }
-    // Translation (same as prompt internationalization)
-    translate = async (content, language) => {
-        return await _translate(content, language, this.defaultLanguage, this.apiKey);
+        return await _translatePrompt({
+            content: prompt, language: language, context: this
+        });
     }
 
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "generaltranslation",
-  "version": "1.1.21",
+  "version": "1.1.22",
   "description": "A language toolkit for AI developers",
   "main": "index.js",
   "scripts": {

package/{translate → prompts}/translate.js

@@ -5,7 +5,7 @@ const _shouldTranslate = item => typeof item?.translate === 'boolean' ? item.tra
 
 // Pre-processes content to send to the API
 // Separates out text that shouldn't be translated.
-const _processContent = (content) => {
+const _processContent = ({ content }) => {
     const processed = [];
     const untranslated = [];
     if (Array.isArray(content)) {
@@ -64,16 +64,24 @@ const _constructContent = ({ content, untranslated = null}) => {
 
 // Get a translation via General Translation API
 // Returns string
-const _translate = async (content, language, defaultLanguage, apiKey) => {
+const _translatePrompt = async ({
+    content, language, context
+}) => {
+    
+    const apiKey = context?.apiKey;
     if (!apiKey) {
         throw new Error('Missing API Key!')
-    }
+    };
+
+    const defaultLanguage = context?.defaultLanguage;
     if (language === defaultLanguage) {
         return _constructContent({ content: content });
-    }
-    const { processed, untranslated } = _processContent(content);
+    };
+
+    const { processed, untranslated } = _processContent({ content });
+    
     try {
-        const response = await fetch('http://translate.gtx.dev', {
+        const response = await fetch(`${context?.baseURL}/prompt`, {
             method: 'POST',
             headers: {
                 'Content-Type': 'application/json',
@@ -96,8 +104,9 @@ const _translate = async (content, language, defaultLanguage, apiKey) => {
         console.error(error)
         return _constructContent({ content: content })
     }
+
 }
 
 module.exports = {
-    _translate
+    _translatePrompt
 }