diff --git a/enterprise/litellm_enterprise/proxy/hooks/managed_files.py b/enterprise/litellm_enterprise/proxy/hooks/managed_files.py index 03a31066e78..dadb4e4c5d4 100644 --- a/enterprise/litellm_enterprise/proxy/hooks/managed_files.py +++ b/enterprise/litellm_enterprise/proxy/hooks/managed_files.py @@ -896,7 +896,12 @@ class _PROXY_LiteLLMManagedFiles(CustomLogger, BaseFileEndpoints): return spellings: Final = list(provider_file_id_spellings(file_id)) managed_file = await _managed_file_table(self.prisma_client).find_first( - where={"OR": [{"unified_file_id": {"in": spellings}}, {"flat_model_file_ids": {"hasSome": spellings}}]} + where={ + "OR": [ + {"unified_file_id": {"in": spellings}}, # bounded-ok: at most four spellings of one file id + {"flat_model_file_ids": {"hasSome": spellings}}, + ] + } ) if managed_file is None: return