Skip to content

Commit 6625f7f

Browse files
committed
fix google test
1 parent a7094a1 commit 6625f7f

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/models/test_google.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff 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
)

0 commit comments

Comments
 (0)