generaltranslation 1.1.19 → 1.1.21

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,7 +4,7 @@
 // ----- IMPORTS ----- //
 
 const { _getLanguageName, _getLanguageCode } = require('./codes/codes.js');
-const { _getModelList, _getModelInfo, _getModelLanguages, _isLanguageSupported } = require('./models/models.js');
+const { _getModelList, _getModelInfo, _getAllModelInfo, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.js');
 const { _translate } = require('./translate/translate.js');
 
 // ----- CORE CLASS ----- // 
@@ -26,18 +26,20 @@ 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);
+    getPrompt = async (prompt, language) => {
+        return await _translate(prompt, language, this.defaultLanguage, this.apiKey);
     }
     // Translation (same as prompt internationalization)
-    translate = async (content, code) => {
-        return await _translate(content, code, this.defaultLanguage, this.apiKey);
+    translate = async (content, language) => {
+        return await _translate(content, language, this.defaultLanguage, this.apiKey);
     }
-    
 
 }
 
@@ -48,5 +50,8 @@ 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.isLanguageSupported = _isLanguageSupported;
+module.exports.getModelsByDeveloper = _getModelsByDeveloper;
+module.exports.getModelsByLanguage = _getModelsByLanguage;

package/models/models.js

@@ -1,5 +1,46 @@
 // ----- 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 => {
+    if (!developer) return null;
+    developer = developer?.toLowerCase();
+    try {
+        const result = await fetch(`https://models.gtx.dev/developer/${developer}`);
+        const models = await result.json();
+        return models;
+    } catch (error) {
+        console.error(error)
+        return [];
+    }
+}
+
+// Get all models certified for a given language
+// Returns an empty array if none
+const _getModelsByLanguage = async language => {
+    if (!language) return null;
+    language = language?.toLowerCase();
+    try {
+        const result = await fetch(`https://models.gtx.dev/language/${language}`);
+        const models = await result.json();
+        return models;
+    } catch (error) {
+        console.error(error)
+        return [];
+    }
+}
+
 // Get all models
 // Returns an array of model names
 const _getModelList = async () => {
@@ -47,5 +88,8 @@ module.exports = {
     _getModelList,
     _getModelInfo,
     _getModelLanguages,
-    _isLanguageSupported
+    _isLanguageSupported,
+    _getModelsByDeveloper,
+    _getModelsByLanguage,
+    _getAllModelInfo
 }

package/package.json

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

package/translate/translate.js

@@ -64,11 +64,11 @@ const _constructContent = ({ content, untranslated = null}) => {
 
 // Get a translation via General Translation API
 // Returns string
-const _translate = async (content, code, defaultLanguage, apiKey) => {
+const _translate = async (content, language, defaultLanguage, apiKey) => {
     if (!apiKey) {
         throw new Error('Missing API Key!')
     }
-    if (code === defaultLanguage) {
+    if (language === defaultLanguage) {
         return _constructContent({ content: content });
     }
     const { processed, untranslated } = _processContent(content);
@@ -81,7 +81,7 @@ const _translate = async (content, code, defaultLanguage, apiKey) => {
             },
             body: JSON.stringify({
                 content: processed,
-                targetLanguage: code,
+                targetLanguage: language,
                 defaultLanguage: defaultLanguage
             })
         })