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

Get agents tests working

Browse source
This commit is contained in:
Ashwin Bharambe 2024-11-04 10:14:55 -08:00 • committed by Ashwin Bharambe
parent 62dd3b376c
commit 66b658dcce
8 changed files with 352 additions and 269 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/tests/memory/fixtures.py
View file

@ -20,7 +20,7 @@ from ..env import get_env_or_fail
@pytest.fixture(scope="session")
def meta_reference() -> ProviderFixture:
def memory_meta_reference() -> ProviderFixture:
return ProviderFixture(
provider=Provider(
provider_id="meta-reference",
@ -31,7 +31,7 @@ def meta_reference() -> ProviderFixture:
@pytest.fixture(scope="session")
def pgvector() -> ProviderFixture:
def memory_pgvector() -> ProviderFixture:
return ProviderFixture(
provider=Provider(
provider_id="pgvector",
@ -48,7 +48,7 @@ def pgvector() -> ProviderFixture:
@pytest.fixture(scope="session")
def weaviate() -> ProviderFixture:
def memory_weaviate() -> ProviderFixture:
return ProviderFixture(
provider=Provider(
provider_id="weaviate",
@ -76,7 +76,7 @@ PROVIDER_PARAMS = [
)
async def memory_stack(request):
fixture_name = request.param
fixture = request.getfixturevalue(fixture_name)
fixture = request.getfixturevalue(f"memory_{fixture_name}")
impls = await resolve_impls_for_test_v2(
[Api.memory],