generaltranslation 1.1.20 → 1.1.22

package/README.md

@@ -102,7 +102,7 @@ const gt = new GT({
 });
 ```
 
-### async getPrompt(prompt, code)
+### async getPrompt(prompt, language)
 
 Translates prompt into the language represented by an ISO-639 language code. Designed for translating prompts into other languages, to internationalize responses from AI models.
 

package/index.js

@@ -4,8 +4,8 @@
 // ----- IMPORTS ----- //
 
 const { _getLanguageName, _getLanguageCode } = require('./codes/codes.js');
-const { _getModelList, _getModelInfo, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.js');
-const { _translate } = require('./translate/translate.js');
+const { _getModelList, _getModelInfo, _getAllModelInfo, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.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
@@ -26,20 +28,19 @@ class GT {
     // Model information functions
     getModelList = _getModelList; // returns array of supported model names
     getModelInfo = _getModelInfo; // returns model object
+    getAllModelInfo =  _getAllModelInfo; // returns all info on all models
     getModelLanguages = _getModelLanguages; // e.g. 'mistral-7b' => ['en']
     isLanguageSupported = _isLanguageSupported; // e.g. ('mistral-7b', 'en') => true
     getModelsByLanguage = _getModelsByLanguage; // returns array of model names
     getModelsByDeveloper = _getModelsByDeveloper; // returns array of model names
 
-    // Prompt internationalization
-    getPrompt = async (prompt, code) => {
-        return await _translate(prompt, code, this.defaultLanguage, this.apiKey);
+    // Prompt I18N
+    getPrompt = async (prompt, language) => {
+        return await _translatePrompt({
+            content: prompt, language: language, context: this
+        });
     }
-    // Translation (same as prompt internationalization)
-    translate = async (content, code) => {
-        return await _translate(content, code, this.defaultLanguage, this.apiKey);
-    }
-    
+
 }
 
 // ----- EXPORTS ----- //
@@ -49,6 +50,7 @@ module.exports.getLanguageCode = _getLanguageCode;
 module.exports.getLanguageName = _getLanguageName;
 module.exports.getModelList = _getModelList;
 module.exports.getModelInfo = _getModelInfo;
+module.exports.getAllModelInfo = _getAllModelInfo;
 module.exports.getModelLanguages = _getModelLanguages;
 module.exports.isLanguageSupported = _isLanguageSupported;
 module.exports.getModelsByDeveloper = _getModelsByDeveloper;

package/models/models.js

@@ -1,5 +1,16 @@
 // ----- MODEL INFORMATION ----- //
 
+// Get all info on all models
+const _getAllModelInfo = async () => {
+    try {
+        const result = await fetch(`https://models.gtx.dev/all`);
+        return await result.json();
+    } catch (error) {
+        console.error(error)
+        return null;
+    }
+}
+
 // Get all models by a given developer
 // Returns an empty array if none
 const _getModelsByDeveloper = async developer => {
@@ -74,10 +85,11 @@ const _isLanguageSupported = async (model, code) => {
 }
 
 module.exports = {
-    _getModelsByDeveloper,
-    _getModelsByLanguage,
     _getModelList,
     _getModelInfo,
     _getModelLanguages,
-    _isLanguageSupported
+    _isLanguageSupported,
+    _getModelsByDeveloper,
+    _getModelsByLanguage,
+    _getAllModelInfo
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "generaltranslation",
-  "version": "1.1.20",
+  "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, code, defaultLanguage, apiKey) => {
+const _translatePrompt = async ({
+    content, language, context
+}) => {
+    
+    const apiKey = context?.apiKey;
     if (!apiKey) {
         throw new Error('Missing API Key!')
-    }
-    if (code === defaultLanguage) {
+    };
+
+    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',
@@ -81,7 +89,7 @@ const _translate = async (content, code, defaultLanguage, apiKey) => {
             },
             body: JSON.stringify({
                 content: processed,
-                targetLanguage: code,
+                targetLanguage: language,
                 defaultLanguage: defaultLanguage
             })
         })
@@ -96,8 +104,9 @@ const _translate = async (content, code, defaultLanguage, apiKey) => {
         console.error(error)
         return _constructContent({ content: content })
     }
+
 }
 
 module.exports = {
-    _translate
+    _translatePrompt
 }