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

5
litellm/proxy/utils.py
View file

@ -717,6 +717,7 @@ class ProxyLogging:
async def post_call_success_hook(
self,
data: dict,
response: Union[ModelResponse, EmbeddingResponse, ImageResponse],
user_api_key_dict: UserAPIKeyAuth,
):
@ -738,7 +739,9 @@ class ProxyLogging:
_callback = callback # type: ignore
if _callback is not None and isinstance(_callback, CustomLogger):
await _callback.async_post_call_success_hook(
user_api_key_dict=user_api_key_dict, response=response
user_api_key_dict=user_api_key_dict,
data=data,
response=response,
)
except Exception as e:
raise e