Class SupermemoryClient

evaluation/scripts/utils/client.py:277–316

class SupermemoryClient:
    def __init__(self):
        from supermemory import Supermemory

        self.client = Supermemory(api_key=os.getenv("SUPERMEMORY_API_KEY"))

    def add(self, messages, user_id):
        content = "\n".join(
            [f"{msg['chat_time']} {msg['role']}: {msg['content']}" for msg in messages]
        )
        max_retries = 5
        for attempt in range(max_retries):
            try:
                self.client.memories.add(content=content, container_tag=user_id)
                break
            except Exception as e:
                if attempt < max_retries - 1:
                    time.sleep(2**attempt)
                else:
                    raise e

    def search(self, query, user_id, top_k):
        max_retries = 10
        for attempt in range(max_retries):
            try:
                results = self.client.search.memories(
                    q=query,
                    container_tag=user_id,
                    threshold=0,
                    rerank=True,
                    rewrite_query=True,
                    limit=top_k,
                )
                context = "\n\n".join([r.memory for r in results.results])
                return context
            except Exception as e:
                if attempt < max_retries - 1:
                    time.sleep(2**attempt)
                else:
                    raise e


class MemuClient: