phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

configure script with config

Browse source
This commit is contained in:
Xi Yan 2024-09-10 11:13:01 -07:00
parent 5bf2fe452d
commit 9d5582245d
1 changed files with 15 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

21
llama_toolchain/cli/stack/configure.py
View file

@ -9,10 +9,10 @@ import json
from pathlib import Path from pathlib import Path
import yaml import yaml
from termcolor import cprint
from llama_toolchain.cli.subcommand import Subcommand from llama_toolchain.cli.subcommand import Subcommand
from llama_toolchain.common.config_dirs import BUILDS_BASE_DIR from llama_toolchain.common.config_dirs import BUILDS_BASE_DIR
from termcolor import cprint
from llama_toolchain.core.datatypes import * # noqa: F403 from llama_toolchain.core.datatypes import * # noqa: F403
@ -38,7 +38,7 @@ class StackConfigure(Subcommand):
allowed_ids = [d.distribution_type for d in available_distribution_specs()] allowed_ids = [d.distribution_type for d in available_distribution_specs()]
self.parser.add_argument( self.parser.add_argument(
"distribution", "--distribution",
type=str, type=str,
help='Distribution ("adhoc" or one of: {})'.format(allowed_ids), help='Distribution ("adhoc" or one of: {})'.format(allowed_ids),
) )
@ -46,20 +46,29 @@ class StackConfigure(Subcommand):
"--name", "--name",
type=str, type=str,
help="Name of the build", help="Name of the build",
required=True,
) )
self.parser.add_argument( self.parser.add_argument(
"--type", "--package-type",
type=str, type=str,
default="conda_env", default="conda_env",
choices=[v.value for v in BuildType], choices=[v.value for v in BuildType],
) )
self.parser.add_argument(
"--config-file",
type=str,
help="Path to a config file to use for the build",
)
def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None: def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None:
from llama_toolchain.core.package import BuildType from llama_toolchain.core.package import BuildType
build_type = BuildType(args.type) if args.config_file:
name = args.name build_config = BuildConfig(**yaml.safe_load(f))
build_type = BuildType(build_config.package_type)
else:
build_type = BuildType(args.package_type)
name = args.name
config_file = ( config_file = (
BUILDS_BASE_DIR BUILDS_BASE_DIR
/ args.distribution / args.distribution