fix bedrock

MadcowD · Oct 3, 2024 · a6100a7 · a6100a7
1 parent 9cf8635
commit a6100a7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/ell/providers/bedrock.py b/src/ell/providers/bedrock.py
@@ -13,8 +13,10 @@
 from PIL import Image as PILImage
 
 try:
-    from botocore.client import BaseClient
+
     from botocore.eventstream import (EventStream)
+    from botocore.client import BaseClient
+
     class BedrockProvider(Provider):
         dangerous_disable_validation = True
 
@@ -153,6 +155,7 @@ def translate_from_provider(
 
             # process metadata for ell
             # XXX: Unify an ell metadata format for ell studio.
+            usage = {} if not usage else usage
             usage["prompt_tokens"] = usage.get("inputTokens", 0)
             usage["completion_tokens"] = usage.get("outputTokens", 0)
             usage["total_tokens"] = usage['prompt_tokens'] + usage['completion_tokens']