mirror/OpenLLM
1
0
Fork 0
mirror of https://github.com/bentoml/OpenLLM.git synced 2026-01-23 06:52:42 -05:00
Code Issues Packages Projects Releases Wiki Activity

revert(perf): remove group alias

Browse Source 
There is no need for this feature

Signed-off-by: Aaron <29749331+aarnphm@users.noreply.github.com>
This commit is contained in:
Aaron
2023-05-28 10:04:33 -07:00
parent 454b1edc32
commit ac710dfd54
1 changed files with 1 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
src/openllm/cli.py
View File

@@ -162,45 +162,9 @@ class OpenLLMCommandGroup(BentoMLCommandGroup):
def __init__(self, *args: t.Any, **attrs: t.Any) -> None:
super(OpenLLMCommandGroup, self).__init__(*args, **attrs)
# these two dictionaries will store known aliases for commands and groups
self._alias_groups: dict[str, tuple[str, click.Group]] = {}
self._cached_http: dict[str, t.Any] = {}
self._cached_grpc: dict[str, t.Any] = {}
def group(self, *args: t.Any, **attrs: t.Any) -> t.Callable[[F[P]], click.Group]:
"""Override the default 'cli.group' with supports for aliases for given group."""
from click.utils import make_default_short_help
aliases = attrs.pop("aliases", [])
def wrapper(f: F[P]) -> click.Group:
from gettext import gettext
name = attrs.pop("name", f.__name__)
group = super(OpenLLMCommandGroup, self).group(name, *args, **attrs)(f)
if group.short_help:
text = inspect.cleandoc(group.short_help)
elif group.help:
text = make_default_short_help(group.help, 45)
else:
text = ""
if group.deprecated:
text = gettext("(Deprecated) {text}").format(text=text)
attrs.setdefault("help", inspect.getdoc(f))
attrs.setdefault("short_help", text)
if len(aliases) > 0:
for alias in aliases:
aliased_group = super(OpenLLMCommandGroup, self).group(alias, *args, **attrs)(f)
aliased_group.short_help = text + f" (alias for '{name}')"
self._alias_groups[name] = (alias, aliased_group)
for __, g in self._alias_groups.values():
g.commands = group.commands
return group
return wrapper
def get_command(self, ctx: click.Context, cmd_name: str) -> click.Command | None:
cmd_name = self.resolve_alias(cmd_name)
if ctx.command.name == "start":
@@ -534,7 +498,7 @@ def create_cli() -> click.Group:
ctx.exit()
@cli.group(cls=OpenLLMCommandGroup, context_settings=_CONTEXT_SETTINGS, aliases=["start-http"], name="start")
@cli.group(cls=OpenLLMCommandGroup, context_settings=_CONTEXT_SETTINGS, name="start")
def _():
"""
Start any LLM as a REST server.