@cobrowser/chatgpt 0.7.21 → 0.7.23-beta.1

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 message
+     */
+    output(message: 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,16 @@ class BaseService {
         }
         logger_1.default.error(`error message ${error.message}`);
     }
+    /**
+     * Sanitize ChatGPT response with our XSS validation package
+     *
+     * @param message
+     */
+    output(message) {
+        if (!message) {
+            return;
+        }
+        return xss_validation_1.XSSProtector.sanitize(message);
+    }
 }
 exports.default = BaseService;

package/dist/services/CopilotService/CopilotService.js

@@ -67,7 +67,8 @@ class CopilotService extends BaseService_1.default {
                 }
                 // main prompt used to instruct chatGPT to suggest some quick replies
                 const mainPrompt = `Propose some quick replies for the agent to be used on the following conversation
-          display your response as unordered list:-\n${conversation}`, 
+          Your response must only contains a valid JSON array of strings like this:
+          ["Reply 1", "Reply 2", "Reply 3"]. the conversation is:-\n${conversation}`, 
                 // main prompt will be the whole conversation as string
                 mainPromptMessage = {
                     role: ChatGptMessage_1.ChatGptRole.USER,
@@ -155,9 +156,9 @@ class CopilotService extends BaseService_1.default {
                 const messages = data.flatMap((message) => message === null || message === void 0 ? void 0 : message.content.map((content) => ((content.type === 'text' && message.role === 'assistant') ? content.text : '')));
                 const assistantMessages = messages.filter(message => typeof message === 'object' && 'value' in message && typeof message.value === 'string');
                 if (assistantMessages.length) {
-                    let answers = assistantMessages.map(message => message.value).join('\n');
+                    let answers = assistantMessages.map(message => message.value);
                     return {
-                        firstChoice: answers,
+                        firstChoice: JSON.stringify(answers),
                         threadId: run.thread_id,
                         usageTokens: run.usage,
                     };

package/dist/services/TranslationService/TranslationService.js

@@ -54,7 +54,7 @@ class TranslationService extends BaseService_1.default {
             try {
                 const response = yield this.request.post('', requestBody);
                 return {
-                    firstChoice: this.getResponseFirstChoice(response),
+                    firstChoice: this.output(this.getResponseFirstChoice(response)),
                     usageTokens: this.getResponseUsageTokens(response),
                 };
             }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cobrowser/chatgpt",
-  "version": "0.7.21",
+  "version": "0.7.23-beta.1",
   "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": "^1.0.8",
     "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": "3a426ebc382c861be00fdef6455b611075436ece"
+  "gitHead": "7d9d8c6fe2a6beab099ee60a685d38599f443d25"
 }