fix: use valid S3 entity index names (#5416)

Co-authored-by: Kartik <kartik.labhshetwar@mem0.ai>
This commit is contained in:
Yufeng He
2026-06-12 19:22:32 +05:30
committed by GitHub
co-authored by Kartik
parent f681889b14
commit e3f5ce7b41
3 changed files with 42 additions and 2 deletions
+9
View File
@@ -6,6 +6,7 @@ import pytest
from mem0 import Memory
from mem0.configs.base import MemoryConfig
from mem0.memory.main import _entity_collection_name
from mem0.memory.utils import normalize_facts
@@ -37,6 +38,14 @@ def test_create_memory(memory_client):
assert result["results"][0]["memory"] == data
def test_entity_collection_name_uses_dash_for_s3_vectors():
assert _entity_collection_name("s3_vectors", "test-index") == "test-index-entities"
def test_entity_collection_name_keeps_underscore_for_other_stores():
assert _entity_collection_name("qdrant", "mem0") == "mem0_entities"
def test_get_memory(memory_client):
data = "Name is John Doe."
memory_client.add([{"role": "user", "content": data}], user_id="test_user")