[serve] Initial commit for CLI (#12770)

This commit is contained in:
Edward Oakes
2020-12-11 10:31:29 -06:00
committed by GitHub
parent 74c98ac38e
commit 699ded5328
2 changed files with 49 additions and 2 deletions
+45
View File
@@ -0,0 +1,45 @@
#!/usr/bin/env python
import click
import ray
from ray import serve
from ray.serve.constants import DEFAULT_HTTP_HOST, DEFAULT_HTTP_PORT
@click.group(
help="[EXPERIMENTAL] CLI for managing Serve instances on a Ray cluster.")
@click.option(
"--address",
"-a",
default="auto",
required=False,
type=str,
help="Address of the running Ray cluster to connect to. "
"Defaults to \"auto\".")
def cli(address):
ray.init(address=address)
@cli.command(help="Start a detached Serve instance on the Ray cluster.")
@click.option(
"--http-host",
default=DEFAULT_HTTP_HOST,
required=False,
type=str,
help="Host for HTTP servers to listen on. "
f"Defaults to {DEFAULT_HTTP_HOST}.")
@click.option(
"--http-port",
default=DEFAULT_HTTP_PORT,
required=False,
type=int,
help="Port for HTTP servers to listen on. "
f"Defaults to {DEFAULT_HTTP_PORT}.")
def start(http_host, http_port):
serve.start(detached=True, http_host=http_host, http_port=http_port)
@cli.command(help="Shutdown the running Serve instance on the Ray cluster.")
def shutdown():
serve.connect().shutdown()
+4 -2
View File
@@ -449,8 +449,10 @@ setuptools.setup(
entry_points={
"console_scripts": [
"ray=ray.scripts.scripts:main",
"rllib=ray.rllib.scripts:cli [rllib]", "tune=ray.tune.scripts:cli",
"ray-operator=ray.operator:main"
"rllib=ray.rllib.scripts:cli [rllib]",
"tune=ray.tune.scripts:cli",
"ray-operator=ray.operator:main",
"serve=ray.serve.scripts:cli",
]
},
include_package_data=True,