backend-manager 3.2.165 → 3.2.166

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "backend-manager",
-  "version": "3.2.165",
+  "version": "3.2.166",
   "description": "Quick tools for developing Firebase functions",
   "main": "src/manager/index.js",
   "bin": {

package/src/manager/libraries/openai.js

@@ -263,7 +263,11 @@ OpenAI.prototype.request = function (options) {
 
         if (mode === 'chatgpt') {
           request.url = 'https://api.openai.com/v1/chat/completions';
+
+          // Get history
           const history = options.history.messages.slice(-options.history.limit);
+
+          // Add prompt to history
           history.unshift({
             role: 'system',
             text: prompt,
@@ -344,15 +348,16 @@ OpenAI.prototype.request = function (options) {
         await fetch(request.url, request)
         .then(async (r) => {
           // Set token counts
-          self.tokens.total.count += r?.usage?.total_tokens || 0;
           self.tokens.input.count += r?.usage?.prompt_tokens || 0;
           self.tokens.output.count += r?.usage?.completion_tokens || 0;
+          self.tokens.total.count = self.tokens.input.count + self.tokens.output.count;
 
           // Set token prices
-          self.tokens.total.price = (self.tokens.total.count / 1000) * TOKEN_COST_TABLE[options.model].input;
           self.tokens.input.price = (self.tokens.input.count / 1000) * TOKEN_COST_TABLE[options.model].input;
           self.tokens.output.price = (self.tokens.output.count / 1000) * TOKEN_COST_TABLE[options.model].output;
+          self.tokens.total.price = self.tokens.input.price + self.tokens.output.price;
 
+          // Return
           return resolve(_.get(r, resultPath));
         })
         .catch((e) => {