add total tokens to cohere client embed

openlit · Mar 17, 2024 · 89888b4 · 89888b4
1 parent bbee8a3
commit 89888b4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/cohere.js b/src/cohere.js
@@ -92,6 +92,7 @@ export default function initCohere({ llm, dokuUrl, apiKey, environment, applicat
       model: model,
       prompt: prompt,
       promptTokens: response.meta["billedUnits"]["inputTokens"],
+      totalTokens: response.meta["billedUnits"]["inputTokens"],
     };
 
     await sendData(data, dokuUrl, apiKey);