diff --git a/src/google/adk/models/lite_llm.py b/src/google/adk/models/lite_llm.py index 39514d6f..2dfcbba8 100644 --- a/src/google/adk/models/lite_llm.py +++ b/src/google/adk/models/lite_llm.py @@ -35,6 +35,7 @@ from litellm import acompletion from litellm import ChatCompletionAssistantMessage from litellm import ChatCompletionAssistantToolCall from litellm import ChatCompletionDeveloperMessage +from litellm import ChatCompletionFileObject from litellm import ChatCompletionImageUrlObject from litellm import ChatCompletionMessageToolCall from litellm import ChatCompletionTextObject @@ -262,6 +263,12 @@ 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.")