phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-06 04:34:57 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

moved llist of apis to global variable and imported in test

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-09-16 20:22:18 -04:00
parent e3d98059fc
commit 6db4c39401
2 changed files with 18 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

22
tests/unit/distribution/test_distribution.py
View file

@ -12,7 +12,7 @@ import yaml
from pydantic import BaseModel, Field, ValidationError
from llama_stack.core.datatypes import Api, Provider, StackRunConfig
from llama_stack.core.distribution import get_provider_registry, providable_apis
from llama_stack.core.distribution import INTERNAL_APIS, get_provider_registry, providable_apis
from llama_stack.providers.datatypes import ProviderSpec
@ -153,16 +153,22 @@ class TestProviderRegistry:
assert registry[Api.inference]["test_provider"].api == Api.inference
def test_internal_apis_excluded(self):
"""Test that internal APIs are excluded and other APIs are included in providable APIs."""
apis = providable_apis()
internal_apis = {Api.inspect, Api.providers, Api.prompts}
"""Test that internal APIs are excluded and APIs without provider registries are marked as internal."""
import importlib
for internal_api in internal_apis:
apis = providable_apis()
for internal_api in INTERNAL_APIS:
assert internal_api not in apis, f"Internal API {internal_api} should not be in providable_apis"
included_apis = {Api.inference, Api.safety, Api.agents}
for api in included_apis:
assert api in apis, f"API {api} should be in providable_apis"
for api in apis:
module_name = f"llama_stack.providers.registry.{api.name.lower()}"
try:
importlib.import_module(module_name)
except ImportError as err:
raise AssertionError(
f"API {api} is in providable_apis but has no provider registry module ({module_name})"
) from err
def test_external_remote_providers(self, api_directories, mock_providers, base_config, provider_spec_yaml):
"""Test loading external remote providers from YAML files."""