feat(proxy_server.py): support for pii masking with microsoft presidio

This commit is contained in:
Krrish Dholakia 2024-02-10 20:20:59 -08:00
parent afd0d7cb41
commit 2d845b12ed
3 changed files with 126 additions and 5 deletions

View file

@ -92,7 +92,7 @@ class ProxyLogging:
self,
user_api_key_dict: UserAPIKeyAuth,
data: dict,
call_type: Literal["completion", "embeddings"],
call_type: Literal["completion", "embeddings", "image_generation"],
):
"""
Allows users to modify/reject the incoming request to the proxy, without having to deal with parsing Request body.