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

feat - allow accessing data post success call

Browse source
This commit is contained in:
Ishaan Jaff 2024-08-19 11:35:33 -07:00
parent 6af497e383
commit b4bca8db82
12 changed files with 71 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/hooks/dynamic_rate_limiter.py
View file

@ -254,7 +254,7 @@ class _PROXY_DynamicRateLimitHandler(CustomLogger):
return None
async def async_post_call_success_hook(
self, user_api_key_dict: UserAPIKeyAuth, response
self, data: dict, user_api_key_dict: UserAPIKeyAuth, response
):
try:
if isinstance(response, ModelResponse):
@ -287,7 +287,9 @@ class _PROXY_DynamicRateLimitHandler(CustomLogger):
return response
return await super().async_post_call_success_hook(
user_api_key_dict, response
data=data,
user_api_key_dict=user_api_key_dict,
response=response,
)
except Exception as e:
verbose_proxy_logger.exception(