test_outputs.py 522 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# SPDX-License-Identifier: Apache-2.0

from vllm.outputs import RequestOutput


def test_request_output_forward_compatible():
    output = RequestOutput(request_id="test_request_id",
                           prompt="test prompt",
                           prompt_token_ids=[1, 2, 3],
                           prompt_logprobs=None,
                           outputs=[],
                           finished=False,
                           example_arg_added_in_new_version="some_value")
    assert output is not None