fix: redis plugin testing (#2938)

This commit is contained in:
Andy Li
2025-06-20 13:35:35 -07:00
committed by GitHub
parent 4ec441a4b2
commit b90348e304
2 changed files with 15 additions and 5 deletions

View File

@@ -283,6 +283,12 @@ class NoopAsyncRedisClient(AsyncRedisClient):
async def scard(self, key: str) -> int:
return 0
async def smembers(self, key: str) -> Set[str]:
return set()
async def srem(self, key: str, *members: Union[str, int, float]) -> int:
return 0
async def get_redis_client() -> AsyncRedisClient:
global _client_instance

View File

@@ -1,6 +1,6 @@
import pytest
from letta.data_sources.redis_client import get_redis_client
from letta.data_sources.redis_client import NoopAsyncRedisClient, get_redis_client
from letta.helpers.decorators import experimental
from letta.settings import settings
@@ -86,7 +86,11 @@ async def test_redis_flag(event_loop):
await redis_client.create_inclusion_exclusion_keys(group=group_name)
await redis_client.sadd(include_key, test_user)
assert await _new_feature(user_id=test_user) == "new_feature"
with pytest.raises(Exception):
assert await _new_feature(user_id=test_user + "1")
print("members: ", await redis_client.smembers(include_key))
if not isinstance(redis_client, NoopAsyncRedisClient):
assert await _new_feature(user_id=test_user) == "new_feature"
with pytest.raises(Exception):
await _new_feature(user_id=test_user + "1")
print("members: ", await redis_client.smembers(include_key))
else:
with pytest.raises(Exception):
await _new_feature(user_id=test_user)