@cobrowser/chatgpt 0.7.22 → 0.7.23-beta.2

package/dist/services/BaseService/BaseService.d.ts

@@ -30,5 +30,11 @@ declare class BaseService {
      * @param error
      */
     handleErrors(error: AxiosError): void;
+    /**
+     * Sanitize ChatGPT response with our XSS validation package
+     *
+     * @param data
+     */
+    output(data: string | undefined): string | undefined;
 }
 export default BaseService;

package/dist/services/BaseService/BaseService.js

@@ -5,6 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 const axios_1 = __importDefault(require("axios"));
 const logger_1 = __importDefault(require("../logger"));
+const xss_validation_1 = require("@cobrowser/xss-validation");
 class BaseService {
     /**
      *
@@ -99,5 +100,23 @@ class BaseService {
         }
         logger_1.default.error(`error message ${error.message}`);
     }
+    /**
+     * Sanitize ChatGPT response with our XSS validation package
+     *
+     * @param data
+     */
+    output(data) {
+        if (!data) {
+            return;
+        }
+        const parsed = JSON.parse(data), translation = parsed.data[0].translation, sanitized = xss_validation_1.XSSProtector.sanitize(translation), encoded = xss_validation_1.XSSProtector.encode(sanitized);
+        parsed.data[0].translation = encoded;
+        console.log('data: ', data);
+        console.log('parsed: ', parsed);
+        console.log('sanitized: ', sanitized);
+        console.log('encoded: ', encoded);
+        logger_1.default.info({ data, parsed, sanitized, encoded }, ':: Sanitizing Response ::');
+        return JSON.stringify(parsed);
+    }
 }
 exports.default = BaseService;

package/dist/services/TranslationService/TranslationService.js

@@ -52,9 +52,10 @@ class TranslationService extends BaseService_1.default {
                 ],
             };
             try {
+                logger_1.default.info(`:: TranslationServicee.translate :: Translating...`);
                 const response = yield this.request.post('', requestBody);
                 return {
-                    firstChoice: this.getResponseFirstChoice(response),
+                    firstChoice: this.output(this.getResponseFirstChoice(response)),
                     usageTokens: this.getResponseUsageTokens(response),
                 };
             }
@@ -90,6 +91,7 @@ class TranslationService extends BaseService_1.default {
                 ],
             };
             try {
+                logger_1.default.info(`:: TranslationServicee.findTextLanguage :: Translating...`);
                 const response = yield this.request.post('', requestBody);
                 return {
                     firstChoice: this.getResponseFirstChoice(response),
@@ -128,6 +130,7 @@ class TranslationService extends BaseService_1.default {
                 ],
             };
             try {
+                logger_1.default.info(`:: TranslationServicee.findTextLanguageBasedOnFrequency :: Translating...`);
                 const response = yield this.request.post('', requestBody);
                 return {
                     firstChoice: this.getResponseFirstChoice(response),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cobrowser/chatgpt",
-  "version": "0.7.22",
+  "version": "0.7.23-beta.2",
   "description": "chatgpt services to connect our projects with chatgpt api",
   "keywords": [
     "chatgpt",
@@ -22,6 +22,7 @@
   },
   "dependencies": {
     "@cobrowser/logger": "^2.0.2",
+    "@cobrowser/xss-validation": "^2.0.0",
     "axios": "^1.6.1",
     "axios-mock-adapter": "^1.22.0",
     "openai": "^4.86.2"
@@ -39,5 +40,5 @@
   "bugs": {
     "url": "https://bitbucket.org/cobrowser/cb_utils/issues"
   },
-  "gitHead": "28e27011208c1da21cc0de3aa4cb35246bbd8e05"
+  "gitHead": "15eb4c0a791004426ce3232ffb862fc1bd3de443"
 }