generaltranslation 1.0.12 → 1.0.14

package/models/models.js

@@ -1,30 +1,46 @@
+// ----- IMPORTS ----- //
+
+let Models = null;
+let AliasToModel = null;
+
+// Fetches models and aliases from the server and caches them
+const fetchModelData = async () => {
+    if (!Models || !AliasToModel) {
+        const results = await Promise.all([
+            fetch('https://data.gtx.dev/models.json'),
+            fetch('https://data.gtx.dev/alias_to_model.json')
+        ])
+        const values = await Promise.all([
+            results[0].json(),
+            results[1].json()
+        ])
+        Models = values[0];
+        AliasToModel = values[1];
+    }
+    return [Models, AliasToModel];
+}
+
 // ----- MODEL INFORMATION ----- //
 
 // Get all info about a given model
 // Returns an object or null if invalid
 const getModelInfo = async model => {
+    const [Models, AliasToModel] = await fetchModelData();
     model = model?.toLowerCase();
-    try {
-        const result = await fetch(`https://gtx.dev/model/${model}`);
-        const modelInfo = await result.json();
-        return modelInfo;
-    } catch {
-        return null;
-    }
+    const modelName = AliasToModel ? AliasToModel[model] : model;
+    return Models ? Models[modelName] : null;
 }
 
 // Get all languages known to be compatible with a given LLM
 // Returns an array of languages codes, [] if unknown
 const getModelLanguages = async model => {
-    const modelInfo = await getModelInfo(model);
-    return modelInfo?.languages || [];
+    return (await getModelInfo(model))?.languages || [];
 }
 
 // Returns true if a model is known to be compatible with a language
 // Returns false otherwise
-const isLanguageSupported = async (model, code) => {
-    const modelLanguages = await getModelLanguages(model);
-    return modelLanguages?.includes(code);
+const isLanguageSupported = async(model, code) => {
+    return (await getModelLanguages(model))?.includes(code);
 }
 
 module.exports = {

package/package.json

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