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

feat(proxy_server): adds create-proxy feature

Browse source
This commit is contained in:
Krrish Dholakia 2023-10-12 18:24:09 -07:00
parent 3da89a58ae
commit b28c055896
11 changed files with 246 additions and 124 deletions
Download patch file Download diff file Expand all files Collapse all files

19
litellm/integrations/custom_logger.py
View file

@ -13,6 +13,24 @@ class CustomLogger:
def __init__(self, callback_func):
# Instance variables
self.callback_func = callback_func
def log_input_event(self, model, messages, kwargs, print_verbose):
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"
self.callback_func(
kwargs,
)
print_verbose(
f"Custom Logger - model call details: {kwargs}"
)
except:
traceback.print_exc()
print_verbose(f"Custom Logger Error - {traceback.format_exc()}")
def log_event(self, kwargs, response_obj, start_time, end_time, print_verbose):
# Method definition
@ -20,6 +38,7 @@ class CustomLogger:
print_verbose(
f"Custom Logger - Enters logging function for model {kwargs}"
)
kwargs["log_event_type"] = "post_api_call"
self.callback_func(
kwargs, # kwargs to func
response_obj,