mirror of
https://github.com/supermemoryai/supermemory.git
synced 2026-10-20 06:29:37 +08: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>
87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
|
|
from supermemory_pipecat.utils import (
|
|
deduplicate_memories,
|
|
format_memories_to_text,
|
|
)
|
|
|
|
|
|
class TestSearchResultExtraction(unittest.TestCase):
|
|
"""v4 hybrid search returns chunk hits whose `memory` field is empty."""
|
|
|
|
def test_falls_through_to_chunk_when_memory_is_empty(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[{"memory": "", "chunk": "The user's dog is called Rex"}],
|
|
)
|
|
|
|
self.assertEqual(len(result["search_results"]), 1)
|
|
self.assertIn("The user's dog is called Rex", format_memories_to_text(result))
|
|
|
|
def test_falls_through_to_chunk_when_memory_is_whitespace(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[{"memory": " ", "chunk": "Chunk body"}],
|
|
)
|
|
|
|
self.assertEqual(len(result["search_results"]), 1)
|
|
self.assertIn("Chunk body", format_memories_to_text(result))
|
|
|
|
def test_falls_through_on_models_too(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[SimpleNamespace(memory="", chunk="Chunk body")],
|
|
)
|
|
|
|
self.assertEqual(len(result["search_results"]), 1)
|
|
self.assertIn("Chunk body", format_memories_to_text(result))
|
|
|
|
def test_never_renders_an_empty_bullet(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[
|
|
{
|
|
"memory": "",
|
|
"chunk": "Chunk body",
|
|
"updatedAt": "2020-01-01T00:00:00Z",
|
|
}
|
|
],
|
|
)
|
|
|
|
rendered = format_memories_to_text(result)
|
|
self.assertIn("Chunk body", rendered)
|
|
for line in rendered.splitlines():
|
|
if line.startswith("- "):
|
|
self.assertNotRegex(line, r"^- (\[[^\]]*\] )?$")
|
|
|
|
def test_memory_still_wins_over_chunk(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[{"memory": "Memory body", "chunk": "Chunk body"}],
|
|
)
|
|
|
|
rendered = format_memories_to_text(result)
|
|
self.assertIn("Memory body", rendered)
|
|
self.assertNotIn("Chunk body", rendered)
|
|
|
|
def test_entry_with_no_usable_text_is_still_dropped(self) -> None:
|
|
result = deduplicate_memories(
|
|
static=[],
|
|
dynamic=[],
|
|
search_results=[{"memory": "", "chunk": ""}, {}, None],
|
|
)
|
|
|
|
self.assertEqual(result["search_results"], [])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|