From 8efaa97eeb9a881f58a98f0259367578da03039f Mon Sep 17 00:00:00 2001 From: Andrew Brookins Date: Wed, 1 Apr 2026 20:41:20 -0700 Subject: [PATCH] Fix hsetex FXX enum usage --- agent_memory_server/long_term_memory.py | 3 ++- tests/test_long_term_memory.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/agent_memory_server/long_term_memory.py b/agent_memory_server/long_term_memory.py index 7d21b7d2..ad2709fa 100644 --- a/agent_memory_server/long_term_memory.py +++ b/agent_memory_server/long_term_memory.py @@ -10,6 +10,7 @@ from docket import Timeout from docket.dependencies import Perpetual from redis.asyncio import Redis +from redis.commands.core import HashDataPersistOptions from ulid import ULID from agent_memory_server.config import settings @@ -542,7 +543,7 @@ async def extract_memory_structure( "topics": encode_tag_values(merged_topics), "entities": encode_tag_values(merged_entities), }, - data_persist_option="FXX", + data_persist_option=HashDataPersistOptions.FXX, keepttl=True, ) if result == 0: diff --git a/tests/test_long_term_memory.py b/tests/test_long_term_memory.py index 5ac5d8f4..946049df 100644 --- a/tests/test_long_term_memory.py +++ b/tests/test_long_term_memory.py @@ -3,6 +3,7 @@ from unittest.mock import AsyncMock, MagicMock, Mock, patch import pytest +from redis.commands.core import HashDataPersistOptions from agent_memory_server.filters import Entities, Namespace, SessionId, Topics from agent_memory_server.long_term_memory import ( @@ -343,7 +344,7 @@ async def test_extract_memory_structure(self, mock_async_redis_client): mapping = call_kwargs[1]["mapping"] assert mapping["topics"] == "topic1,topic2" assert mapping["entities"] == "entity1,entity2" - assert call_kwargs[1]["data_persist_option"] == "FXX" + assert call_kwargs[1]["data_persist_option"] == HashDataPersistOptions.FXX assert call_kwargs[1]["keepttl"] is True @pytest.mark.asyncio