Skip to content

Commit 4c1c501

Browse files
[Refactor] [10/N] to simplify the vLLM openai completion serving architecture (vllm-project#32369)
Signed-off-by: chaunceyjiang <chaunceyjiang@gmail.com>
1 parent ae1eba6 commit 4c1c501

43 files changed

Lines changed: 777 additions & 642 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

tests/entrypoints/openai/test_chat_error.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212
from vllm.entrypoints.openai.chat_completion.protocol import ChatCompletionRequest
1313
from vllm.entrypoints.openai.chat_completion.serving import OpenAIServingChat
1414
from vllm.entrypoints.openai.engine.protocol import ErrorResponse
15-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
15+
from vllm.entrypoints.openai.models.protocol import BaseModelPath
16+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
1617
from vllm.outputs import CompletionOutput, RequestOutput
1718
from vllm.tokenizers import get_tokenizer
1819
from vllm.v1.engine.async_llm import AsyncLLM

tests/entrypoints/openai/test_cli_args.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import pytest
77

88
from vllm.entrypoints.openai.cli_args import make_arg_parser, validate_parsed_serve_args
9-
from vllm.entrypoints.openai.serving_models import LoRAModulePath
9+
from vllm.entrypoints.openai.models.protocol import LoRAModulePath
1010
from vllm.utils.argparse_utils import FlexibleArgumentParser
1111

1212
from ...utils import VLLM_PATH

tests/entrypoints/openai/test_completion_error.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,11 @@
99
import pytest
1010

1111
from vllm.config.multimodal import MultiModalConfig
12-
from vllm.entrypoints.openai.engine.protocol import CompletionRequest, ErrorResponse
13-
from vllm.entrypoints.openai.serving_completion import OpenAIServingCompletion
14-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
12+
from vllm.entrypoints.openai.completion.protocol import CompletionRequest
13+
from vllm.entrypoints.openai.completion.serving import OpenAIServingCompletion
14+
from vllm.entrypoints.openai.engine.protocol import ErrorResponse
15+
from vllm.entrypoints.openai.models.protocol import BaseModelPath
16+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
1517
from vllm.outputs import CompletionOutput, RequestOutput
1618
from vllm.tokenizers import get_tokenizer
1719
from vllm.v1.engine.async_llm import AsyncLLM

tests/entrypoints/openai/test_gptoss_structural_tags_integration.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,13 +8,11 @@
88

99
import pytest
1010

11-
from vllm.entrypoints.openai.engine.protocol import (
12-
StructuredOutputsParams,
13-
)
1411
from vllm.entrypoints.tool_server import ToolServer
1512
from vllm.reasoning.gptoss_reasoning_parser import (
1613
GptOssReasoningParser,
1714
)
15+
from vllm.sampling_params import StructuredOutputsParams
1816

1917

2018
class TestGptOssStructuralTagsIntegration:

tests/entrypoints/openai/test_lora_resolvers.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -9,9 +9,11 @@
99
import pytest
1010

1111
from vllm.config.multimodal import MultiModalConfig
12-
from vllm.entrypoints.openai.engine.protocol import CompletionRequest, ErrorResponse
13-
from vllm.entrypoints.openai.serving_completion import OpenAIServingCompletion
14-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
12+
from vllm.entrypoints.openai.completion.protocol import CompletionRequest
13+
from vllm.entrypoints.openai.completion.serving import OpenAIServingCompletion
14+
from vllm.entrypoints.openai.engine.protocol import ErrorResponse
15+
from vllm.entrypoints.openai.models.protocol import BaseModelPath
16+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
1517
from vllm.lora.request import LoRARequest
1618
from vllm.lora.resolver import LoRAResolver, LoRAResolverRegistry
1719
from vllm.tokenizers import get_tokenizer

tests/entrypoints/openai/test_serving_chat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
ErrorResponse,
2121
RequestResponseMetadata,
2222
)
23+
from vllm.entrypoints.openai.models.serving import BaseModelPath, OpenAIServingModels
2324
from vllm.entrypoints.openai.parser.harmony_utils import get_encoding
24-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
2525
from vllm.outputs import CompletionOutput, RequestOutput
2626
from vllm.tokenizers import get_tokenizer
2727
from vllm.tool_parsers import ToolParserManager

tests/entrypoints/openai/test_serving_engine.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from vllm.config import ModelConfig
1111
from vllm.entrypoints.openai.engine.serving import OpenAIServing
12-
from vllm.entrypoints.openai.serving_models import OpenAIServingModels
12+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
1313
from vllm.tokenizers.mistral import MistralTokenizer
1414

1515

tests/entrypoints/openai/test_serving_models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@
1111
from vllm.entrypoints.openai.engine.protocol import (
1212
ErrorResponse,
1313
)
14-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
14+
from vllm.entrypoints.openai.models.protocol import BaseModelPath
15+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
1516
from vllm.entrypoints.serve.lora.protocol import (
1617
LoadLoRAAdapterRequest,
1718
UnloadLoRAAdapterRequest,

tests/v1/engine/test_async_llm.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,8 @@
1919
from vllm.entrypoints.openai.engine.protocol import (
2020
ErrorResponse,
2121
)
22-
from vllm.entrypoints.openai.serving_models import BaseModelPath, OpenAIServingModels
22+
from vllm.entrypoints.openai.models.protocol import BaseModelPath
23+
from vllm.entrypoints.openai.models.serving import OpenAIServingModels
2324
from vllm.inputs import PromptType
2425
from vllm.outputs import RequestOutput
2526
from vllm.platforms import current_platform
Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,2 @@
1+
# SPDX-License-Identifier: Apache-2.0
2+
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project

0 commit comments

Comments
 (0)