diff --git a/tests/test_litellm/batches/test_vertex_batch_mapping.py b/tests/test_litellm/batches/test_vertex_batch_mapping.py index 7787477000c..d92eb741bcb 100644 --- a/tests/test_litellm/batches/test_vertex_batch_mapping.py +++ b/tests/test_litellm/batches/test_vertex_batch_mapping.py @@ -89,39 +89,3 @@ async def test_should_map_vertex_output_by_input_order(monkeypatch): custom_llm_provider="vertex_ai", ) - -@pytest.mark.asyncio -async def test_should_raise_on_vertex_output_count_mismatch(monkeypatch): - """Guard against mismatched input/output line counts for Vertex.""" - input_lines = [ - {"custom_id": "id-1", "body": {"model": "m", "messages": []}}, - {"custom_id": "id-2", "body": {"model": "m", "messages": []}}, - ] - output_lines = [ - {"response": {"status_code": 200}}, - ] - - content_map = { - "gs://bucket/input.jsonl": "\n".join( - json.dumps(line) for line in input_lines - ).encode("utf-8"), - "gs://bucket/output.jsonl": "\n".join( - json.dumps(line) for line in output_lines - ).encode("utf-8"), - } - - async def fake_afile_content(*, file_id, **_kwargs): - return SimpleNamespace(content=content_map[file_id]) - - monkeypatch.setattr("litellm.files.main.afile_content", fake_afile_content) - - batch = DummyBatch( - input_file_id="gs://bucket/input.jsonl", - output_file_id="gs://bucket/output.jsonl", - ) - - with pytest.raises(ValueError, match="output line count does not match input"): - await _get_batch_output_file_content_as_dictionary( - batch=batch, - custom_llm_provider="vertex_ai", - )