phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(router.py): add server cooldown logic

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-22 15:59:41 -08:00
parent 4ece219ec5
commit 3e76d4b422
5 changed files with 67 additions and 82 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/integrations/custom_logger.py
View file

@ -33,9 +33,6 @@ class CustomLogger:
def log_input_event(self, model, messages, kwargs, print_verbose, callback_func):
try:
print_verbose(
f"Custom Logger - Enters logging function for model {kwargs}"
)
kwargs["model"] = model
kwargs["messages"] = messages
kwargs["log_event_type"] = "pre_api_call"
@ -52,9 +49,6 @@ class CustomLogger:
def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose, callback_func):
# Method definition
try:
print_verbose(
f"Custom Logger - Enters logging function for model {kwargs}"
)
kwargs["log_event_type"] = "post_api_call"
callback_func(
kwargs, # kwargs to func