v0.0.9

MadcowD · Sep 23, 2024 · 5243a68 · 5243a68
1 parent fac77b2
commit 5243a68
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ell-ai"
-version = "0.0.7"
+version = "0.0.9"
 description = "ell - the language model programming library"
 authors = ["William Guss <[email protected]>"]
 license = "MIT"

diff --git a/src/ell/lmp/_track.py b/src/ell/lmp/_track.py
@@ -119,8 +119,8 @@ def tracked_func(*fn_args, _get_invocation_id=False, **fn_kwargs) -> str:
                 )
             latency_ms = (utc_now() - _start_time).total_seconds() * 1000
             usage = metadata.get("usage", {"prompt_tokens": 0, "completion_tokens": 0})
-            prompt_tokens= usage.get("prompt_tokens", 0)
-            completion_tokens= usage.get("completion_tokens", 0)
+            prompt_tokens= usage.get("prompt_tokens", 0) if usage else 0
+            completion_tokens= usage.get("completion_tokens", 0) if usage else 0
 
 
             #XXX: cattrs add invocation origin here recursively on all pirmitive types within a message.