From 69d06cfbcd77bd9db53a2ec9ac5923bdf04263b1 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 29 Jun 2024 18:58:44 -0700 Subject: [PATCH] fix(utils.py): fix passing additional param to post-call processing --- litellm/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index e75a5df736..8776854167 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -1351,7 +1351,9 @@ def client(original_function): ).total_seconds() * 1000 # return response latency in ms like openai ### POST-CALL RULES ### - post_call_processing(original_response=result, model=model) + post_call_processing( + original_response=result, model=model, optional_params=kwargs + ) # [OPTIONAL] ADD TO CACHE if (