fix: use valid S3 entity index names (#5416)
Co-authored-by: Kartik <kartik.labhshetwar@mem0.ai>
This commit is contained in:
@@ -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")
|
||||
|
||||
Reference in New Issue
Block a user