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

fix tests

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-20 14:16:53 -07:00
parent 9c9f5f059a
commit 85143e7316
1 changed files with 15 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/unit/core/test_stack_validation.py
View file

@ -10,7 +10,7 @@ from unittest.mock import AsyncMock
import pytest
from llama_stack.apis.models import Model, ModelType
from llama_stack.apis.models import ListModelsResponse, Model, ModelType
from llama_stack.core.datatypes import QualifiedModel, StackRunConfig, StorageConfig, VectorStoresConfig
from llama_stack.core.stack import validate_vector_stores_config
from llama_stack.providers.datatypes import Api
@ -32,10 +32,10 @@ class TestVectorStoresValidation:
),
)
mock_models = AsyncMock()
mock_models.list_models.return_value = []
mock_models.list_models.return_value = ListModelsResponse(data=[])
with pytest.raises(ValueError, match="not found"):
await validate_vector_stores_config(run_config, {Api.models: mock_models})
await validate_vector_stores_config(run_config.vector_stores, {Api.models: mock_models})
async def test_validate_success(self):
"""Test validation passes with valid model."""
@ -52,14 +52,16 @@ class TestVectorStoresValidation:
),
)
mock_models = AsyncMock()
mock_models.list_models.return_value = [
Model(
identifier="p/valid", # Must match provider_id/model_id format
model_type=ModelType.embedding,
metadata={"embedding_dimension": 768},
provider_id="p",
provider_resource_id="valid",
)
]
mock_models.list_models.return_value = ListModelsResponse(
data=[
Model(
identifier="p/valid", # Must match provider_id/model_id format
model_type=ModelType.embedding,
metadata={"embedding_dimension": 768},
provider_id="p",
provider_resource_id="valid",
)
]
)
await validate_vector_stores_config(run_config, {Api.models: mock_models})
await validate_vector_stores_config(run_config.vector_stores, {Api.models: mock_models})