feat: token usage

timkmecl · Apr 11, 2023 · 802a13a · 802a13a
1 parent 5984975
commit 802a13a
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 4 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -2,7 +2,7 @@
   "name": "chatgpt",
   "displayName": "ChatGPT: write and improve code using AI",
   "description": "Use ChatGPT and GPT4 right inside the IDE to enhance and automate your coding with AI-powered assistance (unofficial)",
-  "version": "1.1.1",
+  "version": "1.1.2",
   "publisher": "timkmecl",
   "icon": "resources/extensionIcon.png",
   "license": "MIT",

diff --git a/src/extension.ts b/src/extension.ts
@@ -297,9 +297,13 @@ class ChatGPTViewProvider implements vscode.WebviewViewProvider {
 				}
 
 
-				console.log(response);
+				console.log(res);
 
 				response = res.text;
+				if (res.detail?.usage?.total_tokens) {
+					response += `\n\n---\n*<sub>Tokens used: ${res.detail.usage.total_tokens} (${res.detail.usage.prompt_tokens}+${res.detail.usage.completion_tokens})</sub>*`;
+				}
+
 				if (this._settings.keepConversation){
 					this._conversation = {
 						parentMessageId: res.id

diff --git a/yarn.lock b/yarn.lock
@@ -1141,6 +1141,11 @@
   "resolved" "https://registry.npmjs.org/fs.realpath/-/fs.realpath-1.0.0.tgz"
   "version" "1.0.0"
 
+"fsevents@~2.3.2":
+  "integrity" "sha512-xiqMQR4xAeHTuB9uWm+fFRcIOgKBMiOBP+eXiyT7jsgVCq1bkVygt00oASowB7EdtpOHaaPgKt812P9ab+DDKA=="
+  "resolved" "https://registry.npmjs.org/fsevents/-/fsevents-2.3.2.tgz"
+  "version" "2.3.2"
+
 "function-bind@^1.1.1":
   "integrity" "sha512-yIovAzMX49sF8Yl58fSCWJ5svSLuaibPxXQJFLmBObTuCr0Mf1KiPopGM9NiFjiYBCbfaa2Fh6breQ6ANVTI0A=="
   "resolved" "https://registry.npmjs.org/function-bind/-/function-bind-1.1.1.tgz"