From db6fdebdcd5930868c9070a95cebd961c30b1241 Mon Sep 17 00:00:00 2001
From: Ralph Khreish <35776126+Crunchyman-ralph@users.noreply.github.com>
Date: Mon, 19 May 2025 20:49:16 +0200
Subject: [PATCH] fix: improve ollama object to telemetry structure

---
 src/ai-providers/ollama.js | 18 +++++++++++++++---
 1 file changed, 15 insertions(+), 3 deletions(-)

diff --git a/src/ai-providers/ollama.js b/src/ai-providers/ollama.js
index 28779c48..e9527233 100644
--- a/src/ai-providers/ollama.js
+++ b/src/ai-providers/ollama.js
@@ -3,7 +3,7 @@
  * AI provider implementation for Ollama models using the ollama-ai-provider package.
  */
 
-import { createOllama, ollama } from 'ollama-ai-provider';
+import { createOllama } from 'ollama-ai-provider';
 import { log } from '../../scripts/modules/utils.js'; // Import logging utility
 import { generateObject, generateText, streamText } from 'ai';
 
@@ -48,7 +48,13 @@ async function generateOllamaText({
 			temperature
 		});
 		log('debug', `Ollama generated text: ${result.text}`);
-		return result.text;
+		return {
+			text: result.text,
+			usage: {
+				inputTokens: result.usage.promptTokens,
+				outputTokens: result.usage.completionTokens
+			}
+		};
 	} catch (error) {
 		log(
 			'error',
@@ -138,7 +144,13 @@ async function generateOllamaObject({
 			temperature: temperature,
 			maxRetries: maxRetries
 		});
-		return result.object;
+		return {
+			object: result.object,
+			usage: {
+				inputTokens: result.usage.promptTokens,
+				outputTokens: result.usage.completionTokens
+			}
+		};
 	} catch (error) {
 		log(
 			'error',