fix(utils.py): fix passing additional param to post-call processing

2024-06-29 18:58:44 -07:00 · 2024-06-29 18:58:44 -07:00 · 69d06cfbcd
commit 69d06cfbcd
parent b699d9a8b9
1 changed files with 3 additions and 1 deletions
--- 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 (