From 33386cdce80958286728d4d6d1a1d8757f0c717e Mon Sep 17 00:00:00 2001
From: Ralph Khreish <35776126+Crunchyman-ralph@users.noreply.github.com>
Date: Fri, 3 Oct 2025 16:14:08 +0200
Subject: [PATCH] chore: apply requested changes

---
 src/ai-providers/base-provider.js | 26 ++++++++++++++++++++------
 1 file changed, 20 insertions(+), 6 deletions(-)

diff --git a/src/ai-providers/base-provider.js b/src/ai-providers/base-provider.js
index a81843e1..d9543400 100644
--- a/src/ai-providers/base-provider.js
+++ b/src/ai-providers/base-provider.js
@@ -176,12 +176,19 @@ export class BaseAIProvider {
 				`${this.name} generateText completed successfully for model: ${params.modelId}`
 			);
 
+			const inputTokens =
+				result.usage?.inputTokens ?? result.usage?.promptTokens ?? 0;
+			const outputTokens =
+				result.usage?.outputTokens ?? result.usage?.completionTokens ?? 0;
+			const totalTokens =
+				result.usage?.totalTokens ?? inputTokens + outputTokens;
+
 			return {
 				text: result.text,
 				usage: {
-					inputTokens: result.usage?.inputTokens || 0,
-					outputTokens: result.usage?.outputTokens || 0,
-					totalTokens: result.usage?.totalTokens || 0
+					inputTokens,
+					outputTokens,
+					totalTokens
 				}
 			};
 		} catch (error) {
@@ -296,12 +303,19 @@ export class BaseAIProvider {
 				`${this.name} generateObject completed successfully for model: ${params.modelId}`
 			);
 
+			const inputTokens =
+				result.usage?.inputTokens ?? result.usage?.promptTokens ?? 0;
+			const outputTokens =
+				result.usage?.outputTokens ?? result.usage?.completionTokens ?? 0;
+			const totalTokens =
+				result.usage?.totalTokens ?? inputTokens + outputTokens;
+
 			return {
 				object: result.object,
 				usage: {
-					inputTokens: result.usage?.inputTokens || 0,
-					outputTokens: result.usage?.outputTokens || 0,
-					totalTokens: result.usage?.totalTokens || 0
+					inputTokens,
+					outputTokens,
+					totalTokens
 				}
 			};
 		} catch (error) {