feat(lite_llm): add PDF support

Adds the ability to ingest and process PDF documents in the lite_llm pipeline.
This commit is contained in:
Marcello Maugeri
2025-07-09 10:17:29 +02:00
parent 18cd3844db
commit 2ac8c44668
+11
View File
@@ -38,6 +38,7 @@ from litellm import ChatCompletionDeveloperMessage
from litellm import ChatCompletionImageUrlObject
from litellm import ChatCompletionMessageToolCall
from litellm import ChatCompletionTextObject
from litellm import ChatCompletionFileObject
from litellm import ChatCompletionToolMessage
from litellm import ChatCompletionUserMessage
from litellm import ChatCompletionVideoUrlObject
@@ -262,6 +263,16 @@ def _get_content(
video_url=data_uri,
)
)
elif part.inline_data.mime_type == "application/pdf":
content_objects.append(
ChatCompletionFileObject(
type="file",
file={
"file_data": data_uri,
"format": "pdf"
}
)
)
else:
raise ValueError("LiteLlm(BaseLlm) does not support this content part.")