From 01ae720a7baa806d48b169c84162aea0811bac41 Mon Sep 17 00:00:00 2001 From: Mohit Gaur <56885276+Mohit-Gaur@users.noreply.github.com> Date: Sat, 23 Nov 2024 11:28:58 +0000 Subject: [PATCH] Add integration test for redis memory provider --- llama_stack/providers/tests/memory/fixtures.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/llama_stack/providers/tests/memory/fixtures.py b/llama_stack/providers/tests/memory/fixtures.py index c9559b61c..88c358218 100644 --- a/llama_stack/providers/tests/memory/fixtures.py +++ b/llama_stack/providers/tests/memory/fixtures.py @@ -93,7 +93,23 @@ def memory_chroma() -> ProviderFixture: ) -MEMORY_FIXTURES = ["faiss", "pgvector", "weaviate", "remote", "chroma"] +@pytest.fixture(scope="session") +def memory_redis() -> ProviderFixture: + return ProviderFixture( + providers=[ + Provider( + provider_id="redis", + provider_type="remote::redis", + config=RemoteProviderConfig( + host=get_env_or_fail("REDIS_HOST"), + port=get_env_or_fail("REDIS_PORT"), + ).model_dump(), + ) + ] + ) + + +MEMORY_FIXTURES = ["faiss", "pgvector", "weaviate", "remote", "chroma", "redis"] @pytest_asyncio.fixture(scope="session")