Skip to content

Commit 42d4cea

Browse files
rishsrivClaude
and
Claude
committed
style: apply black formatting
🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <[email protected]>
1 parent 65e536e commit 42d4cea

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

Diff for: defog/llm/utils.py

+10-4
Original file line numberDiff line numberDiff line change
@@ -754,14 +754,17 @@ async def _process_openai_response(
754754
consecutive_exceptions = 0
755755
while True:
756756
# Check if prompt_tokens_details exists in the response
757-
if hasattr(response.usage, 'prompt_tokens_details') and response.usage.prompt_tokens_details is not None:
757+
if (
758+
hasattr(response.usage, "prompt_tokens_details")
759+
and response.usage.prompt_tokens_details is not None
760+
):
758761
cached_tokens = response.usage.prompt_tokens_details.cached_tokens
759762
total_input_tokens += response.usage.prompt_tokens - cached_tokens
760763
total_cached_input_tokens += cached_tokens
761764
else:
762765
# If prompt_tokens_details doesn't exist, assume all tokens are uncached
763766
total_input_tokens += response.usage.prompt_tokens
764-
767+
765768
total_output_tokens += response.usage.completion_tokens
766769
message = response.choices[0].message
767770
if message.tool_calls:
@@ -875,14 +878,17 @@ async def _process_openai_response(
875878

876879
usage = response.usage
877880
# Check if prompt_tokens_details exists in the response
878-
if hasattr(usage, 'prompt_tokens_details') and usage.prompt_tokens_details is not None:
881+
if (
882+
hasattr(usage, "prompt_tokens_details")
883+
and usage.prompt_tokens_details is not None
884+
):
879885
cached_tokens = usage.prompt_tokens_details.cached_tokens
880886
total_cached_input_tokens += cached_tokens
881887
total_input_tokens += usage.prompt_tokens - cached_tokens
882888
else:
883889
# If prompt_tokens_details doesn't exist, assume all tokens are uncached
884890
total_input_tokens += usage.prompt_tokens
885-
891+
886892
total_output_tokens += usage.completion_tokens
887893
return (
888894
content,

0 commit comments

Comments
 (0)