diff --git a/llama_stack/providers/tests/eval/conftest.py b/llama_stack/providers/tests/eval/conftest.py index ecd339c8c..84eae2efa 100644 --- a/llama_stack/providers/tests/eval/conftest.py +++ b/llama_stack/providers/tests/eval/conftest.py @@ -12,11 +12,11 @@ from ..conftest import get_provider_fixture_overrides from ..datasetio.fixtures import DATASETIO_FIXTURES from ..inference.fixtures import INFERENCE_FIXTURES -from ..memory.fixtures import MEMORY_FIXTURES from ..safety.fixtures import SAFETY_FIXTURES from ..scoring.fixtures import SCORING_FIXTURES from ..tools.fixtures import TOOL_RUNTIME_FIXTURES from .fixtures import EVAL_FIXTURES +from ..vector_io.fixtures import VECTOR_IO_FIXTURES DEFAULT_PROVIDER_COMBINATIONS = [ pytest.param( @@ -27,7 +27,7 @@ DEFAULT_PROVIDER_COMBINATIONS = [ "inference": "fireworks", "agents": "meta_reference", "safety": "llama_guard", - "memory": "faiss", + "vector_io": "faiss", "tool_runtime": "memory_and_search", }, id="meta_reference_eval_fireworks_inference", @@ -41,7 +41,7 @@ DEFAULT_PROVIDER_COMBINATIONS = [ "inference": "together", "agents": "meta_reference", "safety": "llama_guard", - "memory": "faiss", + "vector_io": "faiss", "tool_runtime": "memory_and_search", }, id="meta_reference_eval_together_inference", @@ -55,7 +55,7 @@ DEFAULT_PROVIDER_COMBINATIONS = [ "inference": "together", "agents": "meta_reference", "safety": "llama_guard", - "memory": "faiss", + "vector_io": "faiss", "tool_runtime": "memory_and_search", }, id="meta_reference_eval_together_inference_huggingface_datasetio", @@ -85,7 +85,7 @@ def pytest_generate_tests(metafunc): "inference": INFERENCE_FIXTURES, "agents": AGENTS_FIXTURES, "safety": SAFETY_FIXTURES, - "memory": MEMORY_FIXTURES, + "vector_io": VECTOR_IO_FIXTURES, "tool_runtime": TOOL_RUNTIME_FIXTURES, } combinations = (