mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-10-29 06:36:43 +00:00
5323f427d0
Bundles 10 community fixes for the Python wrappers, each commit kept with its original author. Ships supermemory-openai-sdk 1.0.10, supermemory-cartesia 0.1.5 and supermemory-pipecat 0.1.5. - openai-sdk-python: shared OpenAI clients no longer leak memories across tenants (#1503), memory search fails open instead of crashing the chat (#1288), the requests fallback runs off the event loop (#1660), get_profile works for users with no memories (#1759). - cartesia: the wrapped agent runs exactly once per event (#1280, #1670), plain-string and chunk-only search results are kept (#1683, #1693). - pipecat: memory text stays literal when replacing the injected block (#1647), `InputParams` is exported from the package root (#1575). - #1503 was hand-ported onto main's newer middleware. #1751 is left out: the v5 agent-framework migration (#1786) appends to the conversation document by design. Co-Authored-By: abhinav7x94 <204053250+abhinav7x94@users.noreply.github.com> Co-Authored-By: shamAnimates <145093437+shamAnimates@users.noreply.github.com> Co-Authored-By: aloktripathi1 <142301030+aloktripathi1@users.noreply.github.com> Co-Authored-By: Cintu07 <178455858+Cintu07@users.noreply.github.com> Co-Authored-By: Atharvsinh-codez <201381760+Atharvsinh-codez@users.noreply.github.com> Co-Authored-By: abhay-codes07 <182421137+abhay-codes07@users.noreply.github.com> Co-Authored-By: nkootstra <545768+nkootstra@users.noreply.github.com> Co-Authored-By: Agnik47 <140933190+Agnik47@users.noreply.github.com>
125 lines
3.5 KiB
Python
125 lines
3.5 KiB
Python
"""Regression tests: the wrapped agent must run exactly once per event."""
|
|
|
|
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 debug(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 _StreamingAgent:
|
|
"""Inner agent that yields chunks; optionally dies mid-stream."""
|
|
|
|
def __init__(self, fail_after_chunks: bool = False):
|
|
self.fail_after_chunks = fail_after_chunks
|
|
self.run_count = 0
|
|
|
|
async def process(self, env, event):
|
|
self.run_count += 1
|
|
yield "chunk-1"
|
|
yield "chunk-2"
|
|
if self.fail_after_chunks:
|
|
raise RuntimeError("stream dropped")
|
|
|
|
|
|
def _wrap(inner):
|
|
return SupermemoryCartesiaAgent(
|
|
agent=inner,
|
|
api_key="mock_key",
|
|
container_tag="user-123",
|
|
custom_id="conversation-456",
|
|
)
|
|
|
|
|
|
def _user_turn_ended_event():
|
|
return type("UserTurnEnded", (), {})()
|
|
|
|
|
|
class TestSingleAgentRun(unittest.IsolatedAsyncioTestCase):
|
|
async def test_mid_stream_agent_error_is_not_retried(self) -> None:
|
|
inner = _StreamingAgent(fail_after_chunks=True)
|
|
wrapper = _wrap(inner)
|
|
|
|
outputs = []
|
|
with self.assertRaises(RuntimeError):
|
|
async for out in wrapper.process(None, SimpleNamespace()):
|
|
outputs.append(out)
|
|
|
|
self.assertEqual(outputs, ["chunk-1", "chunk-2"])
|
|
self.assertEqual(inner.run_count, 1)
|
|
|
|
async def test_mid_stream_error_on_user_turn_is_not_retried(self) -> None:
|
|
inner = _StreamingAgent(fail_after_chunks=True)
|
|
wrapper = _wrap(inner)
|
|
wrapper._enrich_event_with_memories = AsyncMock(
|
|
side_effect=lambda event: (event, None)
|
|
)
|
|
|
|
outputs = []
|
|
with self.assertRaises(RuntimeError):
|
|
async for out in wrapper.process(None, _user_turn_ended_event()):
|
|
outputs.append(out)
|
|
|
|
self.assertEqual(outputs, ["chunk-1", "chunk-2"])
|
|
self.assertEqual(inner.run_count, 1)
|
|
|
|
async def test_enrichment_failure_still_runs_agent_once(self) -> None:
|
|
inner = _StreamingAgent()
|
|
wrapper = _wrap(inner)
|
|
wrapper._enrich_event_with_memories = AsyncMock(
|
|
side_effect=RuntimeError("enrichment exploded")
|
|
)
|
|
|
|
outputs = [
|
|
out async for out in wrapper.process(None, _user_turn_ended_event())
|
|
]
|
|
|
|
self.assertEqual(outputs, ["chunk-1", "chunk-2"])
|
|
self.assertEqual(inner.run_count, 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|