phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Small rename

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-18 14:41:38 -07:00
parent 71a905e93f
commit 95a96afe34
3 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/distribution/resolver.py
View file

@ -55,7 +55,7 @@ class ProviderWithSpec(Provider):
# TODO: this code is not very straightforward to follow and needs one more round of refactoring # TODO: this code is not very straightforward to follow and needs one more round of refactoring
async def resolve_impls_with_routing(run_config: StackRunConfig) -> Dict[Api, Any]: async def resolve_impls(run_config: StackRunConfig) -> Dict[Api, Any]:
""" """
Does two things: Does two things:
- flatmaps, sorts and resolves the providers in dependency order - flatmaps, sorts and resolves the providers in dependency order

4
llama_stack/distribution/server/server.py
View file

@ -37,7 +37,7 @@ from llama_stack.providers.utils.telemetry.tracing import (
from llama_stack.distribution.datatypes import * # noqa: F403 from llama_stack.distribution.datatypes import * # noqa: F403
from llama_stack.distribution.request_headers import set_request_provider_data from llama_stack.distribution.request_headers import set_request_provider_data
from llama_stack.distribution.resolver import resolve_impls_with_routing from llama_stack.distribution.resolver import resolve_impls
from .endpoints import get_all_api_endpoints from .endpoints import get_all_api_endpoints
@ -276,7 +276,7 @@ def main(
app = FastAPI() app = FastAPI()
impls = asyncio.run(resolve_impls_with_routing(config)) impls = asyncio.run(resolve_impls(config))
if Api.telemetry in impls: if Api.telemetry in impls:
setup_logger(impls[Api.telemetry]) setup_logger(impls[Api.telemetry])

4
llama_stack/providers/tests/resolver.py
View file

@ -14,7 +14,7 @@ import yaml
from llama_stack.distribution.datatypes import * # noqa: F403 from llama_stack.distribution.datatypes import * # noqa: F403
from llama_stack.distribution.configure import parse_and_maybe_upgrade_config from llama_stack.distribution.configure import parse_and_maybe_upgrade_config
from llama_stack.distribution.request_headers import set_request_provider_data from llama_stack.distribution.request_headers import set_request_provider_data
from llama_stack.distribution.resolver import resolve_impls_with_routing from llama_stack.distribution.resolver import resolve_impls
async def resolve_impls_for_test(api: Api, deps: List[Api] = None): async def resolve_impls_for_test(api: Api, deps: List[Api] = None):
@ -36,7 +36,7 @@ async def resolve_impls_for_test(api: Api, deps: List[Api] = None):
providers=chosen, providers=chosen,
) )
run_config = parse_and_maybe_upgrade_config(run_config) run_config = parse_and_maybe_upgrade_config(run_config)
impls = await resolve_impls_with_routing(run_config) impls = await resolve_impls(run_config)
if "provider_data" in config_dict: if "provider_data" in config_dict:
provider_id = chosen[api.value][0].provider_id provider_id = chosen[api.value][0].provider_id