phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 20:14:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Distribution server now functioning

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-02 13:37:40 -07:00
parent 041cafbee3
commit 2cf9915806
21 changed files with 635 additions and 266 deletions
Download patch file Download diff file Expand all files Collapse all files

17
llama_toolchain/cli/distribution/configure.py
View file

@ -18,11 +18,13 @@ from termcolor import cprint
from llama_toolchain.cli.subcommand import Subcommand
from llama_toolchain.distribution.datatypes import Distribution, PassthroughApiAdapter
from llama_toolchain.distribution.registry import available_distributions
from llama_toolchain.distribution.registry import (
available_distributions,
resolve_distribution,
)
from llama_toolchain.utils import DISTRIBS_BASE_DIR
from .utils import run_command
DISTRIBS = available_distributions()
from .utils import run_command
class DistributionConfigure(Subcommand):
@ -43,18 +45,13 @@ class DistributionConfigure(Subcommand):
self.parser.add_argument(
"--name",
type=str,
help="Mame of the distribution to configure",
help="Name of the distribution to configure",
default="local-source",
choices=[d.name for d in available_distributions()],
)
def _run_distribution_configure_cmd(self, args: argparse.Namespace) -> None:
dist = None
for d in DISTRIBS:
if d.name == args.name:
dist = d
break
dist = resolve_distribution(args.name)
if dist is None:
self.parser.error(f"Could not find distribution {args.name}")
return