[Frontend] Add --version flag to CLI (#10369)

Signed-off-by: Russell Bryant <rbryant@redhat.com>
This commit is contained in:
Russell Bryant 2024-11-15 16:13:53 -05:00 committed by GitHub
parent c76ac49d26
commit a067f85e08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -9,6 +9,7 @@ import uvloop
from openai import OpenAI
from openai.types.chat import ChatCompletionMessageParam
import vllm.version
from vllm.engine.arg_utils import EngineArgs
from vllm.entrypoints.openai.api_server import run_server
from vllm.entrypoints.openai.cli_args import (make_arg_parser,
@ -143,6 +144,11 @@ def main():
env_setup()
parser = FlexibleArgumentParser(description="vLLM CLI")
parser.add_argument('-v',
'--version',
action='version',
version=vllm.version.__version__)
subparsers = parser.add_subparsers(required=True, dest="subparser")
serve_parser = subparsers.add_parser(