File tree Expand file tree Collapse file tree
contributing/samples/gepa Expand file tree Collapse file tree Original file line number Diff line number Diff line change 4343from tau_bench .types import EnvRunResult
4444from tau_bench .types import RunConfig
4545import tau_bench_agent as tau_bench_agent_lib
46-
4746import utils
4847
4948
Original file line number Diff line number Diff line change 2525from absl import flags
2626import experiment
2727from google .genai import types
28-
2928import utils
3029
3130_OUTPUT_DIR = flags .DEFINE_string (
Original file line number Diff line number Diff line change 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
Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments