reformat

langgenius · Nov 16, 2024 · 771b927 · 771b927
1 parent 4c16368
commit 771b927
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/api/core/model_runtime/model_providers/anthropic/llm/llm.py b/api/core/model_runtime/model_providers/anthropic/llm/llm.py
@@ -325,11 +325,13 @@ def _handle_chat_generate_response(
                 assistant_prompt_message.tool_calls.append(tool_call)
 
         # calculate num tokens
-        prompt_tokens = (response.usage and response.usage.input_tokens) or \
-            self.get_num_tokens(model, credentials, prompt_messages)
-
-        completion_tokens = (response.usage and response.usage.output_tokens) or \
-            self.get_num_tokens(model, credentials, [assistant_prompt_message])
+        prompt_tokens = (response.usage and response.usage.input_tokens) or self.get_num_tokens(
+            model, credentials, prompt_messages
+        )
+
+        completion_tokens = (response.usage and response.usage.output_tokens) or self.get_num_tokens(
+            model, credentials, [assistant_prompt_message]
+        )
 
         # transform usage
         usage = self._calc_response_usage(model, credentials, prompt_tokens, completion_tokens)