xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-05-16 19:09:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix][API Server] Fix invalid usage of 'ge' and 'le' in port valid… (#13672)

Browse Source
This commit is contained in:
Robin 2025-02-22 14:05:28 +08:00 committed by GitHub
parent 0ffdf8ce0c
commit c6ed93860f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/entrypoints/api_server.py
View File

@ -145,7 +145,7 @@ async def run_server(args: Namespace,
if __name__ == "__main__": if __name__ == "__main__":
parser = FlexibleArgumentParser() parser = FlexibleArgumentParser()
parser.add_argument("--host", type=str, default=None) parser.add_argument("--host", type=str, default=None)
parser.add_argument("--port", type=int, default=8000, ge=1024, le=65535) parser.add_argument("--port", type=parser.check_port, default=8000)
parser.add_argument("--ssl-keyfile", type=str, default=None) parser.add_argument("--ssl-keyfile", type=str, default=None)
parser.add_argument("--ssl-certfile", type=str, default=None) parser.add_argument("--ssl-certfile", type=str, default=None)
parser.add_argument("--ssl-ca-certs", parser.add_argument("--ssl-ca-certs",

11
vllm/utils.py
View File

@ -1194,6 +1194,17 @@ class FlexibleArgumentParser(argparse.ArgumentParser):
return super().parse_args(processed_args, namespace) return super().parse_args(processed_args, namespace)
def check_port(self, value):
try:
value = int(value)
except ValueError:
raise argparse.ArgumentTypeError("Port must be an integer")
if not (1024 <= value <= 65535):
raise argparse.ArgumentTypeError("Port must be between 1024 and 65535")
return value
def _pull_args_from_config(self, args: List[str]) -> List[str]: def _pull_args_from_config(self, args: List[str]) -> List[str]:
"""Method to pull arguments specified in the config file """Method to pull arguments specified in the config file
into the command-line args variable. into the command-line args variable.