Add Total Tokens to anthropic

openlit · Mar 10, 2024 · 32c1639 · 32c1639
1 parent e6faac0
commit 32c1639
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/dokumetry/anthropic.py b/src/dokumetry/anthropic.py
@@ -133,6 +133,7 @@ def stream_generator():
                     "finishReason": response.stop_reason,
                     "response": response.content[0].text
             }
+            data["totalTokens"] = data["completionTokens"] + data["promptTokens"]
 
             send_data(data, doku_url, api_key)
 

diff --git a/src/dokumetry/async_anthropic.py b/src/dokumetry/async_anthropic.py
@@ -132,6 +132,7 @@ async def stream_generator():
                     "finishReason": response.stop_reason,
                     "response": response.content[0].text
             }
+            data["totalTokens"] = data["completionTokens"] + data["promptTokens"]
 
             send_data(data, doku_url, api_key)