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

fix: Fixing prompts import warning

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-09-16 09:11:17 -04:00
parent 65d45c7318
commit 9fc0696052
2 changed files with 15 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

14
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
from llama_stack.core.distribution import get_provider_registry, providable_apis
from llama_stack.providers.datatypes import ProviderSpec
@ -152,6 +152,18 @@ class TestProviderRegistry:
assert registry[Api.inference]["test_provider"].provider_type == "test_provider"
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}
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"
def test_external_remote_providers(self, api_directories, mock_providers, base_config, provider_spec_yaml):
"""Test loading external remote providers from YAML files."""
remote_dir, _ = api_directories