From f17c2a6ff68b60d28506c9518fb5739b2c9b109a Mon Sep 17 00:00:00 2001
From: Xi Yan <xiyan@meta.com>
Date: Wed, 18 Sep 2024 13:37:33 -0700
Subject: [PATCH] remove imports

---
 llama_stack/cli/stack/configure.py | 22 +++++++++++++++-------
 llama_stack/cli/stack/run.py       |  7 +++----
 2 files changed, 18 insertions(+), 11 deletions(-)

diff --git a/llama_stack/cli/stack/configure.py b/llama_stack/cli/stack/configure.py
index 6b3b53f3e..b0aa1c3ab 100644
--- a/llama_stack/cli/stack/configure.py
+++ b/llama_stack/cli/stack/configure.py
@@ -5,17 +5,10 @@
 # the root directory of this source tree.
 
 import argparse
-import json
-from pathlib import Path
-
-import yaml
 
 from llama_stack.cli.subcommand import Subcommand
 from llama_stack.distribution.utils.config_dirs import BUILDS_BASE_DIR
-
-from llama_stack.distribution.utils.exec import run_with_pty
 from llama_stack.distribution.datatypes import *  # noqa: F403
-import os
 
 
 class StackConfigure(Subcommand):
@@ -46,9 +39,17 @@ class StackConfigure(Subcommand):
         )
 
     def _run_stack_configure_cmd(self, args: argparse.Namespace) -> None:
+        import json
+        import os
+        from pathlib import Path
+
         import pkg_resources
 
+        import yaml
+
         from llama_stack.distribution.build import ImageType
+
+        from llama_stack.distribution.utils.exec import run_with_pty
         from termcolor import cprint
 
         docker_image = None
@@ -98,7 +99,14 @@ class StackConfigure(Subcommand):
         build_config: BuildConfig,
         output_dir: Optional[str] = None,
     ):
+        import json
+        import os
+        from pathlib import Path
+
+        import yaml
         from llama_stack.distribution.configure import configure_api_providers
+
+        from llama_stack.distribution.utils.exec import run_with_pty
         from llama_stack.distribution.utils.serialize import EnumEncoder
         from termcolor import cprint
 
diff --git a/llama_stack/cli/stack/run.py b/llama_stack/cli/stack/run.py
index 95d91aa05..4e2009ee2 100644
--- a/llama_stack/cli/stack/run.py
+++ b/llama_stack/cli/stack/run.py
@@ -6,10 +6,6 @@
 
 import argparse
 
-from pathlib import Path
-
-import yaml
-
 from llama_stack.cli.subcommand import Subcommand
 from llama_stack.distribution.datatypes import *  # noqa: F403
 
@@ -46,7 +42,10 @@ class StackRun(Subcommand):
         )
 
     def _run_stack_run_cmd(self, args: argparse.Namespace) -> None:
+        from pathlib import Path
+
         import pkg_resources
+        import yaml
         from llama_stack.distribution.build import ImageType
         from llama_stack.distribution.utils.config_dirs import BUILDS_BASE_DIR