mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-08-19 13:34:05 +00:00
c449b2fe53
Use client.add and search.memories hybrid mode, improve profile memory deduplication for string/pydantic items, and add dedupe unit tests. Co-authored-by: Cursor <cursoragent@cursor.com>
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
from unittest.mock import AsyncMock
|
|
|
|
|
|
def _install_test_stubs() -> None:
|
|
if "loguru" not in sys.modules:
|
|
loguru_module = types.ModuleType("loguru")
|
|
|
|
class _Logger:
|
|
def info(self, *_args, **_kwargs):
|
|
return None
|
|
|
|
def warning(self, *_args, **_kwargs):
|
|
return None
|
|
|
|
def error(self, *_args, **_kwargs):
|
|
return None
|
|
|
|
loguru_module.logger = _Logger()
|
|
sys.modules["loguru"] = loguru_module
|
|
|
|
if "pydantic" not in sys.modules:
|
|
pydantic_module = types.ModuleType("pydantic")
|
|
|
|
class BaseModel:
|
|
def __init__(self, **kwargs):
|
|
for key, value in kwargs.items():
|
|
setattr(self, key, value)
|
|
|
|
def Field(*, default=None, **_kwargs):
|
|
return default
|
|
|
|
pydantic_module.BaseModel = BaseModel
|
|
pydantic_module.Field = Field
|
|
sys.modules["pydantic"] = pydantic_module
|
|
|
|
|
|
_install_test_stubs()
|
|
|
|
from supermemory_cartesia.agent import SupermemoryCartesiaAgent
|
|
|
|
|
|
class _MockSupermemoryClient:
|
|
def __init__(self, response):
|
|
self.profile = AsyncMock(return_value=response)
|
|
|
|
|
|
class TestSupermemoryCartesiaNullProfile(unittest.IsolatedAsyncioTestCase):
|
|
async def test_retrieve_memories_handles_null_profile(self) -> None:
|
|
agent = SupermemoryCartesiaAgent(
|
|
agent=SimpleNamespace(),
|
|
api_key="mock_key",
|
|
container_tag="user-123",
|
|
custom_id="conversation-456",
|
|
)
|
|
|
|
response = SimpleNamespace(profile=None, search_results=None)
|
|
agent._supermemory_client = _MockSupermemoryClient(response)
|
|
|
|
result = await agent._retrieve_memories("Hello world")
|
|
|
|
self.assertEqual(
|
|
result,
|
|
{
|
|
"profile": {"static": [], "dynamic": []},
|
|
"search_results": [],
|
|
},
|
|
)
|
|
agent._supermemory_client.profile.assert_awaited_once()
|
|
kwargs = agent._supermemory_client.profile.await_args.kwargs
|
|
self.assertEqual(kwargs["container_tag"], "user-123")
|
|
self.assertEqual(kwargs["q"], "Hello world")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|