scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 455

fix(managed files): skip chat messages and content items that are not objects

Browse source
This commit is contained in:
mateo-berri 2026-10-29 09:30:58 -07:00
parent 8cc51b31d7
commit 15e6516fe1
2 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
enterprise/litellm_enterprise/proxy/hooks/managed_files.py
View file

@ -1735,13 +1735,13 @@ class _PROXY_LiteLLMManagedFiles(CustomLogger, BaseFileEndpoints):
"""
file_ids = []
for message in messages:
if message.get("role") == "user":
if isinstance(message, dict) and message.get("role") == "user":
content = message.get("content")
if content:
if isinstance(content, str):
continue
for c in content:
if c.get("type") == "file":
if isinstance(c, dict) and c.get("type") == "file":
file_object_file_field = cast(Dict[str, object], c).get("file")
if not isinstance(file_object_file_field, dict):
continue

2
tests/unit/enterprise/proxy/hooks/test_managed_files.py
View file

@ -337,9 +337,11 @@ def test_get_file_ids_from_messages_leaves_malformed_file_parts_to_request_valid
DualCache(), prisma_client=MagicMock()
)
messages = [
"not-a-message",
{
"role": "user",
"content": [
"not-a-part",
{"type": "file"},
{"type": "file", "file": "file-raw"},
{"type": "file", "file": None},