memtrust-cli 0.3.0__py3-none-any.whl

memtrust/__init__.py

@@ -0,0 +1,11 @@
+"""memtrust: an independent, reproducible benchmark harness for agent-memory backends."""
+
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("memtrust")
+except PackageNotFoundError:
+    # Not installed at all (e.g. importing directly from a raw source
+    # checkout with no `pip install -e .` run) -- there is no installed
+    # package metadata to read version() from in this case.
+    __version__ = "0.0.0+unknown"

memtrust/adapters/__init__.py

@@ -0,0 +1,62 @@
+"""Backend adapters. Each module maps one vendor's real API/SDK shape onto
+the shared MemoryBackendAdapter interface defined in base.py.
+"""
+
+from memtrust.adapters.base import (
+    BackendAPIError,
+    BackendNotConfiguredError,
+    ConflictSignal,
+    DeleteResult,
+    MemoryBackendAdapter,
+    MemoryRecord,
+    QueryResult,
+    StoreResult,
+    UpdateResult,
+)
+from memtrust.adapters.mem0_adapter import Mem0Adapter, Mem0SelfHostedAdapter
+from memtrust.adapters.mem0_direct_adapter import Mem0DirectAdapter
+from memtrust.adapters.mempalace_adapter import MemPalaceAdapter
+from memtrust.adapters.openviking_adapter import OpenVikingAdapter
+from memtrust.adapters.zep_graphiti_adapter import ZepGraphitiAdapter
+from memtrust.adapters.zep_graphiti_selfhosted_adapter import ZepGraphitiSelfHostedAdapter
+
+#: Registry the CLI resolves --backends names against. Keys are the
+#: user-facing backend names used on the command line.
+#:
+#: "mem0_selfhosted", "mem0_direct", and "graphiti_selfhosted" are
+#: intentionally not part of cli.ALL_BACKENDS (the set "all" expands to) --
+#: none targets a single hosted vendor API the way the other four do
+#: ("mem0_selfhosted"/"graphiti_selfhosted" are self-run local
+#: deployments, "mem0_direct" is a self-assembled in-process
+#: embedder/vector-store stack), so all three are opt-in only, never
+#: auto-included. See docs/methodology.md for their confidence levels.
+ADAPTER_REGISTRY: dict[str, type[MemoryBackendAdapter]] = {
+    "mempalace": MemPalaceAdapter,
+    "mem0": Mem0Adapter,
+    "mem0_selfhosted": Mem0SelfHostedAdapter,
+    "mem0_direct": Mem0DirectAdapter,
+    "zep": ZepGraphitiAdapter,
+    "graphiti": ZepGraphitiAdapter,
+    "graphiti_selfhosted": ZepGraphitiSelfHostedAdapter,
+    "openviking": OpenVikingAdapter,
+}
+
+__all__ = [
+    "ADAPTER_REGISTRY",
+    "BackendAPIError",
+    "BackendNotConfiguredError",
+    "ConflictSignal",
+    "DeleteResult",
+    "MemoryBackendAdapter",
+    "MemoryRecord",
+    "QueryResult",
+    "StoreResult",
+    "UpdateResult",
+    "Mem0Adapter",
+    "Mem0SelfHostedAdapter",
+    "Mem0DirectAdapter",
+    "MemPalaceAdapter",
+    "OpenVikingAdapter",
+    "ZepGraphitiAdapter",
+    "ZepGraphitiSelfHostedAdapter",
+]