scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 83

⚡️ perf: Remove test violation on each stream chunk

Browse source 
Signed-off-by: Lunik <lunik@tiwabbit.fr>
This commit is contained in:
Lunik 2024-05-03 20:51:40 +02:00
parent e7405f105c
commit 9ba9b3891f
No known key found for this signature in database
GPG key ID: AEC152C8AC2C85EE
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/hooks/azure_content_safety.py
View file

@ -148,10 +148,10 @@ class _PROXY_AzureContentSafety(
content=response.choices[0].message.content, source="output"
)
async def async_post_call_streaming_hook(
self,
user_api_key_dict: UserAPIKeyAuth,
response: str,
):
self.print_verbose(f"Inside Azure Content-Safety Call-Stream Hook")
await self.test_violation(content=response, source="output")
#async def async_post_call_streaming_hook(
# self,
# user_api_key_dict: UserAPIKeyAuth,
# response: str,
#):
# self.print_verbose(f"Inside Azure Content-Safety Call-Stream Hook")
# await self.test_violation(content=response, source="output")