28594_usage_tracking

@@ -124,6 +124,9 @@ function App() {
         });
         userModifiedInput = "";
       }
+      if(data.usage) {
+        btutil_setChatUsage('chatbot+', data.usage.prompt_tokens, data.usage.total_tokens);
+      }
       chatLogTurboNew.push({ role: "user", content: userModifiedInput });
       chatLogTurboNew.push({ role: "assistant", content: parsedData });
       setChatLogTurbo(chatLogTurboNew);

@@ -127,17 +127,12 @@ app.post('/api', async (req, res) => {
 			usage.prompt_tokens = (enc.encode(query_prompt)).length;
 			usage.completion_tokens = (enc.encode(input)).length;
 			usage.total_tokens = usage.prompt_tokens + usage.completion_tokens;
-
-			// TOKEN USAGE
-			axios.post(`${process.env.API_URL}/e/set-chat-usage`,
-				{ aiwp_logged_in: req.cookies[user_secret_id], app: 'chatbot', prompt_token: usage.prompt_tokens, total_token: usage.total_tokens },
-				{ headers: { 'content-type': 'application/x-www-form-urlencoded' }
-			});
 		} catch (e) {
 			console.log('Error encoding prompt text', e);
 		}
 
 		res.json({
+			usage: usage,
 			message: anchorme({
 				input,
 				options: {
@@ -203,18 +198,13 @@ async function runGPTTurbo(req, res) {
 			usage.prompt_tokens = (enc.encode(query_prompt)).length;
 			usage.completion_tokens = (enc.encode(input)).length;
 			usage.total_tokens = usage.prompt_tokens + usage.completion_tokens;
-
-			// TOKEN USAGE
-			axios.post(`${process.env.API_URL}/e/set-chat-usage`,
-				{ aiwp_logged_in: req.cookies[user_secret_id], app: 'chatbot', prompt_token: usage.prompt_tokens, total_token: usage.total_tokens },
-				{ headers: { 'content-type': 'application/x-www-form-urlencoded' }
-			});
 		} catch (e) {
 			console.log('Error encoding prompt text', e);
 		}
 
 		res.json({
 			prompt: JSON.parse(message),
+			usage: usage,
 			message: anchorme({
 				input,
 				options: {