You've already forked adk-python
mirror of
https://github.com/encounter/adk-python.git
synced 2026-07-09 18:19:28 -07:00
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:
@@ -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.")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user