mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-24 06:35:01 +08:00
37 lines
928 B
Python
37 lines
928 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
import pytest
|
|
import pytest_asyncio
|
|
|
|
from tests.utils import RemoteOpenAIServer
|
|
|
|
# Use a small reasoning model to test the responses API.
|
|
MODEL_NAME = "Qwen/Qwen3-1.7B"
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def default_server_args():
|
|
return [
|
|
"--max-model-len",
|
|
"8192",
|
|
"--enforce-eager", # For faster startup.
|
|
"--reasoning-parser",
|
|
"deepseek_r1",
|
|
]
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def server_with_store(default_server_args):
|
|
with RemoteOpenAIServer(
|
|
MODEL_NAME,
|
|
default_server_args,
|
|
env_dict={"VLLM_ENABLE_RESPONSES_API_STORE": "1"},
|
|
) as remote_server:
|
|
yield remote_server
|
|
|
|
|
|
@pytest_asyncio.fixture
|
|
async def client(server_with_store):
|
|
async with server_with_store.get_async_client() as async_client:
|
|
yield async_client
|