diff --git a/src/google/adk/models/lite_llm.py b/src/google/adk/models/lite_llm.py index 430d2e20..2dfcbba8 100644 --- a/src/google/adk/models/lite_llm.py +++ b/src/google/adk/models/lite_llm.py @@ -35,10 +35,10 @@ 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 -from litellm import ChatCompletionFileObject from litellm import ChatCompletionToolMessage from litellm import ChatCompletionUserMessage from litellm import ChatCompletionVideoUrlObject @@ -265,13 +265,9 @@ def _get_content( ) elif part.inline_data.mime_type == "application/pdf": content_objects.append( - ChatCompletionFileObject( - type="file", - file={ - "file_data": data_uri, - "format": "pdf" - } - ) + ChatCompletionFileObject( + type="file", file={"file_data": data_uri, "format": "pdf"} + ) ) else: raise ValueError("LiteLlm(BaseLlm) does not support this content part.")