Change to keep original usage metadata

Don't overwrite usage, change to keep original usage metadata.
MadcowD · Jan 17, 2025 · 8ecee8b · 8ecee8b
1 parent 3267c06
commit 8ecee8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ell/providers/anthropic.py b/src/ell/providers/anthropic.py
@@ -150,7 +150,7 @@ def translate_from_provider(
             usage["completion_tokens"] = usage.get("output_tokens", 0)
             usage["total_tokens"] = usage['prompt_tokens'] + usage['completion_tokens']
 
-            metadata["usage"] = usage
+            metadata["usage"] = {**usage, **metadata.get("usage", {})}
             return tracked_results, metadata
 
     # XXX: Make a singleton.