generaltranslation 1.1.10 → 1.1.12

package/README.md

@@ -8,13 +8,13 @@ Note: this package is in active development.
 
 ## Getting Started
 
-In your terminal:
+<b>In your terminal:</b>
 
 ```
 npm i generaltranslation
 ```
 
-In your code:
+<b>In your code:</b>
 
 ```
 import GT from 'generaltranslation'
@@ -23,6 +23,12 @@ import GT from 'generaltranslation'
 const gt = new GT()
 ```
 
+or, import functions directly:
+
+```
+import { getLanguageName } from 'generaltranslation'
+```
+
 ## Convert between languages and ISO-639 codes
 
 ### getLanguageName(codes)
@@ -49,7 +55,7 @@ const codes = gt.getLanguageCodes(['French', 'Spanish'])
 console.log(codes) // ['fr', 'es']
 ```
 
-## Which languages do AI models speak?
+## Which languages do AI models understand?
 
 We continually benchmark AI models and add new models as they are released. That means these functions have to be <code>async</code>. This information is provided as a public service. It's completely free and requires no API key.
 
@@ -68,7 +74,7 @@ main();
 
 ### async getModelLanguages(model)
 
-Get all languages known to be compatible with a given AI model. Returns an array of languages codes, or [] if the model is unknown.
+Get all languages known to be compatible with a given AI model. Returns an array of languages codes, or null if the model is unknown.
 
 ```
 async function main() {
@@ -79,7 +85,7 @@ async function main() {
 main();
 ```
 
-### async isSupportedLanguage(model, language)
+### async isSupportedLanguage(model, code)
 
 Returns true if a model is known to be compatible with a given language, represented by an ISO-639 language code. Returns false otherwise.
 
@@ -94,4 +100,62 @@ main();
 
 ## API
 
-Coming soon!
+For these functions, you need to sign up for an API key at <a href='https://generaltranslation.com' target='_blank'>generaltranslation.com</a>.
+
+Add the API key to your code like this:
+
+```
+import GT from 'generaltranslation'
+
+const gt = new GT({
+	apiKey: process.env.GT_API_KEY // looks like 'gtx-XXX'
+});
+```
+
+### async gt.getPrompt(prompt, code)
+
+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.
+
+Just wrap `gt.getPrompt` around your prompt and go. All of the following are valid:
+
+```
+const translatedPrompt = await gt.getPrompt('Tell me a story', 'es');
+```
+
+```
+const first = 'Tell me a story ';
+const second = 'about a cat'
+
+const translatedPrompt = await gt.getPrompt([
+    first, second
+], 'es');
+```
+
+To mark text that shouldn't be translated, wrap it in `{ text: "", translate: false }`. Items marked as `translate: false` are never sent to our API. For example:
+
+```
+const prompt = 'Tell me a story about ';
+const input = 'gatos con espadas'
+
+const translatedPrompt = await gt.getPrompt([
+    prompt, { text: input, translate: false }
+], 'es');
+```
+
+For type consistency, you can also make everything in the prompt parameter an object:
+
+```
+const prompt = 'Tell me a story about ';
+const input = 'gatos con espadas'
+
+const translatedPrompt = await gt.getPrompt([
+    { text: prompt }, 
+    { text: input, translate: false }
+], 'es');
+```
+
+This also works:
+
+```
+const translatedPrompt = await gt.getPrompt({ text: 'Tell me a story' }, 'es');
+```

package/index.js

@@ -29,8 +29,8 @@ class GT {
     isLanguageSupported = _isLanguageSupported; // e.g. ('mistral-7b', 'en') => true
 
     // Prompt internationalization
-    getPrompt = async (prompt, language) => {
-        return await _getPrompt(prompt, language, this.defaultLanguage, this.apiKey);
+    getPrompt = async (prompt, code) => {
+        return await _getPrompt(prompt, code, this.defaultLanguage, this.apiKey);
     }
 
 }

package/models/models.js

@@ -29,10 +29,10 @@ const _getModelInfo = async model => {
 }
 
 // Get all languages known to be compatible with a given LLM
-// Returns an array of languages codes, or [] if unknown
+// Returns an array of languages codes, or null if unknown
 const _getModelLanguages = async model => {
     const modelInfo = await _getModelInfo(model);
-    return modelInfo?.languages || [];
+    return modelInfo?.languages;
 }
 
 // Returns true if a model is known to be compatible with a language

package/package.json

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

package/prompts/prompts.js

@@ -64,32 +64,37 @@ const _constructPrompt = ({translated, redacted = null}) => {
 
 // Get a translated prompt via General Translation API
 // Returns prompt string
-const _getPrompt = async (prompt, language, defaultLanguage, apiKey) => {
-    if (!apiKey) {
-        throw new Error('Missing API Key!')
-    }
-    if (language === defaultLanguage) {
-        return _constructPrompt(prompt);
-    }
-    const { processed, redacted } = _processPrompt(prompt);
-    const response = await fetch('http://prompts.gtx.dev/internationalize', {
-        method: 'POST',
-        headers: {
-            'Content-Type': 'application/json',
-            'gtx-api-key': apiKey
-        },
-        body: JSON.stringify({
-            prompt: processed,
-            language: language,
-            defaultLanguage: defaultLanguage
+const _getPrompt = async (prompt, code, defaultLanguage, apiKey) => {
+    try {
+        if (!apiKey) {
+            throw new Error('Missing API Key!')
+        }
+        if (code === defaultLanguage) {
+            return _constructPrompt(prompt);
+        }
+        const { processed, redacted } = _processPrompt(prompt);
+        const response = await fetch('http://prompts.gtx.dev/internationalize', {
+            method: 'POST',
+            headers: {
+                'Content-Type': 'application/json',
+                'gtx-api-key': apiKey
+            },
+            body: JSON.stringify({
+                prompt: processed,
+                language: code,
+                defaultLanguage: defaultLanguage
+            })
         })
-    })
-    if (!response.ok) {
-        const result = await response.text();
-        throw new Error(`${result || response.status}`);
-    } else {
-        const result = await response.json();
-        return _constructPrompt({translated: result, redacted: redacted});
+        if (!response.ok) {
+            const result = await response.text();
+            throw new Error(`${result || response.status}`);
+        } else {
+            const result = await response.json();
+            return _constructPrompt({translated: result, redacted: redacted});
+        }
+    } catch (error) {
+        console.error(error)
+        return _constructPrompt({ translated: prompt })
     }
 }