reverting

deepset-ai · Aug 19, 2024 · 70191ff · 70191ff
1 parent 91297c6
commit 70191ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...ock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py b/...ock/src/haystack_integrations/components/generators/amazon_bedrock/chat/chat_generator.py
@@ -215,8 +215,8 @@ def run(
                     if "output_tokens" in response.meta["usage"]:
                         response.meta["usage"]["completion_tokens"] = response.meta["usage"].pop("output_tokens")
                 else:
-                    response.meta["usage"] = {}
                     if "prompt_token_count" in response.meta:
+                        response.meta["usage"] = {}
                         response.meta["usage"]["prompt_tokens"] = response.meta.pop("prompt_token_count")
                     if "generation_token_count" in response.meta:
                         response.meta["usage"]["completion_tokens"] = response.meta.pop("generation_token_count")