scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 89

fix flaky test

Browse source
This commit is contained in:
mubashir1osmani 2025-09-24 02:29:30 -04:00
parent 9a8a8f2d47
commit b0ac2f2dc3
1 changed files with 3 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
tests/pass_through_tests/test_anthropic_passthrough.py
View file

@ -43,8 +43,9 @@ async def test_anthropic_basic_completion_with_headers():
json.dumps(response_json, indent=4, default=str),
)
reported_usage = response_json.get("usage", None)
anthropic_api_input_tokens = reported_usage.get("input_tokens", None)
anthropic_api_output_tokens = reported_usage.get("output_tokens", None)
# fix null checks for reported_usage
anthropic_api_input_tokens = reported_usage.get("input_tokens", None) if reported_usage else None
anthropic_api_output_tokens = reported_usage.get("output_tokens", None) if reported_usage else None
litellm_call_id = response_headers.get("x-litellm-call-id")
print(f"LiteLLM Call ID: {litellm_call_id}")