phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 01:52:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

support non value for test config

Browse source
This commit is contained in:
Sixian Yi 2025-01-15 13:45:55 -08:00
parent 26d9804efd
commit 11b2fdd3d8
4 changed files with 15 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama_stack/providers/tests/test_config_helper.py
View file

@ -18,10 +18,10 @@ class APITestConfig(BaseModel):
class Fixtures(BaseModel):
# provider fixtures can be either a mark or a dictionary of api -> providers
provider_fixtures: List[Dict[str, str]]
provider_fixtures: List[Dict[str, str]] = Field(default_factory=list)
inference_models: List[str] = Field(default_factory=list)
safety_shield: Optional[str]
embedding_model: Optional[str]
safety_shield: Optional[str] = Field(default_factory=None)
embedding_model: Optional[str] = Field(default_factory=None)
fixtures: Fixtures
tests: List[str] = Field(default_factory=list)
@ -32,8 +32,8 @@ class APITestConfig(BaseModel):
class TestConfig(BaseModel):
inference: APITestConfig
agent: APITestConfig
memory: APITestConfig
agent: Optional[APITestConfig] = Field(default=None)
memory: Optional[APITestConfig] = Field(default=None)
CONFIG_CACHE = None