@cobrowser/chatgpt 0.5.3 → 0.5.4

package/dist/models/ChatGptUsageTokens.d.ts

@@ -0,0 +1,6 @@
+interface ChatGptUsageTokens {
+    prompt_tokens: Number;
+    completion_tokens: Number;
+    total_tokens: Number;
+}
+export default ChatGptUsageTokens;

package/dist/models/ChatGptUsageTokens.js

@@ -0,0 +1,3 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+;

package/dist/models/CopilotSuggestion.d.ts

@@ -0,0 +1,6 @@
+import ChatGptUsageTokens from "./ChatGptUsageTokens";
+interface CopilotSuggestion {
+    firstChoice?: string;
+    usageTokens?: ChatGptUsageTokens;
+}
+export default CopilotSuggestion;

package/dist/models/CopilotSuggestion.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

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

@@ -1,4 +1,5 @@
 import { AxiosError, AxiosInstance, AxiosResponse } from 'axios';
+import ChatGptUsageTokens from '../../models/ChatGptUsageTokens';
 declare class BaseService {
     request: AxiosInstance;
     readonly BASE_CHAT_COMPLETION_URL = "https://api.openai.com/v1/chat/completions";
@@ -16,6 +17,11 @@ declare class BaseService {
      * @param response
      */
     getResponseFirstChoice(response: AxiosResponse): string | undefined;
+    /**
+     * Filter the response from chatGpt API and only get the first choice
+     * @param response
+     */
+    getResponseUsageTokens(response: AxiosResponse): ChatGptUsageTokens | undefined;
     /**
      * function to handle requests errors
      * @param error

package/dist/services/BaseService/BaseService.js

@@ -60,6 +60,19 @@ class BaseService {
         }
         return undefined;
     }
+    /**
+     * Filter the response from chatGpt API and only get the first choice
+     * @param response
+     */
+    getResponseUsageTokens(response) {
+        var _a;
+        if ((_a = response.data) === null || _a === void 0 ? void 0 : _a.usage) {
+            const usage = response.data.usage;
+            logger_1.default.info(`ChatGpt response usage tokens: ${JSON.stringify(usage)}`);
+            return usage;
+        }
+        return undefined;
+    }
     /**
      * function to handle requests errors
      * @param error

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

@@ -1,5 +1,6 @@
 import BaseService from '../BaseService/BaseService';
 import ChatGptMessage from '../../models/ChatGptMessage';
+import CopilotSuggestion from '../../models/CopilotSuggestion';
 import Item from '../../models/Item';
 export declare class CopilotService extends BaseService {
     constructor(apiUrl?: string, model?: string, maxNumberOfChoices?: number);
@@ -15,7 +16,7 @@ export declare class CopilotService extends BaseService {
      * this will make it easy to ask chatGpt for quick replies
      * @param conversation
      */
-    suggest(conversation: string): Promise<string | undefined>;
+    suggest(conversation: string): Promise<CopilotSuggestion | undefined>;
     /**
      * this will request chatGPT to recommend product or item based on the whole conversation
      * @param conversation the whole chat conversation

package/dist/services/CopilotService/CopilotService.js

@@ -53,7 +53,10 @@ class CopilotService extends BaseService_1.default {
                     messages: [mainPromptMessage],
                 };
                 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);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cobrowser/chatgpt",
-  "version": "0.5.3",
+  "version": "0.5.4",
   "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": "e68ebdaa78a548e653392a939a508d3e55e47494"
+  "gitHead": "e1faace27b50c02cb93e2d4c124ba2a36cdf589a"
 }