phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-27 14:38:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update: add validation on non-string type

Browse source 
Signed-off-by: Wen Zhou <wenzhou@redhat.com>
This commit is contained in:
Wen Zhou 2025-06-30 10:12:48 +02:00
parent 8563c76f88
commit 0b051c037b
2 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -50,7 +50,7 @@ class TestProviderMetrics:
@pytest.mark.asyncio
async def test_provider_with_invalid_metrics_in_config(self):
"""Test that invalid metrics in config fails when access /providers."""
"""Test invalid metrics in config fails when access /providers."""
run_config = StackRunConfig(
image_name="test_image",
providers={
@ -69,6 +69,27 @@ class TestProviderMetrics:
with pytest.raises(ValidationError):
await provider_impl.list_providers()
@pytest.mark.asyncio
async def test_provider_with_invalid_metrics_in_config2(self):
"""Test invalid metrics2 in config fails when access /providers."""
run_config = StackRunConfig(
image_name="test_image",
providers={
"inference": [
Provider(
provider_id="test_provider",
provider_type="test_type",
config={"url": "http://localhost:8000", "metrics": 123},
)
]
},
)
provider_config = ProviderImplConfig(run_config=run_config)
provider_impl = ProviderImpl(provider_config, {})
with pytest.raises(ValidationError):
await provider_impl.list_providers()
@pytest.mark.asyncio
async def test_provider_without_metrics_in_config(self):
"""Test provider without metrics in config returns None."""