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

Fix tool tests

Browse source
This commit is contained in:
Ashwin Bharambe 2025-01-22 20:31:18 -08:00
parent 0bff6e1658
commit 6c205e1d5a
2 changed files with 4 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

7
llama_stack/providers/tests/tools/conftest.py
View file

@ -8,8 +8,8 @@ import pytest
from ..conftest import get_provider_fixture_overrides
from ..inference.fixtures import INFERENCE_FIXTURES
from ..memory.fixtures import MEMORY_FIXTURES
from ..safety.fixtures import SAFETY_FIXTURES
from ..vector_io.fixtures import VECTOR_IO_FIXTURES
from .fixtures import TOOL_RUNTIME_FIXTURES
DEFAULT_PROVIDER_COMBINATIONS = [
@ -17,7 +17,7 @@ DEFAULT_PROVIDER_COMBINATIONS = [
{
"inference": "together",
"safety": "llama_guard",
"memory": "faiss",
"vector_io": "faiss",
"tool_runtime": "memory_and_search",
},
id="together",
@ -39,12 +39,11 @@ def pytest_generate_tests(metafunc):
available_fixtures = {
"inference": INFERENCE_FIXTURES,
"safety": SAFETY_FIXTURES,
"memory": MEMORY_FIXTURES,
"vector_io": VECTOR_IO_FIXTURES,
"tool_runtime": TOOL_RUNTIME_FIXTURES,
}
combinations = (
get_provider_fixture_overrides(metafunc.config, available_fixtures)
or DEFAULT_PROVIDER_COMBINATIONS
)
print(combinations)
metafunc.parametrize("tools_stack", combinations, indirect=True)

2
llama_stack/providers/tests/tools/test_tools.py
View file

@ -88,7 +88,7 @@ class TestTools:
tools_impl = tools_stack.impls[Api.tool_runtime]
# Register memory bank
await vector_dbs_impl.register(
await vector_dbs_impl.register_vector_db(
vector_db_id="test_bank",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,