Skip to content

Commit 063e7e3

Browse files
style: apply pyink/isort formatting to LiteLLM and GEPA samples
- Wrap long lines in lite_llm.py _MIME_TO_FILENAME and test_litellm.py - Remove trailing blank lines in GEPA experiment scripts Made-with: Cursor
1 parent 1db35a2 commit 063e7e3

4 files changed

Lines changed: 9 additions & 5 deletions

File tree

contributing/samples/gepa/experiment.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,6 @@
4343
from tau_bench.types import EnvRunResult
4444
from tau_bench.types import RunConfig
4545
import tau_bench_agent as tau_bench_agent_lib
46-
4746
import utils
4847

4948

contributing/samples/gepa/run_experiment.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,6 @@
2525
from absl import flags
2626
import experiment
2727
from google.genai import types
28-
2928
import utils
3029

3130
_OUTPUT_DIR = flags.DEFINE_string(

src/google/adk/models/lite_llm.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,12 @@
145145
"application/pdf": "document.pdf",
146146
"application/json": "document.json",
147147
"application/msword": "document.doc",
148-
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": "document.docx",
149-
"application/vnd.openxmlformats-officedocument.presentationml.presentation": "document.pptx",
148+
"application/vnd.openxmlformats-officedocument.wordprocessingml.document": (
149+
"document.docx"
150+
),
151+
"application/vnd.openxmlformats-officedocument.presentationml.presentation": (
152+
"document.pptx"
153+
),
150154
"application/x-sh": "script.sh",
151155
}
152156

tests/unittests/models/test_litellm.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3972,7 +3972,9 @@ async def test_get_content_pdf_openai_uses_display_name_as_filename(mocker):
39723972
mock_acreate_file = AsyncMock(return_value=mock_file_response)
39733973
mocker.patch.object(litellm, "acreate_file", new=mock_acreate_file)
39743974

3975-
part = types.Part.from_bytes(data=b"test_pdf_data", mime_type="application/pdf")
3975+
part = types.Part.from_bytes(
3976+
data=b"test_pdf_data", mime_type="application/pdf"
3977+
)
39763978
part.inline_data.display_name = "my_report.pdf"
39773979
content = await _get_content([part], provider="openai")
39783980

0 commit comments

Comments
 (0)