diff --git a/src/google/adk/models/lite_llm.py b/src/google/adk/models/lite_llm.py index 39514d6f..430d2e20 100644 --- a/src/google/adk/models/lite_llm.py +++ b/src/google/adk/models/lite_llm.py @@ -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.")