scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-17 23:51:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 67

Addressing feedback.

Browse source
This commit is contained in:
Josh Bonczkowski 2026-03-17 14:44:02 -04:00
parent f5cc45f04c
commit d21a31d40c
2 changed files with 14 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/newrelic/newrelic.py
View file

@ -329,9 +329,9 @@ class NewRelicLogger(CustomLogger):
return {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
return {
"prompt_tokens": usage.get("prompt_tokens", 0),
"completion_tokens": usage.get("completion_tokens", 0),
"total_tokens": usage.get("total_tokens", 0),
"prompt_tokens": usage.get("prompt_tokens") or 0,
"completion_tokens": usage.get("completion_tokens") or 0,
"total_tokens": usage.get("total_tokens") or 0,
}
def _get_finish_reason(self, response_obj: ModelResponse) -> str:

11
tests/test_litellm/integrations/newrelic/test_newrelic.py
View file

@ -426,6 +426,17 @@ class TestExtractUsage:
usage = self.logger._extract_usage(response)
assert usage == {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
def test_explicit_none_token_fields_return_zeros(self):
response = {
"usage": {
"prompt_tokens": None,
"completion_tokens": None,
"total_tokens": None,
}
}
usage = self.logger._extract_usage(response)
assert usage == {"prompt_tokens": 0, "completion_tokens": 0, "total_tokens": 0}
class TestGetFinishReason:
def setup_method(self):