@cobrowser/chatgpt 0.5.4 → 0.5.6

package/dist/services/TranslationService/TranslationService.d.ts

@@ -8,5 +8,6 @@ export declare class TranslationService extends BaseService {
      * @param toLanguage the desired language to translate to ( English is the default )
      * @param fromLanguage if we want to specify the language of the text, we can use this ( optional)
      */
-    translate(text: string, toLanguage?: string, fromLanguage?: string): Promise<string | undefined>;
+    translate(text: string, toLanguage?: string, fromLanguage?: string): Promise<any | undefined>;
+    findTextLanguage(text: string): Promise<any | undefined>;
 }

package/dist/services/TranslationService/TranslationService.js

@@ -51,7 +51,39 @@ class TranslationService extends BaseService_1.default {
             };
             try {
                 const response = yield this.request.post('', requestBody);
-                return this.getResponseFirstChoice(response);
+                return {
+                    firstChoice: this.getResponseFirstChoice(response),
+                    usageTokens: this.getResponseUsageTokens(response),
+                };
+            }
+            catch (e) {
+                this.handleErrors(e);
+                return Promise.reject(e);
+            }
+        });
+    }
+    findTextLanguage(text) {
+        return __awaiter(this, void 0, void 0, function* () {
+            if (!text) {
+                logger_1.default.error('translation text should be provided');
+                return Promise.reject(new Error('text must be provided'));
+            }
+            // instruction on what to do with the text
+            const translateText = `In which language is this text in '${text}', return language code in json`, 
+            // the body we need to send to the request
+            requestBody = {
+                model: this.chatGptModel,
+                messages: [{
+                        role: ChatGptMessage_1.ChatGptRole.USER,
+                        content: translateText,
+                    }],
+            };
+            try {
+                const response = yield this.request.post('', requestBody);
+                return {
+                    firstChoice: this.getResponseFirstChoice(response),
+                    usageTokens: this.getResponseUsageTokens(response),
+                };
             }
             catch (e) {
                 this.handleErrors(e);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cobrowser/chatgpt",
-  "version": "0.5.4",
+  "version": "0.5.6",
   "description": "chatgpt services to connect our projects with chatgpt api",
   "keywords": [
     "chatgpt",
@@ -38,5 +38,5 @@
   "bugs": {
     "url": "https://bitbucket.org/cobrowser/cb_utils/issues"
   },
-  "gitHead": "e1faace27b50c02cb93e2d4c124ba2a36cdf589a"
+  "gitHead": "bb937a81d830e69871f1741ef3c4620204978f32"
 }