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

address comments

Browse source
This commit is contained in:
Sixian Yi 2025-01-16 12:05:00 -08:00
parent cd1a155926
commit 847f290752
1 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

20
llama_stack/providers/tests/conftest.py
View file

@ -51,13 +51,13 @@ class AgentsApiTestConfig(APITestConfig):
class TestConfig(BaseModel): class TestConfig(BaseModel):
inference: APITestConfig = Field(default=None) inference: Optional[APITestConfig] = None
agents: AgentsApiTestConfig = Field(default=None) agents: Optional[AgentsApiTestConfig] = None
memory: MemoryApiTestConfig = Field(default=None) memory: Optional[MemoryApiTestConfig] = None
def get_test_config_from_config_file(config): def get_test_config_from_config_file(metafunc_config):
config_file = config.getoption("--config") config_file = metafunc_config.getoption("--config")
if config_file is None: if config_file is None:
return None return None
@ -71,17 +71,17 @@ def get_test_config_from_config_file(config):
return TestConfig(**config) return TestConfig(**config)
def get_test_config_for_api(config, api): def get_test_config_for_api(metafunc_config, api):
test_config = get_test_config_from_config_file(config) test_config = get_test_config_from_config_file(metafunc_config)
if test_config is None: if test_config is None:
return None return None
return getattr(test_config, api) return getattr(test_config, api)
def get_provider_fixture_overrides_from_test_config( def get_provider_fixture_overrides_from_test_config(
config, api, default_provider_fixture_combination metafunc_config, api, default_provider_fixture_combinations
): ):
api_config = get_test_config_for_api(config, api) api_config = get_test_config_for_api(metafunc_config, api)
if api_config is None: if api_config is None:
return None return None
@ -99,7 +99,7 @@ def get_provider_fixture_overrides_from_test_config(
) )
if len(fixture_combo_ids) > 0: if len(fixture_combo_ids) > 0:
for default_fixture in default_provider_fixture_combination: for default_fixture in default_provider_fixture_combinations:
if default_fixture.id in fixture_combo_ids: if default_fixture.id in fixture_combo_ids:
custom_provider_fixture_combos.append(default_fixture) custom_provider_fixture_combos.append(default_fixture)
return custom_provider_fixture_combos return custom_provider_fixture_combos