@cobrowser/chatgpt 0.7.4 → 0.7.5

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

@@ -3,6 +3,7 @@ import ChatGptUsageTokens from '../../models/ChatGptUsageTokens';
 declare class BaseService {
     request: AxiosInstance;
     readonly BASE_CHAT_COMPLETION_URL = "https://api.openai.com/v1/chat/completions";
+    openaiApiKey: string | undefined;
     chatGptModel: string;
     maxNumberOfChatCompletionResult: number;
     /**

package/dist/services/BaseService/BaseService.js

@@ -27,6 +27,12 @@ class BaseService {
             writable: true,
             value: 'https://api.openai.com/v1/chat/completions'
         });
+        Object.defineProperty(this, "openaiApiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         Object.defineProperty(this, "chatGptModel", {
             enumerable: true,
             configurable: true,
@@ -46,6 +52,7 @@ class BaseService {
                 Authorization: `Bearer ${apiKey} `,
             },
         });
+        this.openaiApiKey = apiKey;
         this.chatGptModel = model;
         this.maxNumberOfChatCompletionResult = maxNumberOfChoices;
     }

package/dist/services/CopilotService/CopilotService.d.ts

@@ -3,7 +3,8 @@ import BaseService from '../BaseService/BaseService';
 import ChatGptMessage from '../../models/ChatGptMessage';
 import Item from '../../models/Item';
 export declare class CopilotService extends BaseService {
-    constructor(apiKey?: string, model?: string, apiUrl?: string, maxNumberOfChoices?: number);
+    assistantId: string | undefined;
+    constructor(apiKey?: string, model?: string, assistantId?: string, apiUrl?: string, maxNumberOfChoices?: number);
     /**
      * a method to be used to suggest some quick replies for the agent based on the conversation
      *  will be added as a single string in the format
@@ -23,4 +24,10 @@ export declare class CopilotService extends BaseService {
      * @param items the items to recommend from
      */
     recommend(conversation: ChatGptMessage[], items: Item[]): Promise<string | undefined>;
+    /**
+     * Get the reply of the assistant if the assistant_id is available
+     *
+     * @param message
+     */
+    getAssistantReply(message: string): Promise<string | undefined>;
 }

package/dist/services/CopilotService/CopilotService.js

@@ -16,9 +16,17 @@ exports.CopilotService = void 0;
 const BaseService_1 = __importDefault(require("../BaseService/BaseService"));
 const logger_1 = __importDefault(require("../logger"));
 const ChatGptMessage_1 = require("../../models/ChatGptMessage");
+const openai_1 = __importDefault(require("openai"));
 class CopilotService extends BaseService_1.default {
-    constructor(apiKey, model, apiUrl, maxNumberOfChoices) {
+    constructor(apiKey, model, assistantId, apiUrl, maxNumberOfChoices) {
         super(apiKey, model, apiUrl, maxNumberOfChoices);
+        Object.defineProperty(this, "assistantId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.assistantId = assistantId;
     }
     /**
      * a method to be used to suggest some quick replies for the agent based on the conversation
@@ -39,6 +47,20 @@ class CopilotService extends BaseService_1.default {
                 return Promise.reject(new Error('conversation must be provided'));
             }
             try {
+                if (this.assistantId) {
+                    // Here we get the last customer message as we dont need the whole conversation
+                    // as we are doing with the other approach
+                    const conversationArray = conversation.split('\n');
+                    const lastCustomerMessage = conversationArray.reverse().find(e => e.includes('customer:'));
+                    if (!lastCustomerMessage) {
+                        logger_1.default.error('last customer message not found');
+                        return Promise.reject(new Error('last customer message not found'));
+                    }
+                    const assistantReply = yield this.getAssistantReply(lastCustomerMessage);
+                    return {
+                        firstChoice: assistantReply
+                    };
+                }
                 // 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}`, 
@@ -95,5 +117,33 @@ class CopilotService extends BaseService_1.default {
             }
         });
     }
+    /**
+     * Get the reply of the assistant if the assistant_id is available
+     *
+     * @param message
+     */
+    getAssistantReply(message) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            const openai = new openai_1.default({ apiKey: this.openaiApiKey });
+            let reply = undefined;
+            const thread = yield openai.beta.threads.create();
+            yield openai.beta.threads.messages.create(thread.id, { role: "user", content: message });
+            let run = yield openai.beta.threads.runs.create(thread.id, { assistant_id: (_a = this.assistantId) !== null && _a !== void 0 ? _a : '' });
+            while (['queued', 'in_progress', 'cancelling'].includes(run.status)) {
+                yield new Promise(resolve => setTimeout(resolve, 500));
+                run = yield openai.beta.threads.runs.retrieve(run.thread_id, run.id);
+            }
+            if (run.status === 'completed') {
+                const { data } = yield openai.beta.threads.messages.list(run.thread_id);
+                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 assistantMessage = messages.find(message => typeof message === 'object' && 'value' in message && typeof message.value === 'string');
+                if (assistantMessage) {
+                    reply = assistantMessage.value;
+                }
+            }
+            return reply;
+        });
+    }
 }
 exports.CopilotService = CopilotService;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cobrowser/chatgpt",
-  "version": "0.7.4",
+  "version": "0.7.5",
   "description": "chatgpt services to connect our projects with chatgpt api",
   "keywords": [
     "chatgpt",
@@ -23,7 +23,8 @@
   "dependencies": {
     "@cobrowser/logger": "^1.0.0",
     "axios": "^1.6.1",
-    "axios-mock-adapter": "^1.22.0"
+    "axios-mock-adapter": "^1.22.0",
+    "openai": "^4.32.0"
   },
   "directories": {
     "dist": "dist"
@@ -38,5 +39,5 @@
   "bugs": {
     "url": "https://bitbucket.org/cobrowser/cb_utils/issues"
   },
-  "gitHead": "e71a58b5be231da26741a5ded6e51db3b86aa419"
+  "gitHead": "5a4bff51323eed4e0c17a865e7783d8dfc0323e1"
 }