generaltranslation 1.1.21 → 1.1.23

package/README.md

@@ -102,23 +102,23 @@ const gt = new GT({
 });
 ```
 
-### async getPrompt(prompt, language)
+### async translatePrompt(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.
 
-Just wrap `getPrompt` around your prompt and go. 
+Just wrap `translatePrompt` around your prompt and go. 
 
 All of the following are valid:
 
 ```
-const translatedPrompt = await gt.getPrompt('Tell me a story', 'es');
+const translatedPrompt = await gt.translatePrompt('Tell me a story', 'es');
 ```
 
 ```
 const first = 'Tell me a story ';
 const second = 'about a cat'
 
-const translatedPrompt = await gt.getPrompt([
+const translatedPrompt = await gt.translatePrompt([
     first, second
 ], 'es');
 ```
@@ -129,7 +129,7 @@ To mark text that shouldn't be translated, wrap it in `{ text: "", translate: fa
 const prompt = 'Tell me a story about ';
 const input = 'gatos con espadas'
 
-const translatedPrompt = await gt.getPrompt([
+const translatedPrompt = await gt.translatePrompt([
     prompt, { text: input, translate: false }
 ], 'es');
 ```
@@ -140,7 +140,7 @@ For type consistency, you can also make everything in the prompt parameter an ob
 const prompt = 'Tell me a story about ';
 const input = 'gatos con espadas'
 
-const translatedPrompt = await gt.getPrompt([
+const translatedPrompt = await gt.translatePrompt([
     { text: prompt }, 
     { text: input, translate: false }
 ], 'es');
@@ -149,5 +149,5 @@ const translatedPrompt = await gt.getPrompt([
 This also works:
 
 ```
-const translatedPrompt = await gt.getPrompt({ text: 'Tell me a story' }, 'es');
+const translatedPrompt = await gt.translatePrompt({ text: 'Tell me a story' }, 'es');
 ```

package/index.js

@@ -5,7 +5,7 @@
 
 const { _getLanguageName, _getLanguageCode } = require('./codes/codes.js');
 const { _getModelList, _getModelInfo, _getAllModelInfo, _getModelLanguages, _isLanguageSupported, _getModelsByDeveloper, _getModelsByLanguage, } = require('./models/models.js');
-const { _translate } = require('./translate/translate.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
@@ -32,13 +34,11 @@ class GT {
     getModelsByLanguage = _getModelsByLanguage; // returns array of model names
     getModelsByDeveloper = _getModelsByDeveloper; // returns array of model names
 
-    // Prompt internationalization
-    getPrompt = async (prompt, language) => {
-        return await _translate(prompt, language, this.defaultLanguage, this.apiKey);
-    }
-    // Translation (same as prompt internationalization)
-    translate = async (content, language) => {
-        return await _translate(content, language, this.defaultLanguage, this.apiKey);
+    // Prompt I18N
+    translatePrompt = async (prompt, language) => {
+        return await _translatePrompt({
+            content: prompt, language: language, context: this
+        });
     }
 
 }

package/package.json

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