scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-20 00:11:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 134

refactor(spend_tracking): type the get_logging_payload parameters

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-09-17 01:02:03 +00:00
parent f76e8b3984
commit e06c81665f
2 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/db/db_spend_update_writer.py
View file

@ -257,8 +257,8 @@ class DBSpendUpdateWriter:
# Completion object fields
kwargs: dict | None,
completion_response: object,
start_time: datetime | None,
end_time: datetime | None,
start_time: datetime,
end_time: datetime,
response_cost: float | None,
) -> bool:
"""Record the request's spend, answering whether its cost still needs charging.

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

@ -378,7 +378,11 @@ def _looks_like_model_name(model: str) -> bool:
def get_logging_payload(
kwargs, response_obj, start_time, end_time, llm_router: "Router | None" = None
kwargs: dict | None,
response_obj: object,
start_time: datetime,
end_time: datetime,
llm_router: "Router | None" = None,
) -> SpendLogsPayload:
if kwargs is None:
kwargs = {}