scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-16 04:29:31 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 139

style(presidio): organize imports and adhere to ruff check

Browse source
This commit is contained in:
musi22 2026-09-21 23:39:32 +05:30
parent 36e21f22c7
commit ac8ec7bb75
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/proxy/guardrails/guardrail_hooks/presidio.py
View file

@ -9,12 +9,12 @@
import asyncio
import json
import threading
from collections.abc import AsyncGenerator, AsyncIterable, Awaitable, Mapping, Sequence
from contextlib import asynccontextmanager
from datetime import datetime
from functools import reduce
import json
import threading
from types import MappingProxyType
from typing import TYPE_CHECKING, Any, Final, Literal, Optional, Protocol, TypedDict, cast
@ -1628,7 +1628,7 @@ class _OPTIONAL_PresidioPIIMasking(CustomGuardrail):
if not getattr(assembled_model_response, "usage", None) and chunks:
last_chunk_usage: Final = getattr(chunks[-1], "usage", None)
if last_chunk_usage:
setattr(assembled_model_response, "usage", last_chunk_usage)
assembled_model_response.usage = last_chunk_usage
def get_presidio_settings_from_request_data(self, data: dict) -> PresidioPerRequestConfig | None:
if "metadata" in data: