scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-28 01:32:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 43

Merge c6d5ed94c6 into 2dccc0dc79

Browse source
This commit is contained in:
justavibedev 2026-09-23 14:43:39 +00:00 • committed by GitHub
commit aefbb5405d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 14 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
litellm/proxy/spend_tracking/spend_tracking_utils.py
View file

@ -264,7 +264,7 @@ def get_spend_logs_id(call_type: str, response_obj: dict, kwargs: dict) -> str |
kwargs.get("litellm_call_id"),
)
resolved_id: Final = next(
(candidate for candidate in candidate_ids if isinstance(candidate, str) and candidate), None
(candidate for candidate in candidate_ids if isinstance(candidate, str) and candidate not in ("", "None")), None
)
if resolved_id is not None and call_type == CallTypes.aretrieve_batch.value:
return f"{resolved_id}{BATCH_COST_REQUEST_ID_SUFFIX}"

13
tests/test_litellm/proxy/spend_tracking/test_spend_tracking_utils.py
View file

@ -4083,6 +4083,19 @@ def test_get_spend_logs_id_prefers_the_response_id_over_the_standard_logging_id(
)
def test_get_spend_logs_id_falls_back_when_the_standard_logging_id_is_none():
ids = [
get_spend_logs_id(
"acompletion",
{"id": None},
{"litellm_call_id": call_id, "standard_logging_object": {"id": "None"}},
)
for call_id in ("call-id-1", "call-id-2")
]
assert ids == ["call-id-1", "call-id-2"]
@pytest.mark.asyncio
async def test_spend_log_request_id_is_the_message_id_a_bridged_streaming_caller_was_streamed():
"""A streaming /v1/messages call against a non-Anthropic model is served a msg_ id the