scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-02 02:11:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 102

fix(attachment_registry): preserve config-loaded policies across DB sync

Browse source 
Signed-off-by: Grégoire Bellon-Gervais <gregoire.bellon-gervais@docaposte.fr>
This commit is contained in:
Grégoire Bellon-Gervais 2026-06-04 16:18:57 +02:00
parent 9196098e9e
commit 1f18188b86
No known key found for this signature in database
GPG key ID: 2AA58B478665D0AF
1 changed files with 11 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
litellm/proxy/policy_engine/attachment_registry.py
View file

@ -41,6 +41,7 @@ class AttachmentRegistry:
def __init__(self):
self._attachments: List[PolicyAttachment] = []
self._config_attachments: List[PolicyAttachment] = []
self._initialized: bool = False
def load_attachments(self, attachments_config: List[Dict[str, Any]]) -> None:
@ -51,11 +52,13 @@ class AttachmentRegistry:
attachments_config: List of attachment dictionaries from YAML.
"""
self._attachments = []
self._config_attachments = []
for attachment_data in attachments_config:
try:
attachment = self._parse_attachment(attachment_data)
self._attachments.append(attachment)
self._config_attachments.append(attachment)
verbose_proxy_logger.debug(
f"Loaded attachment for policy: {attachment.policy}"
)
@ -210,6 +213,7 @@ class AttachmentRegistry:
Clear all attachments from the registry.
"""
self._attachments = []
self._config_attachments = []
self._initialized = False
def add_attachment(self, attachment: PolicyAttachment) -> None:
@ -452,14 +456,17 @@ class AttachmentRegistry:
"""
Sync policy attachments from the database to in-memory registry.
Preserves config-loaded attachments so a DB sync with no results does not
wipe them. DB entries are merged on top of config attachments.
Args:
prisma_client: The Prisma client instance
"""
try:
attachments = await self.get_all_attachments_from_db(prisma_client)
# Clear existing attachments and reload from DB
self._attachments = []
# Start with config-loaded attachments (preserve YAML-defined ones)
self._attachments = list(self._config_attachments)
for attachment_response in attachments:
attachment = PolicyAttachment(
@ -480,7 +487,8 @@ class AttachmentRegistry:
self._initialized = True
verbose_proxy_logger.info(
f"Synced {len(attachments)} attachments from DB to in-memory registry"
f"Synced {len(attachments)} DB attachments + "
f"{len(self._config_attachments)} config attachments to in-memory registry"
)
except Exception as e:
verbose_proxy_logger.exception(f"Error syncing attachments from DB: {e}")