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

fix(guardrails/purview): propagate litellm_call_id as correlation_id to Purview

Browse source
This commit is contained in:
Sameer Kankute 2026-04-02 11:36:36 +05:30
parent 592e418012
commit 246d9f6e12
No known key found for this signature in database
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/proxy/guardrails/guardrail_hooks/microsoft_purview/purview_dlp.py
View file

@ -8,6 +8,7 @@ Supports three modes:
"""
import asyncio
import uuid
from concurrent.futures import ThreadPoolExecutor
from datetime import datetime
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Type, Union
@ -117,11 +118,13 @@ class MicrosoftPurviewDLPGuardrail(PurviewGuardrailBase, CustomGuardrail):
try:
etag, _ = await self._compute_protection_scopes(user_id)
correlation_id = request_data.get("litellm_call_id") or str(uuid.uuid4())
response = await self._process_content(
user_id=user_id,
text=text,
activity=activity,
etag=etag,
correlation_id=correlation_id,
)
if self._should_block(response):