main
py 18 lines 482 Bytes
Raw
1 # SPDX-License-Identifier: Apache-2.0
2 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
4 import pytest
5
6 from vllm.entrypoints.openai.chat_completion.protocol import ChatCompletionRequest
7
8 from .utils import MockCohereTokenizer
9
10
11 @pytest.fixture(scope="package")
12 def tokenizer() -> MockCohereTokenizer:
13 return MockCohereTokenizer()
14
15
16 @pytest.fixture
17 def request_obj() -> ChatCompletionRequest:
18 return ChatCompletionRequest(messages=[], model="test-model")