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,14 +6,10 @@
import importlib
import inspect
from typing import Any, Dict, List, Set
from llama_stack.providers.datatypes import * # noqa: F403
from llama_stack.distribution.datatypes import * # noqa: F403
import logging
from typing import Any, Dict, List, Set
from llama_stack.apis.agents import Agents
from llama_stack.apis.datasetio import DatasetIO
from llama_stack.apis.datasets import Datasets
@ -32,10 +28,32 @@ from llama_stack.apis.shields import Shields
from llama_stack.apis.telemetry import Telemetry
from llama_stack.apis.tools import ToolGroups, ToolRuntime
from llama_stack.distribution.client import get_client_impl
from llama_stack.distribution.datatypes import (
AutoRoutedProviderSpec,
DatasetsProtocolPrivate,
EvalTasksProtocolPrivate,
MemoryBanksProtocolPrivate,
ModelsProtocolPrivate,
Provider,
RoutingTableProviderSpec,
ScoringFunctionsProtocolPrivate,
ShieldsProtocolPrivate,
StackRunConfig,
ToolsProtocolPrivate,
)
from llama_stack.distribution.distribution import builtin_automatically_routed_apis
from llama_stack.distribution.store import DistributionRegistry
from llama_stack.distribution.utils.dynamic import instantiate_class_type
from llama_stack.providers.datatypes import (
Api,
InlineProviderSpec,
ProviderSpec,
RemoteProviderConfig,
RemoteProviderSpec,
)
log = logging.getLogger(__name__)