remove distribution types!

llama_toolchain/cli/stack/build.py

@@ -27,7 +27,7 @@ class StackBuild(Subcommand):
 
     def _add_arguments(self):
         self.parser.add_argument(
-            "--config",
+            "config",
             type=str,
             help="Path to a config file to use for the build",
         )
@@ -77,17 +77,20 @@ class StackBuild(Subcommand):
         from llama_toolchain.common.prompt_for_config import prompt_for_config
         from llama_toolchain.core.dynamic import instantiate_class_type
 
-        if args.config:
-            with open(args.config, "r") as f:
-                try:
-                    build_config = BuildConfig(**yaml.safe_load(f))
-                except Exception as e:
-                    self.parser.error(f"Could not parse config file {args.config}: {e}")
-                    return
-                if args.name:
-                    build_config.name = args.name
-                self._run_stack_build_command_from_build_config(build_config)
+        if not args.config:
+            self.parser.error(
+                "No config file specified. Please use `llama stack build /path/to/*-build.yaml`. Example config files can be found in llama_toolchain/configs/distributions"
+            )
             return
 
-        build_config = prompt_for_config(BuildConfig, None)
-        self._run_stack_build_command_from_build_config(build_config)
+        with open(args.config, "r") as f:
+            try:
+                build_config = BuildConfig(**yaml.safe_load(f))
+            except Exception as e:
+                self.parser.error(f"Could not parse config file {args.config}: {e}")
+                return
+            if args.name:
+                build_config.name = args.name
+            self._run_stack_build_command_from_build_config(build_config)
+
+        return

llama_toolchain/cli/stack/configure.py

@@ -128,7 +128,6 @@ class StackConfigure(Subcommand):
             )
 
         config.providers = configure_api_providers(config.providers)
-        config.distribution_type = build_config.distribution_spec.distribution_type
         config.docker_image = (
             package_name if build_config.image_type == "docker" else None
         )

llama_toolchain/cli/stack/list_distributions.py

@@ -1,55 +0,0 @@
-# Copyright (c) Meta Platforms, Inc. and affiliates.
-# All rights reserved.
-#
-# This source code is licensed under the terms described in the LICENSE file in
-# the root directory of this source tree.
-
-import argparse
-import json
-
-from llama_toolchain.cli.subcommand import Subcommand
-
-
-class StackListDistributions(Subcommand):
-    def __init__(self, subparsers: argparse._SubParsersAction):
-        super().__init__()
-        self.parser = subparsers.add_parser(
-            "list-distributions",
-            prog="llama stack list-distributions",
-            description="Show available Llama Stack Distributions",
-            formatter_class=argparse.RawTextHelpFormatter,
-        )
-        self._add_arguments()
-        self.parser.set_defaults(func=self._run_distribution_list_cmd)
-
-    def _add_arguments(self):
-        pass
-
-    def _run_distribution_list_cmd(self, args: argparse.Namespace) -> None:
-        from llama_toolchain.cli.table import print_table
-        from llama_toolchain.core.distribution_registry import (
-            available_distribution_specs,
-        )
-
-        # eventually, this should query a registry at llama.meta.com/llamastack/distributions
-        headers = [
-            "Distribution Type",
-            "Providers",
-            "Description",
-        ]
-
-        rows = []
-        for spec in available_distribution_specs():
-            providers = {k: v for k, v in spec.providers.items()}
-            rows.append(
-                [
-                    spec.distribution_type,
-                    json.dumps(providers, indent=2),
-                    spec.description,
-                ]
-            )
-        print_table(
-            rows,
-            headers,
-            separate_rows=True,
-        )

llama_toolchain/cli/stack/run.py

@@ -69,9 +69,6 @@ class StackRun(Subcommand):
         with open(config_file, "r") as f:
             config = PackageConfig(**yaml.safe_load(f))
 
-        if not config.distribution_type:
-            raise ValueError("Build config appears to be corrupt.")
-
         if config.docker_image:
             script = pkg_resources.resource_filename(
                 "llama_toolchain",

llama_toolchain/cli/stack/stack.py

@@ -11,7 +11,6 @@ from llama_toolchain.cli.subcommand import Subcommand
 from .build import StackBuild
 from .configure import StackConfigure
 from .list_apis import StackListApis
-from .list_distributions import StackListDistributions
 from .list_providers import StackListProviders
 from .run import StackRun
 
@@ -31,6 +30,5 @@ class StackParser(Subcommand):
         StackBuild.create(subparsers)
         StackConfigure.create(subparsers)
         StackListApis.create(subparsers)
-        StackListDistributions.create(subparsers)
         StackListProviders.create(subparsers)
         StackRun.create(subparsers)