pre-commit lint

llama_stack/distribution/configure.py

@@ -9,6 +9,10 @@ from typing import Any
 from pydantic import BaseModel
 
 from llama_stack.distribution.datatypes import *  # noqa: F403
+from prompt_toolkit import prompt
+from prompt_toolkit.validation import Validator
+from termcolor import cprint
+
 from llama_stack.apis.memory.memory import MemoryBankType
 from llama_stack.distribution.distribution import (
     api_providers,
@@ -21,9 +25,6 @@ from llama_stack.distribution.utils.prompt_for_config import prompt_for_config
 from llama_stack.providers.impls.meta_reference.safety.config import (
     MetaReferenceShieldType,
 )
-from prompt_toolkit import prompt
-from prompt_toolkit.validation import Validator
-from termcolor import cprint
 
 
 def make_routing_entry_type(config_class: Any):

llama_stack/distribution/server/server.py

@@ -435,13 +435,13 @@ def main(yaml_config: str, port: int = 5000, disable_ipv6: bool = False):
         apis_to_serve = set(config.apis_to_serve)
     else:
         apis_to_serve = set(impls.keys())
-    
+
     for api_str in apis_to_serve:
         api = Api(api_str)
 
         endpoints = all_endpoints[api]
         impl = impls[api]
-        
+
         provider_spec = specs[api]
         if (
             isinstance(provider_spec, RemoteProviderSpec)

llama_stack/distribution/utils/config_dirs.py

@@ -8,7 +8,9 @@ import os
 from pathlib import Path
 
 
-LLAMA_STACK_CONFIG_DIR = Path(os.getenv("LLAMA_STACK_CONFIG_DIR", os.path.expanduser("~/.llama/")))
+LLAMA_STACK_CONFIG_DIR = Path(
+    os.getenv("LLAMA_STACK_CONFIG_DIR", os.path.expanduser("~/.llama/"))
+)
 
 DISTRIBS_BASE_DIR = LLAMA_STACK_CONFIG_DIR / "distributions"
 

llama_stack/distribution/utils/dynamic.py

@@ -8,7 +8,6 @@ import importlib
 from typing import Any, Dict
 
 from llama_stack.distribution.datatypes import *  # noqa: F403
-from termcolor import cprint
 
 
 def instantiate_class_type(fully_qualified_name):