phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-23 01:19:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix tests

Browse source
This commit is contained in:
Xi Yan 2024-12-26 18:56:23 -08:00
parent 50764d76a7
commit 562ef41ff8
3 changed files with 20 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/distribution/inspect.py
View file

@ -8,10 +8,9 @@ from typing import Dict, List
from pydantic import BaseModel
from llama_stack.apis.inspect import Inspect
from llama_stack.distribution.datatypes import HealthInfo, ProviderInfo, RouteInfo
from llama_stack.apis.inspect import HealthInfo, Inspect, ProviderInfo, RouteInfo
from llama_stack.distribution.datatypes import StackRunConfig
from llama_stack.distribution.server.endpoints import get_all_api_endpoints
from llama_stack.providers.datatypes import StackRunConfig
class DistributionInspectConfig(BaseModel):

14
llama_stack/distribution/resolver.py
View file

@ -31,16 +31,9 @@ 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
@ -48,10 +41,17 @@ from llama_stack.distribution.utils.dynamic import instantiate_class_type
from llama_stack.providers.datatypes import (
Api,
DatasetsProtocolPrivate,
EvalTasksProtocolPrivate,
InlineProviderSpec,
MemoryBanksProtocolPrivate,
ModelsProtocolPrivate,
ProviderSpec,
RemoteProviderConfig,
RemoteProviderSpec,
ScoringFunctionsProtocolPrivate,
ShieldsProtocolPrivate,
ToolsProtocolPrivate,
)
log = logging.getLogger(__name__)