This commit is contained in:
Xi Yan 2024-12-26 18:42:57 -08:00
parent a6091fa158
commit b936503784
5 changed files with 67 additions and 26 deletions

View file

@ -6,10 +6,14 @@
import logging
import textwrap
from typing import Any
from llama_stack.distribution.datatypes import * # noqa: F403
from typing import Any, Dict
from llama_stack.distribution.datatypes import (
DistributionSpec,
LLAMA_STACK_RUN_CONFIG_VERSION,
Provider,
StackRunConfig,
)
from llama_stack.distribution.distribution import (
builtin_automatically_routed_apis,
get_provider_registry,
@ -17,10 +21,7 @@ from llama_stack.distribution.distribution import (
from llama_stack.distribution.utils.dynamic import instantiate_class_type
from llama_stack.distribution.utils.prompt_for_config import prompt_for_config
from llama_stack.apis.models import * # noqa: F403
from llama_stack.apis.shields import * # noqa: F403
from llama_stack.apis.memory_banks import * # noqa: F403
from llama_stack.providers.datatypes import Api, ProviderSpec
logger = logging.getLogger(__name__)