feat(llm_guard.py): enable key-specific llm guard check

This commit is contained in:
Krrish Dholakia 2024-03-26 17:21:51 -07:00
parent bec093675c
commit e10eb8f6fe
8 changed files with 26 additions and 3 deletions

View file

@ -141,6 +141,7 @@ class ProxyLogging:
async def during_call_hook(
self,
data: dict,
user_api_key_dict: UserAPIKeyAuth,
call_type: Literal[
"completion",
"embeddings",
@ -157,7 +158,9 @@ class ProxyLogging:
try:
if isinstance(callback, CustomLogger):
await callback.async_moderation_hook(
data=new_data, call_type=call_type
data=new_data,
user_api_key_dict=user_api_key_dict,
call_type=call_type,
)
except Exception as e:
raise e