File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed
Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -2046,7 +2046,8 @@ async def test_google_url_input(
20462046 content = ['What is the main content of this URL?' , Is (url )],
20472047 timestamp = IsDatetime (),
20482048 ),
2049- ]
2049+ ],
2050+ run_id = IsStr (),
20502051 ),
20512052 ModelResponse (
20522053 parts = [TextPart (content = Is (expected_output ))],
@@ -2057,6 +2058,7 @@ async def test_google_url_input(
20572058 provider_details = {'finish_reason' : 'STOP' },
20582059 provider_response_id = IsStr (),
20592060 finish_reason = 'stop' ,
2061+ run_id = IsStr (),
20602062 ),
20612063 ]
20622064 )
@@ -2086,7 +2088,8 @@ async def test_google_url_input_force_download(
20862088 content = ['What is the main content of this URL?' , Is (video_url )],
20872089 timestamp = IsDatetime (),
20882090 ),
2089- ]
2091+ ],
2092+ run_id = IsStr (),
20902093 ),
20912094 ModelResponse (
20922095 parts = [TextPart (content = Is (output ))],
@@ -2097,6 +2100,7 @@ async def test_google_url_input_force_download(
20972100 provider_response_id = IsStr (),
20982101 provider_name = 'google-vertex' ,
20992102 finish_reason = 'stop' ,
2103+ run_id = IsStr (),
21002104 ),
21012105 ]
21022106 )
You can’t perform that action at this time.
0 commit comments