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

pip install helptext

Browse source
This commit is contained in:
Dalton Flanagan 2024-11-08 08:18:32 -05:00
parent 164aeca848
commit 33ea496cf9
5 changed files with 53 additions and 134 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -17,6 +17,7 @@ from llama_stack.distribution.configure import parse_and_maybe_upgrade_config
from llama_stack.distribution.distribution import get_provider_registry
from llama_stack.distribution.request_headers import set_request_provider_data
from llama_stack.distribution.resolver import resolve_impls
from llama_stack.distribution.build import print_pip_install_help
from llama_stack.distribution.store import CachedDiskDistributionRegistry
from llama_stack.providers.utils.kvstore import kvstore_impl, SqliteKVStoreConfig
@ -37,7 +38,12 @@ async def resolve_impls_for_test_v2(
sqlite_file = tempfile.NamedTemporaryFile(delete=False, suffix=".db")
dist_kvstore = await kvstore_impl(SqliteKVStoreConfig(db_path=sqlite_file.name))
dist_registry = CachedDiskDistributionRegistry(dist_kvstore)
impls = await resolve_impls(run_config, get_provider_registry(), dist_registry)
try:
impls = await resolve_impls(run_config, get_provider_registry(), dist_registry)
except ModuleNotFoundError as e:
print_pip_install_help(providers)
print(f"ModuleNotFoundError: {e}") # Add explicit print of error
raise e
if provider_data:
set_request_provider_data(
@ -66,7 +72,11 @@ async def resolve_impls_for_test(api: Api, deps: List[Api] = None):
providers=chosen,
)
run_config = parse_and_maybe_upgrade_config(run_config)
impls = await resolve_impls(run_config, get_provider_registry())
try:
impls = await resolve_impls(run_config, get_provider_registry())
except ModuleNotFoundError as e:
print_pip_install_help(providers)
print(f"ModuleNotFoundError: {e}") # Add explicit print of error
if "provider_data" in config_dict:
provider_id = chosen[api.value][0].provider_id