phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

Merge pull request #2706 from BerriAI/litellm_key_llm_guardrails

Browse source 
feat(llm_guard.py): enable key-specific llm guard check
This commit is contained in:
Krish Dholakia 2024-03-26 19:02:11 -07:00 • committed by GitHub
commit c1f8d346b8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
11 changed files with 141 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/proxy/proxy_server.py
View file

@ -3168,7 +3168,9 @@ async def chat_completion(
tasks = []
tasks.append(
proxy_logging_obj.during_call_hook(data=data, call_type="completion")
proxy_logging_obj.during_call_hook(
data=data, user_api_key_dict=user_api_key_dict, call_type="completion"
)
)
start_time = time.time()