generaltranslation 1.1.18 → 1.1.20

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, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.js');
 const { _translate } = require('./translate/translate.js');
 
 // ----- CORE CLASS ----- // 
@@ -28,6 +28,8 @@ class GT {
     getModelInfo = _getModelInfo; // returns model object
     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) => {
@@ -38,7 +40,6 @@ class GT {
         return await _translate(content, code, this.defaultLanguage, this.apiKey);
     }
     
-
 }
 
 // ----- EXPORTS ----- //
@@ -47,5 +48,8 @@ module.exports = GT;
 module.exports.getLanguageCode = _getLanguageCode;
 module.exports.getLanguageName = _getLanguageName;
 module.exports.getModelList = _getModelList;
+module.exports.getModelInfo = _getModelInfo;
 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,35 @@
 // ----- MODEL INFORMATION ----- //
 
+// 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 () => {
@@ -44,6 +74,8 @@ const _isLanguageSupported = async (model, code) => {
 }
 
 module.exports = {
+    _getModelsByDeveloper,
+    _getModelsByLanguage,
     _getModelList,
     _getModelInfo,
     _getModelLanguages,

package/package.json

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