diff --git a/litellm/proxy/custom_logger.py b/litellm/proxy/custom_logger.py index 54428372357..06d5fc12767 100644 --- a/litellm/proxy/custom_logger.py +++ b/litellm/proxy/custom_logger.py @@ -15,8 +15,13 @@ class MyCustomHandler(CustomLogger): # log: key, user, model, prompt, response, tokens, cost print("\n kwargs\n") print(kwargs) + ### Access kwargs passed to litellm.completion() model = kwargs["model"] messages = kwargs["messages"] + user = kwargs.get("user") + ################################################# + + ### Calculate cost ####################### cost = litellm.completion_cost(completion_response=response_obj) response = response_obj # tokens used in response @@ -26,6 +31,7 @@ class MyCustomHandler(CustomLogger): f""" Model: {model}, Messages: {messages}, + User: {user}, Usage: {usage}, Cost: {cost}, Response: {response}